1 |
|
|
2 |
|
|
3 |
|
|
4 |
|
|
5 |
|
|
6 |
|
|
7 |
|
|
8 |
|
|
9 |
|
|
10 |
|
|
11 |
|
|
12 |
|
|
13 |
|
|
14 |
|
|
15 |
|
|
16 |
|
|
17 |
|
|
18 |
|
package org.apache.camel.component.jpa; |
19 |
|
|
20 |
|
import javax.persistence.EntityManager; |
21 |
|
import javax.persistence.Query; |
22 |
|
import java.util.Arrays; |
23 |
|
import java.util.Collection; |
24 |
|
import java.util.Map; |
25 |
|
import java.util.Set; |
26 |
|
|
27 |
|
|
28 |
|
|
29 |
|
|
30 |
|
|
31 |
|
|
32 |
3 |
public abstract class QueryBuilder implements QueryFactory { |
33 |
|
ParameterBuilder parameterBuilder; |
34 |
|
|
35 |
|
|
36 |
|
|
37 |
|
|
38 |
|
|
39 |
|
|
40 |
|
|
41 |
|
public static QueryBuilder query(final String query) { |
42 |
2 |
return new QueryBuilder() { |
43 |
|
protected Query makeQueryObject(EntityManager entityManager) { |
44 |
2 |
return entityManager.createQuery(query); |
45 |
|
} |
46 |
|
|
47 |
|
@Override |
48 |
2 |
public String toString() { |
49 |
0 |
return "Query: " + query + " params: " + getParameterDescription(); |
50 |
|
} |
51 |
|
}; |
52 |
|
} |
53 |
|
|
54 |
|
|
55 |
|
|
56 |
|
|
57 |
|
public static QueryBuilder namedQuery(final String namedQuery) { |
58 |
1 |
return new QueryBuilder() { |
59 |
|
protected Query makeQueryObject(EntityManager entityManager) { |
60 |
3 |
return entityManager.createNamedQuery(namedQuery); |
61 |
|
} |
62 |
|
|
63 |
|
@Override |
64 |
1 |
public String toString() { |
65 |
0 |
return "Named: " + namedQuery + getParameterDescription(); |
66 |
|
} |
67 |
|
}; |
68 |
|
} |
69 |
|
|
70 |
|
|
71 |
|
|
72 |
|
|
73 |
|
public static QueryBuilder nativeQuery(final String nativeQuery) { |
74 |
0 |
return new QueryBuilder() { |
75 |
|
protected Query makeQueryObject(EntityManager entityManager) { |
76 |
0 |
return entityManager.createNativeQuery(nativeQuery); |
77 |
|
} |
78 |
|
|
79 |
|
@Override |
80 |
0 |
public String toString() { |
81 |
0 |
return "NativeQuery: " + nativeQuery + getParameterDescription(); |
82 |
|
} |
83 |
|
}; |
84 |
|
} |
85 |
|
|
86 |
|
|
87 |
|
|
88 |
|
|
89 |
|
|
90 |
|
|
91 |
|
|
92 |
|
public QueryBuilder parameters(Object... parameters) { |
93 |
0 |
return parameters(Arrays.asList(parameters)); |
94 |
|
} |
95 |
|
|
96 |
|
|
97 |
|
|
98 |
|
|
99 |
|
|
100 |
|
|
101 |
|
|
102 |
|
public QueryBuilder parameters(final Collection parameters) { |
103 |
0 |
checkNoParametersConfigured(); |
104 |
0 |
parameterBuilder = new ParameterBuilder() { |
105 |
|
public void populateQuery(EntityManager entityManager, Query query) { |
106 |
0 |
int counter = 0; |
107 |
0 |
for (Object parameter : parameters) { |
108 |
0 |
query.setParameter(counter++, parameter); |
109 |
0 |
} |
110 |
0 |
} |
111 |
|
|
112 |
|
@Override |
113 |
0 |
public String toString() { |
114 |
0 |
return "Parameters: " + parameters; |
115 |
|
} |
116 |
|
}; |
117 |
0 |
return this; |
118 |
|
} |
119 |
|
|
120 |
|
|
121 |
|
|
122 |
|
|
123 |
|
|
124 |
|
|
125 |
|
|
126 |
|
public QueryBuilder parameters(final Map<String, Object> parameterMap) { |
127 |
0 |
checkNoParametersConfigured(); |
128 |
0 |
parameterBuilder = new ParameterBuilder() { |
129 |
|
public void populateQuery(EntityManager entityManager, Query query) { |
130 |
0 |
Set<Map.Entry<String, Object>> entries = parameterMap.entrySet(); |
131 |
0 |
for (Map.Entry<String, Object> entry : entries) { |
132 |
0 |
query.setParameter(entry.getKey(), entry.getValue()); |
133 |
0 |
} |
134 |
0 |
} |
135 |
|
|
136 |
|
@Override |
137 |
0 |
public String toString() { |
138 |
0 |
return "Parameters: " + parameterMap; |
139 |
|
} |
140 |
|
}; |
141 |
0 |
return this; |
142 |
|
} |
143 |
|
|
144 |
|
protected void checkNoParametersConfigured() { |
145 |
0 |
if (parameterBuilder != null) { |
146 |
0 |
throw new IllegalArgumentException("Cannot add parameters to a QueryBuilder which already has parameters configured"); |
147 |
|
} |
148 |
0 |
} |
149 |
|
|
150 |
|
public Query createQuery(EntityManager entityManager) { |
151 |
5 |
Query query = makeQueryObject(entityManager); |
152 |
5 |
populateQuery(entityManager, query); |
153 |
5 |
return query; |
154 |
|
} |
155 |
|
|
156 |
|
protected String getParameterDescription() { |
157 |
0 |
if (parameterBuilder == null) { |
158 |
0 |
return ""; |
159 |
|
} |
160 |
|
else { |
161 |
0 |
return " " + parameterBuilder.toString(); |
162 |
|
} |
163 |
|
} |
164 |
|
|
165 |
|
protected void populateQuery(EntityManager entityManager, Query query) { |
166 |
5 |
if (parameterBuilder != null) { |
167 |
0 |
parameterBuilder.populateQuery(entityManager, query); |
168 |
|
} |
169 |
5 |
} |
170 |
|
|
171 |
|
protected abstract Query makeQueryObject(EntityManager entityManager); |
172 |
|
|
173 |
|
|
174 |
|
|
175 |
|
|
176 |
0 |
protected abstract static class ParameterBuilder { |
177 |
|
public abstract void populateQuery(EntityManager entityManager, Query query); |
178 |
|
} |
179 |
|
} |