Blame view

src/main/java/eu/omp/irap/vespa/epntapclient/RequestCtrl.java 4.06 KB
70b98e7e   Nathanael Jourdane   Add listeners.
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
/*
 * 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;
70b98e7e   Nathanael Jourdane   Add listeners.
23
24
25
26
27
28
29
30
31
32
33
34
35

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 {

70b98e7e   Nathanael Jourdane   Add listeners.
36
37
38
	/** 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";

0d36b4b0   Nathanael Jourdane   Use RequestPanelC...
39
	/** The parameters fields for the request. */
70b98e7e   Nathanael Jourdane   Add listeners.
40
41
	protected Map<String, Object> paramValues = new HashMap<>();

70b98e7e   Nathanael Jourdane   Add listeners.
42
43

	public RequestCtrl() {
5672c106   Nathanael Jourdane   Fix Sonar issues.
44
		/* Subclasses initializes their own attributes, we don't want to initialize them twice. */
70b98e7e   Nathanael Jourdane   Add listeners.
45
46
47
48
49
50
51
52
	}

	/**
	 * @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.
	 */
5672c106   Nathanael Jourdane   Fix Sonar issues.
53
	public String getQuery(String tableName, int nbRow) {
70b98e7e   Nathanael Jourdane   Add listeners.
54
		StringJoiner addJoin = new StringJoiner(" AND ");
5672c106   Nathanael Jourdane   Fix Sonar issues.
55
		for (Map.Entry<String, Object> param : paramValues.entrySet()) {
70b98e7e   Nathanael Jourdane   Add listeners.
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
			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;
5672c106   Nathanael Jourdane   Fix Sonar issues.
71

70b98e7e   Nathanael Jourdane   Add listeners.
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
		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;
	}

}