summaryrefslogtreecommitdiff
path: root/src/main/java/dev/dnpm/oshelper/atc/services/OnkostarAgentCodeService.java
blob: 90528aaf029be4ca9d6a1baa67483ca771f19ef1 (plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
/*
 * This file is part of onkostar-plugin-dnpm
 *
 * Copyright (C) 2023-2026 the original author or authors.
 *
 * This program is free software: you can redistribute it and/or modify
 * it under the terms of the GNU Lesser General Public License as published by
 * the Free Software Foundation, either version 3 of the License, or
 * (at your option) any later version.
 *
 * This program is distributed in the hope that it will be useful,
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
 * GNU Lesser General Public License for more details.
 *
 * You should have received a copy of the GNU Lesser General Public License
 * along with this program.  If not, see <http://www.gnu.org/licenses/>.
 */

package dev.dnpm.oshelper.atc.services;

import dev.dnpm.oshelper.atc.AgentCode;
import dev.dnpm.oshelper.atc.AtcCode;
import dev.dnpm.oshelper.atc.UnregisteredCode;
import java.util.List;
import java.util.Optional;
import javax.sql.DataSource;
import org.springframework.jdbc.core.JdbcTemplate;

/**
 * Implementation of {@link AgentCodeService} that uses database to query for unregistered agents
 *
 * @author Paul-Christian Volkmer
 * @since 2.0.0
 */
public class OnkostarAgentCodeService implements AgentCodeService {

  private final JdbcTemplate jdbcTemplate;

  public OnkostarAgentCodeService(final DataSource dataSource) {
    this.jdbcTemplate = new JdbcTemplate(dataSource);
  }

  /**
   * Queries source for agents code starting with or name containing query string. If size is zero,
   * all available results will be returned.
   *
   * @param query The query string
   * @param size Maximal amount of responses
   * @return A list with agent codes
   */
  @Override
  public List<AgentCode> findAgentCodes(final String query, final int size) {
    var sql =
        "SELECT code, shortdesc, synonyms\n"
            + "    FROM property_catalogue\n"
            + "    JOIN property_catalogue_version ON (property_catalogue_version.datacatalog_id = property_catalogue.id)\n"
            + "    JOIN property_catalogue_version_entry p ON (p.property_version_id = property_catalogue_version.id)\n"
            + "    WHERE name = 'OS.Substanzen'\n"
            + "    AND (LOWER(code) LIKE ? OR LOWER(shortdesc) LIKE ? OR LOWER(synonyms) LIKE ?)";

    if (size > 0) {
      sql = sql + " LIMIT " + size;
    }

    return jdbcTemplate.query(
        sql,
        new Object[] {query + "%", "%" + query + "%", "%" + query + "%"},
        (resultSet, i) -> {
          var code = resultSet.getString("code");
          var shortdesc = resultSet.getString("shortdesc");
          var synonyms = resultSet.getString("synonyms");
          if (AtcCode.isAtcCode(code)) {
            return new AtcCode(code, shortdesc);
          }
          return new UnregisteredCode(code, shortdesc, synonyms);
        });
  }

  /**
   * Queries source for agents with code matching code string.
   *
   * @param code The code string
   * @return An optional agent code if found
   */
  @Override
  public Optional<AgentCode> getAgentCode(final String code) {
    var sql =
        "SELECT code, shortdesc, synonyms\n"
            + "    FROM property_catalogue\n"
            + "    JOIN property_catalogue_version ON (property_catalogue_version.datacatalog_id = property_catalogue.id)\n"
            + "    JOIN property_catalogue_version_entry p ON (p.property_version_id = property_catalogue_version.id)\n"
            + "    WHERE name = 'OS.Substanzen'\n"
            + "    AND LOWER(code) = ?";

    final var result =
        jdbcTemplate.query(
            sql,
            new Object[] {code},
            (resultSet, i) -> {
              var c = resultSet.getString("code");
              var shortdesc = resultSet.getString("shortdesc");
              var synonyms = resultSet.getString("synonyms");
              if (AtcCode.isAtcCode(c)) {
                return new AtcCode(c, shortdesc);
              }
              return new UnregisteredCode(c, shortdesc, synonyms);
            });

    if (result.size() == 1) {
      return Optional.of(result.get(0));
    }
    return Optional.empty();
  }
}