001    //$HeadURL: svn+ssh://rbezema@svn.wald.intevation.org/deegree/base/branches/2.2_testing/src/org/deegree/io/GenericSQLIDGenerator.java $
002    /*----------------    FILE HEADER  ------------------------------------------
003    
004     This file is part of deegree.
005     Copyright (C) 2001-2008 by:
006     EXSE, Department of Geography, University of Bonn
007     http://www.giub.uni-bonn.de/deegree/
008     lat/lon GmbH
009     http://www.lat-lon.de
010    
011     This library is free software; you can redistribute it and/or
012     modify it under the terms of the GNU Lesser General Public
013     License as published by the Free Software Foundation; either
014     version 2.1 of the License, or (at your option) any later version.
015    
016     This library is distributed in the hope that it will be useful,
017     but WITHOUT ANY WARRANTY; without even the implied warranty of
018     MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE.  See the GNU
019     Lesser General Public License for more details.
020    
021     You should have received a copy of the GNU Lesser General Public
022     License along with this library; if not, write to the Free Software
023     Foundation, Inc., 59 Temple Place, Suite 330, Boston, MA  02111-1307  USA
024    
025     Contact:
026    
027     Andreas Poth
028     lat/lon GmbH
029     Aennchenstr. 19
030     53115 Bonn
031     Germany
032     E-Mail: poth@lat-lon.de
033    
034     Prof. Dr. Klaus Greve
035     Department of Geography
036     University of Bonn
037     Meckenheimer Allee 166
038     53115 Bonn
039     Germany
040     E-Mail: greve@giub.uni-bonn.de
041    
042     
043     ---------------------------------------------------------------------------*/
044    package org.deegree.io;
045    
046    import java.sql.Connection;
047    import java.sql.ResultSet;
048    import java.sql.SQLException;
049    import java.sql.Statement;
050    
051    import org.deegree.framework.util.DataBaseIDGenerator;
052    
053    /**
054     * Primary key generator for generic JDBC-connections.
055     * <p>
056     * NOTE: At the moment, every application has to take care of locking the table to prevent problems
057     * in multithreaded or multihosted applications.
058     * <p>
059     * 
060     * @author <a href="mailto:mschneider@lat-lon.de">Markus Schneider</a>
061     * @version $Revision: 9342 $ $Date: 2007-12-27 13:32:57 +0100 (Do, 27 Dez 2007) $
062     */
063    class GenericSQLIDGenerator implements DataBaseIDGenerator {
064    
065        Connection con;
066    
067        int fieldType;
068    
069        boolean isNumeric;
070    
071        String tableName;
072    
073        String fieldName;
074    
075        /**
076         * Constructs a new GenericSQLIDGenerator.
077         * <p>
078         * 
079         * @param con
080         * @param fieldType
081         */
082        GenericSQLIDGenerator( Connection con, String tableName, String fieldName, int fieldType, boolean isNumeric ) {
083            this.con = con;
084            this.fieldName = fieldName;
085            this.tableName = tableName;
086            this.fieldType = fieldType;
087            this.isNumeric = isNumeric;
088        }
089    
090        /**
091         * Returns the successor to the given id (string). Valid characters in the id-string are: 0-9,
092         * a-z, A-Z. Every other character may result in an exception.
093         * <p>
094         * 
095         * @param lastId
096         * @return the successor to the given id (string).
097         */
098        private String incrementId( String lastId ) {
099            char[] chars = lastId.toCharArray();
100            for ( int i = chars.length - 1; i >= 0; i-- ) {
101                char c = chars[i];
102                if ( c >= '0' && c <= '9' ) {
103                    if ( c == '9' ) {
104                        c = 'a';
105                    } else {
106                        c++;
107                    }
108                } else if ( c >= 'a' && c <= 'z' ) {
109                    if ( c == 'z' ) {
110                        c = 'A';
111                    } else {
112                        c++;
113                    }
114                } else if ( c >= 'A' && c <= 'Z' ) {
115                    if ( c == 'Z' ) {
116                        c = '0';
117                    } else {
118                        c++;
119                    }
120                }
121                chars[i] = c;
122                if ( c != '0' ) {
123                    break;
124                }
125            }
126            String newId = new String( chars );
127            if ( chars[0] == '0' ) {
128                newId = '1' + newId;
129            }
130            return newId;
131        }
132    
133        /**
134         * Generates a new id, suitable as a primary key for the next dataset.
135         * <p>
136         * 
137         * @return Id, the object type depends on the database field used as primary key
138         */
139        public Object generateUniqueId()
140                                throws SQLException {
141    
142            Object id = null;
143    
144            // retrieve last id
145            Statement stmt = con.createStatement();
146            ResultSet rs = stmt.executeQuery( "SELECT MAX(" + fieldName + ") FROM " + tableName );
147    
148            if ( rs.next() ) {
149                if ( isNumeric ) {
150                    id = new Integer( rs.getInt( 1 ) + 1 );
151                } else {
152                    String oldId = rs.getString( 1 );
153                    if ( oldId != null ) {
154                        id = incrementId( oldId );
155                    } else {
156                        id = "0";
157                    }
158                }
159            } else {
160                if ( isNumeric ) {
161                    id = new Integer( 0 );
162                } else {
163                    id = "0";
164                }
165            }
166            rs.close();
167            stmt.close();
168            return id;
169        }
170    }