private static void initializeSerialFields(Connection connection, DBFeatureType featureType)
throws java.sql.SQLException, DataException {
DBAttributeDescriptor attr;
ArrayList serialCandidates = new ArrayList();
Iterator iter = featureType.iterator();
while (iter.hasNext()) {
attr = (DBAttributeDescriptor) iter.next();
if (attr.getSqlTypeName().equals("int4")
&& attr.getTableName() != null
&& attr.getTableName().length() > 0) {
serialCandidates.add(attr);
}
}
if (serialCandidates.size() == 0) {
return;
}
Statement st = connection.createStatement();
StringBuffer sqlSeq =
new StringBuffer(
"select table_catalog,table_schema,table_name,column_name from information_schema.columns where column_default like 'nextval(%' and ( ");
iter = serialCandidates.iterator();
String sql;
int i;
for (i = 0; i < serialCandidates.size() - 1; i++) {
attr = (DBAttributeDescriptor) serialCandidates.get(i);
addConditionForSerialField(attr, sqlSeq);
sqlSeq.append(" or ");
}
attr = (DBAttributeDescriptor) serialCandidates.get(i);
addConditionForSerialField(attr, sqlSeq);
sqlSeq.append(")");
sql = sqlSeq.toString();
ResultSet rs = st.executeQuery(sql);
while (rs.next()) {
iter = serialCandidates.iterator();
while (iter.hasNext()) {
attr = (DBAttributeDescriptor) iter.next();
if (rs.getString("column_name").equals(attr.getName())) {
attr.setAutoIncrement(true);
serialCandidates.remove(attr);
break;
}
}
}
}
private static DBAttributeDescriptor getAttributeFromJDBC(
DefaultFeatureType fType, Connection conn, ResultSetMetaData rsMetadata, int colIndex)
throws SQLException {
DBAttributeDescriptor column = (DBAttributeDescriptor) fType.createAttributeDescriptor();
try {
column.loading();
column.setName(rsMetadata.getColumnName(colIndex));
column.setCaseSensitive(rsMetadata.isCaseSensitive(colIndex));
column.setSqlType(rsMetadata.getColumnType(colIndex));
column.setAllowNull(rsMetadata.isNullable(colIndex) == ResultSetMetaData.columnNullable);
column.setAutoIncrement(rsMetadata.isAutoIncrement(colIndex));
column.setReadOnly(rsMetadata.isReadOnly(colIndex));
column.setWritable(rsMetadata.isWritable(colIndex));
column.setClassName(rsMetadata.getColumnClassName(colIndex));
column.setCatalogName(rsMetadata.getCatalogName(colIndex));
column.setDefinitelyWritable(rsMetadata.isDefinitelyWritable(colIndex));
column.setLabel(rsMetadata.getColumnLabel(colIndex));
column.setSchemaName(rsMetadata.getSchemaName(colIndex));
column.setTableName(rsMetadata.getTableName(colIndex));
column.setCatalogName(rsMetadata.getCatalogName(colIndex));
column.setSqlTypeName(rsMetadata.getColumnTypeName(colIndex));
column.setSearchable(rsMetadata.isSearchable(colIndex));
column.setSigned(rsMetadata.isSigned(colIndex));
column.setCurrency(rsMetadata.isCurrency(colIndex));
column.setPrecision(rsMetadata.getPrecision(colIndex));
column.setSize(rsMetadata.getColumnDisplaySize(colIndex));
switch (rsMetadata.getColumnType(colIndex)) {
case java.sql.Types.INTEGER:
column.setType(FeatureAttributeDescriptor.INT);
break;
case java.sql.Types.BIGINT:
column.setType(FeatureAttributeDescriptor.LONG);
break;
case java.sql.Types.REAL:
column.setType(FeatureAttributeDescriptor.DOUBLE);
break;
case java.sql.Types.DOUBLE:
column.setType(FeatureAttributeDescriptor.DOUBLE);
break;
case java.sql.Types.CHAR:
column.setType(FeatureAttributeDescriptor.STRING);
break;
case java.sql.Types.VARCHAR:
column.setType(FeatureAttributeDescriptor.STRING);
break;
case java.sql.Types.FLOAT:
column.setType(FeatureAttributeDescriptor.FLOAT);
break;
case java.sql.Types.DECIMAL:
column.setType(FeatureAttributeDescriptor.FLOAT);
break;
case java.sql.Types.DATE:
column.setType(FeatureAttributeDescriptor.DATE);
break;
case java.sql.Types.TIME:
column.setType(FeatureAttributeDescriptor.TIME);
break;
case java.sql.Types.TIMESTAMP:
column.setType(FeatureAttributeDescriptor.TIMESTAMP);
break;
case java.sql.Types.BOOLEAN:
column.setType(FeatureAttributeDescriptor.BOOLEAN);
break;
case java.sql.Types.OTHER:
if (column.getSqlTypeName().equalsIgnoreCase("geometry")) {
column.setType(FeatureAttributeDescriptor.GEOMETRY);
break;
}
// No hacemos break para que se quede en default
default:
column.setType(FeatureAttributeDescriptor.OBJECT);
break;
}
column.stopLoading();
} catch (java.sql.SQLException e) {
throw new SQLException("", "load attribute definition", e);
} catch (IsNotAttributeSettingException e) {
e.printStackTrace();
}
return column;
}
