fix mapping between entity props and document props

Sage-Bionetworks · Oct 12, 2024 · 74355c0 · 74355c0
1 parent 81cbd04
commit 74355c0
Show file tree

Hide file tree

Showing 3 changed files with 8 additions and 2 deletions.
diff --git a/...pi-spring/src/main/java/org/sagebionetworks/agora/api/model/entity/DataVersionEntity.java b/...pi-spring/src/main/java/org/sagebionetworks/agora/api/model/entity/DataVersionEntity.java
@@ -2,14 +2,20 @@
 
 import org.springframework.data.annotation.Id;
 import org.springframework.data.mongodb.core.mapping.Document;
+import org.springframework.data.mongodb.core.mapping.Field;
 
 @Document(collection = "dataversion")
 public class DataVersionEntity {
 
   @Id
   public String id;
 
+  @Field("data_file")
   public String dataFile;
+
+  @Field("data_version")
   public String dataVersion;
+
+  @Field("team_images_id")
   public String teamImagesId;
 }
diff --git a/...g/src/main/java/org/sagebionetworks/agora/api/model/repository/DataVersionRepository.java b/...g/src/main/java/org/sagebionetworks/agora/api/model/repository/DataVersionRepository.java
@@ -6,5 +6,5 @@
 
 public interface DataVersionRepository extends MongoRepository<DataVersionEntity, String> {
   @Query("{}")
-  DataVersionEntity get();
+  DataVersionEntity find();
 }
diff --git a/...ra/api-spring/src/main/java/org/sagebionetworks/agora/api/service/DataVersionService.java b/...ra/api-spring/src/main/java/org/sagebionetworks/agora/api/service/DataVersionService.java
@@ -17,7 +17,7 @@ public DataVersionService(DataVersionRepository dataVersionRepository) {
 
   @Transactional(readOnly = true)
   public DataversionDto getDataVersion() {
-    DataVersionEntity entity = this.dataVersionRepository.get();
+    DataVersionEntity entity = this.dataVersionRepository.find();
 
     return DataversionDto.builder()
       .dataFile(entity.dataFile)