diff options
Diffstat (limited to 'src/main/java/ATCCodes/services/OnkostarAgentCodeService.java')
| -rw-r--r-- | src/main/java/ATCCodes/services/OnkostarAgentCodeService.java | 11 |
1 files changed, 9 insertions, 2 deletions
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 |