Blame view

src/main/java/eu/omp/irap/vespa/epntapclient/RequestCtrl.java 4.28 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
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
/*
 * 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";

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

0d36b4b0   Nathanael Jourdane   Use RequestPanelC...
46
47
	private EpnTapController epnTapCtrl;

70b98e7e   Nathanael Jourdane   Add listeners.
48
49

	public RequestCtrl() {
0d36b4b0   Nathanael Jourdane   Use RequestPanelC...
50
	}
70b98e7e   Nathanael Jourdane   Add listeners.
51

0d36b4b0   Nathanael Jourdane   Use RequestPanelC...
52
53
	public RequestCtrl(EpnTapController epnTapCtrl) {
		this.epnTapCtrl = epnTapCtrl;
70b98e7e   Nathanael Jourdane   Add listeners.
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
	}

	/**
	 * @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;
	}

}