[BI-1720] - cache studies on create

mlm483 · mlm483 · commit 88a4cd5f6b03 · 2023-07-27T17:27:30.000-04:00
diff --git a/src/main/java/org/breedinginsight/brapps/importer/daos/BrAPIStudyDAO.java b/src/main/java/org/breedinginsight/brapps/importer/daos/BrAPIStudyDAO.java
@@ -40,6 +40,7 @@
 import javax.inject.Singleton;
 import javax.validation.constraints.NotNull;
 import java.util.*;
+import java.util.concurrent.Callable;
 import java.util.stream.Collectors;
 
 @Slf4j
@@ -152,7 +153,37 @@ public List<BrAPIStudy> getStudiesByExperimentID(@NotNull UUID experimentID, Pro
 
     public List<BrAPIStudy> createBrAPIStudies(List<BrAPIStudy> brAPIStudyList, UUID programId, ImportUpload upload) throws ApiException {
         StudiesApi api = brAPIEndpointProvider.get(programDAO.getCoreClient(programId), StudiesApi.class);
-        return brAPIDAOUtil.post(brAPIStudyList, upload, api::studiesPost, importDAO::update);
+        List<BrAPIStudy> createdStudies = new ArrayList<>();
+        try {
+            if (!brAPIStudyList.isEmpty()) {
+                Callable<Map<String, BrAPIStudy>> postCallback = () -> {
+                    List<BrAPIStudy> postedStudies = brAPIDAOUtil
+                            .post(brAPIStudyList, upload, api::studiesPost, importDAO::update);
+                    return studyById(postedStudies);
+                };
+                createdStudies.addAll(programStudyCache.post(programId, postCallback));
+            }
+
+            return createdStudies;
+        } catch (Exception e) {
+            throw new InternalServerException("Unknown error has occurred: " + e.getMessage(), e);
+        }
+    }
+
+    /**
+     * @return Map - Key = BI external reference ID, Value = BrAPIStudy
+     * */
+    private Map<String, BrAPIStudy> studyById(List<BrAPIStudy> studies) {
+        Map<String, BrAPIStudy> studyById = new HashMap<>();
+        for (BrAPIStudy study: studies) {
+            BrAPIExternalReference xref = study
+                    .getExternalReferences()
+                    .stream()
+                    .filter(reference -> String.format("%s/%s", referenceSource, ExternalReferenceSource.STUDIES).equalsIgnoreCase(reference.getReferenceSource()))
+                    .findFirst().orElseThrow(() -> new IllegalStateException("No BI external reference found"));
+            studyById.put(xref.getReferenceID(), study);
+        }
+        return studyById;
     }
 
     public List<BrAPIStudy> getStudiesByStudyDbId(Collection<String> studyDbIds, Program program) throws ApiException {
