RequestCtrl.java
4.28 KB
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
116
117
118
119
/*
* This file is a part of EpnTAPClient.
* This program aims to provide EPN-TAP support for software clients, like CASSIS spectrum analyzer.
* See draft specifications: https://voparis-confluence.obspm.fr/pages/viewpage.action?pageId=559861
* Copyright (C) 2016 Institut de Recherche en Astrophysique et Planétologie.
*
* This program is free software: you can
* redistribute it and/or modify it under the terms of the GNU 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 General Public License for more details. You should have received a copy of
* the GNU General Public License along with this program. If not, see
* <http://www.gnu.org/licenses/>.
*/
package eu.omp.irap.vespa.epntapclient;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.logging.Logger;
import com.google.gson.JsonArray;
import com.google.gson.JsonObject;
import eu.omp.irap.vespa.votable.utils.CantSendQueryException;
import eu.omp.irap.vespa.votable.utils.Network;
import eu.omp.irap.vespa.votable.utils.StringJoiner;
/**
* @author N. Jourdane
*/
public class RequestCtrl {
/** The logger for the class RequestCtrl. */
private static final Logger logger = Logger.getLogger(RequestCtrl.class.getName());
/** The URL of the resolver used for the `target name` field. */
private static final String RESOLVER_URL = "http://voparis-registry.obspm.fr/ssodnet/1/autocomplete";
/** The parameters fields for the request. */
protected Map<String, Object> paramValues = new HashMap<>();
private EpnTapController epnTapCtrl;
public RequestCtrl() {
}
public RequestCtrl(EpnTapController epnTapCtrl) {
this.epnTapCtrl = epnTapCtrl;
}
/**
* @param nbRow The maximum number of rows returned.
* @param tableName The name of the target table for the query.
* @param params A map of parameters, for the `WHERE` keywords.
* @return The literal string of the query.
*/
public static String getQuery(String tableName, Map<String, Object> params, int nbRow) {
StringJoiner addJoin = new StringJoiner(" AND ");
for (Map.Entry<String, Object> param : params.entrySet()) {
if (param.getValue() instanceof ArrayList) {
StringJoiner orJoin = new StringJoiner(" OR ");
@SuppressWarnings("unchecked")
List<String> possibleValues = (List<String>) param.getValue();
for (String possibleValue : possibleValues) {
orJoin.add(param.getKey() + " LIKE '" + possibleValue + "'");
}
addJoin.add("(" + orJoin + ")");
} else if (param.getValue() instanceof String) {
addJoin.add(param.getKey() + " LIKE '" + param.getValue() + "'");
} else {
addJoin.add(param.getKey() + " = " + param.getValue().toString());
}
}
String where = addJoin.isEmpty() ? "" : " WHERE " + addJoin;
return "SELECT TOP " + nbRow + " target_name, target_class FROM " + tableName + where;
}
/**
* The method used to get target names propositions by asking to the resolver.
*
* @param begining The beginning of the target_name.
* @return An array of Strings corresponding to the target names got.
* @throws CantSendQueryException
* @throws CantGetXMLException If the resolver do not work.
*/
public static String[] getTargetNames(String begining) throws CantSendQueryException {
Map<String, String> params = new HashMap<>();
params.put("q", "\"" + begining + "\"");
String query = Network.buildQuery(RESOLVER_URL, params);
JsonObject root = Network.readJson(query);
int count = Integer.parseInt(root.get("count").toString());
String[] targetNames = new String[count];
JsonArray hits = root.getAsJsonArray("hits");
for (int i = 0; i < count; i++) {
JsonObject elmt = hits.get(i).getAsJsonObject();
targetNames[i] = elmt.get("name").toString().replace("\"", "");
}
return targetNames;
}
public void updateParameter(String paramName, Object paramValue) {
paramValues.put(paramName, paramValue);
}
public void removeParameter(String paramName) {
paramValues.remove(paramName);
}
public Map<String, Object> getParamValues() {
return paramValues;
}
}