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

History | View | Annotate | Download (18.8 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
@SuppressWarnings("UseSpecificCatch")
36
public class PerformChangesOperation extends AbstractConnectionWritableOperation {
37

    
38
    protected TableReference table;
39
    protected FeatureType featureType;
40
    protected FeatureType featureTypeSource;
41
    protected FeatureType featureTypeTarget;
42
    protected Iterator<FeatureReferenceProviderServices> deleteds;
43
    protected Iterator<FeatureProvider> updateds;
44
    protected Iterator<FeatureProvider> inserteds;
45

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

    
52
    public PerformChangesOperation(JDBCHelper helper,
53
            TableReference table,
54
            FeatureType featureType,
55
            Iterator<FeatureReferenceProviderServices> deleteds,
56
            Iterator<FeatureProvider> inserteds,
57
            Iterator<FeatureProvider> updateds,
58
            Iterator<FeatureStoreProvider.FeatureTypeChanged> featureTypesChanged) {
59
        super(helper);
60
        this.deleteds = deleteds;
61
        this.inserteds = inserteds;
62
        this.updateds = updateds;
63
        this.table = table;
64
        this.featureType = featureType;
65
        if (featureTypesChanged.hasNext()) {
66
            FeatureStoreProvider.FeatureTypeChanged item = featureTypesChanged.next();
67
            this.featureTypeSource = item.getSource();
68
            this.featureTypeTarget = item.getTarget();
69
            typeChanged = true;
70
        } else {
71
            this.featureTypeSource = null;
72
            this.featureTypeTarget = null;
73
            typeChanged = false;
74
        }
75
    }
76

    
77
    public boolean isTypeChanged() {
78
        return typeChanged;
79
    }
80

    
81
    @Override
82
    public Object perform(Connection conn) throws DataException {
83
        if (typeChanged) {
84
            this.performUpdateTable(conn);
85
        }
86
        if (deleteds.hasNext()) {
87
            performDeletes(conn);
88
        }
89
        if (updateds.hasNext()) {
90
            performUpdates(conn);
91
        }
92
        if (inserteds.hasNext()) {
93
            performInserts(conn);
94
        }
95
        return true;
96
    }
97
    
98
    public String getDeleteSQL() {
99
        JDBCSQLBuilderBase sqlbuilder = createSQLBuilder();
100
        return this.getDeleteSQL(sqlbuilder);
101
    }
102
    
103
    public String getDeleteSQL(JDBCSQLBuilderBase sqlbuilder) {
104
        ExpressionBuilder expbuilder = sqlbuilder.expression();
105

    
106
        sqlbuilder.delete().table()
107
                .database(this.table.getDatabase())
108
                .schema(this.table.getSchema())
109
                .name(this.table.getTable());
110
        for (FeatureAttributeDescriptor attr : this.featureType) {
111
            if (attr.isPrimaryKey()) {
112
                sqlbuilder.delete().where().and(
113
                        expbuilder.eq(
114
                                expbuilder.column(attr.getName()),
115
                                expbuilder.parameter(attr.getName()).as_variable()
116
                        )
117
                );
118
            }
119
        }
120
        if (!sqlbuilder.delete().has_where() ) {
121
            throw new RuntimeException("Operation requires missing pk");
122
        }
123
        sqlbuilder.setProperties(
124
                Variable.class, 
125
                PROP_TABLE, table
126
        );
127
        String sql = sqlbuilder.delete().toString();
128
        return sql;
129
    }
130

    
131
    public void performDeletes(Connection conn) throws DataException {
132

    
133
        JDBCSQLBuilderBase sqlbuilder = createSQLBuilder();
134
        String sql = getDeleteSQL(sqlbuilder);
135

    
136
        PreparedStatement st = null;
137
        Disposable paramsDisposer = null;
138
        try {
139
            st = conn.prepareStatement(sql);
140
            while (deleteds.hasNext()) {
141
                FeatureReference reference = (FeatureReference) deleteds.next();
142
                paramsDisposer = sqlbuilder.setParameters(st, reference);
143
                int nAffected = JDBCUtils.executeUpdate(st,sql);
144
                if (nAffected == 0) {
145
                    throw new JDBCUpdateWithoutChangesException(
146
                            sqlbuilder.delete().toString(),
147
                            null
148
                    );
149
                }
150
                if (nAffected > 1) {
151
                    LOGGER.warn("Remove statement affectst to {} rows ( {} )",
152
                            nAffected, sql
153
                    );
154
                }
155
            }
156
        } catch (SQLException e) {
157
            throw new JDBCSQLException(e);
158
        } finally {
159
            JDBCUtils.closeQuietly(st);
160
            DisposeUtils.disposeQuietly(paramsDisposer);
161
        }
162
    }
163
    
164
    public String getInsertSQL() {
165
        JDBCSQLBuilderBase sqlbuilder = createSQLBuilder();
166
        return this.getInsertSQL(sqlbuilder);
167
    }
168
    
169
    public String getInsertSQL(JDBCSQLBuilderBase sqlbuilder) {
170
        GeometryExpressionBuilder expbuilder = sqlbuilder.expression();
171

    
172
        sqlbuilder.insert().table()
173
                .database(this.table.getDatabase())
174
                .schema(this.table.getSchema())
175
                .name(this.table.getTable());
176
        for (FeatureAttributeDescriptor attr : this.featureType) {
177
            if( attr.isAutomatic() || attr.isComputed() ) {
178
                continue;
179
            }
180
            if (attr.getType() == DataTypes.GEOMETRY) {
181
                sqlbuilder.insert().column().name(attr.getName()).with_value(
182
                    expbuilder.parameter(attr.getName()).as_variable()
183
                        .srs(
184
                            expbuilder.parameter().value(
185
                                attr.getSRS()).as_constant()
186
                        )
187
                );
188
            } else {
189
                sqlbuilder.insert().column().name(attr.getName()).with_value(
190
                        expbuilder.parameter(attr.getName())
191
                );
192
            }
193
        }
194

    
195
        sqlbuilder.setProperties(
196
                Variable.class, 
197
                PROP_FEATURE_TYPE, featureType,
198
                PROP_TABLE, table
199
        );
200
        String sql = sqlbuilder.insert().toString();
201
        return sql;
202
    }
203

    
204
    public void performInserts(Connection conn) throws DataException {
205
        JDBCSQLBuilderBase sqlbuilder = createSQLBuilder();
206
        String sql = getInsertSQL(sqlbuilder);
207
        
208
        PreparedStatement st;
209
        Disposable paramsDisposer;
210
        try {
211
            st = conn.prepareStatement(sql);
212
            while (inserteds.hasNext()) {
213
                FeatureProvider feature = inserteds.next();
214
                paramsDisposer = sqlbuilder.setParameters(st, feature);
215
                try {
216
                    if (JDBCUtils.executeUpdate(st,sql) == 0) {
217
                        throw new JDBCExecuteSQLException(
218
                                sqlbuilder.insert().toString(),
219
                                null
220
                        );
221
                    }
222
                } finally {
223
                    DisposeUtils.disposeQuietly(paramsDisposer);
224
                }
225
            }
226
        } catch (JDBCExecuteSQLException ex) {
227
            throw ex;
228
        } catch (Exception ex) {
229
            throw new JDBCExecuteSQLException(sql,ex);
230
        }
231
    }
232

    
233
    public String getUpdateSQL() {
234
        JDBCSQLBuilderBase sqlbuilder = createSQLBuilder();
235
        return this.getUpdateSQL(sqlbuilder);
236
    }
237
    
238
    public String getUpdateSQL(JDBCSQLBuilderBase sqlbuilder) {
239
        GeometryExpressionBuilder expbuilder = sqlbuilder.expression();
240

    
241
        sqlbuilder.update().table()
242
                .database(this.table.getDatabase())
243
                .schema(this.table.getSchema())
244
                .name(this.table.getTable());
245
        for (FeatureAttributeDescriptor attr : this.featureType) {
246
            if (attr.isPrimaryKey()) {
247
                sqlbuilder.update().where().and(
248
                        expbuilder.eq(
249
                                expbuilder.column(attr.getName()),
250
                                expbuilder.parameter(attr.getName()).as_variable()
251
                        )
252
                );
253
                continue;
254
            } 
255
            if ( attr.isAutomatic() || attr.isReadOnly() || attr.isComputed() ) {
256
                continue;
257
            }
258
            if (attr.getType() == DataTypes.GEOMETRY) {
259
                sqlbuilder.update().column().name(attr.getName()).with_value(
260
                    expbuilder.parameter(attr.getName()).as_geometry_variable().srs(
261
                        expbuilder.parameter().value(attr.getSRS()).as_constant()
262
                    )
263
                );
264
            } else {
265
                sqlbuilder.update().column().name(attr.getName()).with_value(
266
                    expbuilder.parameter(attr.getName()).as_variable()
267
                );
268
            }
269
        }
270
        if (!sqlbuilder.update().has_where() ) {
271
            throw new RuntimeException("Operation requires missing pk");
272
        }
273
        sqlbuilder.setProperties(
274
                Variable.class, 
275
                PROP_FEATURE_TYPE, this.featureType,
276
                PROP_TABLE, table
277
        );
278
        String sql = sqlbuilder.update().toString();
279
        return sql;
280
    }
281
    
282
    public void performUpdates(Connection conn) throws DataException {
283

    
284
        JDBCSQLBuilderBase sqlbuilder = createSQLBuilder();
285
        String sql = getInsertSQL(sqlbuilder);
286
        
287
        PreparedStatement st = null;
288
        Disposable paramsDisposer = null;
289
        try {
290
            st = conn.prepareStatement(sql);
291
            while (updateds.hasNext()) {
292
                FeatureProvider featureProvider = (FeatureProvider) updateds.next();
293
                paramsDisposer = sqlbuilder.setParameters(st, featureProvider);
294
                if (JDBCUtils.executeUpdate(st,sql) == 0) {
295
                    throw new JDBCUpdateWithoutChangesException(sql,null);
296
                }
297
            }
298
        } catch (SQLException e) {
299
            throw new JDBCSQLException(e);
300
        } finally {
301
            JDBCUtils.closeQuietly(st);
302
            DisposeUtils.disposeQuietly(paramsDisposer);
303
        }
304
    }
305

    
306
    public List<String> getUpdateTableSQLs() {
307
        JDBCSQLBuilderBase sqlbuilder = createSQLBuilder();
308
        sqlbuilder.alter_table().table()
309
                .database(this.table.getDatabase())
310
                .schema(this.table.getSchema())
311
                .name(this.table.getTable());
312

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

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

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