Statistics
| Revision:

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 @ 44678

History | View | Annotate | Download (23.5 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 java.util.Objects;
10
import org.apache.commons.lang3.tuple.Pair;
11
import org.gvsig.expressionevaluator.ExpressionBuilder.Parameter;
12
import org.gvsig.expressionevaluator.Formatter;
13
import org.gvsig.fmap.dal.DataTypes;
14
import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor;
15
import org.gvsig.fmap.dal.feature.FeatureType;
16
import org.gvsig.fmap.dal.feature.spi.FeatureProvider;
17
import org.gvsig.fmap.dal.store.h2.expressionbuilderformatter.H2SpatialFormatter;
18
import org.gvsig.fmap.dal.store.jdbc2.spi.JDBCSQLBuilderBase;
19
import org.gvsig.fmap.geom.Geometry;
20
import org.gvsig.fmap.geom.exception.CreateGeometryException;
21
import org.gvsig.tools.dispose.Disposable;
22

    
23
public class H2SpatialSQLBuilder extends JDBCSQLBuilderBase {
24

    
25
    protected Formatter formatter = null;
26
    
27
    public H2SpatialSQLBuilder(H2SpatialHelper helper) {
28
        super(helper);
29
        
30
        //
31
        // H2/H2GIS SQL functions reference list
32
        //
33
        // http://www.h2database.com/html/functions.html
34
        // http://www.h2gis.org/docs/1.3/functions/
35
        //
36
        // http://www.h2database.com/html/grammar.html
37
        //
38
        // http://www.h2database.com/html/datatypes.html
39
        //
40
        //
41

    
42
        
43
        this.defaultSchema = "PUBLIC";
44
        this.supportSchemas = true;
45
        this.allowAutomaticValues = true;
46
        this.geometrySupportType = this.helper.getGeometrySupportType();
47
        this.hasSpatialFunctions = this.helper.hasSpatialFunctions();
48

    
49
        this.STMT_DELETE_GEOMETRY_COLUMN_FROM_TABLE_schema_table = null;
50
        this.STMT_DELETE_GEOMETRY_COLUMN_FROM_TABLE_table = null;
51
         
52
        this.STMT_UPDATE_TABLE_STATISTICS_table = "ANALYZE SAMPLE_SIZE 0";
53

    
54
        this.type_boolean = "BOOLEAN";
55
        this.type_byte = "TINYINT";
56
        this.type_bytearray = "BLOB";
57
        this.type_geometry = "GEOMETRY";
58
        this.type_char = "CHAR";
59
        this.type_date = "DATE";
60
        this.type_double = "DOUBLE"; 
61
        this.type_decimal_ps = "DECIMAL({0,Number,##########},{1,Number,##########})";
62
        this.type_decimal_p = "DECIMAL({0,Number,##########})";
63
        this.type_float = "REAL";
64
        this.type_int = "INTEGER";
65
        this.type_long = "BIGINT";
66
        this.type_string = "VARCHAR";
67
        this.type_string_p = "VARCHAR({0,Number,##########})";
68
        this.type_time = "TIME";
69
        this.type_timestamp = "TIMESTAMP";
70
        this.type_version = "VARCHAR";
71
        this.type_URI = "VARCHAR";
72
        this.type_URL = "VARCHAR";
73
        this.type_FILE = "VARCHAR";
74
        this.type_FOLDER = "VARCHAR";        
75
    }
76

    
77
    @Override
78
    protected Formatter formatter() {
79
        if( this.formatter==null ) {
80
            this.formatter = new H2SpatialFormatter(this);
81
        }
82
        return this.formatter;
83
    }
84

    
85
    public class H2SpatialTableNameBuilderBase extends TableNameBuilderBase {
86

    
87
        @Override
88
        public boolean has_database() {
89
            return false;
90
        }
91
        
92
    }
93

    
94
    public class H2SpatialCreateIndexBuilder extends CreateIndexBuilderBase {
95
    
96
        @Override
97
        public List<String> toStrings(Formatter formatter) {
98
            StringBuilder builder = new StringBuilder();
99
            builder.append("CREATE ");
100
            if( this.isUnique ) {
101
                builder.append("UNIQUE ");
102
            }
103
            if( this.isSpatial ) {
104
                builder.append("SPATIAL ");
105
            }
106
            builder.append("INDEX ");
107
            if( this.ifNotExist ) {
108
                builder.append("IF NOT EXISTS ");
109
            }
110
            builder.append(as_identifier(this.indexName));
111
            builder.append(" ON ");
112
            builder.append(this.table.toString(formatter));
113
            builder.append(" ( ");
114
            boolean is_first_column = true;
115
            for( String column : this.columns) {
116
                if( is_first_column ) {
117
                    is_first_column = false;
118
                } else {
119
                    builder.append(", ");
120
                }
121
                builder.append(as_identifier(column));
122
            }
123
            builder.append(" )");
124
            
125
            List<String> sqls = new ArrayList<>();
126
            sqls.add(builder.toString());
127
            return sqls;
128
        }
129

    
130
    }
131
        
132
    protected class H2SpatialAlterTableBuilderBase extends AlterTableBuilderBase {
133
        @Override
134
        public List<String> toStrings(Formatter formatter) {
135
            List<String> sqls = new ArrayList<>();
136
            if( this.isEmpty() ) {
137
                return sqls;
138
            }
139
            for (String column : drops) {
140
                StringBuilder builder = new StringBuilder();
141
                builder.append("ALTER TABLE ");
142
                builder.append(this.table.toString(formatter));
143
                builder.append(" DROP COLUMN IF EXISTS ");
144
                builder.append(as_identifier(column)); 
145
                sqls.add(builder.toString());
146
            }
147
            for (ColumnDescriptor column : adds) {
148
                StringBuilder builder = new StringBuilder();
149
                builder.append("ALTER TABLE ");
150
                builder.append(this.table.toString(formatter));
151
                builder.append(" ADD COLUMN ");
152
                builder.append(as_identifier(column.getName())); 
153
                builder.append(" ");
154
                if( column.getType() == DataTypes.INT && column.isAutomatic() ) {
155
                    builder.append(" SERIAL");
156
                } else {
157
                    builder.append(
158
                        sqltype(
159
                            column.getType(), 
160
                            column.getSize(),
161
                            column.getPrecision(), 
162
                            column.getScale(), 
163
                            column.getGeometryType(), 
164
                            column.getGeometrySubtype()
165
                        )
166
                    );
167
                }
168
                if (column.getDefaultValue() == null) {
169
                    if (column.allowNulls()) {
170
                        builder.append(" DEFAULT NULL");
171
                    }
172
                } else {
173
                    builder.append(" DEFAULT '");
174
                    builder.append(Objects.toString(column.getDefaultValue(),""));
175
                    builder.append("'");
176
                }
177
                if (column.allowNulls()) {
178
                    builder.append(" NULL");
179
                } else {
180
                    builder.append(" NOT NULL");
181
                }
182
                if (column.isPrimaryKey()) {
183
                    builder.append(" PRIMARY KEY");
184
                }
185
                sqls.add(builder.toString());
186
                
187
                if( column.isGeometry() ) {
188
                    String constraint_name = "constraint_" + this.table().getName() + "_" + column.getName()+"_geom";
189
                    String sql;
190
                    if( column.getGeometrySRSId()==null ) {
191
                      sql = MessageFormat.format(
192
                            "ALTER TABLE \"{0}\".\"{1}\" ADD CONSTRAINT IF NOT EXISTS \"{2}\" CHECK NVL2(\"{3}\", ST_GeometryTypeCode(\"{3}\") = {4,number,###} AND ST_CoordDim(\"{3}\") = {5,number,###}, TRUE)",
193
                            this.table().getSchema(),
194
                            this.table().getName(),
195
                            constraint_name,
196
                            column.getName(),
197
                            sqlgeometrytype(column.getGeometryType(),column.getGeometrySubtype()),
198
                            sqlgeometrynumdimension(column.getGeometryType(),column.getGeometrySubtype()),
199
                            column.getGeometrySRSId() 
200
                      );
201
                    } else {
202
                      sql = MessageFormat.format(
203
                            "ALTER TABLE \"{0}\".\"{1}\" ADD CONSTRAINT IF NOT EXISTS \"{2}\" CHECK NVL2(\"{3}\", ST_GeometryTypeCode(\"{3}\") = {4,number,###} AND ST_CoordDim(\"{3}\") = {5,number,###} AND ST_SRID(\"{3}\") = {6,number,#####}, TRUE)",
204
                            this.table().getSchema(),
205
                            this.table().getName(),
206
                            constraint_name,
207
                            column.getName(),
208
                            sqlgeometrytype(column.getGeometryType(),column.getGeometrySubtype()),
209
                            sqlgeometrynumdimension(column.getGeometryType(),column.getGeometrySubtype()),
210
                            column.getGeometrySRSId() 
211
                      );
212
                    }
213
                    sqls.add(sql);
214
                }
215
            }
216
            for (ColumnDescriptor column : alters) {
217
                StringBuilder builder = new StringBuilder();
218
                builder.append("ALTER TABLE ");
219
                builder.append(this.table.toString(formatter));
220
                builder.append(" ALTER COLUMN ");
221
                builder.append(as_identifier(column.getName())); 
222
                builder.append(" ");
223
                builder.append(
224
                    sqltype(
225
                        column.getType(), 
226
                        column.getSize(),
227
                        column.getPrecision(), 
228
                        column.getScale(), 
229
                        column.getGeometryType(), 
230
                        column.getGeometrySubtype()
231
                    )
232
                );
233
                if (column.getDefaultValue() == null) {
234
                    if (column.allowNulls()) {
235
                        builder.append(" DEFAULT NULL");
236
                    }
237
                } else {
238
                    builder.append(" DEFAULT '");
239
                    builder.append(column.getDefaultValue().toString());
240
                    builder.append("'");
241
                }
242
                if( column.isAutomatic() ) {
243
                    builder.append(" AUTO_INCREMENT");
244
                }
245
                sqls.add(builder.toString());
246
                if( column.isGeometry() ) {
247
                    String sql;
248
                    String constraint_name = "constraint_" + this.table().getName() + "_" + column.getName()+"_dim";
249
                    sql = MessageFormat.format(
250
                        "ALTER TABLE \"{0}\".\"{1}\" ADD CONSTRAINT IF NOT EXISTS \"{2}\" CHECK ST_CoordDim(\"{3}\") = {4}",
251
                        this.table().getSchema(),
252
                        this.table().getName(),
253
                        constraint_name,
254
                        column.getName(),
255
                        sqlgeometrynumdimension(column.getGeometryType(),column.getGeometrySubtype())
256
                    );
257
                    sqls.add(sql);
258
                }
259
            }
260
            for (Pair<String,String> pair : renames) {
261
                StringBuilder builder = new StringBuilder();
262
                builder.append("ALTER TABLE ");
263
                builder.append(this.table.toString(formatter));
264
                builder.append(" RENAME COLUMN ");
265
                builder.append(as_identifier(pair.getLeft())); 
266
                builder.append(" TO ");
267
                builder.append(as_identifier(pair.getRight())); 
268
                sqls.add(builder.toString());
269
            }
270
            return sqls;
271
        }
272

    
273
    }
274
    
275
    protected class H2SpatialCreateTableBuilder extends CreateTableBuilderBase {
276

    
277
        @Override
278
        public List<String> toStrings(Formatter formatter) {
279

    
280
            List<String> sqls = new ArrayList<>();
281
            StringBuilder builder = new StringBuilder();
282

    
283
            builder.append("CREATE TABLE ");
284
            builder.append(this.table.toString(formatter));
285
            builder.append(" (");
286
            boolean first = true;
287
            for (ColumnDescriptor column : columns) {
288

    
289
                if (first) {
290
                    first = false;
291
                } else {
292
                    builder.append(", ");
293
                }
294
                builder.append(as_identifier(column.getName()));
295
                builder.append(" ");
296
                builder.append(
297
                    sqltype(
298
                        column.getType(), 
299
                        column.getSize(),
300
                        column.getPrecision(), 
301
                        column.getScale(), 
302
                        column.getGeometryType(), 
303
                        column.getGeometrySubtype()
304
                    )
305
                );
306
                if( column.isGeometry() ) {
307
                    //
308
                    // https://github.com/orbisgis/h2gis/wiki/1.-Spatial-data#geometry-columns-view
309
                    // https://github.com/orbisgis/h2gis/blob/master/h2gis-functions/src/main/java/org/h2gis/functions/spatial/type/GeometryTypeFromConstraint.java
310
                    //
311
                    if( column.getGeometrySRSId()==null ) {
312
                      builder.append( 
313
                          MessageFormat.format(
314
                            " CHECK NVL2(\"{0}\", ST_GeometryTypeCode(\"{0}\") = {1} AND ST_CoordDim(\"{0}\") = {2}, TRUE)",
315
                            column.getName(),
316
                            sqlgeometrytype(column.getGeometryType(),column.getGeometrySubtype()),
317
                            sqlgeometrynumdimension(column.getGeometryType(),column.getGeometrySubtype())                        )
318
                      );
319
                    } else {
320
                      builder.append( 
321
                          MessageFormat.format(
322
                            " CHECK NVL2(\"{0}\", ST_GeometryTypeCode(\"{0}\") = {1,number,###} AND ST_CoordDim(\"{0}\") = {2,number,###} AND ST_SRID(\"{0}\") = {3,number,#####}, TRUE)",
323
                            column.getName(),
324
                            sqlgeometrytype(column.getGeometryType(),column.getGeometrySubtype()),
325
                            sqlgeometrynumdimension(column.getGeometryType(),column.getGeometrySubtype()),
326
                            column.getGeometrySRSId() 
327
                        )
328
                      );
329
                    }
330
                } else {
331
                    if (column.isPrimaryKey()) {
332
                        builder.append(" PRIMARY KEY");
333
                        if( column.isAutomatic() ) {
334
                            builder.append(" AUTO_INCREMENT");
335
                        }
336
                    } else {
337
                        if( column.isAutomatic() ) {
338
                            builder.append(" AUTO_INCREMENT");
339
                        }
340
                        if (column.getDefaultValue() == null) {
341
                            if (column.allowNulls()) {
342
                                builder.append(" DEFAULT NULL");
343
                            }
344
                        } else {
345
                            if( column.getType() == DataTypes.DATE ) {
346
                                builder.append(" DEFAULT ( TIMESTAMP '");
347
                                Date d = (Date) column.getDefaultValue();
348
                                builder.append(MessageFormat.format( "{0,date,yyyy-MM-dd HH:mm:ss.S}",d));
349
                                builder.append("' )");
350
                            } else {
351
                                builder.append(" DEFAULT '");
352
                                builder.append(Objects.toString(column.getDefaultValue(),""));
353
                                builder.append("'");
354
                            }
355
                        }
356
                    }
357
                }
358
                if (!column.allowNulls()) {
359
                    builder.append(" NOT NULL");
360
                }
361
            }
362
            builder.append(" )");
363
            sqls.add(builder.toString());
364
            return sqls;
365
        }
366
    }
367

    
368
    public class H2SpatialSelectBuilderBase extends SelectBuilderBase {
369
        
370
        @Override
371
        protected boolean isValid(StringBuilder message) {
372
            if( message == null ) {
373
                message = new StringBuilder();
374
            }
375
            if( this.has_offset() && !this.has_order_by() ) {
376
                // Algunos gestores de BBDD requieren que se especifique un
377
                // orden para poder usar OFFSET. Como eso parece buena idea para
378
                // asegurar que siempre tengamos los mismo resultados, lo exijimos
379
                // siempre.
380
                message.append("Can't use OFFSET without an ORDER BY.");
381
                return false;
382
            }
383
            return true;
384
        }        
385
        
386
        @Override
387
        public String toString(Formatter formatter) {
388
            StringBuilder builder = new StringBuilder();
389
            if( !isValid(builder) ) {
390
                throw new IllegalStateException(builder.toString());
391
            }
392
            builder.append("SELECT ");
393
            if( this.distinct ) {
394
                builder.append("DISTINCT ");
395
            }
396
            boolean first = true;
397
            for (SelectColumnBuilder column : columns) {
398
                if (first) {
399
                    first = false;
400
                } else {
401
                    builder.append(", ");
402
                }
403
                builder.append(column.toString(formatter));
404
            }
405

    
406
            if ( this.has_from() ) {
407
                builder.append(" FROM ");
408
                builder.append(this.from.toString(formatter));
409
            }
410
            if( this.has_group_by() ) {
411
                builder.append(" GROUP BY ");
412
                builder.append(this.groupColumn.get(0).toString(formatter));
413
                for (int i = 1; i < groupColumn.size(); i++) {
414
                    builder.append(", ");
415
                    builder.append(this.groupColumn.get(i).toString(formatter));
416
                }
417
            }
418
            if ( this.has_where() ) {
419
                builder.append(" WHERE ");
420
                builder.append(this.where.toString(formatter));
421
            }
422
            
423
            if( this.has_order_by() ) {
424
                builder.append(" ORDER BY ");
425
                first = true;
426
                for (OrderByBuilder item : this.order_by) {
427
                    if (first) {
428
                        first = false;
429
                    } else {
430
                        builder.append(", ");
431
                    }
432
                    builder.append(item.toString(formatter));                    
433
                }   
434
            }
435
            
436
            if ( this.has_limit() && this.has_offset() ) {
437
                builder.append(" LIMIT ");
438
                builder.append(this.limit);
439
                builder.append(" OFFSET ");
440
                builder.append(this.offset);
441
                
442
            } else if ( this.has_limit()) {
443
                builder.append(" LIMIT ");
444
                builder.append(this.limit);
445

    
446
            } else if ( this.has_offset() ) {
447
                builder.append(" LIMIT -1 OFFSET ");
448
                builder.append(this.offset);    
449
            }
450
            return builder.toString();
451

    
452
        }
453
    }
454

    
455
    @Override
456
    public Object sqlgeometrydimension(int type, int subtype) {
457
        //'XY' or 2: 2D points, identified by X and Y coordinates
458
        //'XYM': 2D points, identified by X and Y coordinates. Each point stores an M-value (measure) as well
459
        //'XYZ' or 3: 3D points, identified by X, Y and Z coordinates
460
        //'XYZM' or 4: 3D points, identified by X, Y and Z coordinates. Each point stores an M-value (measure) as well
461
        switch(subtype) {
462
            case Geometry.SUBTYPES.GEOM2D:
463
            default:
464
                return "XY";
465
            case Geometry.SUBTYPES.GEOM2DM:
466
                return "XYM";
467
            case Geometry.SUBTYPES.GEOM3D:
468
                return "XYZ";
469
            case Geometry.SUBTYPES.GEOM3DM:
470
                return "XYZM";
471
        }
472
    }
473

    
474
    @Override
475
    public String sqltype(int type, int size, int precision, int scale, int geomtype, int geomSubtype) {
476
        if( type!=DataTypes.GEOMETRY ) {
477
            return super.sqltype(type, size, precision, scale, geomtype, geomSubtype);
478
        }
479
        return "GEOMETRY(1)";
480
    }
481
    
482
    @Override
483
    public Object sqlgeometrytype(int geomtype, int geomsubtype) {
484
        //
485
        // https://github.com/orbisgis/h2gis/wiki/1.-Spatial-data#geometry-columns-view
486
        //
487
        switch(geomtype) {
488
        case Geometry.TYPES.POINT:
489
            return 1; 
490
        case Geometry.TYPES.MULTIPOINT:
491
            return 4; 
492
        case Geometry.TYPES.LINE:
493
            return 2; 
494
        case Geometry.TYPES.MULTILINE:
495
            return 5; 
496
        case Geometry.TYPES.POLYGON:
497
            return 3; 
498
        case Geometry.TYPES.MULTIPOLYGON:
499
            return 6;
500
        case Geometry.TYPES.GEOMETRY:
501
        default:
502
            return 0; // "GEOMETRY";
503
        }
504
    }
505
    
506
    public Object sqlgeometrynumdimension(int type, int subtype) {
507
        int dimensions=2;
508
        switch(subtype) {
509
        case Geometry.SUBTYPES.GEOM3D:
510
            dimensions = 3;
511
            break;
512
        case Geometry.SUBTYPES.GEOM2D:
513
            dimensions = 2;
514
            break;
515
        case Geometry.SUBTYPES.GEOM2DM:
516
            dimensions = 2; // ??????
517
            break;
518
        case Geometry.SUBTYPES.GEOM3DM:
519
            dimensions = 3; // ??????
520
            break;
521
        }
522
        return dimensions;
523
    }
524
    
525
    @Override
526
    public H2SpatialHelper getHelper() {
527
        return (H2SpatialHelper) this.helper;
528
    }
529
   
530
    @Override
531
    public Disposable setParameters(PreparedStatement st, FeatureProvider feature) {
532
        try {
533
            FeatureType featureType = feature.getType();
534
            List<Object> values = new ArrayList<>();
535
            for (Parameter parameter : this.parameters()) {
536
                if (parameter.is_constant()) {
537
                    values.add(parameter.value());
538
                } else {
539
                    String name = parameter.name();
540
                    FeatureAttributeDescriptor descriptor = featureType.getAttributeDescriptor(name);
541
                    switch( descriptor.getType() ) {
542
                    case DataTypes.DATE:
543
                        Date value = (Date)(feature.get(name));
544
                        if( value == null ) {
545
                            values.add(null);
546
                        } else {
547
                            values.add(new java.sql.Date(value.getTime()));
548
                        }
549
                        break;
550
                    case DataTypes.GEOMETRY:
551
                        Geometry geom = this.forceGeometryType(
552
                            descriptor.getGeomType(),
553
                            (Geometry)(feature.get(name))
554
                        );
555
                        values.add(geom);
556
                        break;
557
                    default:
558
                        values.add(feature.get(name));
559
                        break;
560
                    }
561
                }
562
            }
563
            return this.setStatementParameters(st, values, this.geometry_support_type());
564
        } catch (SQLException | CreateGeometryException ex) {
565
            String f = "unknow";
566
            try {
567
                f = feature.toString();
568
            } catch (Exception ex2) {
569
                // Do nothing
570
            }
571
            throw new RuntimeException("Can't set parameters to prepared statement from the feature (" + f + ")", ex);
572
        }
573
    }
574

    
575
    @Override
576
    protected AlterTableBuilder createAlterTableBuilder() {
577
        return new H2SpatialAlterTableBuilderBase();
578
    }
579
            
580
    @Override
581
    public TableNameBuilder createTableNameBuilder() {
582
        return new H2SpatialTableNameBuilderBase();
583
    }
584
    
585
    @Override
586
    protected CreateTableBuilder createCreateTableBuilder() {
587
        return new H2SpatialCreateTableBuilder();
588
    }
589

    
590
    @Override
591
    protected SelectBuilder createSelectBuilder() {
592
        return new H2SpatialSelectBuilderBase();
593
    }
594

    
595
    @Override
596
    protected CreateIndexBuilder createCreateIndexBuilder() {
597
        return new H2SpatialCreateIndexBuilder();
598
    }
599

    
600
}