Statistics
| Revision:

svn-gvsig-desktop / trunk / libraries / libFMap / src / com / iver / cit / gvsig / fmap / drivers / DefaultJDBCDriver.java @ 28962

History | View | Annotate | Download (32.3 KB)

1 11971 caballero
/* gvSIG. Sistema de Informaci?n Geogr?fica de la Generalitat Valenciana
2
 *
3
 * Copyright (C) 2004 IVER T.I. and Generalitat Valenciana.
4
 *
5
 * This program is free software; you can redistribute it and/or
6
 * modify it under the terms of the GNU General Public License
7
 * as published by the Free Software Foundation; either version 2
8
 * of the License, or (at your option) any later version.
9
 *
10
 * This program is distributed in the hope that it will be useful,
11
 * but WITHOUT ANY WARRANTY; without even the implied warranty of
12
 * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the
13
 * GNU General Public License for more details.
14
 *
15
 * You should have received a copy of the GNU General Public License
16
 * along with this program; if not, write to the Free Software
17
 * Foundation, Inc., 59 Temple Place - Suite 330, Boston, MA  02111-1307,USA.
18
 *
19
 * For more information, contact:
20
 *
21
 *  Generalitat Valenciana
22
 *   Conselleria d'Infraestructures i Transport
23
 *   Av. Blasco Ib??ez, 50
24
 *   46010 VALENCIA
25
 *   SPAIN
26
 *
27
 *      +34 963862235
28
 *   gvsig@gva.es
29
 *      www.gvsig.gva.es
30
 *
31
 *    or
32
 *
33
 *   IVER T.I. S.A
34
 *   Salamanca 50
35
 *   46005 Valencia
36
 *   Spain
37
 *
38
 *   +34 963163400
39
 *   dac@iver.es
40
 */
41
package com.iver.cit.gvsig.fmap.drivers;
42
43
import java.awt.geom.Rectangle2D;
44
import java.sql.DatabaseMetaData;
45
import java.sql.Driver;
46
import java.sql.DriverManager;
47
import java.sql.ResultSet;
48
import java.sql.ResultSetMetaData;
49
import java.sql.SQLException;
50
import java.sql.Statement;
51
import java.sql.Types;
52 12247 caballero
import java.util.ArrayList;
53 11971 caballero
import java.util.Hashtable;
54
import java.util.TreeMap;
55
56
import org.apache.log4j.Logger;
57
58 12247 caballero
import com.hardcode.driverManager.DriverEvent;
59
import com.hardcode.driverManager.DriverEventListener;
60 11971 caballero
import com.hardcode.gdbms.driver.exceptions.BadFieldDriverException;
61
import com.hardcode.gdbms.driver.exceptions.ReadDriverException;
62
import com.hardcode.gdbms.driver.exceptions.ReloadDriverException;
63
import com.hardcode.gdbms.engine.data.DataSourceFactory;
64
import com.hardcode.gdbms.engine.data.driver.DriverException;
65
import com.hardcode.gdbms.engine.data.driver.ObjectDriver;
66
import com.hardcode.gdbms.engine.values.Value;
67
import com.hardcode.gdbms.engine.values.ValueFactory;
68
import com.iver.cit.gvsig.exceptions.expansionfile.ExpansionFileReadException;
69
import com.iver.cit.gvsig.fmap.Messages;
70
import com.iver.cit.gvsig.fmap.core.IFeature;
71
import com.iver.cit.gvsig.fmap.core.IGeometry;
72
import com.iver.cit.gvsig.fmap.drivers.db.utils.ConnectionWithParams;
73
import com.iver.cit.gvsig.fmap.drivers.db.utils.SingleVectorialDBConnectionManager;
74
import com.iver.cit.gvsig.fmap.layers.SelectableDataSource;
75
import com.iver.cit.gvsig.fmap.layers.XMLException;
76
import com.iver.utiles.XMLEntity;
77
import com.iver.utiles.swing.JPasswordDlg;
78
79
80
81
/**
82
 * Clase abstracta qu
83
 */
