create Trial and Sample components

Author: dmeidlin

src/main/java/io/swagger/model/core/BatchTypes.java

@@ -9,7 +9,10 @@
  */
 public enum BatchTypes {
   GERMPLASM("germplasm"),
-  LISTS("lists");
+  LISTS("lists"),
+  TRIALS("trials"),
+  SAMPLES("samples");
+
 
   private String value;
 

src/main/java/org/brapi/test/BrAPITestServer/factory/core/TrialComponent.java

@@ -0,0 +1,54 @@
+package org.brapi.test.BrAPITestServer.factory.core;
+
+import io.swagger.model.Metadata;
+import io.swagger.model.core.BatchTypes;
+import io.swagger.model.core.Trial;
+import io.swagger.model.core.TrialSearchRequest;
+import org.brapi.test.BrAPITestServer.factory.BrAPIComponent;
+import org.brapi.test.BrAPITestServer.model.entity.SearchRequestEntity;
+import org.brapi.test.BrAPITestServer.service.core.TrialService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import java.util.List;
+import java.util.stream.Collectors;
+
+@Component
+public class TrialComponent implements BrAPIComponent<Trial, TrialSearchRequest> {
+    private final TrialService trialService;
+
+    @Autowired
+    public TrialComponent(TrialService trialService) {
+        this.trialService = trialService;
+    }
+
+    @Override
+    public List<Trial> findEntities(TrialSearchRequest request, Metadata metadata) {
+        return trialService.findTrials(request, metadata);
+    }
+
+    @Override
+    public BatchTypes getBatchType() {
+        return BatchTypes.TRIALS;
+    }
+
+    @Override
+    public SearchRequestEntity.SearchRequestTypes getSearchType() {
+        return SearchRequestEntity.SearchRequestTypes.TRIALS;
+    }
+
+    @Override
+    public List<String> collectDbIds(List<Trial> entities) {
+        return entities.stream().map(Trial::getTrialDbId).collect(Collectors.toList());
+    }
+
+    @Override
+    public void deleteBatchData(List<String> dbIds) {
+        trialService.deleteTrialBatch(dbIds);
+    }
+
+    @Override
+    public void softDeleteBatchData(List<String> dbIds) {
+        trialService.softDeleteTrialBatch(dbIds);
+    }
+}

src/main/java/org/brapi/test/BrAPITestServer/factory/geno/SampleComponent.java

@@ -0,0 +1,54 @@
+package org.brapi.test.BrAPITestServer.factory.geno;
+
+import io.swagger.model.Metadata;
+import io.swagger.model.core.BatchTypes;
+import io.swagger.model.geno.Sample;
+import io.swagger.model.geno.SampleSearchRequest;
+import org.brapi.test.BrAPITestServer.factory.BrAPIComponent;
+import org.brapi.test.BrAPITestServer.model.entity.SearchRequestEntity;
+import org.brapi.test.BrAPITestServer.service.geno.SampleService;
+import org.springframework.beans.factory.annotation.Autowired;
+import org.springframework.stereotype.Component;
+
+import java.util.List;
+import java.util.stream.Collectors;
+
+@Component
+public class SampleComponent implements BrAPIComponent<Sample, SampleSearchRequest> {
+    private final SampleService sampleService;
+
+    @Autowired
+    public SampleComponent(SampleService sampleService) {
+        this.sampleService = sampleService;
+    }
+
+    @Override
+    public List<Sample> findEntities(SampleSearchRequest request, Metadata metadata) {
+        return sampleService.findSamples(request, metadata);
+    }
+
+    @Override
+    public BatchTypes getBatchType() {
+        return BatchTypes.SAMPLES;
+    }
+
+    @Override
+    public SearchRequestEntity.SearchRequestTypes getSearchType() {
+        return SearchRequestEntity.SearchRequestTypes.SAMPLES;
+    }
+
+    @Override
+    public List<String> collectDbIds(List<Sample> entities) {
+        return entities.stream().map(Sample::getSampleDbId).collect(Collectors.toList());
+    }
+
+    @Override
+    public void deleteBatchData(List<String> dbIds) {
+        sampleService.deleteSampleBatch(dbIds);
+    }
+
+    @Override
+    public void softDeleteBatchData(List<String> dbIds) {
+        sampleService.softDeleteSampleBatch(dbIds);
+    }
+}

src/main/java/org/brapi/test/BrAPITestServer/service/core/TrialService.java

@@ -155,6 +155,14 @@ public List<Trial> saveTrials(@Valid List<TrialNewRequest> body) throws BrAPISer
 		return savedTrials;
 	}
 
+	public void deleteTrialBatch(List<String> trialDbIds) {
+		trialRepository.deleteAllByIdInBatch(trialDbIds);
+	}
+
+	public void softDeleteTrialBatch(List<String> trialDbIds) {
+		trialRepository.updateSoftDeletedStatusBatch(trialDbIds, true);
+	}
+
 	public void softDeleteTrial(String trialDbId) throws BrAPIServerDbIdNotFoundException {
 		int updatedCount = trialRepository.updateSoftDeletedStatus(trialDbId, true);
 		if (updatedCount == 0) {

src/main/java/org/brapi/test/BrAPITestServer/service/geno/SampleService.java

@@ -147,6 +147,14 @@ public List<Sample> saveSamples(List<SampleNewRequest> body) throws BrAPIServerE
 		return savedSamples;
 	}
 
+	public void deleteSampleBatch(List<String> sampleDbIds) {
+		sampleRepository.deleteAllByIdInBatch(sampleDbIds);
+	}
+
+	public void softDeleteSampleBatch(List<String> sampleDbIds) {
+		sampleRepository.updateSoftDeletedStatusBatch(sampleDbIds, true);
+	}
+
 	public void deleteSample(String sampleDbId) throws BrAPIServerException {
 		sampleRepository.deleteAllByIdInBatch(Arrays.asList(sampleDbId));
 	}