ID Improvement - Get

metaisbeta · Feb 14, 2022 · 9badb45 · 9badb45
1 parent 3e025da
commit 9badb45
Show file tree

Hide file tree

Showing 2 changed files with 6 additions and 13 deletions.
diff --git a/src/main/java/io/github/phillima/avisualizer/repository/AvisualizerRepository.java b/src/main/java/io/github/phillima/avisualizer/repository/AvisualizerRepository.java
@@ -9,7 +9,7 @@
 import java.util.UUID;
 
 @Repository
-public interface AvisualizerRepository extends MongoRepository<AvisualizerEntity, UUID> {
+public interface AvisualizerRepository extends MongoRepository<AvisualizerEntity, String> {
 
     @Query("{ 'hash' : ?0 }")
     List<AvisualizerEntity> findByHash(String hash);

diff --git a/src/main/java/io/github/phillima/avisualizer/service/AvisualizerService.java b/src/main/java/io/github/phillima/avisualizer/service/AvisualizerService.java
@@ -92,7 +92,7 @@ public String returnSV(String projectID) throws URISyntaxException, IOException
         if(StringUtils.isEmpty(projectID)){
             return returnFileSV(SV_DEFAULT);
         }
-        Optional<AvisualizerEntity> response= this.repository.findById(UUID.fromString(projectID));
+        Optional<AvisualizerEntity> response= this.repository.findById(projectID);
         if(response.isPresent()){
             return response.get().getSv();
         }
@@ -103,7 +103,7 @@ public String returnCV(String projectID) throws URISyntaxException, IOException
         if(StringUtils.isEmpty(projectID)){
             return returnFileSV(SV_DEFAULT);
         }
-        Optional<AvisualizerEntity> response= this.repository.findById(UUID.fromString(projectID));
+        Optional<AvisualizerEntity> response= this.repository.findById(projectID);
         if(response.isPresent()){
             return response.get().getCv();
         }
@@ -114,7 +114,7 @@ public String returnPV(String projectID) throws URISyntaxException, IOException
         if(StringUtils.isEmpty(projectID)){
             return returnFileSV(SV_DEFAULT);
         }
-        Optional<AvisualizerEntity> response= this.repository.findById(UUID.fromString(projectID));
+        Optional<AvisualizerEntity> response= this.repository.findById(projectID);
         if(response.isPresent()){
             return response.get().getPv();
         }
@@ -145,21 +145,14 @@ public AvisualizerEntity saveModel(AvisualizerModel model){
         return this.repository.save(entity);
     }
 
-    public AvisualizerEntity getAllInformation(String projectUUID) throws URISyntaxException, IOException {
-        Optional<AvisualizerEntity> response = this.repository.findById(UUID.fromString(projectUUID));
+    public AvisualizerEntity getAllInformation(String projectID) {
+        Optional<AvisualizerEntity> response = this.repository.findById(projectID);
         if(response.isPresent()){
             return response.get();
         }
         return new AvisualizerEntity();
     }
 
-    private Long processProjectID(Long projectID){
-        if(Objects.isNull(projectID)){
-            return -1L;
-        }
-        return projectID;
-    }
-
     public void saveError(ErrorModel model){
         ErrorEntity entity = new ErrorEntity();