84
public abstract class DefaultJDBCDriver implements IVectorialJDBCDriver, ObjectDriver {
85
    private static Logger logger = Logger.getLogger(SelectableDataSource.class.getName());
86
    protected static Hashtable poolPassw = new Hashtable();
87
88
    protected IConnection conn;
89
    // protected String tableName;
90
    // protected String whereClause;
91
    // protected String fields;
92
    // protected String sqlOrig;
93
    protected DBLayerDefinition lyrDef = null;
94
    protected ResultSet rs;
95
    protected boolean bCursorActivo = false;
96
    protected Statement st;
97
    protected int numReg=-1;
98
99
    protected Rectangle2D fullExtent = null;
100
101
    // protected String strFID_FieldName;
102
    // protected String idFID_FieldName;
103
104
    protected Hashtable hashRelate;
105
106
107
    protected ResultSetMetaData metaData = null;
108
    protected Rectangle2D workingArea;
109
    protected String driverClass;
110
    protected String userName;
111 12247 caballero
    protected String dbUrl;
112 11971 caballero
    protected String className;
113
    protected String catalogName;
114 12247 caballero
    protected String schema;
115 11971 caballero
    protected String tableName;
116
    protected String[] fields;
117
    protected String FIDfield;
118
    protected String geometryField;
119
    protected String whereClause;
120
    protected String strSRID;
121
        //private double flatness;
122
123
    protected String host, port, dbName, connName;
124
125 12247 caballero
    protected ArrayList driverEventListeners = new ArrayList();
126 11971 caballero
127 17163 vcaballero
    /* (non-Javadoc)
128
     * @see com.iver.cit.gvsig.fmap.drivers.IVectorialDatabaseDriver#getConnection()
129
     */
130
    public IConnection getConnection()
131 11971 caballero
        {
132
            return conn;
133
        }
134 17163 vcaballero
        /* (non-Javadoc)
135
         * @see com.iver.cit.gvsig.fmap.drivers.IVectorialDatabaseDriver#getFields()
136
         */
137 11971 caballero
        public String[] getFields()
138
        {
139
        /* StringTokenizer tokenizer = new StringTokenizer(fields, ",");
140
        String[] arrayFields = new String[tokenizer.countTokens()];
141
        int i=0;
142
        while (tokenizer.hasMoreTokens())
143
        {
144
            arrayFields[i] = tokenizer.nextToken();
145
            i++;
146
        }
147
            return arrayFields; */
148
        return lyrDef.getFieldNames();
149
150
        }
151
    /**
152
     * First, the geometry field. After, the rest of fields
153
     * @return
154
     */
155
    public String getTotalFields()
156
    {
157
        String strAux = getGeometryField(getLyrDef().getFieldGeometry());
158
        String[] fieldNames = getLyrDef().getFieldNames();
159
        for (int i=0; i< fieldNames.length; i++)
160
        {
161
            strAux = strAux + ", " + fieldNames[i];
162
        }
163
        return strAux;
164
    }
165
166
        /* (non-Javadoc)
167
         * @see com.iver.cit.gvsig.fmap.drivers.VectorialDatabaseDriver#getWhereClause()
168
         */
169
        public String getWhereClause()
170
        {
171
            return lyrDef.getWhereClause().toUpperCase();
172
        }
173 17163 vcaballero
        /* (non-Javadoc)
174
         * @see com.iver.cit.gvsig.fmap.drivers.IVectorialDatabaseDriver#getTableName()
175
         */
176 11971 caballero
        public String getTableName()
177
        {
178
            return lyrDef.getTableName();
179
        }
180
181
182
        /**
183
         * @throws DriverIOException
184
         * @throws DriverException
185
         * @see com.iver.cit.gvsig.fmap.layers.ReadableVectorial#getShapeCount()
186
         */
187
        public int getShapeCount() throws ReadDriverException {
188
                    if (numReg == -1)
189
                    {
190
                        try
191
                    {
192
                            Statement s = ((ConnectionJDBC)conn).getConnection().createStatement();
193
                            ResultSet r = s.executeQuery("SELECT COUNT(*) AS NUMREG FROM " + lyrDef.getTableName() + " " + getCompleteWhere());
194
                            r.next();
195
                            numReg = r.getInt(1);
196
                            System.err.println("numReg = " + numReg);
197
                    }
198
                        catch (SQLException e)
199
                        {
200
                            throw new ReadDriverException(getName(),e);
201
                        }
202
                    }
203
204
            return numReg;
205
        }
206
207 17163 vcaballero
         /* (non-Javadoc)
208
         * @see com.iver.cit.gvsig.fmap.drivers.VectorialDriver#getFullExtent()
209
         */
210
        public Rectangle2D getFullExtent() throws ExpansionFileReadException, ReadDriverException{
211 11971 caballero
                // Por defecto recorremos todas las geometrias.
212
                // Las bases de datos como PostGIS pueden y deben
213
                // sobreescribir este m?todo.
214
                if (fullExtent == null)
215
                {
216
//                    try
217
//                    {
218
                        IFeatureIterator itGeom = getFeatureIterator("SELECT " +
219
                                getGeometryField(getLyrDef().getFieldGeometry()) + ", " + getLyrDef().getFieldID() + " FROM " +
220
                                getLyrDef().getComposedTableName() +  " " + getCompleteWhere());
221
                        IGeometry geom;
222
                        int cont = 0;
223
                        while (itGeom.hasNext())
224
                        {
225
                            geom = itGeom.next().getGeometry();
226
                            if (cont==0)
227
                                fullExtent = geom.getBounds2D();
228
                            else
229
                                fullExtent.add(geom.getBounds2D());
230
                            cont++;
231
                        }
232
//                    }
233
//                    catch (DriverException e) {
234
//                        // TODO Auto-generated catch block
235
//                        e.printStackTrace();
236
//                    }
237
238
                }
239
                return fullExtent;
240
            }
241
242
243 17163 vcaballero
        /* (non-Javadoc)
244
         * @see com.iver.cit.gvsig.fmap.drivers.VectorialDriver#getShapeType()
245 11971 caballero
         */
246
        public int getShapeType() {
247
        /* IGeometry geom;
248
        if (shapeType == -1)
249
        {
250
                shapeType = FShape.MULTI;
251
                try {
252
                        geom = getShape(0);
253
                        if (geom != null)
254
                                shapeType = geom.getGeometryType();
255
                } catch (IOException e) {
256
                        // e.printStackTrace();
257
                }
258
        }
259
        return shapeType; */
260
                return lyrDef.getShapeType();
261
        }
262
263 17163 vcaballero
        /* (non-Javadoc)
264
         * @see com.hardcode.gdbms.engine.data.driver.ReadAccess#getFieldType(int)
265
         */
266 11971 caballero
        public int getFieldType(int idField) throws ReadDriverException{
267
            try {
268
                int i = idField + 2; // idField viene basado en 1, y
269
                                        // adem?s nos saltamos el campo de geometry
270 12299 caballero
                int type=metaData.getColumnType(i);
271
                if (type == Types.VARCHAR)
272 11971 caballero
                return Types.VARCHAR;
273 12299 caballero
                    if (type == Types.FLOAT)
274 11971 caballero
                        return Types.FLOAT;
275 12299 caballero
                    if (type == Types.REAL)
276
                        return Types.FLOAT;
277
                    if (type == Types.DOUBLE)
278 11971 caballero
                        return Types.DOUBLE;
279 12299 caballero
                    if (type == Types.INTEGER)
280 11971 caballero
                        return Types.INTEGER;
281 12299 caballero
                    if (type == Types.SMALLINT)
282 11971 caballero
                        return Types.SMALLINT;
283 12299 caballero
                    if (type == Types.TINYINT)
284 11971 caballero
                        return Types.TINYINT;
285 12299 caballero
                    if (type == Types.BIGINT)
286 11971 caballero
                        return Types.BIGINT;
287 12299 caballero
                    if (type == Types.BIT)
288 11971 caballero
                        return Types.BIT;
289 12299 caballero
                    if (type == Types.DATE)
290 11971 caballero
                        return Types.DATE;
291 12299 caballero
            if (type == Types.DECIMAL)
292 11971 caballero
                return Types.DOUBLE;
293 12299 caballero
            if (type == Types.NUMERIC)
294 11971 caballero
                return Types.DOUBLE;
295 12299 caballero
            if (type == Types.DATE)
296 11971 caballero
                return Types.DATE;
297 12299 caballero
            if (type == Types.TIME)
298 11971 caballero
                return Types.TIME;
299 12299 caballero
            if (type == Types.TIMESTAMP)
300 11971 caballero
                return Types.TIMESTAMP;
301 14543 jmvivo
            if (type == Types.NUMERIC)
302
                        return Types.DOUBLE;
303 15685 vcaballero
304 12299 caballero
        } catch (SQLException e) {
305 11971 caballero
                    throw new BadFieldDriverException(getName(),e,String.valueOf(idField));
306
            }
307
            return Types.OTHER;
308 12299 caballero
         }
309 11971 caballero
    /**
310
     * Obtiene el valor que se encuentra en la fila y columna indicada
311
     * Esta es la implementaci?n por defecto. Si lo del absolute
312
     * no va bien, como es el caso del PostGis, el driver lo
313
     * tiene que reimplementar
314
     *
315
     * @param rowIndex fila
316
     * @param fieldId columna
317
     *
318
     * @return subclase de Value con el valor del origen de datos
319
     *
320
     * @throws DriverException Si se produce un error accediendo al DataSource
321
     */
322
    public Value getFieldValue(long rowIndex, int idField)
323
        throws ReadDriverException
324
        {
325
                int i = (int) (rowIndex + 1);
326
                int fieldId = idField+2;
327
                try {
328
                    rs.absolute(i);
329
                if (metaData.getColumnType(fieldId) == Types.VARCHAR)
330
                {
331
                    String strAux = rs.getString(fieldId);
332
                    if (strAux == null) strAux = "";
333
                    return ValueFactory.createValue(strAux);
334
                }
335
                        if (metaData.getColumnType(fieldId) == Types.FLOAT)
336
                            return ValueFactory.createValue(rs.getFloat(fieldId));
337
                        if (metaData.getColumnType(fieldId) == Types.DOUBLE)
338
                            return ValueFactory.createValue(rs.getDouble(fieldId));
339
                        if (metaData.getColumnType(fieldId) == Types.INTEGER)
340
                            return ValueFactory.createValue(rs.getInt(fieldId));
341
                        if (metaData.getColumnType(fieldId) == Types.BIGINT)
342
                            return ValueFactory.createValue(rs.getLong(fieldId));
343
                        if (metaData.getColumnType(fieldId) == Types.BIT)
344
                            return ValueFactory.createValue(rs.getBoolean(fieldId));
345
                        if (metaData.getColumnType(fieldId) == Types.DATE)
346
                            return ValueFactory.createValue(rs.getDate(fieldId));
347
                } catch (SQLException e) {
348
                throw new BadFieldDriverException(getName(),e,String.valueOf(fieldId));
349
                }
350
                return null;
351
352
353
        }
354
355
    /**
356
     * Obtiene el n?mero de campos del DataSource
357
     *
358
     * @return
359
     *
360
     * @throws DriverException Si se produce alg?n error accediendo al
361
     *         DataSource
362
     */
363
    public int getFieldCount() throws ReadDriverException
364
    {
365
        try {
366
            // Suponemos que el primer campo es el de las geometries, y no lo
367
            // contamos
368
            return rs.getMetaData().getColumnCount()-1;
369
        } catch (SQLException e) {
370
            throw new ReadDriverException(getName(),e);
371
        }
372
373
    }
374
375
    /**
376
     * Devuelve el nombre del campo fieldId-?simo
377
     *
378
     * @param fieldId ?ndice del campo cuyo nombre se quiere obtener
379
     *
380
     * @return
381
     * @throws com.hardcode.gdbms.engine.data.driver.DriverException
382
     *
383
     * @throws DriverException Si se produce alg?n error accediendo al
384
     *         DataSource
385
     */
386
    public String getFieldName(int fieldId) throws ReadDriverException
387
    {
388
        try {
389
            return rs.getMetaData().getColumnName(fieldId+2);
390
        } catch (SQLException e) {
391
            throw new ReadDriverException(getName(),e);
392
        }
393
    }
394
395
    /**
396
     * Obtiene el n?mero de registros del DataSource
397
     *
398
     * @return
399
     * @throws ReadDriverException
400
     *
401
     * @throws DriverException Si se produce alg?n error accediendo al
402
     *         DataSource
403
     */
404
    public long getRowCount() throws ReadDriverException{
405
        return getShapeCount();
406
    }
407
408 17163 vcaballero
    /* (non-Javadoc)
409
     * @see com.iver.cit.gvsig.fmap.drivers.IVectorialDatabaseDriver#close()
410
     */
411 11971 caballero
    public void close() {
412
    }
413
414
    /**
415
     * Recorre el recordset creando una tabla Hash que usaremos para
416
     * relacionar el n?mero de un registro con su identificador ?nico.
417
     * Debe ser llamado en el setData justo despu?s de crear el recorset
418
     * principal
419
     * @throws SQLException
420
     */
421 12351 caballero
    protected void doRelateID_FID() throws DBException
422 11971 caballero
    {
423 12351 caballero
        try {
424
                        hashRelate = new Hashtable();
425 11971 caballero
426 12351 caballero
                        String strSQL = "SELECT " + getLyrDef().getFieldID() + " FROM "
427
                                        + getLyrDef().getComposedTableName() + " "
428
                                        + getCompleteWhere() + " ORDER BY "
429
                                        + getLyrDef().getFieldID();
430
                        Statement s = ((ConnectionJDBC) getConnection()).getConnection()
431
                                        .createStatement(ResultSet.TYPE_SCROLL_INSENSITIVE,
432
                                                        ResultSet.CONCUR_READ_ONLY);
433
                        ResultSet r = s.executeQuery(strSQL);
434
                        int id = 0;
435
                        int gid;
436
                        int index = 0;
437
                        while (r.next()) {
438
                                String aux = r.getString(1);
439
                                Value val = ValueFactory.createValue(aux);
440
                                hashRelate.put(val, new Integer(index));
441
                                System.out.println("ASOCIANDO CLAVE " + aux + " CON VALOR "
442
                                                + index);
443
                                index++;
444
                        }
445
                        numReg = index;
446
                        r.close();
447
                        // rs.beforeFirst();
448
                } catch (SQLException e) {
449
                        throw new DBException(e);
450
                }
451 11971 caballero
452
    }
453
454 12351 caballero
    /*
455
         * (non-Javadoc)
456
         *
457
         * @see com.iver.cit.gvsig.fmap.drivers.VectorialDatabaseDriver#getRowIndexByFID(java.lang.Object)
458
         */
459 11971 caballero
    public int getRowIndexByFID(IFeature FID)
460
    {
461
        int resul;
462
        // Object obj = FID.getAttribute(lyrDef.getIdFieldID());
463
        String theId = FID.getID();
464
        Value aux = ValueFactory.createValue(theId);
465
        // System.err.println("Mirando si existe " + aux.toString());
466
        if (hashRelate.containsKey(aux))
467
        {
468
                Integer rowIndex = (Integer) hashRelate.get(aux);
469
                resul = rowIndex.intValue();
470
                // System.err.println("Row asociada a " + aux.toString() + ":" + resul);
471
                return resul;
472
        }
473
        else
474
                return -1;
475
    }
476
477
    /* (non-Javadoc)
478
     * @see com.iver.cit.gvsig.fmap.drivers.VectorialDatabaseDriver#setXMLEntity(com.iver.utiles.XMLEntity)
479
     */
480
    public void setXMLEntity(XMLEntity xml) throws XMLException
481
    {
482
483
        className = xml.getStringProperty("className");
484
        // dbUrl = xml.getStringProperty("dbURL");
485
        catalogName = xml.getStringProperty("catalog");
486
        userName =xml.getStringProperty("username");
487
        driverClass =xml.getStringProperty("driverclass");
488
        tableName = xml.getStringProperty("tablename");
489 12247 caballero
        if (xml.contains("schema")){
490
                schema = xml.getStringProperty("schema");
491
        }
492 11971 caballero
        fields = xml.getStringArrayProperty("fields");
493
        FIDfield = xml.getStringProperty("FID");
494
        geometryField = xml.getStringProperty("THE_GEOM");
495
        whereClause = xml.getStringProperty("whereclause");
496
        strSRID = xml.getStringProperty("SRID");
497
498 12247 caballero
       if (xml.contains("host"))
499
        {
500
                host = xml.getStringProperty("host");
501
                port = xml.getStringProperty("port");
502
                dbName = xml.getStringProperty("dbName");
503
                connName = xml.getStringProperty("connName");
504
        }
505
        else
506
        {
507
                // Por compatibilidad con versiones anteriores
508
                dbUrl = xml.getStringProperty("dbURL");
509
                extractParamsFromDbUrl(dbUrl);
510 11971 caballero
511 12247 caballero
        }
512 14543 jmvivo
        if (!xml.contains("literalDBName")){
513
                dbName = dbName.toLowerCase();
514
        }else{
515
                if(!xml.getBooleanProperty("literalDBName")){
516
                        dbName = dbName.toLowerCase();
517
                }
518
        }
519 11971 caballero
        if (xml.contains("minXworkArea"))
520
        {
521
            double x = xml.getDoubleProperty("minXworkArea");
522
            double y = xml.getDoubleProperty("minYworkArea");
523
            double H = xml.getDoubleProperty("HworkArea");
524
            double W = xml.getDoubleProperty("WworkArea");
525
            workingArea = new Rectangle2D.Double(x,y,W,H);
526
        }
527
528
        DBLayerDefinition lyrDef = new DBLayerDefinition();
529
        lyrDef.setCatalogName(catalogName);
530 12247 caballero
        lyrDef.setSchema(schema);
531 11971 caballero
        lyrDef.setTableName(tableName);
532
        lyrDef.setFieldGeometry(geometryField);
533
        lyrDef.setFieldNames(fields);
534
        lyrDef.setFieldID(FIDfield);
535
        lyrDef.setWhereClause(whereClause);
536
        // lyrDef.setClassToInstantiate(driverClass);
537
        if (workingArea != null)
538
            lyrDef.setWorkingArea(workingArea);
539
540
        lyrDef.setSRID_EPSG(strSRID);
541
542
        setLyrDef(lyrDef);
543
544
    }
545 12247 caballero
   private void extractParamsFromDbUrl(String dbUrl2) {
546
            //jdbc:postgres://localhost:5431/latin1
547
            int iDbName = dbUrl2.lastIndexOf('/');
548
                dbName = dbUrl2.substring(iDbName+1);
549
                int iLast2points = dbUrl2.lastIndexOf(':');
550
                port = dbUrl2.substring(iLast2points+1, iDbName);
551
                int iHost = dbUrl2.indexOf("//");
552
                host = dbUrl2.substring(iHost + 2, iLast2points);
553
                connName = dbUrl2;
554
        }
555 17163 vcaballero
    /* (non-Javadoc)
556
     * @see com.iver.cit.gvsig.fmap.drivers.IVectorialDatabaseDriver#load()
557
     */
558 11971 caballero
    public void load() throws ReadDriverException {
559
            try {
560 12247 caballero
                    if (driverClass != null)
561
                    Class.forName(driverClass);
562 11971 caballero
563
            String _drvName = getName();
564
565 14543 jmvivo
            String keyPool = _drvName + "_" + host
566
            + "_" + port + "_" + dbName
567
            + "_" + userName;
568 11971 caballero
569
            IConnection newConn = null;
570
            String clave = null;
571
            ConnectionWithParams cwp = null;
572
573
            if (poolPassw.containsKey(keyPool)) {
574
575
                clave = (String) poolPassw.get(keyPool);
576
                cwp = SingleVectorialDBConnectionManager.instance().getConnection(
577
                                         _drvName, userName, clave, connName,
578
                                         host, port, dbName, true);
579
580
            } else {
581
582
                cwp = SingleVectorialDBConnectionManager.instance().getConnection(
583
                                             _drvName, userName, null, connName,
584
                                             host, port, dbName, false);
585
586
                if (cwp.isConnected()) {
587
588
                        poolPassw.put(keyPool, cwp.getPw());
589
590
                } else {
591
592
                    JPasswordDlg dlg = new JPasswordDlg();
593
                    String strMessage = Messages.getString("conectar_jdbc");
594
                    String strPassword = Messages.getString("password");
595
                    dlg.setMessage(strMessage
596
                                    + " ["
597
                                    + _drvName + ", "
598
                                    + host + ", "
599
                                    + port + ", "
600
                                    + dbName + ", "
601
                                    + userName + "]. "
602
                                    + strPassword
603
                                    + "?");
604 15685 vcaballero
                    dlg.setVisible(true);
605 11971 caballero
                    clave = dlg.getPassword();
606
                    if (clave == null)
607
                        return;
608
                    poolPassw.put(keyPool, clave);
609
610
                    cwp.connect(clave);
611
                }
612
            }
613
614
            newConn = cwp.getConnection();
615 12247 caballero
            //((ConnectionJDBC)newConn).getConnection().setAutoCommit(false);
616 11971 caballero
617
            DBLayerDefinition lyrDef = new DBLayerDefinition();
618
            if (getLyrDef() == null) {
619
                    lyrDef.setCatalogName(catalogName);
620 12247 caballero
                    lyrDef.setSchema(schema);
621 11971 caballero
                    lyrDef.setTableName(tableName);
622
                    lyrDef.setFieldNames(fields);
623
                    lyrDef.setFieldID(FIDfield);
624
                    lyrDef.setFieldGeometry(geometryField);
625
                    lyrDef.setWhereClause(whereClause);
626
                    // lyrDef.setClassToInstantiate(driverClass);
627
                    if (workingArea != null)
628
                        lyrDef.setWorkingArea(workingArea);
629
630
                    lyrDef.setSRID_EPSG(strSRID);
631
            } else {
632
                    lyrDef = getLyrDef();
633
            }
634
635
            setData(newConn, lyrDef);
636
        } catch (ClassNotFoundException e) {
637
//            logger.debug(e);
638
//            DriverJdbcNotFoundExceptionType type =
639
//                    new DriverJdbcNotFoundExceptionType();
640
//            type.setDriverJdbcClassName(driverClass);
641
//            type.setLayerName(this.getTableName());
642
            throw new ReadDriverException(getName(),e);
643 12247 caballero
        } catch (DBException e) {
644 11971 caballero
                        throw new ReadDriverException(getName(),e);
645
                }
646
    }
647
    /* (non-Javadoc)
648
     * @see com.iver.cit.gvsig.fmap.drivers.VectorialDatabaseDriver#getXMLEntity()
649
     */
650
    public XMLEntity getXMLEntity()
651
    {
652
        XMLEntity xml = new XMLEntity();
653
        xml.putProperty("className",this.getClass().getName());
654
        try {
655
                 DatabaseMetaData metadata = ((ConnectionJDBC)getConnection()).getConnection().getMetaData();
656
            xml.putProperty("catalog", getLyrDef().getCatalogName());
657
658
            // TODO: NO DEBEMOS GUARDAR EL NOMBRE DE USUARIO Y CONTRASE?A
659
            // AQUI. Hay que utilizar un pool de conexiones
660
            // y pedir al usuario que conecte a la base de datos
661
            // en la primera capa. En el resto, usar la conexi?n
662
            // creada con anterioridad.
663
            String userName = metadata.getUserName();
664
            int aux = userName.indexOf("@");
665
            if (aux != -1)
666
                userName = userName.substring(0,aux);
667
            xml.putProperty("username", userName);
668
669
            Driver drv = DriverManager.getDriver(metadata.getURL());
670
            // System.out.println(drv.getClass().getName());
671
            xml.putProperty("driverclass", drv.getClass().getName());
672 12247 caballero
                         xml.putProperty("schema", lyrDef.getSchema());
673 11971 caballero
            xml.putProperty("tablename", getTableName());
674
            xml.putProperty("fields", lyrDef.getFieldNames());
675
            xml.putProperty("FID", lyrDef.getFieldID());
676
            xml.putProperty("THE_GEOM", lyrDef.getFieldGeometry());
677
            xml.putProperty("whereclause", getWhereClause());
678
            xml.putProperty("SRID", lyrDef.getSRID_EPSG());
679
680
            ConnectionWithParams cwp =
681
                    SingleVectorialDBConnectionManager.instance().findConnection(getConnection());
682
683 12247 caballero
             //FIXME:(Chema) Estos cambios los hago porque da errores as persistencia
684
            if (cwp != null){
685
                    xml.putProperty("host", cwp.getHost());
686
                    xml.putProperty("port", cwp.getPort());
687
                    xml.putProperty("dbName", cwp.getDb());
688 14543 jmvivo
                    xml.putProperty("literalDBName",true);
689 12247 caballero
                    xml.putProperty("connName", cwp.getName());
690
            } else {
691
                    xml.putProperty("dbURL",metadata.getURL());
692
            }
693
            // Chema
694 11971 caballero
695
            if (getWorkingArea() != null)
696
            {
697
                xml.putProperty("minXworkArea", getWorkingArea().getMinX());
698
                xml.putProperty("minYworkArea", getWorkingArea().getMinY());
699
                xml.putProperty("HworkArea", getWorkingArea().getHeight());
700
                xml.putProperty("WworkArea", getWorkingArea().getWidth());
701
            }
702
703
        } catch (SQLException e) {
704
            // TODO Auto-generated catch block
705
            e.printStackTrace();
706
        }
707
708
709
        return xml;
710
711
    }
712
713 17163 vcaballero
    /* (non-Javadoc)
714
     * @see com.iver.cit.gvsig.fmap.drivers.IVectorialDatabaseDriver#setWorkingArea(java.awt.geom.Rectangle2D)
715 11971 caballero
     */
716
    public void setWorkingArea(Rectangle2D rect) {
717
        this.workingArea = rect;
718
    }
719
720 17163 vcaballero
    /* (non-Javadoc)
721
     * @see com.iver.cit.gvsig.fmap.drivers.IVectorialDatabaseDriver#getWorkingArea()
722 11971 caballero
     */
723
    public Rectangle2D getWorkingArea() {
724
        return workingArea;
725
    }
726
727
    /* (non-Javadoc)
728
     * @see com.hardcode.gdbms.engine.data.driver.GDBMSDriver#setDataSourceFactory(com.hardcode.gdbms.engine.data.DataSourceFactory)
729
     */
730
    public void setDataSourceFactory(DataSourceFactory arg0) {
731
        // TODO Auto-generated method stub
732
733
    }
734
735 17163 vcaballero
    /* (non-Javadoc)
736
     * @see com.iver.cit.gvsig.fmap.drivers.IVectorialDatabaseDriver#getLyrDef()
737 11971 caballero
     */
738
    public DBLayerDefinition getLyrDef() {
739 12247 caballero
       if (this.conn != null){
740
                    if (lyrDef.getConnection() != this.conn){
741
                            lyrDef.setConnection(this.conn);
742
                    }
743
            }
744 11971 caballero
        return lyrDef;
745
    }
746
747
    /**
748
     * @param lyrDef The lyrDef to set.
749
     */
750
    public void setLyrDef(DBLayerDefinition lyrDef) {
751
        this.lyrDef = lyrDef;
752
    }
753
754
    abstract public String getSqlTotal();
755
756
    /**
757
     * @return Returns the completeWhere. WITHOUT order by clause!!
758
     */
759
    abstract public String getCompleteWhere();
760
761
    /* (non-Javadoc)
762
     * @see com.iver.cit.gvsig.fmap.drivers.VectorialDriver#reLoad()
763
     */
764
    public void reload() throws ReloadDriverException
765
    {
766
                try {
767 12247 caballero
                    if ((conn == null) || (((ConnectionJDBC)conn).getConnection().isClosed()))
768
                    {
769
                            this.load();
770
                    }
771 11971 caballero
772 12247 caballero
                    ((ConnectionJDBC)conn).getConnection().commit();
773 11971 caballero
774
                    setData(conn, lyrDef);
775
                } catch (SQLException e) {
776
                        throw new ReloadDriverException(getName(),e);
777
                } catch (ReadDriverException e) {
778
                        throw new ReloadDriverException(getName(),e);
779 12351 caballero
                } catch (DBException e) {
780
                        throw new ReloadDriverException(getName(),e);
781 11971 caballero
                }
782
783
    }
784
785 17163 vcaballero
    /* (non-Javadoc)
786
     * @see com.hardcode.gdbms.engine.data.driver.ReadAccess#getFieldWidth(int)
787
     */
788 11971 caballero
    public int getFieldWidth(int fieldId)
789
    {
790
            int i = -1;
791
            try {
792
                    int aux = fieldId + 2; // idField viene basado en 1, y
793
                        i = rs.getMetaData().getColumnDisplaySize(aux);
794
                } catch (SQLException e) {
795
                        e.printStackTrace();
796
                }
797
                // SUN define que getColumnDisplaySize devuelve numeros negativos cuando el campo es de tipo Text o Vartext
798
                // sin ancho. Nosotros vamos a devolver 255 para que fucione, por lo menos al exportar a DBF.
799
                // Nota: Si se truncan cadenas, este es el sitio que lo provoca.
800
                if (i <0) i=255;
801
                return i;
802
    }
803
804
//        public void setFlatness(double flatness) {
805
//                this.flatness = flatness;
806
//        }
807
808
    // -----------------------------------------------------------
809
    // ----  EXT JDBC NUEVA                           ---
810
    // -----------------------------------------------------------
811
812 17163 vcaballero
    /* (non-Javadoc)
813
     * @see com.iver.cit.gvsig.fmap.drivers.IVectorialDatabaseDriver#getConnectionString(java.lang.String, java.lang.String, java.lang.String, java.lang.String, java.lang.String)
814 11971 caballero
     */
815
    public String getConnectionString(
816
                    String host,
817
                    String port,
818
                    String dbname,
819
                    String user,
820
                    String pw) {
821
822
                String resp = getConnectionStringBeginning() + "//" + host.toLowerCase();
823
824
                if (dbname.trim().length() > 0) {
825
                        resp += ":" + port;
826
                } else {
827
                        resp += ":" + getDefaultPort();
828
                }
829
830 14543 jmvivo
                resp += "/" + dbname;//.toLowerCase();
831 11971 caballero
                return resp;
832
    }
833
834
    /**
835
     * Gets available table names. Should be overwritten by subclasses if its
836
     * not compatible or if it can be refined
837
     *
838
     * @param conn connection object
839
     * @param catalog catalog name
840
     * @return array of strings with available table names
841
     * @throws SQLException
842
     */
843
    public String[] getTableNames(IConnection conn, String catalog) throws DBException {
844
            try {
845
            DatabaseMetaData dbmd;
846
                        dbmd = ((ConnectionJDBC)conn).getConnection().getMetaData();
847
848
        String[] types = {"TABLE", "VIEW"};
849
                ResultSet rs = dbmd.getTables(catalog, null, null, types);
850
                TreeMap ret = new TreeMap();
851
                while (rs.next()){
852 12247 caballero
                        // ret.put(rs.getString("TABLE_NAME"), rs.getString("TABLE_NAME"));
853
                        // As suggested by Jorge Agudo, to allow charging tables from other schemas
854
                        ret.put((rs.getString("TABLE_SCHEM")!=null?(rs.getString("TABLE_SCHEM") + "."): "") + rs.getString("TABLE_NAME"), (rs.getString("TABLE_SCHEM")!=null?(rs.getString("TABLE_SCHEM") + "."): "") + rs.getString("TABLE_NAME"));
855 11971 caballero
                }
856
                rs.close();
857
                return (String[]) ret.keySet().toArray(new String[0]);
858
            } catch (SQLException e) {
859
                        throw new DBException(e);
860
                }
861
    }
862
863
864
    /**
865
     *       Gets all field names of a given table
866
     * @param conn connection object
867
     * @param table_name table name
868
     * @return all field names of the given table
869
     * @throws SQLException
870
     */
871
    public String[] getAllFields(IConnection conn, String table_name) throws DBException {
872
            try {
873
                Statement st = ((ConnectionJDBC)conn).getConnection().createStatement();
874
                ResultSet rs = st.executeQuery("select * from " + table_name + " LIMIT 1");
875
                ResultSetMetaData rsmd = rs.getMetaData();
876
                String[] ret = new String[rsmd.getColumnCount()];
877
878
                for (int i = 0; i < ret.length; i++) {
879
                        ret[i] = rsmd.getColumnName(i+1);
880
                }
881
                rs.close(); st.close();
882
                return ret;
883
            } catch (SQLException e) {
884
                        throw new DBException(e);
885
                }
886
    }
887
888
    /**
889
     *       Gets all field type names of a given table
890
     * @param conn connection object
891
     * @param table_name table name
892
     * @return all field type names of the given table
893
     * @throws SQLException
894
     */
895
    public String[] getAllFieldTypeNames(IConnection conn, String table_name) throws DBException {
896
try {
897
                Statement st = ((ConnectionJDBC)conn).getConnection().createStatement();
898
                ResultSet rs = st.executeQuery("select * from " + table_name + " LIMIT 1");
899
                ResultSetMetaData rsmd = rs.getMetaData();
900
                String[] ret = new String[rsmd.getColumnCount()];
901
902
                for (int i = 0; i < ret.length; i++) {
903
                        ret[i] = rsmd.getColumnTypeName(i+1);
904
                }
905
                rs.close(); st.close();
906
                return ret;
907
} catch (SQLException e) {
908
        throw new DBException(e);
909
}
910
    }
911
912
    /**
913
     * Gets the table's possible id fields. By default, all fields can be id.
914
     * It should be overwritten by subclasses.
915
     *
916
     * @param conn conenction object
917
     * @param table_name table name
918
     * @return the table's possible id fields
919
     * @throws SQLException
920
     */
921
    public String[] getIdFieldsCandidates(IConnection conn, String table_name) throws DBException {
922
            return getAllFields(conn, table_name);
923
    }
924
925
    /**
926
     * Gets the table's possible geometry fields. By default, all fields can be geometry
927
     * fields. It should be overwritten by subclasses.
928
     *
929
     * @param conn conenction object
930
     * @param table_name table name
931
     * @return the table's possible geometry fields
932
     * @throws SQLException
933
     */
934
    public String[] getGeometryFieldsCandidates(IConnection conn, String table_name) throws DBException {
935
            return getAllFields(conn, table_name);
936
    }
937
938
    /**
939
     * Tells if it's an empty table (with no records)
940
     * @param conn conenction object
941
     * @param tableName rtable name
942
     * @return whether it's an empty table (with no records) or not
943
     */
944
        public boolean isEmptyTable(IConnection conn, String tableName) {
945
946
                boolean res = true;
947
948
                try {
949
                        Statement st = ((ConnectionJDBC)conn).getConnection().createStatement();
950
                        ResultSet rs = null;
951
                        rs = st.executeQuery("select * from " + tableName + " LIMIT 1");
952
                        res = !rs.next();
953
                        rs.close(); st.close();
954
                } catch (Exception ex) {
955
                        res = true;
956
                }
957
                return res;
958
        }
959
960
961
962
        /**
963
         * Utility method to allow the driver to do something with the geometry field.
964
         * By default does nothing.
965
         *
966
         * @param flds user-selected fields
967
         * @param geomField geometry field
968
         * @return new user-selected fields
969
         */
970
        public String[] manageGeometryField(String[] flds, String geomField) {
971
                return flds;
972
        }
973
974
        /**
975
         * Empty method called when the layer is going to be removed from the view.
976
         * Subclasses can overwrite it if needed.
977
         *
978
         */
979
        public void remove() {
980
981
        }
982
983 12247 caballero
        public void addDriverEventListener(DriverEventListener listener) {
984
                if (!driverEventListeners.contains(listener))
985
                        driverEventListeners.add(listener);
986 11971 caballero
987 12247 caballero
        }
988 11971 caballero
989 12247 caballero
        public void removeDriverEventListener(DriverEventListener listener) {
990
                driverEventListeners.remove(listener);
991
992
        }
993
994
        public void notifyDriverEndLoaded() {
995
                DriverEvent event = new DriverEvent(DriverEvent.DRIVER_EVENT_LOADING_END);
996
                for (int i=0; i < driverEventListeners.size(); i++)
997
                {
998
                        DriverEventListener aux = (DriverEventListener) driverEventListeners.get(i);
999
                        aux.driverNotification(event);
1000
                }
1001
        }
1002 17342 vcaballero
        public static String removePrefix(String str) {
1003 12247 caballero
1004 17342 vcaballero
                int colon_ind = str.indexOf(":");
1005
                if (colon_ind != -1) {
1006
                        return str.substring(colon_ind + 1);
1007
                } else {
1008
                        return str;
1009
                }
1010
        }
1011 11971 caballero
}