From e10c7fba2d7f5358c12b90308f68b84269d1a0c5 Mon Sep 17 00:00:00 2001
From: Paul-Christian Volkmer
Date: Tue, 14 Mar 2023 11:45:59 +0100
Subject: Issue #9: Entspricht der Code aus OS dem ATC-Schema, verwende
 Substanz als ATCCode

---
 src/main/java/ATCCodes/services/OnkostarAgentCodeService.java | 11 +++++++++--
 1 file changed, 9 insertions(+), 2 deletions(-)

(limited to 'src/main/java/ATCCodes/services')

diff --git a/src/main/java/ATCCodes/services/OnkostarAgentCodeService.java b/src/main/java/ATCCodes/services/OnkostarAgentCodeService.java
index f29f074..e0c75f4 100644
--- a/src/main/java/ATCCodes/services/OnkostarAgentCodeService.java
+++ b/src/main/java/ATCCodes/services/OnkostarAgentCodeService.java
@@ -1,6 +1,7 @@
 package ATCCodes.services;
 
 import ATCCodes.AgentCode;
+import ATCCodes.AtcCode;
 import ATCCodes.UnregisteredCode;
 import org.springframework.jdbc.core.JdbcTemplate;
 import org.springframework.stereotype.Service;
@@ -47,8 +48,14 @@ public class OnkostarAgentCodeService implements AgentCodeService {
         return jdbcTemplate.query(
                 sql,
                 new Object[]{query + "%", "%" + query + "%", "%" + query + "%"},
-                (resultSet, i) ->
-                        new UnregisteredCode(resultSet.getString("code"), resultSet.getString("shortdesc"))
+                (resultSet, i) -> {
+                    var code = resultSet.getString("code");
+                    var shortdesc = resultSet.getString("shortdesc");
+                    if (AtcCode.isAtcCode(code)) {
+                        return new AtcCode(code, shortdesc);
+                    }
+                    return new UnregisteredCode(code, shortdesc);
+                }
         );
     }
 }
\ No newline at end of file
-- 
cgit v1.2.3