-
Notifications
You must be signed in to change notification settings - Fork 14
/
SparqlQueryHolder.java
128 lines (109 loc) · 4.31 KB
/
SparqlQueryHolder.java
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
120
121
122
123
124
125
126
127
128
/**
* Copyright (C) 2016 Czech Technical University in Prague
*
* 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 cz.cvut.kbss.jopa.query.sparql;
import cz.cvut.kbss.jopa.model.query.Parameter;
import cz.cvut.kbss.jopa.query.QueryHolder;
import cz.cvut.kbss.jopa.query.QueryParameter;
import java.util.*;
public class SparqlQueryHolder implements QueryHolder {
// Original query string
private final String query;
private final Map<Parameter<?>, QueryParameter<?>> parameterSet;
private final Map<Object, QueryParameter<?>> identifiersToParameters;
// These parameters are in order matching the query parts and can appear multiple times in the list
private final List<QueryParameter<?>> parameters;
private final List<String> queryParts;
public SparqlQueryHolder(String query, List<String> parts, List<QueryParameter<?>> parameters) {
this.query = query;
this.parameters = parameters;
this.queryParts = parts;
this.parameterSet = new HashMap<>();
parameters.forEach(p -> parameterSet.put(p, p));
this.identifiersToParameters = new HashMap<>(parameterSet.size());
parameterSet.values().forEach(p -> identifiersToParameters.put(p.getIdentifier(), p));
}
@Override
public String getQuery() {
return query;
}
@Override
public Set<Parameter<?>> getParameters() {
return Collections.unmodifiableSet(parameterSet.keySet());
}
@Override
public QueryParameter<?> getParameter(String name) {
if (!identifiersToParameters.containsKey(name)) {
throw unknownParameter(name);
}
return identifiersToParameters.get(name);
}
private static IllegalArgumentException unknownParameter(Object p) {
return new IllegalArgumentException("Parameter '" + p + "' does not exist in this query.");
}
@Override
public Parameter<?> getParameter(int position) {
if (!identifiersToParameters.containsKey(position)) {
throw unknownParameter(position);
}
return identifiersToParameters.get(position);
}
@Override
public Object getParameterValue(Parameter<?> parameter) {
assert getInternalParameter(parameter).getValue() != null;
return getInternalParameter(parameter).getValue().getValue();
}
private QueryParameter<?> getInternalParameter(Parameter<?> p) {
Objects.requireNonNull(p);
if (!parameterSet.containsKey(p)) {
throw unknownParameter(p);
}
return parameterSet.get(p);
}
@Override
public <T> void setParameter(Parameter<T> parameter, Object value) {
Objects.requireNonNull(value);
getInternalParameter(parameter).setValue(value);
}
@Override
public <T> void setParameter(Parameter<T> parameter, String value, String language) {
Objects.requireNonNull(value);
getInternalParameter(parameter).setValue(value, language);
}
@Override
public void clearParameter(Parameter<?> parameter) {
getInternalParameter(parameter).resetValue();
}
@Override
public void clearParameters() {
parameterSet.values().forEach(QueryParameter::resetValue);
}
@Override
public String assembleQuery() {
final StringBuilder sb = new StringBuilder();
for (int i = 0; i < parameters.size(); i++) {
sb.append(queryParts.get(i));
final String paramValue = parameters.get(i).getValue().getQueryString();
sb.append(paramValue);
}
if (queryParts.size() > parameters.size()) {
sb.append(queryParts.get(parameters.size()));
}
return sb.toString();
}
@Override
public String toString() {
return query;
}
}