Add implementation of QuantityDataRepository#createReferenceValueOutp…

…uts()
52North · Jan 19, 2018 · 6213bcb · 6213bcb
1 parent 1f2b9f3
commit 6213bcb
Showing 1 changed file with 20 additions and 0 deletions.
diff --git a/dao/src/main/java/org/n52/series/db/da/QuantityDataRepository.java b/dao/src/main/java/org/n52/series/db/da/QuantityDataRepository.java
@@ -31,6 +31,7 @@
 
 import java.math.BigDecimal;
 import java.math.RoundingMode;
+import java.util.ArrayList;
 import java.util.Date;
 import java.util.HashMap;
 import java.util.List;
@@ -41,8 +42,10 @@
 import org.n52.io.request.IoParameters;
 import org.n52.io.response.dataset.Data;
 import org.n52.io.response.dataset.DatasetMetadata;
+import org.n52.io.response.dataset.ReferenceValueOutput;
 import org.n52.io.response.dataset.quantity.QuantityValue;
 import org.n52.series.db.DataAccessException;
+import org.n52.series.db.beans.ProcedureEntity;
 import org.n52.series.db.beans.QuantityDataEntity;
 import org.n52.series.db.beans.QuantityDatasetEntity;
 import org.n52.series.db.beans.ServiceEntity;
@@ -57,6 +60,23 @@ public Class<QuantityDatasetEntity> getDatasetEntityType() {
         return QuantityDatasetEntity.class;
     }
 
+    @Override
+    public ReferenceValueOutput<QuantityValue>[] createReferenceValueOutputs(QuantityDatasetEntity datasetEntity, DbQuery query) {
+        Set<QuantityDatasetEntity> referenceValues = datasetEntity.getReferenceValues();
+        List<ReferenceValueOutput<QuantityValue>> outputs = new ArrayList<>();
+        for (QuantityDatasetEntity referenceSeriesEntity : referenceValues) {
+            ReferenceValueOutput<QuantityValue> refenceValueOutput = new ReferenceValueOutput<>();
+            ProcedureEntity procedure = referenceSeriesEntity.getProcedure();
+            refenceValueOutput.setLabel(procedure.getNameI18n(query.getLocale()));
+            refenceValueOutput.setReferenceValueId(referenceSeriesEntity.getPkid().toString());
+
+            QuantityDataEntity lastValue = referenceSeriesEntity.getLastValue();
+            refenceValueOutput.setLastValue(createSeriesValueFor(lastValue, referenceSeriesEntity, query));
+            outputs.add(refenceValueOutput);
+        }
+        return outputs.toArray(new ReferenceValueOutput[outputs.size()]);
+    }
+
     @Override
     protected Data<QuantityValue> assembleDataWithReferenceValues(QuantityDatasetEntity timeseries,
                                                            DbQuery dbQuery,