svn-gvsig-desktop / trunk / org.gvsig.desktop / org.gvsig.desktop.compat.cdc / org.gvsig.fmap.dal / org.gvsig.fmap.dal.db / org.gvsig.fmap.dal.db.h2 / src / main / java / org / gvsig / fmap / dal / store / h2 / H2SpatialSQLBuilder.java @ 44198
History | View | Annotate | Download (21.3 KB)
1 |
package org.gvsig.fmap.dal.store.h2; |
---|---|
2 |
|
3 |
import java.sql.PreparedStatement; |
4 |
import java.sql.SQLException; |
5 |
import java.text.MessageFormat; |
6 |
import java.util.ArrayList; |
7 |
import java.util.Date; |
8 |
import java.util.List; |
9 |
import org.apache.commons.lang3.tuple.Pair; |
10 |
import org.gvsig.expressionevaluator.ExpressionBuilder.Parameter; |
11 |
import org.gvsig.expressionevaluator.Formatter; |
12 |
import org.gvsig.fmap.dal.DataTypes; |
13 |
import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor; |
14 |
import org.gvsig.fmap.dal.feature.FeatureType; |
15 |
import org.gvsig.fmap.dal.feature.spi.FeatureProvider; |
16 |
import org.gvsig.fmap.dal.store.h2.expressionbuilderformatter.H2SpatialFormatter; |
17 |
import org.gvsig.fmap.dal.store.jdbc2.spi.JDBCSQLBuilderBase; |
18 |
import org.gvsig.fmap.geom.Geometry; |
19 |
import org.gvsig.fmap.geom.exception.CreateGeometryException; |
20 |
import org.gvsig.tools.dispose.Disposable; |
21 |
|
22 |
public class H2SpatialSQLBuilder extends JDBCSQLBuilderBase { |
23 |
|
24 |
protected Formatter formatter = null; |
25 |
|
26 |
public H2SpatialSQLBuilder(H2SpatialHelper helper) {
|
27 |
super(helper);
|
28 |
|
29 |
//
|
30 |
// H2/H2GIS SQL functions reference list
|
31 |
//
|
32 |
// http://www.h2database.com/html/functions.html
|
33 |
// http://www.h2gis.org/docs/1.3/functions/
|
34 |
//
|
35 |
// http://www.h2database.com/html/grammar.html
|
36 |
//
|
37 |
// http://www.h2database.com/html/datatypes.html
|
38 |
//
|
39 |
//
|
40 |
|
41 |
|
42 |
this.defaultSchema = ""; |
43 |
this.supportSchemas = false; |
44 |
this.allowAutomaticValues = true; |
45 |
this.geometrySupportType = this.helper.getGeometrySupportType(); |
46 |
this.hasSpatialFunctions = this.helper.hasSpatialFunctions(); |
47 |
|
48 |
this.STMT_DELETE_GEOMETRY_COLUMN_FROM_TABLE_schema_table = null; |
49 |
this.STMT_DELETE_GEOMETRY_COLUMN_FROM_TABLE_table = null; |
50 |
|
51 |
this.STMT_UPDATE_TABLE_STATISTICS_table = "ANALYZE SAMPLE_SIZE 0"; |
52 |
|
53 |
// config.set(SQLConfig.ST_AsBinary, "NVL2({0}, ST_AsBinary({0}), null)");
|
54 |
|
55 |
this.type_boolean = "BOOLEAN"; |
56 |
this.type_byte = "TINYINT"; |
57 |
this.type_bytearray = "TINYINT"; |
58 |
this.type_geometry = "GEOMETRY"; |
59 |
this.type_char = "CHAR"; |
60 |
this.type_date = "DATE"; |
61 |
this.type_double = "DOUBLE"; |
62 |
this.type_numeric_p = "DECIMAL({0})"; |
63 |
this.type_numeric_ps = "DECIMAL({0},{1})"; |
64 |
this.type_bigdecimal = "DOUBLE"; |
65 |
this.type_float = "REAL"; |
66 |
this.type_int = "INTEGER"; |
67 |
this.type_long = "BIGINT"; |
68 |
this.type_string = "VARCHAR"; |
69 |
this.type_string_p = "VARCHAR({0})"; |
70 |
this.type_time = "TIME"; |
71 |
this.type_timestamp = "TIMESTAMP"; |
72 |
this.type_version = "VARCHAR"; |
73 |
this.type_URI = "VARCHAR"; |
74 |
this.type_URL = "VARCHAR"; |
75 |
this.type_FILE = "VARCHAR"; |
76 |
this.type_FOLDER = "VARCHAR"; |
77 |
} |
78 |
|
79 |
@Override
|
80 |
protected Formatter formatter() { |
81 |
if( this.formatter==null ) { |
82 |
this.formatter = new H2SpatialFormatter(this); |
83 |
} |
84 |
return this.formatter; |
85 |
} |
86 |
|
87 |
public class H2SpatialTableNameBuilderBase extends TableNameBuilderBase { |
88 |
|
89 |
@Override
|
90 |
public boolean has_schema() { |
91 |
return false; |
92 |
} |
93 |
|
94 |
@Override
|
95 |
public boolean has_database() { |
96 |
return false; |
97 |
} |
98 |
|
99 |
} |
100 |
|
101 |
public class H2SpatialCreateIndexBuilder extends CreateIndexBuilderBase { |
102 |
|
103 |
@Override
|
104 |
public List<String> toStrings(Formatter formatter) { |
105 |
StringBuilder builder = new StringBuilder(); |
106 |
builder.append("CREATE ");
|
107 |
if( this.isUnique ) { |
108 |
builder.append("UNIQUE ");
|
109 |
} |
110 |
if( this.isSpatial ) { |
111 |
builder.append("SPATIAL ");
|
112 |
} |
113 |
builder.append("INDEX ");
|
114 |
if( this.ifNotExist ) { |
115 |
builder.append("IF NOT EXISTS ");
|
116 |
} |
117 |
builder.append(as_identifier(this.indexName));
|
118 |
builder.append(" ON ");
|
119 |
builder.append(this.table.toString(formatter));
|
120 |
builder.append(" ( ");
|
121 |
boolean is_first_column = true; |
122 |
for( String column : this.columns) { |
123 |
if( is_first_column ) {
|
124 |
is_first_column = false;
|
125 |
} else {
|
126 |
builder.append(", ");
|
127 |
} |
128 |
builder.append(as_identifier(column)); |
129 |
} |
130 |
builder.append(" )");
|
131 |
|
132 |
List<String> sqls = new ArrayList<>(); |
133 |
sqls.add(builder.toString()); |
134 |
return sqls;
|
135 |
} |
136 |
|
137 |
} |
138 |
|
139 |
protected class H2SpatialAlterTableBuilderBase extends AlterTableBuilderBase { |
140 |
@Override
|
141 |
public List<String> toStrings(Formatter formatter) { |
142 |
List<String> sqls = new ArrayList<>(); |
143 |
if( this.isEmpty() ) { |
144 |
return sqls;
|
145 |
} |
146 |
for (String column : drops) { |
147 |
StringBuilder builder = new StringBuilder(); |
148 |
builder.append("ALTER TABLE ");
|
149 |
builder.append(this.table.toString(formatter));
|
150 |
builder.append(" DROP COLUMN IF EXISTS ");
|
151 |
builder.append(as_identifier(column)); |
152 |
sqls.add(builder.toString()); |
153 |
} |
154 |
for (ColumnDescriptor column : adds) {
|
155 |
StringBuilder builder = new StringBuilder(); |
156 |
builder.append("ALTER TABLE ");
|
157 |
builder.append(this.table.toString(formatter));
|
158 |
builder.append(" ADD COLUMN ");
|
159 |
builder.append(as_identifier(column.getName())); |
160 |
builder.append(" ");
|
161 |
if( column.getType() == DataTypes.INT && column.isAutomatic() ) {
|
162 |
builder.append(" SERIAL");
|
163 |
} else {
|
164 |
builder.append( |
165 |
sqltype( |
166 |
column.getType(), |
167 |
column.getPrecision(), |
168 |
column.getSize(), |
169 |
column.getGeometryType(), |
170 |
column.getGeometrySubtype() |
171 |
) |
172 |
); |
173 |
} |
174 |
if (column.getDefaultValue() == null) { |
175 |
if (column.allowNulls()) {
|
176 |
builder.append(" DEFAULT NULL");
|
177 |
} |
178 |
} else {
|
179 |
builder.append(" DEFAULT '");
|
180 |
builder.append(column.getDefaultValue().toString()); |
181 |
builder.append("'");
|
182 |
} |
183 |
if (column.allowNulls()) {
|
184 |
builder.append(" NULL");
|
185 |
} else {
|
186 |
builder.append(" NOT NULL");
|
187 |
} |
188 |
if (column.isPrimaryKey()) {
|
189 |
builder.append(" PRIMARY KEY");
|
190 |
} |
191 |
sqls.add(builder.toString()); |
192 |
|
193 |
if( column.isGeometry() ) {
|
194 |
String sql;
|
195 |
String constraint_name = "constraint_" + this.table().getName() + "_" + column.getName()+"_dim"; |
196 |
sql = MessageFormat.format(
|
197 |
"ALTER TABLE \"{0}\" ADD CONSTRAINT IF NOT EXISTS \"{1}\" CHECK ST_CoordDim(\"{2}\") = {3}",
|
198 |
this.table().getName(),
|
199 |
constraint_name, |
200 |
column.getName(), |
201 |
sqlgeometrynumdimension(column.getGeometryType(),column.getGeometrySubtype()) |
202 |
); |
203 |
sqls.add(sql); |
204 |
} |
205 |
} |
206 |
for (ColumnDescriptor column : alters) {
|
207 |
StringBuilder builder = new StringBuilder(); |
208 |
builder.append("ALTER TABLE ");
|
209 |
builder.append(this.table.toString(formatter));
|
210 |
builder.append(" ALTER COLUMN ");
|
211 |
builder.append(as_identifier(column.getName())); |
212 |
builder.append(" ");
|
213 |
builder.append( |
214 |
sqltype( |
215 |
column.getType(), |
216 |
column.getPrecision(), |
217 |
column.getSize(), |
218 |
column.getGeometryType(), |
219 |
column.getGeometrySubtype() |
220 |
) |
221 |
); |
222 |
if (column.getDefaultValue() == null) { |
223 |
if (column.allowNulls()) {
|
224 |
builder.append(" DEFAULT NULL");
|
225 |
} |
226 |
} else {
|
227 |
builder.append(" DEFAULT '");
|
228 |
builder.append(column.getDefaultValue().toString()); |
229 |
builder.append("'");
|
230 |
} |
231 |
if( column.isAutomatic() ) {
|
232 |
builder.append(" AUTO_INCREMENT");
|
233 |
} |
234 |
sqls.add(builder.toString()); |
235 |
if( column.isGeometry() ) {
|
236 |
String sql;
|
237 |
String constraint_name = "constraint_" + this.table().getName() + "_" + column.getName()+"_dim"; |
238 |
sql = MessageFormat.format(
|
239 |
"ALTER TABLE \"{0}\" ADD CONSTRAINT IF NOT EXISTS \"{1}\" CHECK ST_CoordDim(\"{2}\") = {3}",
|
240 |
this.table().getName(),
|
241 |
constraint_name, |
242 |
column.getName(), |
243 |
sqlgeometrynumdimension(column.getGeometryType(),column.getGeometrySubtype()) |
244 |
); |
245 |
sqls.add(sql); |
246 |
} |
247 |
} |
248 |
for (Pair<String,String> pair : renames) { |
249 |
StringBuilder builder = new StringBuilder(); |
250 |
builder.append("ALTER TABLE ");
|
251 |
builder.append(this.table.toString(formatter));
|
252 |
builder.append(" RENAME COLUMN ");
|
253 |
builder.append(as_identifier(pair.getLeft())); |
254 |
builder.append(" TO ");
|
255 |
builder.append(as_identifier(pair.getRight())); |
256 |
sqls.add(builder.toString()); |
257 |
} |
258 |
return sqls;
|
259 |
} |
260 |
|
261 |
} |
262 |
|
263 |
protected class H2SpatialCreateTableBuilder extends CreateTableBuilderBase { |
264 |
|
265 |
@Override
|
266 |
public List<String> toStrings(Formatter formatter) { |
267 |
|
268 |
List<String> sqls = new ArrayList<>(); |
269 |
StringBuilder builder = new StringBuilder(); |
270 |
|
271 |
builder.append("CREATE TABLE ");
|
272 |
builder.append(this.table.toString(formatter));
|
273 |
builder.append(" (");
|
274 |
boolean first = true; |
275 |
for (ColumnDescriptor column : columns) {
|
276 |
|
277 |
if (first) {
|
278 |
first = false;
|
279 |
} else {
|
280 |
builder.append(", ");
|
281 |
} |
282 |
builder.append(as_identifier(column.getName())); |
283 |
builder.append(" ");
|
284 |
builder.append( |
285 |
sqltype( |
286 |
column.getType(), |
287 |
column.getPrecision(), |
288 |
column.getSize(), |
289 |
column.getGeometryType(), |
290 |
column.getGeometrySubtype() |
291 |
) |
292 |
); |
293 |
if( !column.isGeometry() ) {
|
294 |
if (column.isPrimaryKey()) {
|
295 |
builder.append(" PRIMARY KEY");
|
296 |
} |
297 |
if( column.isAutomatic() ) {
|
298 |
builder.append(" AUTO_INCREMENT");
|
299 |
} |
300 |
if (column.getDefaultValue() == null) { |
301 |
if (column.allowNulls()) {
|
302 |
builder.append(" DEFAULT NULL");
|
303 |
} |
304 |
} else {
|
305 |
if( column.getType() == DataTypes.DATE ) {
|
306 |
builder.append(" DEFAULT ( TIMESTAMP '");
|
307 |
Date d = (Date) column.getDefaultValue(); |
308 |
builder.append(MessageFormat.format( "{0,date,yyyy-MM-dd HH:mm:ss.S}",d)); |
309 |
builder.append("' )");
|
310 |
} else {
|
311 |
builder.append(" DEFAULT '");
|
312 |
builder.append(column.getDefaultValue().toString()); |
313 |
builder.append("'");
|
314 |
} |
315 |
} |
316 |
} |
317 |
if (!column.allowNulls()) {
|
318 |
builder.append(" NOT NULL");
|
319 |
} |
320 |
} |
321 |
builder.append(" )");
|
322 |
sqls.add(builder.toString()); |
323 |
for (ColumnDescriptor column : columns) {
|
324 |
if( column.isGeometry() ) {
|
325 |
String sql;
|
326 |
String constraint_name = "constraint_" + this.table().getName() + "_" + column.getName()+"_dim"; |
327 |
sql = MessageFormat.format(
|
328 |
"ALTER TABLE \"{0}\" ADD CONSTRAINT IF NOT EXISTS \"{1}\" CHECK ST_CoordDim(\"{2}\") = {3}",
|
329 |
this.table().getName(),
|
330 |
constraint_name, |
331 |
column.getName(), |
332 |
sqlgeometrynumdimension(column.getGeometryType(),column.getGeometrySubtype()) |
333 |
); |
334 |
if( column.getGeometrySRSId()!=null ) { |
335 |
String sql2;
|
336 |
String constraint_name2 = "constraint_" + this.table().getName() + "_" + column.getName()+"_srid"; |
337 |
sql2 = MessageFormat.format(
|
338 |
"ALTER TABLE \"{0}\" ADD CONSTRAINT IF NOT EXISTS \"{1}\" CHECK ST_SRID(\"{2}\") = {3,number,#####}",
|
339 |
this.table().getName(),
|
340 |
constraint_name2, |
341 |
column.getName(), |
342 |
column.getGeometrySRSId() |
343 |
); |
344 |
sqls.add(sql2); |
345 |
} |
346 |
sqls.add(sql); |
347 |
} |
348 |
} |
349 |
|
350 |
return sqls;
|
351 |
} |
352 |
} |
353 |
|
354 |
public class H2SpatialSelectBuilderBase extends SelectBuilderBase { |
355 |
|
356 |
@Override
|
357 |
protected boolean isValid(StringBuilder message) { |
358 |
if( message == null ) { |
359 |
message = new StringBuilder(); |
360 |
} |
361 |
if( this.has_offset() && !this.has_order_by() ) { |
362 |
// Algunos gestores de BBDD requieren que se especifique un
|
363 |
// orden para poder usar OFFSET. Como eso parece buena idea para
|
364 |
// asegurar que siempre tengamos los mismo resultados, lo exijimos
|
365 |
// siempre.
|
366 |
message.append("Can't use OFFSET without an ORDER BY.");
|
367 |
return false; |
368 |
} |
369 |
return true; |
370 |
} |
371 |
|
372 |
@Override
|
373 |
public String toString(Formatter formatter) { |
374 |
StringBuilder builder = new StringBuilder(); |
375 |
if( !isValid(builder) ) {
|
376 |
throw new IllegalStateException(builder.toString()); |
377 |
} |
378 |
builder.append("SELECT ");
|
379 |
if( this.distinct ) { |
380 |
builder.append("DISTINCT ");
|
381 |
} |
382 |
boolean first = true; |
383 |
for (SelectColumnBuilder column : columns) {
|
384 |
if (first) {
|
385 |
first = false;
|
386 |
} else {
|
387 |
builder.append(", ");
|
388 |
} |
389 |
builder.append(column.toString(formatter)); |
390 |
} |
391 |
|
392 |
if ( this.has_from() ) { |
393 |
builder.append(" FROM ");
|
394 |
builder.append(this.from.toString(formatter));
|
395 |
} |
396 |
if ( this.has_where() ) { |
397 |
builder.append(" WHERE ");
|
398 |
builder.append(this.where.toString(formatter));
|
399 |
} |
400 |
|
401 |
if( this.has_order_by() ) { |
402 |
builder.append(" ORDER BY ");
|
403 |
first = true;
|
404 |
for (OrderByBuilder item : this.order_by) { |
405 |
if (first) {
|
406 |
first = false;
|
407 |
} else {
|
408 |
builder.append(", ");
|
409 |
} |
410 |
builder.append(item.toString(formatter)); |
411 |
} |
412 |
} |
413 |
|
414 |
if ( this.has_limit() && this.has_offset() ) { |
415 |
builder.append(" LIMIT ");
|
416 |
builder.append(this.limit);
|
417 |
builder.append(" OFFSET ");
|
418 |
builder.append(this.offset);
|
419 |
|
420 |
} else if ( this.has_limit()) { |
421 |
builder.append(" LIMIT ");
|
422 |
builder.append(this.limit);
|
423 |
|
424 |
} else if ( this.has_offset() ) { |
425 |
builder.append(" LIMIT -1 OFFSET ");
|
426 |
builder.append(this.offset);
|
427 |
} |
428 |
return builder.toString();
|
429 |
|
430 |
} |
431 |
} |
432 |
|
433 |
@Override
|
434 |
public Object sqlgeometrydimension(int type, int subtype) { |
435 |
//'XY' or 2: 2D points, identified by X and Y coordinates
|
436 |
//'XYM': 2D points, identified by X and Y coordinates. Each point stores an M-value (measure) as well
|
437 |
//'XYZ' or 3: 3D points, identified by X, Y and Z coordinates
|
438 |
//'XYZM' or 4: 3D points, identified by X, Y and Z coordinates. Each point stores an M-value (measure) as well
|
439 |
switch(subtype) {
|
440 |
case Geometry.SUBTYPES.GEOM2D:
|
441 |
default:
|
442 |
return "XY"; |
443 |
case Geometry.SUBTYPES.GEOM2DM:
|
444 |
return "XYM"; |
445 |
case Geometry.SUBTYPES.GEOM3D:
|
446 |
return "XYZ"; |
447 |
case Geometry.SUBTYPES.GEOM3DM:
|
448 |
return "XYZM"; |
449 |
} |
450 |
} |
451 |
|
452 |
@Override
|
453 |
public String sqltype(int type, int p, int s, int geomtype, int geomSubtype) { |
454 |
if( type!=DataTypes.GEOMETRY ) {
|
455 |
return super.sqltype(type, p, s, geomtype, geomSubtype); |
456 |
} |
457 |
//
|
458 |
// https://github.com/orbisgis/h2gis/wiki/1.-Spatial-data#geometry-columns-view
|
459 |
//
|
460 |
switch(geomtype) {
|
461 |
case Geometry.TYPES.POINT:
|
462 |
return "POINT"; |
463 |
case Geometry.TYPES.MULTIPOINT:
|
464 |
return "MULTIPOINT"; |
465 |
case Geometry.TYPES.LINE:
|
466 |
return "LINESTRING"; |
467 |
case Geometry.TYPES.MULTILINE:
|
468 |
return "MULTILINESTRING"; |
469 |
case Geometry.TYPES.POLYGON:
|
470 |
return "POLYGON"; |
471 |
case Geometry.TYPES.MULTIPOLYGON:
|
472 |
return "MULTIPOLYGON"; |
473 |
default:
|
474 |
return "GEOMETRY"; |
475 |
} |
476 |
} |
477 |
|
478 |
public Object sqlgeometrynumdimension(int type, int subtype) { |
479 |
int dimensions=2; |
480 |
switch(subtype) {
|
481 |
case Geometry.SUBTYPES.GEOM3D:
|
482 |
dimensions = 3;
|
483 |
break;
|
484 |
case Geometry.SUBTYPES.GEOM2D:
|
485 |
dimensions = 2;
|
486 |
break;
|
487 |
case Geometry.SUBTYPES.GEOM2DM:
|
488 |
dimensions = 2; // ?????? |
489 |
break;
|
490 |
case Geometry.SUBTYPES.GEOM3DM:
|
491 |
dimensions = 3; // ?????? |
492 |
break;
|
493 |
} |
494 |
return dimensions;
|
495 |
} |
496 |
|
497 |
@Override
|
498 |
public H2SpatialHelper getHelper() {
|
499 |
return (H2SpatialHelper) this.helper; |
500 |
} |
501 |
|
502 |
@Override
|
503 |
public Disposable setParameters(PreparedStatement st, FeatureProvider feature) { |
504 |
try {
|
505 |
FeatureType featureType = feature.getType(); |
506 |
List<Object> values = new ArrayList<>(); |
507 |
for (Parameter parameter : this.parameters()) { |
508 |
if (parameter.is_constant()) {
|
509 |
values.add(parameter.value()); |
510 |
} else {
|
511 |
String name = parameter.name();
|
512 |
FeatureAttributeDescriptor descriptor = featureType.getAttributeDescriptor(name); |
513 |
switch( descriptor.getType() ) {
|
514 |
case DataTypes.DATE:
|
515 |
Date value = (Date)(feature.get(name)); |
516 |
if( value == null ) { |
517 |
values.add(null);
|
518 |
} else {
|
519 |
values.add(value.getTime()); |
520 |
} |
521 |
break;
|
522 |
case DataTypes.GEOMETRY:
|
523 |
Geometry geom = this.forceGeometryType(
|
524 |
descriptor.getGeomType(), |
525 |
(Geometry)(feature.get(name)) |
526 |
); |
527 |
values.add(geom); |
528 |
break;
|
529 |
default:
|
530 |
values.add(feature.get(name)); |
531 |
break;
|
532 |
} |
533 |
} |
534 |
} |
535 |
return this.setStatementParameters(st, values, this.geometry_support_type()); |
536 |
} catch (SQLException | CreateGeometryException ex) { |
537 |
String f = "unknow"; |
538 |
try {
|
539 |
f = feature.toString(); |
540 |
} catch (Exception ex2) { |
541 |
// Do nothing
|
542 |
} |
543 |
throw new RuntimeException("Can't set parameters to prepared statement from the feature (" + f + ")", ex); |
544 |
} |
545 |
} |
546 |
|
547 |
@Override
|
548 |
protected AlterTableBuilder createAlterTableBuilder() {
|
549 |
return new H2SpatialAlterTableBuilderBase(); |
550 |
} |
551 |
|
552 |
@Override
|
553 |
protected TableNameBuilder createTableNameBuilder() {
|
554 |
return new H2SpatialTableNameBuilderBase(); |
555 |
} |
556 |
|
557 |
@Override
|
558 |
protected CreateTableBuilder createCreateTableBuilder() {
|
559 |
return new H2SpatialCreateTableBuilder(); |
560 |
} |
561 |
|
562 |
@Override
|
563 |
protected SelectBuilder createSelectBuilder() {
|
564 |
return new H2SpatialSelectBuilderBase(); |
565 |
} |
566 |
|
567 |
@Override
|
568 |
protected CreateIndexBuilder createCreateIndexBuilder() {
|
569 |
return new H2SpatialCreateIndexBuilder(); |
570 |
} |
571 |
|
572 |
} |