svn-gvsig-desktop / trunk / org.gvsig.desktop / org.gvsig.desktop.compat.cdc / org.gvsig.fmap.dal / org.gvsig.fmap.dal.impl / src / test / java / org / gvsig / fmap / dal / feature / BaseTestFeatureStore.java @ 40435
History | View | Annotate | Download (51.6 KB)
1 | 40435 | jjdelcerro | /* gvSIG. Geographic Information System of the Valencian Government
|
---|---|---|---|
2 | *
|
||
3 | * Copyright (C) 2007-2008 Infrastructures and Transports Department
|
||
4 | * of the Valencian Government (CIT)
|
||
5 | *
|
||
6 | * This program is free software; you can redistribute it and/or
|
||
7 | * modify it under the terms of the GNU General Public License
|
||
8 | * as published by the Free Software Foundation; either version 2
|
||
9 | * of the License, or (at your option) any later version.
|
||
10 | *
|
||
11 | * This program is distributed in the hope that it will be useful,
|
||
12 | * but WITHOUT ANY WARRANTY; without even the implied warranty of
|
||
13 | * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the
|
||
14 | * GNU General Public License for more details.
|
||
15 | *
|
||
16 | * You should have received a copy of the GNU General Public License
|
||
17 | * along with this program; if not, write to the Free Software
|
||
18 | * Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston,
|
||
19 | * MA 02110-1301, USA.
|
||
20 | *
|
||
21 | */
|
||
22 | |||
23 | /*
|
||
24 | * AUTHORS (In addition to CIT):
|
||
25 | * 2009 IVER T.I. S.A. {{Task}}
|
||
26 | */
|
||
27 | |||
28 | /**
|
||
29 | *
|
||
30 | */
|
||
31 | package org.gvsig.fmap.dal.feature; |
||
32 | |||
33 | import java.util.ArrayList; |
||
34 | import java.util.Collections; |
||
35 | import java.util.Iterator; |
||
36 | import java.util.List; |
||
37 | import java.util.Random; |
||
38 | import java.util.TreeSet; |
||
39 | |||
40 | import org.gvsig.fmap.dal.DALLocator; |
||
41 | import org.gvsig.fmap.dal.DataManager; |
||
42 | import org.gvsig.fmap.dal.DataServerExplorer; |
||
43 | import org.gvsig.fmap.dal.DataStoreParameters; |
||
44 | import org.gvsig.fmap.dal.DataTypes; |
||
45 | import org.gvsig.fmap.dal.exception.DataEvaluatorRuntimeException; |
||
46 | import org.gvsig.fmap.dal.exception.DataException; |
||
47 | import org.gvsig.fmap.dal.feature.FeatureQueryOrder.FeatureQueryOrderMember; |
||
48 | import org.gvsig.fmap.dal.feature.testmulithread.StoreTask; |
||
49 | import org.gvsig.fmap.dal.resource.ResourceManager; |
||
50 | import org.gvsig.fmap.geom.Geometry; |
||
51 | import org.gvsig.tools.ToolsLocator; |
||
52 | import org.gvsig.tools.dispose.DisposableIterator; |
||
53 | import org.gvsig.tools.dynobject.DynClass; |
||
54 | import org.gvsig.tools.dynobject.DynField; |
||
55 | import org.gvsig.tools.dynobject.DynObject; |
||
56 | import org.gvsig.tools.evaluator.Evaluator; |
||
57 | import org.gvsig.tools.evaluator.EvaluatorData; |
||
58 | import org.gvsig.tools.evaluator.EvaluatorException; |
||
59 | import org.gvsig.tools.evaluator.EvaluatorFieldsInfo; |
||
60 | import org.gvsig.tools.junit.AbstractLibraryAutoInitTestCase; |
||
61 | import org.gvsig.tools.persistence.PersistentState; |
||
62 | import org.slf4j.Logger; |
||
63 | import org.slf4j.LoggerFactory; |
||
64 | |||
65 | /**
|
||
66 | * @author jmvivo
|
||
67 | *
|
||
68 | */
|
||
69 | public abstract class BaseTestFeatureStore extends |
||
70 | AbstractLibraryAutoInitTestCase { |
||
71 | |||
72 | private static Logger logger = null; |
||
73 | |||
74 | protected DataManager dataManager = null; |
||
75 | private static Random rnd; |
||
76 | |||
77 | public Logger getLogger() { |
||
78 | if (logger == null) { |
||
79 | logger = LoggerFactory.getLogger(this.getClass());
|
||
80 | } |
||
81 | return logger;
|
||
82 | } |
||
83 | |||
84 | public abstract boolean usesResources(); |
||
85 | |||
86 | public abstract boolean hasExplorer(); |
||
87 | |||
88 | public FeatureQuery getDefaultQuery(FeatureStore store)
|
||
89 | throws DataException {
|
||
90 | FeatureQuery query = store.createFeatureQuery(); |
||
91 | FeatureAttributeDescriptor[] key = store.getDefaultFeatureType()
|
||
92 | .getPrimaryKey(); |
||
93 | for (int i = 0; i < key.length; i++) { |
||
94 | query.getOrder().add(key[i].getName(), true);
|
||
95 | } |
||
96 | |||
97 | return query;
|
||
98 | } |
||
99 | |||
100 | |||
101 | public abstract DataStoreParameters getDefaultDataStoreParameters() |
||
102 | throws DataException;
|
||
103 | |||
104 | |||
105 | protected void doSetUp() throws Exception { |
||
106 | dataManager = DALLocator.getDataManager(); |
||
107 | } |
||
108 | |||
109 | |||
110 | //=================================================
|
||
111 | |||
112 | |||
113 | public void printFeature(Feature feature, int maxColSize) { |
||
114 | printFeature(feature, true, maxColSize);
|
||
115 | } |
||
116 | |||
117 | |||
118 | public void printFeature(Feature feature, boolean showColName,int maxColSize) { |
||
119 | FeatureType fType = feature.getType(); |
||
120 | if (showColName){
|
||
121 | this.printFeatureTypeColNames(feature.getType(), maxColSize);
|
||
122 | } |
||
123 | StringBuffer row = new StringBuffer(); |
||
124 | Iterator iter = fType.iterator();
|
||
125 | FeatureAttributeDescriptor attr; |
||
126 | |||
127 | while (iter.hasNext()) {
|
||
128 | attr = (FeatureAttributeDescriptor) iter.next(); |
||
129 | row.append(truncateOrFillString(feature.get(attr.getName()), |
||
130 | maxColSize + 1, ' ')); |
||
131 | } |
||
132 | System.out.println(row.toString());
|
||
133 | } |
||
134 | |||
135 | public String truncateOrFillString(Object str, int max, char fillWith) { |
||
136 | if (str == null) { |
||
137 | return truncateOrFillString("{null}", max, fillWith); |
||
138 | } |
||
139 | return truncateOrFillString(str.toString(), max, fillWith);
|
||
140 | } |
||
141 | |||
142 | public String truncateOrFillString(String str, int max, char fillWith) { |
||
143 | if (str.length() > max) {
|
||
144 | return str.substring(0, max - 1) + " "; |
||
145 | } else {
|
||
146 | StringBuffer strB = new StringBuffer(str); |
||
147 | while (strB.length() < max) {
|
||
148 | strB.append(fillWith); |
||
149 | } |
||
150 | return strB.toString();
|
||
151 | } |
||
152 | |||
153 | } |
||
154 | |||
155 | public void printFeatureTypeColNames(FeatureType fType, int maxColSize) { |
||
156 | Iterator iter = fType.iterator();
|
||
157 | FeatureAttributeDescriptor attr; |
||
158 | StringBuffer colNames = new StringBuffer(); |
||
159 | StringBuffer typeNames = new StringBuffer(); |
||
160 | StringBuffer sep = new StringBuffer(); |
||
161 | if (maxColSize < 1){ |
||
162 | maxColSize = 15;
|
||
163 | } |
||
164 | while (iter.hasNext()) {
|
||
165 | attr = (FeatureAttributeDescriptor) iter.next(); |
||
166 | colNames.append(truncateOrFillString(attr.getName(), maxColSize + 1, ' ')); |
||
167 | typeNames.append(truncateOrFillString("(" + attr.getDataTypeName() + ")", |
||
168 | maxColSize + 1, ' ')); |
||
169 | sep.append(truncateOrFillString("", maxColSize, '=')); |
||
170 | sep.append(' ');
|
||
171 | } |
||
172 | |||
173 | System.out.println(""); |
||
174 | System.out.println(""); |
||
175 | System.out.println(colNames.toString());
|
||
176 | System.out.println(typeNames.toString());
|
||
177 | System.out.println(sep.toString());
|
||
178 | } |
||
179 | |||
180 | protected FeatureAttributeDescriptor getFirstAttributeOfType(
|
||
181 | FeatureType ftype, int dataType) {
|
||
182 | return getFirstAttributeOfType(ftype, new int[] { dataType }); |
||
183 | } |
||
184 | |||
185 | protected FeatureAttributeDescriptor getFirstAttributeOfType(
|
||
186 | FeatureType ftype, int[] dataTypes) { |
||
187 | FeatureAttributeDescriptor attr; |
||
188 | Iterator iter = ftype.iterator();
|
||
189 | int i;
|
||
190 | while (iter.hasNext()) {
|
||
191 | attr = (FeatureAttributeDescriptor) iter.next(); |
||
192 | for (i = 0; i < dataTypes.length; i++) { |
||
193 | if (attr.getType() == dataTypes[i]) {
|
||
194 | return attr;
|
||
195 | } |
||
196 | } |
||
197 | } |
||
198 | return null; |
||
199 | } |
||
200 | |||
201 | protected boolean compareDynObject(DynObject obj1, DynObject obj2) { |
||
202 | DynClass dynClass = obj1.getDynClass(); |
||
203 | if (!dynClass.getName().equals(obj2.getDynClass().getName())) {
|
||
204 | return false; |
||
205 | } |
||
206 | |||
207 | DynField[] fields = dynClass.getDeclaredDynFields();
|
||
208 | String fieldName;
|
||
209 | Object v1, v2;
|
||
210 | for (int i = 0; i < fields.length; i++) { |
||
211 | fieldName = fields[i].getName(); |
||
212 | v1 = obj1.getDynValue(fieldName); |
||
213 | v2 = obj2.getDynValue(fieldName); |
||
214 | if (v1 == v2) {
|
||
215 | continue;
|
||
216 | } else if (v1 != null) { |
||
217 | if (!v1.equals(v2)) {
|
||
218 | return false; |
||
219 | } |
||
220 | } |
||
221 | } |
||
222 | |||
223 | return true; |
||
224 | } |
||
225 | |||
226 | protected boolean compareStores(FeatureStore store1, FeatureStore store2) |
||
227 | throws DataException {
|
||
228 | if (store1.getParameters().getClass() != store2.getParameters()
|
||
229 | .getClass()) { |
||
230 | return false; |
||
231 | } |
||
232 | if (!compareDynObject(store1.getParameters(), store2.getParameters())) {
|
||
233 | return false; |
||
234 | } |
||
235 | |||
236 | if (store1.getEnvelope() != store2.getEnvelope()) {
|
||
237 | if (store1.getEnvelope() != null) { |
||
238 | return store1.getEnvelope().equals(store2.getEnvelope());
|
||
239 | } else {
|
||
240 | return false; |
||
241 | } |
||
242 | } |
||
243 | |||
244 | if (!store1.getName().equals(store2.getName())) {
|
||
245 | return false; |
||
246 | } |
||
247 | if (((FeatureSelection) store1.getSelection()).getSize() != ((FeatureSelection) store2
|
||
248 | .getSelection()).getSize()) { |
||
249 | return false; |
||
250 | } |
||
251 | DisposableIterator iter1 = ((FeatureSelection) store1.getSelection()) |
||
252 | .fastIterator(); |
||
253 | DisposableIterator iter2 = ((FeatureSelection) store2.getSelection()) |
||
254 | .fastIterator(); |
||
255 | if (!compareFeatureIterators(iter1, iter2)) {
|
||
256 | return false; |
||
257 | } |
||
258 | iter1.dispose(); |
||
259 | iter2.dispose(); |
||
260 | |||
261 | if (store1.getFeatureTypes().size() != store2.getFeatureTypes().size()) {
|
||
262 | return false; |
||
263 | } |
||
264 | Iterator iterTypes1 = store1.getFeatureTypes().iterator();
|
||
265 | Iterator iterTypes2 = store2.getFeatureTypes().iterator();
|
||
266 | while (iterTypes1.hasNext()) {
|
||
267 | if (!compareTypes((FeatureType) iterTypes1.next(),
|
||
268 | (FeatureType) iterTypes2 |
||
269 | .next())) { |
||
270 | return false; |
||
271 | } |
||
272 | } |
||
273 | if (!compareTypes(store1.getDefaultFeatureType(), store2
|
||
274 | .getDefaultFeatureType())) { |
||
275 | return false; |
||
276 | } |
||
277 | |||
278 | if (store1.getLocks() != null) { |
||
279 | if (store1.getLocks().getLocksCount() != store2.getLocks()
|
||
280 | .getLocksCount()) { |
||
281 | return false; |
||
282 | } |
||
283 | if (!compareFeatureIterators(store1.getLocks().getLocks(), store2
|
||
284 | .getLocks().getLocks())) { |
||
285 | return false; |
||
286 | } |
||
287 | |||
288 | } else if (store2.getLocks() != null) { |
||
289 | return false; |
||
290 | } |
||
291 | |||
292 | return true; |
||
293 | } |
||
294 | |||
295 | protected boolean compareFeatureIterators(Iterator iter1, Iterator iter2) { |
||
296 | Feature feature; |
||
297 | Feature ffeature; |
||
298 | while (iter1.hasNext()) {
|
||
299 | feature = (Feature) iter1.next(); |
||
300 | ffeature = (Feature) iter2.next(); |
||
301 | if (!this.compareFeatures(feature, ffeature)) { |
||
302 | return false; |
||
303 | } |
||
304 | } |
||
305 | |||
306 | if (!iter2.hasNext()) {
|
||
307 | return true; |
||
308 | } else {
|
||
309 | getLogger().warn("size !=");
|
||
310 | return false; |
||
311 | } |
||
312 | |||
313 | } |
||
314 | |||
315 | protected boolean compareFeatureIterators(Iterator iter1, Iterator iter2, |
||
316 | String[] attrNames) { |
||
317 | Feature feature; |
||
318 | Feature ffeature; |
||
319 | while (iter1.hasNext()) {
|
||
320 | feature = (Feature) iter1.next(); |
||
321 | ffeature = (Feature) iter2.next(); |
||
322 | if (!this.compareFeatures(feature, ffeature, attrNames)) { |
||
323 | return false; |
||
324 | } |
||
325 | } |
||
326 | |||
327 | return !iter2.hasNext();
|
||
328 | |||
329 | } |
||
330 | |||
331 | |||
332 | |||
333 | protected boolean compareTypes(FeatureType ft1, FeatureType ft2) { |
||
334 | if (ft1.size() != ft2.size()) {
|
||
335 | getLogger().warn("size !=");
|
||
336 | return false; |
||
337 | } |
||
338 | if (ft1.getDefaultGeometryAttributeIndex() != ft2
|
||
339 | .getDefaultGeometryAttributeIndex()) { |
||
340 | getLogger().warn( |
||
341 | "getDefaultGeometryAttributeIndex "
|
||
342 | + ft1.getDefaultGeometryAttributeIndex() + |
||
343 | " !="+ ft2.getDefaultGeometryAttributeIndex());
|
||
344 | return false; |
||
345 | } |
||
346 | if (ft1.getDefaultGeometryAttributeIndex() > -1) { |
||
347 | if (ft1.getDefaultSRS() != null) { |
||
348 | if (!ft1.getDefaultSRS().equals(ft2.getDefaultSRS())) {
|
||
349 | getLogger().warn("getDefaultSRS !=");
|
||
350 | return false; |
||
351 | } |
||
352 | |||
353 | } else {
|
||
354 | if (ft2.getDefaultSRS() != null) { |
||
355 | getLogger().warn("getDefaultSRS !=");
|
||
356 | return false; |
||
357 | } |
||
358 | } |
||
359 | } |
||
360 | |||
361 | if (ft1.getDefaultGeometryAttributeName() != null) { |
||
362 | if (!ft1.getDefaultGeometryAttributeName().equals(
|
||
363 | ft2.getDefaultGeometryAttributeName())) { |
||
364 | getLogger().warn("getDefaultGeometryAttributeName !=");
|
||
365 | |||
366 | return false; |
||
367 | } |
||
368 | } else {
|
||
369 | if (ft2.getDefaultGeometryAttributeName() != null) { |
||
370 | getLogger().warn("getDefaultGeometryAttributeName !=");
|
||
371 | return false; |
||
372 | } |
||
373 | } |
||
374 | |||
375 | |||
376 | |||
377 | FeatureAttributeDescriptor attr1, attr2; |
||
378 | for (int i = 0; i < ft1.size(); i++) { |
||
379 | attr1 = ft1.getAttributeDescriptor(i); |
||
380 | attr2 = ft2.getAttributeDescriptor(i); |
||
381 | |||
382 | if (!compareAttributes(attr1, attr2)) {
|
||
383 | return false; |
||
384 | } |
||
385 | |||
386 | } |
||
387 | return true; |
||
388 | |||
389 | } |
||
390 | |||
391 | protected boolean compareAttributes(FeatureAttributeDescriptor attr1, |
||
392 | FeatureAttributeDescriptor attr2) { |
||
393 | if (attr1 == null || attr2 == null) { |
||
394 | getLogger().warn("attr1 == null || attr2 == null");
|
||
395 | return false; |
||
396 | } |
||
397 | if (!attr1.getName().equals(attr2.getName())) {
|
||
398 | getLogger().warn( |
||
399 | "name '" + attr1.getName() + "' != '" + attr2.getName() |
||
400 | + "'");
|
||
401 | return false; |
||
402 | } |
||
403 | |||
404 | if (attr1.getDataType() != attr2.getDataType()) {
|
||
405 | getLogger().warn( |
||
406 | attr1.getName() + ":" +
|
||
407 | "dataType '" + attr1.getDataTypeName() + "'[" |
||
408 | + attr1.getDataType() + "] != '"
|
||
409 | + attr2.getDataTypeName() + "'["
|
||
410 | + attr2.getDataType() + "]");
|
||
411 | return false; |
||
412 | } |
||
413 | |||
414 | if (attr1.getSize() != attr2.getSize()) {
|
||
415 | getLogger().warn( |
||
416 | attr1.getName() + ":" +
|
||
417 | "size " + attr1.getSize() + " != " + attr2.getSize()); |
||
418 | return false; |
||
419 | } |
||
420 | |||
421 | if (attr1.getPrecision() != attr2.getPrecision()) {
|
||
422 | getLogger().warn( |
||
423 | attr1.getName() + ":" +
|
||
424 | "precision " + attr1.getPrecision() + " != " |
||
425 | + attr1.getPrecision()); |
||
426 | return false; |
||
427 | } |
||
428 | |||
429 | if (attr1.getGeometryType() != attr2.getGeometryType()) {
|
||
430 | getLogger().warn( |
||
431 | attr1.getName() + ":" +
|
||
432 | "GeometryType " + attr1.getGeometryType() + " != " |
||
433 | + attr2.getGeometryType()); |
||
434 | return false; |
||
435 | } |
||
436 | |||
437 | if (attr1.getGeometrySubType() != attr2.getGeometrySubType()) {
|
||
438 | getLogger().warn( |
||
439 | attr1.getName() + ":" +
|
||
440 | "GeometrySubType " + attr1.getGeometrySubType() + " != " |
||
441 | + attr2.getGeometrySubType()); |
||
442 | |||
443 | return false; |
||
444 | } |
||
445 | |||
446 | if (attr1.getSRS() != null) { |
||
447 | if (!attr1.getSRS().equals(attr2.getSRS())) {
|
||
448 | getLogger().warn( |
||
449 | attr1.getName() + ":" +
|
||
450 | "srs " + attr1.getSRS() + " != " + attr2.getSRS()); |
||
451 | return false; |
||
452 | } |
||
453 | } else {
|
||
454 | if (attr2.getSRS() != null) { |
||
455 | getLogger().warn( |
||
456 | attr1.getName() + ":" +
|
||
457 | "srs " + attr1.getSRS() + " != " + attr2.getSRS()); |
||
458 | return false; |
||
459 | } |
||
460 | } |
||
461 | |||
462 | return true; |
||
463 | } |
||
464 | |||
465 | protected boolean compareFeatures(Feature f1, Feature f2, |
||
466 | String[] attrsNames) { |
||
467 | FeatureAttributeDescriptor attr1; |
||
468 | FeatureAttributeDescriptor attr2; |
||
469 | Object v1, v2;
|
||
470 | for (int i = 0; i < attrsNames.length; i++) { |
||
471 | attr1 = f1.getType().getAttributeDescriptor(attrsNames[i]); |
||
472 | attr2 = f2.getType().getAttributeDescriptor(attrsNames[i]); |
||
473 | if (attr1 != attr2) {
|
||
474 | if (!compareAttributes(attr1, attr1)) {
|
||
475 | return false; |
||
476 | } |
||
477 | } |
||
478 | v1 = f1.get(attr1.getName()); |
||
479 | v2 = f2.get(attr2.getName()); |
||
480 | if (!compareFeatureValue(v1, v2, attr1)) {
|
||
481 | return false; |
||
482 | } |
||
483 | } |
||
484 | |||
485 | return true; |
||
486 | } |
||
487 | |||
488 | protected boolean compareFeatures(Feature f1, Feature f2) { |
||
489 | if (!compareTypes(f1.getType(), f2.getType())) {
|
||
490 | return false; |
||
491 | } |
||
492 | Iterator iter = f1.getType().iterator();
|
||
493 | FeatureAttributeDescriptor attr; |
||
494 | Object v1, v2;
|
||
495 | while (iter.hasNext()) {
|
||
496 | attr = (FeatureAttributeDescriptor) iter.next(); |
||
497 | v1 = f1.get(attr.getName()); |
||
498 | v2 = f2.get(attr.getName()); |
||
499 | if (!compareFeatureValue(v1, v2, attr)) {
|
||
500 | return false; |
||
501 | } |
||
502 | } |
||
503 | |||
504 | return true; |
||
505 | |||
506 | } |
||
507 | |||
508 | protected boolean compareFeatureValue(Object v1, Object v2, |
||
509 | FeatureAttributeDescriptor attr) { |
||
510 | |||
511 | if ((v1 == null || v2 == null) && !attr.allowNull() ){ |
||
512 | getLogger().warn("null and !allowNull:"
|
||
513 | + attr.getName()); |
||
514 | return false; |
||
515 | } |
||
516 | |||
517 | if (v1 == v2) {
|
||
518 | return true; |
||
519 | } else if (v1 == null) { |
||
520 | getLogger().warn(" v1 == null and v2 != null:"
|
||
521 | + attr.getName()); |
||
522 | return false; |
||
523 | } else if (v2 == null) { |
||
524 | getLogger().warn("v2 == null and v1 != null:"
|
||
525 | + attr.getName()); |
||
526 | return false; |
||
527 | |||
528 | } |
||
529 | switch (attr.getType()) {
|
||
530 | case DataTypes.GEOMETRY:
|
||
531 | Geometry geom1 = (Geometry) v1; |
||
532 | Geometry geom2 = (Geometry) v2; |
||
533 | if (!geom1.equals(geom2)) {
|
||
534 | getLogger().warn(" v1 != v2 (Geom):" + attr.getName());
|
||
535 | return false; |
||
536 | |||
537 | } |
||
538 | return true; |
||
539 | case DataTypes.DOUBLE:
|
||
540 | double diff = ((Double) v1).doubleValue() |
||
541 | - ((Double) v1).doubleValue();
|
||
542 | if (!(Math.abs(diff) < 0.000001)) { |
||
543 | getLogger().warn(" v1 != v2 (Dobule):" + attr.getName());
|
||
544 | return false; |
||
545 | |||
546 | } |
||
547 | return true; |
||
548 | |||
549 | case DataTypes.OBJECT:
|
||
550 | if (!v1.equals(v2)) {
|
||
551 | getLogger().warn( |
||
552 | " v1 != v2 (object):" + attr.getName() + " [ignored]"); |
||
553 | return false; |
||
554 | } |
||
555 | return true; |
||
556 | |||
557 | default:
|
||
558 | if (!v1.equals(v2)) {
|
||
559 | getLogger() |
||
560 | .warn( |
||
561 | " v1 != v2:" + attr.getName() + ": " + v1 |
||
562 | + " != " + v2);
|
||
563 | return false; |
||
564 | } |
||
565 | } |
||
566 | return true; |
||
567 | |||
568 | } |
||
569 | |||
570 | |||
571 | //------------------------------------------------
|
||
572 | |||
573 | public void testSimpleIteration(FeatureStore store) { |
||
574 | this.testSimpleIteration(store, null); |
||
575 | } |
||
576 | |||
577 | protected String[] getRandomAttibuteList(FeatureType fType) { |
||
578 | String[] attrNames = new String[fType.size()]; |
||
579 | Iterator iter = fType.iterator();
|
||
580 | int i = 0; |
||
581 | while (iter.hasNext()) {
|
||
582 | attrNames[i] = ((FeatureAttributeDescriptor) iter.next()).getName(); |
||
583 | i++; |
||
584 | } |
||
585 | return this.getRandomAttibuteList(attrNames); |
||
586 | } |
||
587 | |||
588 | protected Random getRandom(){ |
||
589 | if (rnd == null){ |
||
590 | rnd = new Random(); |
||
591 | rnd.setSeed(System.currentTimeMillis());
|
||
592 | } |
||
593 | return rnd;
|
||
594 | } |
||
595 | |||
596 | protected String[] getRandomAttibuteList(String[] attrNames) { |
||
597 | int nAttributes = getRandom().nextInt(
|
||
598 | attrNames.length + (attrNames.length / 2)) + 1; |
||
599 | TreeSet set = new TreeSet(); |
||
600 | for (int i = 0; i < nAttributes; i++) { |
||
601 | set.add(attrNames[getRandom().nextInt(attrNames.length)]); |
||
602 | } |
||
603 | return (String[]) set.toArray(new String[0]); |
||
604 | } |
||
605 | |||
606 | public void testIterationFastAndStandart(FeatureStore store) |
||
607 | throws Exception { |
||
608 | this.testIterationFastAndStandart(store, null); |
||
609 | |||
610 | FeatureQuery query = this.getDefaultQuery(store);
|
||
611 | // Random Attribute list
|
||
612 | query.setAttributeNames(getRandomAttibuteList(store |
||
613 | .getDefaultFeatureType())); |
||
614 | this.testIterationFastAndStandart(store, query);
|
||
615 | |||
616 | // Sorted
|
||
617 | FeatureAttributeDescriptor attr = getFirstAttributeOfType(store |
||
618 | .getDefaultFeatureType(), new int[] { DataTypes.INT, |
||
619 | DataTypes.LONG, DataTypes.STRING, DataTypes.DOUBLE, DataTypes.FLOAT }); |
||
620 | { |
||
621 | // asure that attr is in query attributes
|
||
622 | boolean attrFound = false; |
||
623 | String[] curAttrs = query.getAttributeNames(); |
||
624 | for (int i = 0; i < curAttrs.length; i++) { |
||
625 | if (curAttrs[i].equals(attr.getName())) {
|
||
626 | attrFound = true;
|
||
627 | break;
|
||
628 | |||
629 | } |
||
630 | } |
||
631 | if (!attrFound) {
|
||
632 | String[] newAttrs = new String[curAttrs.length + 1]; |
||
633 | for (int i = 0; i < curAttrs.length; i++) { |
||
634 | newAttrs[i] = curAttrs[i]; |
||
635 | } |
||
636 | newAttrs[curAttrs.length] = attr.getName(); |
||
637 | query.setAttributeNames(newAttrs); |
||
638 | } |
||
639 | } |
||
640 | |||
641 | |||
642 | query.getOrder().add(attr.getName(), true);
|
||
643 | this.testIterationFastAndStandart(store, query);
|
||
644 | |||
645 | // Filter
|
||
646 | query = this.getDefaultQuery(store);
|
||
647 | |||
648 | query.setFilter(new Evaluator(){
|
||
649 | private EvaluatorFieldsInfo evaluatorFieldsInfo = new EvaluatorFieldsInfo(); |
||
650 | |||
651 | public Object evaluate(EvaluatorData data) |
||
652 | throws EvaluatorException {
|
||
653 | // TODO Auto-generated method stub
|
||
654 | return Boolean.TRUE; |
||
655 | } |
||
656 | |||
657 | public String getSQL() { |
||
658 | return "true = true"; |
||
659 | } |
||
660 | |||
661 | public String getDescription() { |
||
662 | // TODO Auto-generated method stub
|
||
663 | return null; |
||
664 | } |
||
665 | |||
666 | public String getName() { |
||
667 | return "AlwaysTrue"; |
||
668 | } |
||
669 | |||
670 | public EvaluatorFieldsInfo getFieldsInfo() {
|
||
671 | return evaluatorFieldsInfo;
|
||
672 | } |
||
673 | |||
674 | }); |
||
675 | this.testIterationFastAndStandart(store, query);
|
||
676 | |||
677 | // Filter + Sorted
|
||
678 | query.getOrder().add(attr.getName(), true);
|
||
679 | this.testIterationFastAndStandart(store, query);
|
||
680 | } |
||
681 | |||
682 | public void testSimpleIteration(FeatureStore store, FeatureQuery query) { |
||
683 | FeatureSet set; |
||
684 | try {
|
||
685 | |||
686 | if (query == null) { |
||
687 | query = this.getDefaultQuery(store);
|
||
688 | } |
||
689 | set = store.getFeatureSet(query); |
||
690 | FeatureType type = set.getDefaultFeatureType(); |
||
691 | |||
692 | DisposableIterator it = set.iterator(); |
||
693 | Feature feature; |
||
694 | printFeatureTypeColNames(type, 15);
|
||
695 | while (it.hasNext()) {
|
||
696 | |||
697 | feature = (Feature) it.next(); |
||
698 | printFeature(feature, false, 15); |
||
699 | } |
||
700 | |||
701 | it.dispose(); |
||
702 | set.dispose(); |
||
703 | |||
704 | } catch (DataException e3) {
|
||
705 | e3.printStackTrace(); |
||
706 | fail(); |
||
707 | return;
|
||
708 | } |
||
709 | |||
710 | } |
||
711 | |||
712 | public void testIterationFastAndStandart(FeatureStore store, |
||
713 | FeatureQuery query) { |
||
714 | FeatureSet set; |
||
715 | try {
|
||
716 | |||
717 | if (query == null) { |
||
718 | query = this.getDefaultQuery(store);
|
||
719 | } |
||
720 | set = store.getFeatureSet(query); |
||
721 | |||
722 | DisposableIterator it = set.iterator(); |
||
723 | DisposableIterator fit = set.fastIterator(); |
||
724 | |||
725 | assertTrue(this.compareFeatureIterators(it, fit));
|
||
726 | |||
727 | it.dispose(); |
||
728 | fit.dispose(); |
||
729 | set.dispose(); |
||
730 | |||
731 | } catch (DataException e3) {
|
||
732 | e3.printStackTrace(); |
||
733 | fail(); |
||
734 | return;
|
||
735 | } |
||
736 | |||
737 | } |
||
738 | |||
739 | public void testSimpleIteration(DataStoreParameters parameters) |
||
740 | throws Exception { |
||
741 | FeatureStore store = null;
|
||
742 | store = (FeatureStore) dataManager.openStore(parameters |
||
743 | .getDataStoreName(), parameters); |
||
744 | |||
745 | this.testSimpleIteration(store);
|
||
746 | |||
747 | store.dispose(); |
||
748 | |||
749 | } |
||
750 | |||
751 | public void testIterationFastAndStandart(DataStoreParameters parameters) |
||
752 | throws Exception { |
||
753 | FeatureStore store = null;
|
||
754 | store = (FeatureStore) dataManager.openStore(parameters |
||
755 | .getDataStoreName(), parameters); |
||
756 | |||
757 | this.testIterationFastAndStandart(store);
|
||
758 | |||
759 | store.dispose(); |
||
760 | |||
761 | } |
||
762 | |||
763 | /**
|
||
764 | *
|
||
765 | * @param count
|
||
766 | * if (< 0) list.size() >= 1 else list.size() == count
|
||
767 | * @throws Exception
|
||
768 | */
|
||
769 | public void testExplorerList(int count) throws Exception { |
||
770 | FeatureStore store = null;
|
||
771 | DataStoreParameters params = getDefaultDataStoreParameters(); |
||
772 | store = (FeatureStore) dataManager.openStore(params.getDataStoreName(), |
||
773 | params); |
||
774 | |||
775 | DataServerExplorer explorer; |
||
776 | explorer = store.getExplorer(); |
||
777 | |||
778 | if (count < 0) { |
||
779 | assertTrue(explorer.list().size() >= 1);
|
||
780 | } else {
|
||
781 | assertTrue(explorer.list().size() == count); |
||
782 | } |
||
783 | |||
784 | store.dispose(); |
||
785 | |||
786 | explorer.dispose(); |
||
787 | } |
||
788 | |||
789 | //=================================================
|
||
790 | //=================================================
|
||
791 | |||
792 | |||
793 | |||
794 | |||
795 | public void testIterationFastAndStandart() throws Exception { |
||
796 | this.testIterationFastAndStandart(this.getDefaultDataStoreParameters()); |
||
797 | } |
||
798 | |||
799 | public void testSimpleIteration() throws Exception { |
||
800 | this.testSimpleIteration(this.getDefaultDataStoreParameters()); |
||
801 | } |
||
802 | |||
803 | public void testInitializeStore() throws Exception { |
||
804 | DataStoreParameters params = getDefaultDataStoreParameters(); |
||
805 | FeatureStore store = (FeatureStore) dataManager.openStore(params |
||
806 | .getDataStoreName(), params); |
||
807 | |||
808 | assertNotNull(store.getMetadataID()); |
||
809 | assertNotNull(store.getName()); |
||
810 | assertEquals(store.getEnvelope(), store.getDynValue("Envelope"));
|
||
811 | assertTrue("Feature count not > 0", store.getFeatureCount() > 0); |
||
812 | if (store.isLocksSupported()) {
|
||
813 | assertNotNull(store.getLocks()); |
||
814 | } else {
|
||
815 | assertNull(store.getLocks()); |
||
816 | } |
||
817 | store.dispose(); |
||
818 | } |
||
819 | |||
820 | |||
821 | public void testExplorer() throws Exception { |
||
822 | if (!this.hasExplorer()) { |
||
823 | return;
|
||
824 | } |
||
825 | this.testExplorerList(-1); |
||
826 | |||
827 | } |
||
828 | |||
829 | public void testSelection() throws Exception { |
||
830 | DataStoreParameters parameters = this.getDefaultDataStoreParameters();
|
||
831 | |||
832 | FeatureStore store = (FeatureStore) dataManager.openStore(parameters |
||
833 | .getDataStoreName(), parameters); |
||
834 | FeatureSet set = store.getFeatureSet(); |
||
835 | |||
836 | assertTrue(store.getFeatureSelection().isEmpty()); |
||
837 | store.setSelection(set); |
||
838 | assertFalse(store.getFeatureSelection().isEmpty()); |
||
839 | |||
840 | assertEquals(set.getSize(), store.getFeatureSelection().getSize()); |
||
841 | |||
842 | DisposableIterator iter = set.iterator(); |
||
843 | while (iter.hasNext()) {
|
||
844 | assertTrue(store.getFeatureSelection().isSelected( |
||
845 | (Feature) iter.next())); |
||
846 | } |
||
847 | iter.dispose(); |
||
848 | |||
849 | store.getFeatureSelection().reverse(); |
||
850 | assertTrue(store.getFeatureSelection().isEmpty()); |
||
851 | assertEquals(0, store.getFeatureSelection().getSize());
|
||
852 | iter = set.iterator(); |
||
853 | while (iter.hasNext()) {
|
||
854 | assertFalse(store.getFeatureSelection().isSelected( |
||
855 | (Feature) iter.next())); |
||
856 | } |
||
857 | iter.dispose(); |
||
858 | |||
859 | store.getFeatureSelection().reverse(); |
||
860 | assertEquals(set.getSize(), store.getFeatureSelection().getSize()); |
||
861 | assertFalse(store.getFeatureSelection().isEmpty()); |
||
862 | |||
863 | set.dispose(); |
||
864 | |||
865 | } |
||
866 | |||
867 | public void testCustomFTypeSet() throws Exception { |
||
868 | DataStoreParameters dbfParameters = this
|
||
869 | .getDefaultDataStoreParameters(); |
||
870 | |||
871 | FeatureStore store = (FeatureStore) dataManager.openStore(dbfParameters |
||
872 | .getDataStoreName(), dbfParameters); |
||
873 | |||
874 | testCustomFTypeSet(store); |
||
875 | |||
876 | store.dispose(); |
||
877 | } |
||
878 | |||
879 | public void testCustomFTypeSet(FeatureStore store) throws Exception{ |
||
880 | |||
881 | FeatureSet set, set1; |
||
882 | FeatureQuery query; |
||
883 | DisposableIterator iter, iter1; |
||
884 | Iterator attrIter;
|
||
885 | FeatureAttributeDescriptor attr; |
||
886 | |||
887 | set = store.getFeatureSet(this.getDefaultQuery(store));
|
||
888 | attrIter = store.getDefaultFeatureType().iterator(); |
||
889 | |||
890 | String[] names; |
||
891 | while (attrIter.hasNext()) {
|
||
892 | attr = (FeatureAttributeDescriptor) attrIter.next(); |
||
893 | int fieldIndex = attr.getIndex();
|
||
894 | |||
895 | query = this.getDefaultQuery(store);
|
||
896 | String fieldName = store.getDefaultFeatureType()
|
||
897 | .getAttributeDescriptor(fieldIndex).getName(); |
||
898 | |||
899 | names = new String[] { fieldName }; |
||
900 | query.setAttributeNames(names); |
||
901 | set1 = store.getFeatureSet(query); |
||
902 | |||
903 | if (getRandom().nextBoolean()) {
|
||
904 | iter = set.fastIterator(); |
||
905 | } else {
|
||
906 | iter = set.iterator(); |
||
907 | } |
||
908 | if (getRandom().nextBoolean()) {
|
||
909 | iter1 = set1.fastIterator(); |
||
910 | } else {
|
||
911 | iter1 = set1.iterator(); |
||
912 | } |
||
913 | |||
914 | assertTrue(compareFeatureIterators(iter, iter1, names)); |
||
915 | |||
916 | iter.dispose(); |
||
917 | iter1.dispose(); |
||
918 | set1.dispose(); |
||
919 | } |
||
920 | |||
921 | int ntimes = getRandom().nextInt(10) + 5; |
||
922 | FeatureType type = store.getDefaultFeatureType(); |
||
923 | query = this.getDefaultQuery(store);
|
||
924 | for (int i = 0; i < ntimes; i++) { |
||
925 | names = getRandomAttibuteList(type); |
||
926 | |||
927 | query.setAttributeNames(names); |
||
928 | set1 = store.getFeatureSet(query); |
||
929 | |||
930 | if (getRandom().nextBoolean()) {
|
||
931 | iter = set.fastIterator(); |
||
932 | } else {
|
||
933 | iter = set.iterator(); |
||
934 | } |
||
935 | if (getRandom().nextBoolean()) {
|
||
936 | iter1 = set1.fastIterator(); |
||
937 | } else {
|
||
938 | iter1 = set1.iterator(); |
||
939 | } |
||
940 | |||
941 | assertTrue(compareFeatureIterators(iter, iter1, names)); |
||
942 | |||
943 | iter.dispose(); |
||
944 | iter1.dispose(); |
||
945 | |||
946 | iter1 = set1.fastIterator(); |
||
947 | assertTrue(checksAttributesPositions(iter1, names)); |
||
948 | iter1.dispose(); |
||
949 | |||
950 | iter1 = set1.iterator(); |
||
951 | assertTrue(checksAttributesPositions(iter1, names)); |
||
952 | iter1.dispose(); |
||
953 | |||
954 | set1.dispose(); |
||
955 | |||
956 | |||
957 | } |
||
958 | |||
959 | |||
960 | |||
961 | set.dispose(); |
||
962 | |||
963 | } |
||
964 | |||
965 | protected boolean checksAttributesPositions(DisposableIterator iter, |
||
966 | String[] names) { |
||
967 | Feature feature; |
||
968 | FeatureType type; |
||
969 | FeatureAttributeDescriptor attr; |
||
970 | while (iter.hasNext()) {
|
||
971 | feature = (Feature) iter.next(); |
||
972 | type = feature.getType(); |
||
973 | for (int i = 0; i < names.length; i++) { |
||
974 | attr = type.getAttributeDescriptor(i); |
||
975 | if (!names[i].equals(attr.getName())) {
|
||
976 | getLogger().error( |
||
977 | "Error in attribute {} (expected: '{}' have: '{}'",
|
||
978 | new Object[] { new Integer(i), names[i], |
||
979 | attr.getName() }); |
||
980 | return false; |
||
981 | } |
||
982 | } |
||
983 | } |
||
984 | return true; |
||
985 | } |
||
986 | |||
987 | public void testPersistence() throws Exception { |
||
988 | if (ToolsLocator.getPersistenceManager() == null) { |
||
989 | fail("Default Persistence Manager not register");
|
||
990 | } |
||
991 | DataStoreParameters params = this.getDefaultDataStoreParameters();
|
||
992 | |||
993 | FeatureStore store = (FeatureStore) dataManager.openStore(params |
||
994 | .getDataStoreName(), params); |
||
995 | |||
996 | testSimpleIteration(store); |
||
997 | |||
998 | PersistentState state = ToolsLocator.getPersistenceManager().getState( |
||
999 | store); |
||
1000 | |||
1001 | FeatureStore store2 = (FeatureStore) ToolsLocator |
||
1002 | .getPersistenceManager().create(state); |
||
1003 | |||
1004 | testSimpleIteration(store2); |
||
1005 | |||
1006 | assertTrue(compareStores(store, store2)); |
||
1007 | |||
1008 | store.dispose(); |
||
1009 | store2.dispose(); |
||
1010 | |||
1011 | } |
||
1012 | |||
1013 | |||
1014 | public void testSort() throws Exception { |
||
1015 | DataStoreParameters dbfParameters = this
|
||
1016 | .getDefaultDataStoreParameters(); |
||
1017 | |||
1018 | FeatureStore store = (FeatureStore) dataManager.openStore(dbfParameters |
||
1019 | .getDataStoreName(), dbfParameters); |
||
1020 | |||
1021 | testSort(store); |
||
1022 | |||
1023 | store.dispose(); |
||
1024 | |||
1025 | } |
||
1026 | |||
1027 | public void testSort(FeatureStore store) throws Exception{ |
||
1028 | FeatureSet set1; |
||
1029 | FeatureQuery query; |
||
1030 | DisposableIterator iter1; |
||
1031 | Iterator attrIter;
|
||
1032 | FeatureAttributeDescriptor attr; |
||
1033 | |||
1034 | attrIter = store.getDefaultFeatureType().iterator(); |
||
1035 | |||
1036 | String[] names; |
||
1037 | while (attrIter.hasNext()) {
|
||
1038 | attr = (FeatureAttributeDescriptor) attrIter.next(); |
||
1039 | |||
1040 | if (attr.getType() == DataTypes.GEOMETRY) {
|
||
1041 | continue;
|
||
1042 | } |
||
1043 | query = this.getDefaultQuery(store);
|
||
1044 | String fieldName = attr.getName();
|
||
1045 | |||
1046 | |||
1047 | |||
1048 | names = new String[] { fieldName }; |
||
1049 | query.setAttributeNames(names); |
||
1050 | query.getOrder().add(fieldName, getRandom().nextBoolean()); |
||
1051 | |||
1052 | set1 = store.getFeatureSet(query); |
||
1053 | if (getRandom().nextBoolean()) {
|
||
1054 | iter1 = set1.fastIterator(); |
||
1055 | } else {
|
||
1056 | iter1 = set1.iterator(); |
||
1057 | } |
||
1058 | |||
1059 | assertTrue(checkSort(iter1, query)); |
||
1060 | |||
1061 | iter1.dispose(); |
||
1062 | set1.dispose(); |
||
1063 | } |
||
1064 | |||
1065 | int ntimes = getRandom().nextInt(10) + 5; |
||
1066 | FeatureType type = store.getDefaultFeatureType(); |
||
1067 | query = this.getDefaultQuery(store);
|
||
1068 | for (int i = 0; i < ntimes; i++) { |
||
1069 | names = getRandomAttibuteList(type); |
||
1070 | |||
1071 | int nShortFields = getRandom().nextInt(names.length) + 1; |
||
1072 | query.getOrder().clear(); |
||
1073 | for (int j = 0; j < nShortFields; j++) { |
||
1074 | attr = store.getDefaultFeatureType().getAttributeDescriptor(names[getRandom().nextInt(names.length)]); |
||
1075 | if (attr.getType() == DataTypes.INT
|
||
1076 | || attr.getType() == DataTypes.LONG |
||
1077 | || attr.getType() == DataTypes.DOUBLE |
||
1078 | || attr.getType() == DataTypes.STRING |
||
1079 | || attr.getType() == DataTypes.DATE |
||
1080 | || attr.getType() == DataTypes.BOOLEAN |
||
1081 | || attr.getType() == DataTypes.BYTE |
||
1082 | || attr.getType() == DataTypes.FLOAT) { |
||
1083 | |||
1084 | query.getOrder().add(attr.getName(), |
||
1085 | getRandom().nextBoolean()); |
||
1086 | } |
||
1087 | } |
||
1088 | |||
1089 | query.setAttributeNames(names); |
||
1090 | set1 = store.getFeatureSet(query); |
||
1091 | |||
1092 | // if (getRandom().nextBoolean()) {
|
||
1093 | iter1 = set1.fastIterator(); |
||
1094 | // } else {
|
||
1095 | // iter1 = set1.iterator();
|
||
1096 | // }
|
||
1097 | |||
1098 | assertTrue(checkSort(iter1, query)); |
||
1099 | |||
1100 | iter1.dispose(); |
||
1101 | set1.dispose(); |
||
1102 | |||
1103 | } |
||
1104 | |||
1105 | } |
||
1106 | |||
1107 | |||
1108 | public boolean checkSort(Iterator iter, FeatureQuery query) { |
||
1109 | |||
1110 | FeatureQueryOrderMember order; |
||
1111 | Feature prevFeature = null;
|
||
1112 | Feature currFeature = null;
|
||
1113 | boolean isFirst = true; |
||
1114 | Comparable v1, v2;
|
||
1115 | Object o1, o2;
|
||
1116 | int v;
|
||
1117 | FeatureQueryOrder queryOrder = query.getOrder(); |
||
1118 | |||
1119 | Iterator orderIter;
|
||
1120 | |||
1121 | //for debug only
|
||
1122 | /*
|
||
1123 | System.out.println("\nCheck order:");
|
||
1124 | Iterator orderIter = queryOrder.iterator();
|
||
1125 | while (orderIter.hasNext()) {
|
||
1126 | order = (FeatureQueryOrderMember) orderIter.next();
|
||
1127 | System.out.print(order.getAttributeName() + " ");
|
||
1128 | if (order.getAscending()) {
|
||
1129 | System.out.print("Asc, ");
|
||
1130 | } else {
|
||
1131 | System.out.print("Desc, ");
|
||
1132 | }
|
||
1133 | }
|
||
1134 | System.out.println(";");
|
||
1135 | */
|
||
1136 | |||
1137 | while (iter.hasNext()) {
|
||
1138 | currFeature = (Feature) iter.next(); |
||
1139 | if (isFirst) {
|
||
1140 | prevFeature = currFeature.getCopy(); |
||
1141 | // printFeature(prevFeature, true, 15);
|
||
1142 | isFirst = false;
|
||
1143 | continue;
|
||
1144 | } |
||
1145 | // printFeature(currFeature, false, 15);
|
||
1146 | orderIter = queryOrder.iterator(); |
||
1147 | while (orderIter.hasNext()) {
|
||
1148 | order = (FeatureQueryOrderMember) orderIter.next(); |
||
1149 | if (order.hasEvaluator()) {
|
||
1150 | try {
|
||
1151 | o1 = order.getEvaluator().evaluate( |
||
1152 | (EvaluatorData) prevFeature); |
||
1153 | o2 = order.getEvaluator().evaluate( |
||
1154 | (EvaluatorData) currFeature); |
||
1155 | } catch (EvaluatorException e) {
|
||
1156 | throw new DataEvaluatorRuntimeException(e); |
||
1157 | } |
||
1158 | } else {
|
||
1159 | |||
1160 | o1 = prevFeature.get(order.getAttributeName()); |
||
1161 | o2 = currFeature.get(order.getAttributeName()); |
||
1162 | } |
||
1163 | if (o1 instanceof Comparable && o2 instanceof Comparable) { |
||
1164 | v1 = (Comparable) o1;
|
||
1165 | v2 = (Comparable) o2;
|
||
1166 | } else {
|
||
1167 | // uncomparable objets
|
||
1168 | break;
|
||
1169 | } |
||
1170 | if (v1 == null) { |
||
1171 | if (v2 == null) { |
||
1172 | break;
|
||
1173 | } else {
|
||
1174 | v = 1;
|
||
1175 | } |
||
1176 | } else {
|
||
1177 | v = v1.compareTo(v2); |
||
1178 | } |
||
1179 | if (v != 0) { |
||
1180 | if (!order.getAscending()) {
|
||
1181 | v = -v; |
||
1182 | } |
||
1183 | } |
||
1184 | if (v < 0) { |
||
1185 | break;
|
||
1186 | } else if (v > 0) { |
||
1187 | // Checks for ignore case short
|
||
1188 | if (v1 instanceof String && v2 instanceof String) { |
||
1189 | v1 = ((String)v1).toLowerCase();
|
||
1190 | v2 = ((String)v2).toLowerCase();
|
||
1191 | v = v1.compareTo(v2); |
||
1192 | if (v != 0) { |
||
1193 | if (!order.getAscending()) {
|
||
1194 | v = -v; |
||
1195 | } |
||
1196 | } |
||
1197 | if (v < 0) { |
||
1198 | getLogger() |
||
1199 | .warn("Short compartor String ok with ignore case");
|
||
1200 | break;
|
||
1201 | } else if (v > 0) { |
||
1202 | return false; |
||
1203 | } |
||
1204 | } else {
|
||
1205 | return false; |
||
1206 | } |
||
1207 | } |
||
1208 | } |
||
1209 | prevFeature = currFeature.getCopy(); |
||
1210 | } |
||
1211 | |||
1212 | return true; |
||
1213 | } |
||
1214 | |||
1215 | protected void fullStoreIteratorTest(FeatureStore store) throws Exception{ |
||
1216 | testIterationFastAndStandart(store); |
||
1217 | testCustomFTypeSet(store); |
||
1218 | testSort(store); |
||
1219 | } |
||
1220 | |||
1221 | |||
1222 | public void testTransformsData() throws Exception { |
||
1223 | DataStoreParameters dbfParameters = this
|
||
1224 | .getDefaultDataStoreParameters(); |
||
1225 | |||
1226 | FeatureStore store = (FeatureStore) dataManager.openStore(dbfParameters |
||
1227 | .getDataStoreName(), dbfParameters); |
||
1228 | |||
1229 | FeatureStore store1 = (FeatureStore) dataManager.openStore( |
||
1230 | dbfParameters.getDataStoreName(), dbfParameters); |
||
1231 | |||
1232 | FeatureStoreTransform transform = new StringsToLowerTransform();
|
||
1233 | transform.setFeatureStore(store); |
||
1234 | |||
1235 | store.getTransforms().add(transform); |
||
1236 | |||
1237 | FeatureSet set, set1; |
||
1238 | DisposableIterator iter, iter1; |
||
1239 | Iterator iterAttr;
|
||
1240 | FeatureAttributeDescriptor attr; |
||
1241 | Feature feature, feature1; |
||
1242 | int i, ntimes;
|
||
1243 | Object v1, v2;
|
||
1244 | |||
1245 | |||
1246 | fullStoreIteratorTest(store); |
||
1247 | |||
1248 | set = store.getFeatureSet(); |
||
1249 | set1 = store1.getFeatureSet(); |
||
1250 | ntimes = getRandom().nextInt(3) + 1; |
||
1251 | for (i = 0; i < ntimes; i++) { |
||
1252 | if (getRandom().nextBoolean()) {
|
||
1253 | iter = set.fastIterator(); |
||
1254 | } else {
|
||
1255 | iter = set.iterator(); |
||
1256 | } |
||
1257 | if (getRandom().nextBoolean()) {
|
||
1258 | iter1 = set1.fastIterator(); |
||
1259 | } else {
|
||
1260 | iter1 = set1.iterator(); |
||
1261 | } |
||
1262 | while (iter.hasNext()) {
|
||
1263 | feature = (Feature) iter.next(); |
||
1264 | feature1 = (Feature) iter1.next(); |
||
1265 | |||
1266 | iterAttr = set.getDefaultFeatureType().iterator(); |
||
1267 | while (iterAttr.hasNext()) {
|
||
1268 | attr = (FeatureAttributeDescriptor) iterAttr.next(); |
||
1269 | v1 = feature.get(attr.getIndex()); |
||
1270 | v2 = feature1.get(attr.getIndex()); |
||
1271 | if (attr.getType() == DataTypes.STRING) {
|
||
1272 | if (v2 != null) { |
||
1273 | v2 = ((String) v2).toLowerCase();
|
||
1274 | } |
||
1275 | |||
1276 | } |
||
1277 | assertTrue(compareFeatureValue(v1, v2, attr)); |
||
1278 | } |
||
1279 | } |
||
1280 | assertFalse(iter1.hasNext()); |
||
1281 | iter.dispose(); |
||
1282 | iter1.dispose(); |
||
1283 | } |
||
1284 | |||
1285 | |||
1286 | set.dispose(); |
||
1287 | set1.dispose(); |
||
1288 | |||
1289 | |||
1290 | transform = new StringsToUpperTransform();
|
||
1291 | transform.setFeatureStore(store); |
||
1292 | |||
1293 | store.getTransforms().add(transform); |
||
1294 | |||
1295 | fullStoreIteratorTest(store); |
||
1296 | |||
1297 | set = store.getFeatureSet(); |
||
1298 | set1 = store1.getFeatureSet(); |
||
1299 | ntimes = getRandom().nextInt(3) + 1; |
||
1300 | for (i = 0; i < ntimes; i++) { |
||
1301 | if (getRandom().nextBoolean()) {
|
||
1302 | iter = set.fastIterator(); |
||
1303 | } else {
|
||
1304 | iter = set.iterator(); |
||
1305 | } |
||
1306 | if (getRandom().nextBoolean()) {
|
||
1307 | iter1 = set1.fastIterator(); |
||
1308 | } else {
|
||
1309 | iter1 = set1.iterator(); |
||
1310 | } |
||
1311 | while (iter.hasNext()) {
|
||
1312 | feature = (Feature) iter.next(); |
||
1313 | feature1 = (Feature) iter1.next(); |
||
1314 | |||
1315 | iterAttr = set.getDefaultFeatureType().iterator(); |
||
1316 | while (iterAttr.hasNext()) {
|
||
1317 | attr = (FeatureAttributeDescriptor) iterAttr.next(); |
||
1318 | v1 = feature.get(attr.getIndex()); |
||
1319 | v2 = feature1.get(attr.getIndex()); |
||
1320 | if (attr.getType() == DataTypes.STRING) {
|
||
1321 | if (v2 != null) { |
||
1322 | v2 = ((String) v2).toUpperCase();
|
||
1323 | } |
||
1324 | |||
1325 | } |
||
1326 | compareFeatureValue(v1, v2, attr); |
||
1327 | } |
||
1328 | } |
||
1329 | assertFalse(iter1.hasNext()); |
||
1330 | iter.dispose(); |
||
1331 | iter1.dispose(); |
||
1332 | } |
||
1333 | set.dispose(); |
||
1334 | set1.dispose(); |
||
1335 | |||
1336 | |||
1337 | transform = new AddPrefixAttributeName("_"); |
||
1338 | transform.setFeatureStore(store); |
||
1339 | |||
1340 | store.getTransforms().add(transform); |
||
1341 | |||
1342 | fullStoreIteratorTest(store); |
||
1343 | |||
1344 | set = store.getFeatureSet(); |
||
1345 | set1 = store1.getFeatureSet(); |
||
1346 | ntimes = getRandom().nextInt(3) + 1; |
||
1347 | for (i = 0; i < ntimes; i++) { |
||
1348 | if (getRandom().nextBoolean()) {
|
||
1349 | iter = set.fastIterator(); |
||
1350 | } else {
|
||
1351 | iter = set.iterator(); |
||
1352 | } |
||
1353 | if (getRandom().nextBoolean()) {
|
||
1354 | iter1 = set1.fastIterator(); |
||
1355 | } else {
|
||
1356 | iter1 = set1.iterator(); |
||
1357 | } |
||
1358 | while (iter.hasNext()) {
|
||
1359 | feature = (Feature) iter.next(); |
||
1360 | feature1 = (Feature) iter1.next(); |
||
1361 | |||
1362 | iterAttr = set1.getDefaultFeatureType().iterator(); |
||
1363 | while (iterAttr.hasNext()) {
|
||
1364 | attr = (FeatureAttributeDescriptor) iterAttr.next(); |
||
1365 | v1 = feature.get("_" + attr.getName());
|
||
1366 | v2 = feature1.get(attr.getIndex()); |
||
1367 | if (attr.getType() == DataTypes.STRING) {
|
||
1368 | if (v2 != null) { |
||
1369 | v2 = ((String) v2).toUpperCase();
|
||
1370 | } |
||
1371 | |||
1372 | } |
||
1373 | compareFeatureValue(v1, v2, attr); |
||
1374 | } |
||
1375 | } |
||
1376 | assertFalse(iter1.hasNext()); |
||
1377 | iter.dispose(); |
||
1378 | iter1.dispose(); |
||
1379 | } |
||
1380 | set.dispose(); |
||
1381 | set1.dispose(); |
||
1382 | |||
1383 | transform = new AddAttribute("__new__", DataTypes.STRING, "$$OK$$", 10); |
||
1384 | transform.setFeatureStore(store); |
||
1385 | |||
1386 | store.getTransforms().add(transform); |
||
1387 | |||
1388 | fullStoreIteratorTest(store); |
||
1389 | |||
1390 | set = store.getFeatureSet(); |
||
1391 | set1 = store1.getFeatureSet(); |
||
1392 | ntimes = getRandom().nextInt(3) + 1; |
||
1393 | for (i = 0; i < ntimes; i++) { |
||
1394 | if (getRandom().nextBoolean()) {
|
||
1395 | iter = set.fastIterator(); |
||
1396 | } else {
|
||
1397 | iter = set.iterator(); |
||
1398 | } |
||
1399 | if (getRandom().nextBoolean()) {
|
||
1400 | iter1 = set1.fastIterator(); |
||
1401 | } else {
|
||
1402 | iter1 = set1.iterator(); |
||
1403 | } |
||
1404 | while (iter.hasNext()) {
|
||
1405 | feature = (Feature) iter.next(); |
||
1406 | feature1 = (Feature) iter1.next(); |
||
1407 | |||
1408 | iterAttr = set1.getDefaultFeatureType().iterator(); |
||
1409 | while (iterAttr.hasNext()) {
|
||
1410 | attr = (FeatureAttributeDescriptor) iterAttr.next(); |
||
1411 | v1 = feature.get("_" + attr.getName());
|
||
1412 | v2 = feature1.get(attr.getIndex()); |
||
1413 | if (attr.getType() == DataTypes.STRING) {
|
||
1414 | if (v2 != null) { |
||
1415 | v2 = ((String) v2).toUpperCase();
|
||
1416 | } |
||
1417 | |||
1418 | } |
||
1419 | compareFeatureValue(v1, v2, attr); |
||
1420 | } |
||
1421 | compareFeatureValue(feature.get("__new__"), "$$OK$$", set1 |
||
1422 | .getDefaultFeatureType().getAttributeDescriptor( |
||
1423 | "__new__"));
|
||
1424 | } |
||
1425 | assertFalse(iter1.hasNext()); |
||
1426 | iter.dispose(); |
||
1427 | iter1.dispose(); |
||
1428 | } |
||
1429 | set.dispose(); |
||
1430 | set1.dispose(); |
||
1431 | |||
1432 | |||
1433 | transform = new RemoveAttribute("__new__"); |
||
1434 | transform.setFeatureStore(store); |
||
1435 | |||
1436 | store.getTransforms().add(transform); |
||
1437 | |||
1438 | fullStoreIteratorTest(store); |
||
1439 | |||
1440 | set = store.getFeatureSet(); |
||
1441 | set1 = store1.getFeatureSet(); |
||
1442 | ntimes = getRandom().nextInt(3) + 1; |
||
1443 | for (i = 0; i < ntimes; i++) { |
||
1444 | if (getRandom().nextBoolean()) {
|
||
1445 | iter = set.fastIterator(); |
||
1446 | } else {
|
||
1447 | iter = set.iterator(); |
||
1448 | } |
||
1449 | if (getRandom().nextBoolean()) {
|
||
1450 | iter1 = set1.fastIterator(); |
||
1451 | } else {
|
||
1452 | iter1 = set1.iterator(); |
||
1453 | } |
||
1454 | while (iter.hasNext()) {
|
||
1455 | feature = (Feature) iter.next(); |
||
1456 | feature1 = (Feature) iter1.next(); |
||
1457 | |||
1458 | iterAttr = set1.getDefaultFeatureType().iterator(); |
||
1459 | while (iterAttr.hasNext()) {
|
||
1460 | attr = (FeatureAttributeDescriptor) iterAttr.next(); |
||
1461 | v1 = feature.get("_" + attr.getName());
|
||
1462 | v2 = feature1.get(attr.getIndex()); |
||
1463 | if (attr.getType() == DataTypes.STRING) {
|
||
1464 | if (v2 != null) { |
||
1465 | v2 = ((String) v2).toUpperCase();
|
||
1466 | } |
||
1467 | |||
1468 | } |
||
1469 | compareFeatureValue(v1, v2, attr); |
||
1470 | } |
||
1471 | assertNull(feature.getType().getAttributeDescriptor("__new__"));
|
||
1472 | } |
||
1473 | assertFalse(iter1.hasNext()); |
||
1474 | iter.dispose(); |
||
1475 | iter1.dispose(); |
||
1476 | } |
||
1477 | set.dispose(); |
||
1478 | set1.dispose(); |
||
1479 | |||
1480 | store.getTransforms().clear(); |
||
1481 | |||
1482 | compareStores(store, store1); |
||
1483 | |||
1484 | |||
1485 | store.dispose(); |
||
1486 | store1.dispose(); |
||
1487 | } |
||
1488 | |||
1489 | abstract class myTransform extends AbstractFeatureStoreTransform { |
||
1490 | protected FeatureStore store;
|
||
1491 | protected FeatureType orgDefaultFType;
|
||
1492 | protected List orgFTypes; |
||
1493 | |||
1494 | public void applyTransform(Feature source, EditableFeature target) |
||
1495 | throws DataException {
|
||
1496 | |||
1497 | Iterator iter = target.getType().iterator();
|
||
1498 | FeatureAttributeDescriptor attr; |
||
1499 | while (iter.hasNext()) {
|
||
1500 | attr = (FeatureAttributeDescriptor) iter.next(); |
||
1501 | this.setValue(source, target, attr);
|
||
1502 | } |
||
1503 | |||
1504 | } |
||
1505 | |||
1506 | protected void setValue(Feature source, EditableFeature target, |
||
1507 | FeatureAttributeDescriptor attrTarget) throws DataException {
|
||
1508 | target.set(attrTarget.getIndex(), source.get(attrTarget.getName())); |
||
1509 | } |
||
1510 | |||
1511 | public FeatureType getDefaultFeatureType() throws DataException { |
||
1512 | return orgDefaultFType;
|
||
1513 | } |
||
1514 | |||
1515 | public FeatureStore getFeatureStore() {
|
||
1516 | return store;
|
||
1517 | } |
||
1518 | |||
1519 | public List getFeatureTypes() throws DataException { |
||
1520 | return orgFTypes;
|
||
1521 | } |
||
1522 | |||
1523 | public void setFeatureStore(FeatureStore featureStore) { |
||
1524 | this.store = featureStore;
|
||
1525 | try {
|
||
1526 | this.orgDefaultFType = this.store.getDefaultFeatureType(); |
||
1527 | this.orgFTypes = this.store.getFeatureTypes(); |
||
1528 | |||
1529 | } catch (DataException e) {
|
||
1530 | throw new RuntimeException(e); |
||
1531 | } |
||
1532 | |||
1533 | } |
||
1534 | |||
1535 | public FeatureType getSourceFeatureTypeFrom(
|
||
1536 | FeatureType targetFeatureType) { |
||
1537 | |||
1538 | return targetFeatureType;
|
||
1539 | } |
||
1540 | |||
1541 | public String getName() { |
||
1542 | // System.out
|
||
1543 | // .println("\nmyFeatureStore.getname(): " + store.getName());
|
||
1544 | return store.getName();
|
||
1545 | } |
||
1546 | |||
1547 | public String getDescription() { |
||
1548 | // System.out.println("\nmyFeatureStore.getDescription(getfullname): "
|
||
1549 | // + store.getFullName());
|
||
1550 | return store.getFullName();
|
||
1551 | } |
||
1552 | |||
1553 | } |
||
1554 | |||
1555 | abstract class TransformTypeTransform extends myTransform { |
||
1556 | |||
1557 | private FeatureType myDefaultFeatureType = null; |
||
1558 | private List myFeatureTypes = null; |
||
1559 | |||
1560 | public FeatureType getDefaultFeatureType() throws DataException { |
||
1561 | if (this.myDefaultFeatureType == null) { |
||
1562 | this.myDefaultFeatureType = this.transformType(orgDefaultFType); |
||
1563 | } |
||
1564 | |||
1565 | return this.myDefaultFeatureType; |
||
1566 | } |
||
1567 | |||
1568 | protected abstract FeatureType transformType(FeatureType type); |
||
1569 | |||
1570 | protected abstract FeatureType restoreType(FeatureType type); |
||
1571 | |||
1572 | |||
1573 | public List getFeatureTypes() throws DataException { |
||
1574 | if (this.myFeatureTypes == null) { |
||
1575 | ArrayList list = new ArrayList(); |
||
1576 | Iterator iter = orgFTypes.iterator();
|
||
1577 | while (iter.hasNext()) {
|
||
1578 | FeatureType type = (FeatureType) iter.next(); |
||
1579 | if (type.getId().equals(
|
||
1580 | this.getDefaultFeatureType().getId())) {
|
||
1581 | list.add(this.getDefaultFeatureType());
|
||
1582 | } else {
|
||
1583 | list.add(this.transformType(type));
|
||
1584 | } |
||
1585 | } |
||
1586 | this.myFeatureTypes = Collections.unmodifiableList(list); |
||
1587 | } |
||
1588 | return this.myFeatureTypes; |
||
1589 | } |
||
1590 | |||
1591 | public boolean isTransformsOriginalValues() { |
||
1592 | return false; |
||
1593 | } |
||
1594 | |||
1595 | public FeatureType getSourceFeatureTypeFrom(
|
||
1596 | FeatureType targetFeatureType) { |
||
1597 | FeatureType org = null;
|
||
1598 | FeatureType cur = null;
|
||
1599 | Iterator iter = null; |
||
1600 | try {
|
||
1601 | iter = this.getFeatureTypes().iterator();
|
||
1602 | } catch (DataException e) {
|
||
1603 | new RuntimeException(e); |
||
1604 | } |
||
1605 | while (iter.hasNext()) {
|
||
1606 | cur = (FeatureType) iter.next(); |
||
1607 | if (cur.getId().equals(targetFeatureType.getId())) {
|
||
1608 | org = cur; |
||
1609 | break;
|
||
1610 | } |
||
1611 | } |
||
1612 | if (org == null) { |
||
1613 | throw new RuntimeException(); |
||
1614 | } |
||
1615 | |||
1616 | return this.restoreType(org); |
||
1617 | } |
||
1618 | |||
1619 | } |
||
1620 | |||
1621 | abstract class TransformAttributeNameTransform extends |
||
1622 | TransformTypeTransform { |
||
1623 | |||
1624 | protected void setValue(Feature source, EditableFeature target, |
||
1625 | FeatureAttributeDescriptor attrTarget) throws DataException {
|
||
1626 | target.set(attrTarget.getIndex(), source.get(this
|
||
1627 | .restoreAttributeName(attrTarget.getName()))); |
||
1628 | } |
||
1629 | |||
1630 | |||
1631 | protected FeatureType transformType(FeatureType type) {
|
||
1632 | EditableFeatureType result = type.getEditable(); |
||
1633 | Iterator iter = result.iterator();
|
||
1634 | EditableFeatureAttributeDescriptor attr; |
||
1635 | while (iter.hasNext()) {
|
||
1636 | attr = (EditableFeatureAttributeDescriptor) iter.next(); |
||
1637 | attr.setName(transformAttributeName(attr.getName())); |
||
1638 | } |
||
1639 | return result.getNotEditableCopy();
|
||
1640 | } |
||
1641 | |||
1642 | protected abstract String transformAttributeName(String source); |
||
1643 | |||
1644 | protected abstract String restoreAttributeName(String target); |
||
1645 | |||
1646 | protected FeatureType restoreType(FeatureType type) {
|
||
1647 | EditableFeatureType result; |
||
1648 | if (type instanceof EditableFeatureType) { |
||
1649 | result = (EditableFeatureType) type.getCopy(); |
||
1650 | } else {
|
||
1651 | result = type.getEditable(); |
||
1652 | } |
||
1653 | Iterator iter = result.iterator();
|
||
1654 | EditableFeatureAttributeDescriptor attr; |
||
1655 | while (iter.hasNext()) {
|
||
1656 | attr = (EditableFeatureAttributeDescriptor) iter.next(); |
||
1657 | |||
1658 | attr.setName(restoreAttributeName(attr.getName())); |
||
1659 | } |
||
1660 | return result.getNotEditableCopy();
|
||
1661 | } |
||
1662 | |||
1663 | public FeatureType getSourceFeatureTypeFrom(
|
||
1664 | FeatureType targetFeatureType) { |
||
1665 | FeatureType org = null;
|
||
1666 | FeatureType cur = null;
|
||
1667 | Iterator iter = null; |
||
1668 | iter = this.orgFTypes.iterator();
|
||
1669 | while (iter.hasNext()) {
|
||
1670 | cur = (FeatureType) iter.next(); |
||
1671 | if (cur.getId().equals(targetFeatureType.getId())) {
|
||
1672 | org = cur; |
||
1673 | break;
|
||
1674 | } |
||
1675 | } |
||
1676 | if (cur == null) { |
||
1677 | throw new RuntimeException(); |
||
1678 | } |
||
1679 | EditableFeatureType r = org.getEditable(); |
||
1680 | iter = r.iterator(); |
||
1681 | FeatureAttributeDescriptor attr; |
||
1682 | while (iter.hasNext()) {
|
||
1683 | attr = (FeatureAttributeDescriptor) iter.next(); |
||
1684 | if (targetFeatureType.getIndex(transformAttributeName(attr
|
||
1685 | .getName())) == -1) {
|
||
1686 | iter.remove(); |
||
1687 | } |
||
1688 | } |
||
1689 | return r.getNotEditableCopy();
|
||
1690 | } |
||
1691 | |||
1692 | } |
||
1693 | |||
1694 | |||
1695 | class AddPrefixAttributeName extends TransformAttributeNameTransform { |
||
1696 | |||
1697 | private String prefix; |
||
1698 | |||
1699 | AddPrefixAttributeName(String prefix) {
|
||
1700 | this.prefix = prefix;
|
||
1701 | } |
||
1702 | |||
1703 | protected String restoreAttributeName(String target) { |
||
1704 | return target.substring(getPrefix().length(), target.length());
|
||
1705 | } |
||
1706 | |||
1707 | private String getPrefix() { |
||
1708 | return prefix;
|
||
1709 | } |
||
1710 | |||
1711 | protected String transformAttributeName(String source) { |
||
1712 | return getPrefix() + source;
|
||
1713 | } |
||
1714 | |||
1715 | } |
||
1716 | |||
1717 | class AddAttribute extends TransformTypeTransform { |
||
1718 | |||
1719 | private String name; |
||
1720 | private int type; |
||
1721 | private Object defValue; |
||
1722 | private int size; |
||
1723 | private Evaluator eval;
|
||
1724 | |||
1725 | AddAttribute(String name) {
|
||
1726 | this(name, DataTypes.STRING, null, 15); |
||
1727 | } |
||
1728 | |||
1729 | AddAttribute(String name, int type) { |
||
1730 | this(name, type, null, 15); |
||
1731 | } |
||
1732 | |||
1733 | AddAttribute(String name, int type, Object defValue) { |
||
1734 | this(name, type, defValue, 15); |
||
1735 | } |
||
1736 | |||
1737 | AddAttribute(String name, int type, Evaluator evaluator, int size) { |
||
1738 | this.name = name;
|
||
1739 | this.type = type;
|
||
1740 | this.defValue = null; |
||
1741 | this.size = size;
|
||
1742 | this.eval = evaluator;
|
||
1743 | } |
||
1744 | |||
1745 | AddAttribute(String name, int type, Object defValue, int size) { |
||
1746 | this.name = name;
|
||
1747 | this.type = type;
|
||
1748 | this.defValue = defValue;
|
||
1749 | this.size = size;
|
||
1750 | this.eval = null; |
||
1751 | } |
||
1752 | |||
1753 | |||
1754 | |||
1755 | protected FeatureType restoreType(FeatureType type) {
|
||
1756 | EditableFeatureType result; |
||
1757 | if (type instanceof EditableFeatureType) { |
||
1758 | result = (EditableFeatureType) type.getCopy(); |
||
1759 | } else {
|
||
1760 | result = type.getEditable(); |
||
1761 | } |
||
1762 | result.remove(this.name);
|
||
1763 | return result.getNotEditableCopy();
|
||
1764 | } |
||
1765 | |||
1766 | protected FeatureType transformType(FeatureType type) {
|
||
1767 | EditableFeatureType result = type.getEditable(); |
||
1768 | |||
1769 | EditableFeatureAttributeDescriptor att; |
||
1770 | if (this.eval == null) { |
||
1771 | att = result.add(name, this.type)
|
||
1772 | .setDefaultValue(this.defValue);
|
||
1773 | } else {
|
||
1774 | att = result.add(name, this.type, this.eval); |
||
1775 | } |
||
1776 | att.setSize(size); |
||
1777 | |||
1778 | |||
1779 | return result.getNotEditableCopy();
|
||
1780 | } |
||
1781 | |||
1782 | protected void setValue(Feature source, EditableFeature target, |
||
1783 | FeatureAttributeDescriptor attrTarget) throws DataException {
|
||
1784 | if (attrTarget.getName().equals(this.name)) { |
||
1785 | target.set(attrTarget.getIndex(), attrTarget.getDefaultValue()); |
||
1786 | } else {
|
||
1787 | target.set(attrTarget.getIndex(), source.get(attrTarget |
||
1788 | .getName())); |
||
1789 | } |
||
1790 | } |
||
1791 | } |
||
1792 | |||
1793 | |||
1794 | class RemoveAttribute extends TransformTypeTransform { |
||
1795 | |||
1796 | private String attributeName; |
||
1797 | private String fTypeIdToRemoveAttribute = null; |
||
1798 | private FeatureAttributeDescriptor attr;
|
||
1799 | |||
1800 | RemoveAttribute(String attributeName) {
|
||
1801 | this.attributeName = attributeName;
|
||
1802 | } |
||
1803 | |||
1804 | RemoveAttribute(String attributeName, FeatureType fType) {
|
||
1805 | this.attributeName = attributeName;
|
||
1806 | } |
||
1807 | |||
1808 | protected FeatureType restoreType(FeatureType type) {
|
||
1809 | if (!type.getId().equals(this.fTypeIdToRemoveAttribute)) { |
||
1810 | return type;
|
||
1811 | } |
||
1812 | EditableFeatureType result = type.getEditable(); |
||
1813 | |||
1814 | EditableFeatureAttributeDescriptor att; |
||
1815 | if (this.attr.getEvaluator() == null) { |
||
1816 | att = result.add(attributeName, this.attr.getType())
|
||
1817 | .setDefaultValue(this.attr.getDefaultValue());
|
||
1818 | } else {
|
||
1819 | att = result.add(attributeName, this.attr.getType(),
|
||
1820 | this.attr.getEvaluator());
|
||
1821 | } |
||
1822 | att.setSize(this.attr.getSize());
|
||
1823 | att.setAllowNull(this.attr.allowNull());
|
||
1824 | att.setGeometryType(this.attr.getGeometryType());
|
||
1825 | att.setSRS(this.attr.getSRS());
|
||
1826 | att.setPrecision(this.attr.getPrecision());
|
||
1827 | // TODO
|
||
1828 | |||
1829 | |||
1830 | return result.getNotEditableCopy();
|
||
1831 | } |
||
1832 | |||
1833 | protected FeatureType transformType(FeatureType type) {
|
||
1834 | if (!type.getId().equals(this.fTypeIdToRemoveAttribute)) { |
||
1835 | return type;
|
||
1836 | } |
||
1837 | |||
1838 | EditableFeatureType result; |
||
1839 | if (type instanceof EditableFeatureType) { |
||
1840 | result = (EditableFeatureType) type.getCopy(); |
||
1841 | } else {
|
||
1842 | result = type.getEditable(); |
||
1843 | } |
||
1844 | result.remove(this.attributeName);
|
||
1845 | return result.getNotEditableCopy();
|
||
1846 | |||
1847 | } |
||
1848 | |||
1849 | public void setFeatureStore(FeatureStore featureStore) { |
||
1850 | Iterator iter;
|
||
1851 | try {
|
||
1852 | iter = featureStore.getFeatureTypes().iterator(); |
||
1853 | } catch (DataException e) {
|
||
1854 | throw new RuntimeException(e); |
||
1855 | } |
||
1856 | FeatureType type; |
||
1857 | FeatureAttributeDescriptor attrTmp; |
||
1858 | while (iter.hasNext()) {
|
||
1859 | type = (FeatureType) iter.next(); |
||
1860 | attrTmp = type.getAttributeDescriptor(this.attributeName);
|
||
1861 | if (attrTmp != null) { |
||
1862 | this.fTypeIdToRemoveAttribute = type.getId();
|
||
1863 | this.attr = attrTmp;
|
||
1864 | break;
|
||
1865 | } |
||
1866 | } |
||
1867 | super.setFeatureStore(featureStore);
|
||
1868 | } |
||
1869 | |||
1870 | public void applyTransform(Feature source, EditableFeature target) |
||
1871 | throws DataException {
|
||
1872 | // TODO Auto-generated method stub
|
||
1873 | super.applyTransform(source, target);
|
||
1874 | } |
||
1875 | |||
1876 | |||
1877 | |||
1878 | } |
||
1879 | |||
1880 | |||
1881 | public class StringsToLowerTransform extends myTransform { |
||
1882 | |||
1883 | public boolean isTransformsOriginalValues() { |
||
1884 | return true; |
||
1885 | } |
||
1886 | |||
1887 | protected void setValue(Feature source, EditableFeature target, |
||
1888 | FeatureAttributeDescriptor attr) throws DataException {
|
||
1889 | if (attr.getType() == DataTypes.STRING) {
|
||
1890 | String v = ((String) source.get(attr.getName())); |
||
1891 | if (v != null){ |
||
1892 | v = v.toLowerCase(); |
||
1893 | } else if (!attr.allowNull()) { |
||
1894 | v = (String) attr.getDefaultValue();
|
||
1895 | v = v.toLowerCase(); |
||
1896 | } |
||
1897 | target.set(attr.getName(), v); |
||
1898 | } else {
|
||
1899 | target.set(attr.getName(), source.get(attr.getName())); |
||
1900 | } |
||
1901 | } |
||
1902 | |||
1903 | } |
||
1904 | |||
1905 | class StringsToUpperTransform extends StringsToLowerTransform { |
||
1906 | protected void setValue(Feature source, EditableFeature target, |
||
1907 | FeatureAttributeDescriptor attr) throws DataException {
|
||
1908 | if (attr.getType() == DataTypes.STRING) {
|
||
1909 | String v = ((String) source.get(attr.getName())); |
||
1910 | if (v != null) { |
||
1911 | v = v.toUpperCase(); |
||
1912 | } else if (!attr.allowNull()) { |
||
1913 | v = (String) attr.getDefaultValue();
|
||
1914 | v.toUpperCase(); |
||
1915 | } |
||
1916 | target.set(attr.getName(), v); |
||
1917 | } else {
|
||
1918 | target.set(attr.getName(), source.get(attr.getName())); |
||
1919 | } |
||
1920 | } |
||
1921 | |||
1922 | } |
||
1923 | |||
1924 | |||
1925 | public void testFeatureReference() throws Exception { |
||
1926 | DataStoreParameters dbfParameters = this
|
||
1927 | .getDefaultDataStoreParameters(); |
||
1928 | |||
1929 | FeatureStore store = (FeatureStore) dataManager.openStore(dbfParameters |
||
1930 | .getDataStoreName(), dbfParameters); |
||
1931 | |||
1932 | Feature feature, refered; |
||
1933 | FeatureSet set; |
||
1934 | int nTimes2 = getRandom().nextInt(2) + 1; |
||
1935 | |||
1936 | for (int j = 0; j < nTimes2; j++) { |
||
1937 | set = store.getFeatureSet(); |
||
1938 | DisposableIterator iter; |
||
1939 | |||
1940 | int nTimes = getRandom().nextInt(2) + 3; |
||
1941 | for (int i = 0; i < nTimes; i++) { |
||
1942 | if (getRandom().nextBoolean()) {
|
||
1943 | iter = set.fastIterator(); |
||
1944 | } else {
|
||
1945 | iter = set.fastIterator(); |
||
1946 | } |
||
1947 | while (iter.hasNext()) {
|
||
1948 | feature = (Feature) iter.next(); |
||
1949 | refered = feature.getReference().getFeature(); |
||
1950 | compareFeatures(feature, refered); |
||
1951 | } |
||
1952 | iter.dispose(); |
||
1953 | } |
||
1954 | |||
1955 | set.dispose(); |
||
1956 | } |
||
1957 | |||
1958 | nTimes2 = getRandom().nextInt(5) + 2; |
||
1959 | FeatureQuery query = store.createFeatureQuery(); |
||
1960 | |||
1961 | for (int j = 0; j < nTimes2; j++) { |
||
1962 | DisposableIterator iter; |
||
1963 | |||
1964 | query.setAttributeNames(getRandomAttibuteList(store |
||
1965 | .getDefaultFeatureType())); |
||
1966 | set = store.getFeatureSet(query); |
||
1967 | |||
1968 | int nTimes = getRandom().nextInt(3) + 3; |
||
1969 | for (int i = 0; i < nTimes; i++) { |
||
1970 | if (getRandom().nextBoolean()) {
|
||
1971 | iter = set.fastIterator(); |
||
1972 | } else {
|
||
1973 | iter = set.fastIterator(); |
||
1974 | } |
||
1975 | while (iter.hasNext()) {
|
||
1976 | feature = (Feature) iter.next(); |
||
1977 | refered = feature.getReference().getFeature( |
||
1978 | set.getDefaultFeatureType()); |
||
1979 | compareFeatures(feature, refered); |
||
1980 | } |
||
1981 | iter.dispose(); |
||
1982 | } |
||
1983 | |||
1984 | set.dispose(); |
||
1985 | } |
||
1986 | |||
1987 | |||
1988 | |||
1989 | store.dispose(); |
||
1990 | |||
1991 | } |
||
1992 | |||
1993 | public void testResourcesLocks() throws Exception { |
||
1994 | |||
1995 | if (!this.usesResources()) { |
||
1996 | return;
|
||
1997 | } |
||
1998 | |||
1999 | DataStoreParameters dbfParameters = this
|
||
2000 | .getDefaultDataStoreParameters(); |
||
2001 | |||
2002 | FeatureStore store = (FeatureStore) dataManager.openStore(dbfParameters |
||
2003 | .getDataStoreName(), dbfParameters); |
||
2004 | |||
2005 | int nThreads = getRandom().nextInt(4) + 2; |
||
2006 | List threads = new ArrayList(); |
||
2007 | TaskTestIterators task; |
||
2008 | for (int i = 0; i < nThreads; i++) { |
||
2009 | task = new TaskTestIterators(this, "" + i, store); |
||
2010 | threads.add(task); |
||
2011 | task.start(); |
||
2012 | } |
||
2013 | |||
2014 | Iterator iter;
|
||
2015 | List stillAlives = new ArrayList(); |
||
2016 | stillAlives.addAll(threads); |
||
2017 | |||
2018 | while (!stillAlives.isEmpty()) {
|
||
2019 | iter = stillAlives.iterator(); |
||
2020 | while (iter.hasNext()) {
|
||
2021 | task = (TaskTestIterators) iter.next(); |
||
2022 | if (!task.isAlive()) {
|
||
2023 | iter.remove(); |
||
2024 | } else if ((!task.isFinished()) && task.isOutOfDate()) { |
||
2025 | iter.remove(); |
||
2026 | getLogger().error("task {} outOfDate", task.getName());
|
||
2027 | } else {
|
||
2028 | Thread.yield();
|
||
2029 | Thread.sleep(100); |
||
2030 | } |
||
2031 | } |
||
2032 | } |
||
2033 | |||
2034 | store.dispose(); |
||
2035 | |||
2036 | iter = threads.iterator(); |
||
2037 | while (iter.hasNext()) {
|
||
2038 | task = (TaskTestIterators) iter.next(); |
||
2039 | assertTrue(task.isFinishedOk()); |
||
2040 | } |
||
2041 | } |
||
2042 | |||
2043 | class TaskTestIterators extends StoreTask { |
||
2044 | private BaseTestFeatureStore testInstance;
|
||
2045 | |||
2046 | public TaskTestIterators(BaseTestFeatureStore testInstance,
|
||
2047 | String name, FeatureStore store) {
|
||
2048 | super(name, store);
|
||
2049 | this.testInstance = testInstance;
|
||
2050 | } |
||
2051 | |||
2052 | public TaskTestIterators(BaseTestFeatureStore testInstance,
|
||
2053 | String name, FeatureStore store, int timeToWait) { |
||
2054 | super(name, store, timeToWait);
|
||
2055 | this.testInstance = testInstance;
|
||
2056 | } |
||
2057 | |||
2058 | public void run() { |
||
2059 | if (!this.startProcess()) { |
||
2060 | return;
|
||
2061 | } |
||
2062 | try {
|
||
2063 | this.testInstance.fullStoreIteratorTest(store);
|
||
2064 | |||
2065 | finishedOk(); |
||
2066 | } catch (Throwable e) { |
||
2067 | finishedError(e); |
||
2068 | return;
|
||
2069 | } |
||
2070 | } |
||
2071 | |||
2072 | |||
2073 | } |
||
2074 | |||
2075 | |||
2076 | protected void tearDown() throws Exception { |
||
2077 | super.tearDown();
|
||
2078 | if (!usesResources()) {
|
||
2079 | return;
|
||
2080 | } |
||
2081 | |||
2082 | ResourceManager resMan = DALLocator.getResourceManager(); |
||
2083 | resMan.closeResources(); |
||
2084 | } |
||
2085 | |||
2086 | |||
2087 | |||
2088 | } |