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.jdbc / src / main / java / org / gvsig / fmap / dal / store / jdbc2 / spi / operations / PerformChangesOperation.java @ 44669

History | View | Annotate | Download (19.1 KB)

1
package org.gvsig.fmap.dal.store.jdbc2.spi.operations;
2

    
3
import java.sql.Connection;
4
import java.sql.PreparedStatement;
5
import java.sql.SQLException;
6
import java.sql.Statement;
7
import java.util.ArrayList;
8
import java.util.Iterator;
9
import java.util.List;
10
import org.apache.commons.collections.CollectionUtils;
11
import org.apache.commons.lang3.StringUtils;
12
import org.gvsig.expressionevaluator.ExpressionBuilder;
13
import org.gvsig.expressionevaluator.ExpressionBuilder.Variable;
14
import org.gvsig.expressionevaluator.GeometryExpressionBuilder;
15
import org.gvsig.fmap.dal.DataTypes;
16
import org.gvsig.fmap.dal.exception.DataException;
17
import org.gvsig.fmap.dal.feature.FeatureAttributeDescriptor;
18
import org.gvsig.fmap.dal.feature.FeatureReference;
19
import org.gvsig.fmap.dal.feature.FeatureType;
20
import org.gvsig.fmap.dal.feature.spi.FeatureProvider;
21
import org.gvsig.fmap.dal.feature.spi.FeatureReferenceProviderServices;
22
import org.gvsig.fmap.dal.feature.spi.FeatureStoreProvider;
23
import org.gvsig.fmap.dal.store.jdbc2.JDBCHelper;
24
import org.gvsig.fmap.dal.store.jdbc.exception.JDBCExecuteSQLException;
25
import org.gvsig.fmap.dal.store.jdbc.exception.JDBCSQLException;
26
import org.gvsig.fmap.dal.store.jdbc.exception.JDBCUpdateWithoutChangesException;
27
import org.gvsig.fmap.dal.store.jdbc2.spi.JDBCSQLBuilderBase;
28
import org.gvsig.fmap.dal.store.jdbc2.JDBCUtils;
29
import org.gvsig.fmap.dal.store.jdbc2.OperationsFactory.TableReference;
30
import static org.gvsig.fmap.dal.store.jdbc2.spi.JDBCSQLBuilderBase.PROP_FEATURE_TYPE;
31
import static org.gvsig.fmap.dal.store.jdbc2.spi.JDBCSQLBuilderBase.PROP_TABLE;
32
import org.gvsig.tools.dispose.Disposable;
33
import org.gvsig.tools.dispose.DisposeUtils;
34

    
35
public class PerformChangesOperation extends AbstractConnectionWritableOperation {
36

    
37
    protected TableReference table;
38
    protected FeatureType featureType;
39
    protected Iterator<FeatureReferenceProviderServices> deleteds;
40
    protected Iterator<FeatureStoreProvider.FeatureTypeChanged> featureTypesChanged;
41
    protected Iterator<FeatureProvider> updateds;
42
    protected Iterator<FeatureProvider> inserteds;
43

    
44
    protected boolean typeChanged = false;
45
    
46
    public PerformChangesOperation(JDBCHelper helper) {
47
        this(helper, null, null, null, null, null, null);
48
    }
49

    
50
    public PerformChangesOperation(JDBCHelper helper,
51
            TableReference table,
52
            FeatureType featureType,
53
            Iterator<FeatureReferenceProviderServices> deleteds,
54
            Iterator<FeatureProvider> inserteds,
55
            Iterator<FeatureProvider> updateds,
56
            Iterator<FeatureStoreProvider.FeatureTypeChanged> featureTypesChanged) {
57
        super(helper);
58
        this.deleteds = deleteds;
59
        this.inserteds = inserteds;
60
        this.updateds = updateds;
61
        this.table = table;
62
        this.featureType = featureType;
63
        this.featureTypesChanged = featureTypesChanged;
64
    }
65

    
66
    public boolean isTypeChanged() {
67
        return typeChanged;
68
    }
69

    
70
    @Override
71
    public Object perform(Connection conn) throws DataException {
72
        if (featureTypesChanged.hasNext()) {
73
            FeatureStoreProvider.FeatureTypeChanged item = featureTypesChanged.next();
74
            this.performUpdateTable(conn, table, item.getSource(), item.getTarget());
75
            typeChanged = true;
76
        } else {
77
            typeChanged = false;
78
        }
79
        if (deleteds.hasNext()) {
80
            performDeletes(conn, table, featureType, deleteds);
81
        }
82
        if (updateds.hasNext()) {
83
            performUpdates(conn, table, featureType, updateds);
84
        }
85
        if (inserteds.hasNext()) {
86
            performInserts(conn, table, featureType, inserteds);
87
        }
88
        return true;
89
    }
90
    
91
    protected JDBCSQLBuilderBase buildDeleteSQL(
92
            TableReference table,
93
            FeatureType type
94
        ) {
95
        JDBCSQLBuilderBase sqlbuilder = createSQLBuilder();
96
        ExpressionBuilder expbuilder = sqlbuilder.expression();
97

    
98
        sqlbuilder.delete().table()
99
                .database(this.table.getDatabase())
100
                .schema(this.table.getSchema())
101
                .name(this.table.getTable());
102
        for (FeatureAttributeDescriptor attr : type) {
103
            if (attr.isPrimaryKey()) {
104
                sqlbuilder.delete().where().and(
105
                        expbuilder.eq(
106
                                expbuilder.column(attr.getName()),
107
                                expbuilder.parameter(attr.getName()).as_variable()
108
                        )
109
                );
110
            }
111
        }
112
        if (!sqlbuilder.delete().has_where() ) {
113
            throw new RuntimeException("Operation requires missing pk");
114
        }
115
        return sqlbuilder;
116
    }
117

    
118
    public void performDeletes(Connection conn,
119
            TableReference table,
120
            FeatureType type,
121
            Iterator<FeatureReferenceProviderServices> deleteds) throws DataException {
122

    
123
        JDBCSQLBuilderBase sqlbuilder = buildDeleteSQL(table, type);
124

    
125
        PreparedStatement st = null;
126
        Disposable paramsDisposer = null;
127
        sqlbuilder.setProperties(
128
                Variable.class, 
129
                PROP_TABLE, table
130
        );
131
        String sql = sqlbuilder.delete().toString();
132
        try {
133
            st = conn.prepareStatement(sql);
134
            while (deleteds.hasNext()) {
135
                FeatureReference reference = (FeatureReference) deleteds.next();
136
                paramsDisposer = sqlbuilder.setParameters(st, reference);
137
                int nAffected = JDBCUtils.executeUpdate(st,sql);
138
                if (nAffected == 0) {
139
                    throw new JDBCUpdateWithoutChangesException(
140
                            sqlbuilder.delete().toString(),
141
                            null
142
                    );
143
                }
144
                if (nAffected > 1) {
145
                    LOGGER.warn("Remove statement affectst to {} rows ( {} )",
146
                            nAffected, sql
147
                    );
148
                }
149
            }
150
        } catch (SQLException e) {
151
            throw new JDBCSQLException(e);
152
        } finally {
153
            JDBCUtils.closeQuietly(st);
154
            DisposeUtils.disposeQuietly(paramsDisposer);
155
        }
156
    }
157
    
158
    protected JDBCSQLBuilderBase buildInsertSQL(
159
            TableReference table,
160
            FeatureType type
161
        ) {
162
        JDBCSQLBuilderBase sqlbuilder = createSQLBuilder();
163
        GeometryExpressionBuilder expbuilder = sqlbuilder.expression();
164

    
165
        sqlbuilder.insert().table()
166
                .database(this.table.getDatabase())
167
                .schema(this.table.getSchema())
168
                .name(this.table.getTable());
169
        for (FeatureAttributeDescriptor attr : type) {
170
            if( attr.isAutomatic() || attr.isComputed() ) {
171
                continue;
172
            }
173
            if (attr.getType() == DataTypes.GEOMETRY) {
174
                sqlbuilder.insert().column().name(attr.getName()).with_value(
175
                    expbuilder.parameter(attr.getName()).as_variable()
176
                        .srs(
177
                            expbuilder.parameter().value(
178
                                attr.getSRS()).as_constant()
179
                        )
180
                );
181
            } else {
182
                sqlbuilder.insert().column().name(attr.getName()).with_value(
183
                        expbuilder.parameter(attr.getName())
184
                );
185
            }
186
        }
187
        return sqlbuilder;
188
    }
189

    
190
    public void performInserts(Connection conn,
191
            TableReference table,
192
            FeatureType type,
193
            Iterator<FeatureProvider> inserteds) throws DataException {
194
        JDBCSQLBuilderBase sqlbuilder = buildInsertSQL(table, type);
195

    
196
        PreparedStatement st;
197
        Disposable paramsDisposer;
198
        sqlbuilder.setProperties(
199
                Variable.class, 
200
                PROP_FEATURE_TYPE, type,
201
                PROP_TABLE, table
202
        );
203
        String sql = sqlbuilder.insert().toString();
204
        try {
205
            st = conn.prepareStatement(sql);
206
            while (inserteds.hasNext()) {
207
                FeatureProvider feature = inserteds.next();
208
                paramsDisposer = sqlbuilder.setParameters(st, feature);
209
                try {
210
                    if (JDBCUtils.executeUpdate(st,sql) == 0) {
211
                        throw new JDBCExecuteSQLException(
212
                                sqlbuilder.insert().toString(),
213
                                null
214
                        );
215
                    }
216
                } finally {
217
                    DisposeUtils.disposeQuietly(paramsDisposer);
218
                }
219
            }
220
        } catch (JDBCExecuteSQLException ex) {
221
            throw ex;
222
        } catch (Exception ex) {
223
            throw new JDBCExecuteSQLException(sql,ex);
224
        }
225
    }
226

    
227
    protected JDBCSQLBuilderBase buildUpdateSQL(
228
            TableReference table,
229
            FeatureType type
230
        ) {
231
        JDBCSQLBuilderBase sqlbuilder = createSQLBuilder();
232
        GeometryExpressionBuilder expbuilder = sqlbuilder.expression();
233

    
234
        sqlbuilder.update().table()
235
                .database(this.table.getDatabase())
236
                .schema(this.table.getSchema())
237
                .name(this.table.getTable());
238
        for (FeatureAttributeDescriptor attr : type) {
239
            if (attr.isPrimaryKey()) {
240
                sqlbuilder.update().where().and(
241
                        expbuilder.eq(
242
                                expbuilder.column(attr.getName()),
243
                                expbuilder.parameter(attr.getName()).as_variable()
244
                        )
245
                );
246
            } 
247
            if ( attr.isAutomatic() || attr.isReadOnly() || attr.isComputed() ) {
248
                continue;
249
            }
250
            if (attr.getType() == DataTypes.GEOMETRY) {
251
                sqlbuilder.update().column().name(attr.getName()).with_value(
252
                    expbuilder.parameter(attr.getName()).as_geometry_variable().srs(
253
                        expbuilder.parameter().value(attr.getSRS()).as_constant()
254
                    )
255
                );
256
            } else {
257
                sqlbuilder.update().column().name(attr.getName()).with_value(
258
                    expbuilder.parameter(attr.getName()).as_variable()
259
                );
260
            }
261
        }
262
        if (!sqlbuilder.update().has_where() ) {
263
            throw new RuntimeException("Operation requires missing pk");
264
        }
265
        return sqlbuilder;
266
    }
267
    
268
    public void performUpdates(Connection conn,
269
            TableReference table,
270
            FeatureType type,
271
            Iterator<FeatureProvider> updateds) throws DataException {
272

    
273
        JDBCSQLBuilderBase sqlbuilder = buildUpdateSQL(table, type);
274
        
275
        PreparedStatement st = null;
276
        Disposable paramsDisposer = null;
277
        sqlbuilder.setProperties(
278
                Variable.class, 
279
                PROP_FEATURE_TYPE, type,
280
                PROP_TABLE, table
281
        );
282
        String sql = sqlbuilder.update().toString();
283
        try {
284
            st = conn.prepareStatement(sql);
285
            while (updateds.hasNext()) {
286
                FeatureProvider featureProvider = (FeatureProvider) updateds.next();
287
                paramsDisposer = sqlbuilder.setParameters(st, featureProvider);
288
                if (JDBCUtils.executeUpdate(st,sql) == 0) {
289
                    throw new JDBCUpdateWithoutChangesException(sql,null);
290
                }
291
            }
292
        } catch (SQLException e) {
293
            throw new JDBCSQLException(e);
294
        } finally {
295
            JDBCUtils.closeQuietly(st);
296
            DisposeUtils.disposeQuietly(paramsDisposer);
297
        }
298
    }
299

    
300
    protected List<String> buildUpdateTableSQL(
301
            TableReference table,
302
            FeatureType original,
303
            FeatureType target
304
        ) {
305
        JDBCSQLBuilderBase sqlbuilder = createSQLBuilder();
306
        sqlbuilder.alter_table().table()
307
                .database(this.table.getDatabase())
308
                .schema(this.table.getSchema())
309
                .name(this.table.getTable());
310

    
311
        for (FeatureAttributeDescriptor attrOrgiginal : original) {
312
            FeatureAttributeDescriptor attrTarget = target.getAttributeDescriptor(
313
                    attrOrgiginal.getName()
314
            );
315
            if (attrTarget == null) {
316
                if( attrOrgiginal.isComputed() ) {
317
                    continue;
318
                }
319
                sqlbuilder.alter_table().drop_column(attrOrgiginal.getName());
320
            } else if( !this.areEquals(attrOrgiginal, attrTarget) ) {
321
                if( attrTarget.isComputed() ) {
322
                    continue;
323
                }
324
                if( attrTarget.getType()==DataTypes.GEOMETRY ) {
325
                    sqlbuilder.alter_table().alter_geometry_column(
326
                            attrTarget.getName(),
327
                            attrTarget.getGeomType().getType(),
328
                            attrTarget.getGeomType().getSubType(),
329
                            attrTarget.getSRS(),
330
                            attrTarget.isIndexed(),
331
                            attrTarget.allowNull()
332
                    );
333
                } else {
334
                    sqlbuilder.alter_table().alter_column(
335
                            attrTarget.getName(),
336
                            attrTarget.getType(),
337
                            attrTarget.getSize(),
338
                            attrTarget.getPrecision(),
339
                            attrTarget.getScale(),
340
                            attrTarget.isPrimaryKey(),
341
                            attrTarget.isIndexed(),
342
                            attrTarget.allowNull(),
343
                            attrTarget.isAutomatic(),
344
                            attrTarget.getDefaultValue()
345
                    );
346
                }
347
            }
348
        }
349
        for (FeatureAttributeDescriptor attrTarget : target) {
350
            if( attrTarget.isComputed() ) {
351
                continue;
352
            }
353
            if (original.getAttributeDescriptor(attrTarget.getName()) == null) {
354
                if( attrTarget.getType()==DataTypes.GEOMETRY ) {
355
                    sqlbuilder.alter_table().add_geometry_column(
356
                            attrTarget.getName(),
357
                            attrTarget.getGeomType().getType(),
358
                            attrTarget.getGeomType().getSubType(),
359
                            attrTarget.getSRS(),
360
                            attrTarget.isIndexed(),
361
                            attrTarget.allowNull()
362
                    );
363
                } else {
364
                    sqlbuilder.alter_table().add_column(
365
                            attrTarget.getName(),
366
                            attrTarget.getType(),
367
                            attrTarget.getSize(),
368
                            attrTarget.getPrecision(),
369
                            attrTarget.getScale(),
370
                            attrTarget.isPrimaryKey(),
371
                            attrTarget.isIndexed(),
372
                            attrTarget.allowNull(),
373
                            attrTarget.isAutomatic(),
374
                            attrTarget.getDefaultValue()
375
                    );
376
                }
377
            }
378
        }
379
        sqlbuilder.setProperties(
380
                Variable.class, 
381
                PROP_TABLE, table
382
        );
383
        return sqlbuilder.alter_table().toStrings();
384
    }
385
    
386
    protected boolean areEquals(FeatureAttributeDescriptor attr1, FeatureAttributeDescriptor attr2) {
387
        // No interesa si son o no iguales en general, solo si son iguales en lo 
388
        // que a los atributos usados para crear la columna de la tabla se refiere.
389
        if( !StringUtils.equals(attr1.getName(), attr2.getName()) ) {
390
            return false;
391
        }
392
        if( attr1.getType() != attr2.getType() ) {
393
            return false;
394
        }
395
        if( attr1.getPrecision() != attr2.getPrecision() ) {
396
            return false;
397
        }
398
        if( attr1.getScale() != attr2.getScale() ) {
399
            return false;
400
        }
401
        if( attr1.getSize() != attr2.getSize() ) {
402
            return false;
403
        }
404
        if( attr1.isPrimaryKey() != attr2.isPrimaryKey() ) {
405
            return false;
406
        }        
407
//        if( attr1.isIndexed() != attr2.isIndexed() ) {
408
//            return false;
409
//        }
410
        if( attr1.allowNull() != attr2.allowNull() ) {
411
            return false;
412
        }
413
        if( attr1.isAutomatic() != attr2.isAutomatic() ) {
414
            return false;
415
        }
416
        if( attr1.getDefaultValue() != attr2.getDefaultValue() ) {
417
            if( attr1.getDefaultValue()==null || attr2.getDefaultValue()==null) {
418
                return false;
419
            }
420
            if( !attr1.getDefaultValue().equals(attr2.getDefaultValue()) ) {
421
                return false;
422
            }
423
        }
424
        return true;
425
    }
426

    
427
    protected List<String> buildCreateIndexSQL(
428
            TableReference table,
429
            FeatureType original,
430
            FeatureType target
431
        ) {
432
        ArrayList<String> sqls = new ArrayList<>();
433
        
434
        for (FeatureAttributeDescriptor attrTarget : target) {
435
            boolean createIndex = false;
436
            if( attrTarget.isIndexed() ) {
437
                FeatureAttributeDescriptor attrOriginal = original.getAttributeDescriptor(attrTarget.getName());
438
                if ( attrOriginal == null) {
439
                    createIndex = true;
440
                } else {
441
                    if( attrOriginal.isIndexed() ) {
442
                        createIndex = false;
443
                    } else {
444
                        createIndex = true;
445
                    }
446
                }
447
            }
448
            if( createIndex ) {
449
                JDBCSQLBuilderBase sqlbuilder = createSQLBuilder();
450
                if( attrTarget.getType()==DataTypes.GEOMETRY ) {
451
                    sqlbuilder.create_index().spatial();
452
                }
453
                sqlbuilder.create_index().if_not_exist();
454
                sqlbuilder.create_index().name("idx_" + table + "_" + attrTarget.getName());
455
                sqlbuilder.create_index().column(attrTarget.getName());
456
                sqlbuilder.create_index().table()
457
                    .database(this.table.getDatabase())
458
                    .schema(this.table.getSchema())
459
                    .name(this.table.getTable());
460
                sqlbuilder.setProperties(
461
                        Variable.class, 
462
                        PROP_TABLE, table
463
                );
464
                sqls.addAll(sqlbuilder.create_index().toStrings());
465
            }
466
        }
467
        return sqls;
468
    }
469
    
470
    public void performUpdateTable(Connection conn,
471
            TableReference table,
472
            FeatureType original,
473
            FeatureType target) throws DataException {
474
        
475
        ArrayList<String> sqls = new ArrayList<>();
476
        
477
        sqls.addAll(buildUpdateTableSQL(table, original, target));
478
        sqls.addAll(buildCreateIndexSQL(table, original, target));
479

    
480
        if( !CollectionUtils.isEmpty(sqls) ) {
481
            Statement st = null;
482
            try {
483
                st = conn.createStatement();
484
                for (String sql : sqls) {
485
                    if( !StringUtils.isBlank(sql) ) {
486
                        JDBCUtils.execute(st, sql);
487
                    }
488
                }
489
            } catch (SQLException e) {
490
                throw new JDBCSQLException(e);
491
            } finally {
492
                JDBCUtils.closeQuietly(st);
493
            }
494
        }
495
    }
496
}