feat: add performance status mapping

src/main/java/dev/pcvolkmer/onco/datamapper/mapper/KpaEcogDataMapper.java

@@ -0,0 +1,78 @@
+package dev.pcvolkmer.onco.datamapper.mapper;
+
+import dev.pcvolkmer.mv64e.mtb.EcogCoding;
+import dev.pcvolkmer.mv64e.mtb.EcogCodingCode;
+import dev.pcvolkmer.mv64e.mtb.PerformanceStatus;
+import dev.pcvolkmer.mv64e.mtb.Reference;
+import dev.pcvolkmer.onco.datamapper.ResultSet;
+import dev.pcvolkmer.onco.datamapper.datacatalogues.EcogCatalogue;
+
+import java.io.IOException;
+import java.util.Arrays;
+import java.util.Comparator;
+import java.util.List;
+import java.util.stream.Collectors;
+
+/**
+ * Mapper class to load and map prozedur data from database table 'dk_dnpm_uf_ecog'
+ *
+ * @author Paul-Christian Volkmer
+ * @since 0.1
+ */
+public class KpaEcogDataMapper implements SubformDataMapper<PerformanceStatus> {
+
+    private final EcogCatalogue catalogue;
+
+    public KpaEcogDataMapper(final EcogCatalogue catalogue) {
+        this.catalogue = catalogue;
+    }
+
+    /**
+     * Loads and maps Prozedur related by database id
+     *
+     * @param id The database id of the procedure data set
+     * @return The loaded data set
+     */
+    @Override
+    public PerformanceStatus getById(final int id) {
+        var data = catalogue.getById(id);
+        return this.map(data);
+    }
+
+    @Override
+    public List<PerformanceStatus> getByParentId(final int parentId) {
+        return catalogue.getAllByParentId(parentId)
+                .stream()
+                .map(this::map)
+                .sorted(Comparator.comparing(PerformanceStatus::getEffectiveDate))
+                .collect(Collectors.toList());
+    }
+
+    private PerformanceStatus map(final ResultSet resultSet) {
+        var builder = PerformanceStatus.builder();
+        builder
+                .id(resultSet.getProcedureId().toString())
+                .patient(Reference.builder().id(resultSet.getString("patient_id")).build())
+                .effectiveDate(resultSet.getDate("datum"))
+                .value(getEcogCoding(resultSet.getString("ecog")))
+        ;
+
+        return builder.build();
+    }
+
+    private EcogCoding getEcogCoding(final String value) {
+        if (value == null || !Arrays.stream(EcogCodingCode.values()).map(EcogCodingCode::toValue).collect(Collectors.toSet()).contains(value)) {
+            return null;
+        }
+
+        var resultBuilder = EcogCoding.builder();
+        try {
+            resultBuilder.code(EcogCodingCode.forValue(value));
+        } catch (IOException e) {
+            throw new IllegalStateException("No valid code found");
+        }
+
+        return resultBuilder.build();
+    }
+
+}

src/main/java/dev/pcvolkmer/onco/datamapper/mapper/KpaTumorausbreitungDataMapper.java

@@ -46,12 +46,6 @@ public class KpaTumorausbreitungDataMapper implements SubformDataMapper<TumorSta
     }
 
     private TumorStaging map(final ResultSet resultSet) {
-        var diseases = catalogue.getDiseases(resultSet.getProcedureId());
-
-        if (diseases.size() != 1) {
-            throw new IllegalStateException(String.format("No unique disease for procedure %s", resultSet.getProcedureId()));
-        }
-
         var builder = TumorStaging.builder();
         builder
                 .date(resultSet.getDate("zeitpunkt"))
@@ -60,7 +54,6 @@ public class KpaTumorausbreitungDataMapper implements SubformDataMapper<TumorSta
                 .tnmClassification(getTnmClassification(resultSet))
         ;
 
-
         return builder.build();
     }
 

src/test/java/dev/pcvolkmer/onco/datamapper/mapper/KpaEcogDataMapperTest.java

@@ -0,0 +1,75 @@
+package dev.pcvolkmer.onco.datamapper.mapper;
+
+import dev.pcvolkmer.mv64e.mtb.EcogCoding;
+import dev.pcvolkmer.mv64e.mtb.EcogCodingCode;
+import dev.pcvolkmer.mv64e.mtb.PerformanceStatus;
+import dev.pcvolkmer.mv64e.mtb.Reference;
+import dev.pcvolkmer.onco.datamapper.ResultSet;
+import dev.pcvolkmer.onco.datamapper.datacatalogues.EcogCatalogue;
+import org.junit.jupiter.api.BeforeEach;
+import org.junit.jupiter.api.Test;
+import org.junit.jupiter.api.extension.ExtendWith;
+import org.mockito.Mock;
+import org.mockito.junit.jupiter.MockitoExtension;
+
+import java.time.Instant;
+import java.util.Date;
+import java.util.List;
+import java.util.Map;
+
+import static org.assertj.core.api.Assertions.assertThat;
+import static org.mockito.ArgumentMatchers.anyInt;
+import static org.mockito.ArgumentMatchers.anyString;
+import static org.mockito.Mockito.doAnswer;
+import static org.mockito.Mockito.when;
+
+@ExtendWith(MockitoExtension.class)
+class KpaEcogDataMapperTest {
+
+    EcogCatalogue catalogue;
+
+    KpaEcogDataMapper dataMapper;
+
+    @BeforeEach
+    void setUp(@Mock EcogCatalogue catalogue) {
+        this.catalogue = catalogue;
+        this.dataMapper = new KpaEcogDataMapper(catalogue);
+    }
+
+    @Test
+    void shouldMapResultSet(@Mock ResultSet resultSet) {
+        var testData = Map.of(
+                "id", "1",
+                "patient_id", "42",
+                "datum", new java.sql.Date(Date.from(Instant.parse("2000-01-01T12:00:00Z")).getTime()),
+                "ecog", "1"
+        );
+
+        doAnswer(invocationOnMock -> {
+            var columnName = invocationOnMock.getArgument(0, String.class);
+            return testData.get(columnName);
+        }).when(resultSet).getString(anyString());
+
+        doAnswer(invocationOnMock -> {
+            var columnName = invocationOnMock.getArgument(0, String.class);
+            return testData.get(columnName);
+        }).when(resultSet).getDate(anyString());
+
+        when(resultSet.getProcedureId()).thenReturn(1);
+
+        doAnswer(invocationOnMock -> List.of(resultSet))
+                .when(catalogue)
+                .getAllByParentId(anyInt());
+
+        var actualList = this.dataMapper.getByParentId(1);
+        assertThat(actualList).hasSize(1);
+
+        var actual = actualList.get(0);
+        assertThat(actual).isInstanceOf(PerformanceStatus.class);
+        assertThat(actual.getId()).isEqualTo("1");
+        assertThat(actual.getPatient()).isEqualTo(Reference.builder().id("42").build());
+        assertThat(actual.getEffectiveDate()).isEqualTo(new java.sql.Date(Date.from(Instant.parse("2000-01-01T12:00:00Z")).getTime()));
+        assertThat(actual.getValue()).isEqualTo(EcogCoding.builder().code(EcogCodingCode.CODE_1).build());
+    }
+
+}