/*
* GeoTools - The Open Source Java GIS Toolkit
* http://geotools.org
*
* (C) 2002-2009, Open Source Geospatial Foundation (OSGeo)
*
* This library is free software; you can redistribute it and/or
* modify it under the terms of the GNU Lesser General Public
* License as published by the Free Software Foundation;
* version 2.1 of the License.
*
* This library is distributed in the hope that it will be useful,
* but WITHOUT ANY WARRANTY; without even the implied warranty of
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
* Lesser General Public License for more details.
*/
package org.geotools.data.postgis;
import java.io.IOException;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.sql.Types;
import java.util.Map;
import org.geotools.factory.Hints;
import org.geotools.jdbc.ColumnMetadata;
import org.geotools.jdbc.JDBCDataStore;
import org.geotools.jdbc.PreparedFilterToSQL;
import org.geotools.jdbc.PreparedStatementSQLDialect;
import org.opengis.feature.simple.SimpleFeatureType;
import org.opengis.feature.type.GeometryDescriptor;
import com.vividsolutions.jts.geom.Envelope;
import com.vividsolutions.jts.geom.Geometry;
import com.vividsolutions.jts.geom.GeometryFactory;
import com.vividsolutions.jts.geom.LinearRing;
import com.vividsolutions.jts.io.WKBWriter;
/**
*
*
* @source $URL$
*/
public class PostGISPSDialect extends PreparedStatementSQLDialect {
private PostGISDialect delegate;
public PostGISPSDialect(JDBCDataStore store, PostGISDialect delegate) {
super(store);
this.delegate = delegate;
}
@Override
public boolean isAggregatedSortSupported(String function) {
return "distinct".equalsIgnoreCase(function);
}
@Override
public boolean includeTable(String schemaName, String tableName, Connection cx)
throws SQLException {
return delegate.includeTable(schemaName, tableName, cx);
}
public Envelope decodeGeometryEnvelope(ResultSet rs, int column,
Connection cx) throws SQLException, IOException {
return delegate.decodeGeometryEnvelope(rs, column, cx);
}
@Override
public Geometry decodeGeometryValue(GeometryDescriptor descriptor, ResultSet rs, int column,
GeometryFactory factory, Connection cx) throws IOException, SQLException {
return delegate.decodeGeometryValue(descriptor, rs, column, factory, cx);
}
public Geometry decodeGeometryValue(GeometryDescriptor descriptor,
ResultSet rs, String column, GeometryFactory factory, Connection cx)
throws IOException, SQLException {
return delegate
.decodeGeometryValue(descriptor, rs, column, factory, cx);
}
@Override
public void encodeGeometryColumn(GeometryDescriptor gatt, String prefix, int srid,
StringBuffer sql) {
delegate.encodeGeometryColumn(gatt, prefix, srid, sql);
}
@Override
public void encodeGeometryColumn(GeometryDescriptor gatt, String prefix,
int srid, Hints hints, StringBuffer sql) {
delegate.encodeGeometryColumn(gatt, prefix, srid, hints, sql);
}
public void encodeGeometryEnvelope(String tableName, String geometryColumn,
StringBuffer sql) {
delegate.encodeGeometryEnvelope(tableName, geometryColumn, sql);
}
public void encodePrimaryKey(String column, StringBuffer sql) {
delegate.encodePrimaryKey(column, sql);
}
public Integer getGeometrySRID(String schemaName, String tableName,
String columnName, Connection cx) throws SQLException {
return delegate.getGeometrySRID(schemaName, tableName, columnName, cx);
}
public String getGeometryTypeName(Integer type) {
return delegate.getGeometryTypeName(type);
}
public Class<?> getMapping(ResultSet columnMetaData, Connection cx)
throws SQLException {
return delegate.getMapping(columnMetaData, cx);
}
@Override
public boolean lookupGeneratedValuesPostInsert() {
return delegate.lookupGeneratedValuesPostInsert();
}
public Object getNextAutoGeneratedValue(String schemaName,
String tableName, String columnName, Connection cx)
throws SQLException {
return delegate.getNextAutoGeneratedValue(schemaName, tableName,
columnName, cx);
}
@Override
public Object getLastAutoGeneratedValue(String schemaName, String tableName, String columnName,
Connection cx) throws SQLException {
return delegate.getLastAutoGeneratedValue(schemaName, tableName, columnName, cx);
}
public Object getNextSequenceValue(String schemaName, String sequenceName,
Connection cx) throws SQLException {
return delegate.getNextSequenceValue(schemaName, sequenceName, cx);
}
public String getSequenceForColumn(String schemaName, String tableName,
String columnName, Connection cx) throws SQLException {
return delegate.getSequenceForColumn(schemaName, tableName, columnName,
cx);
}
public boolean isLooseBBOXEnabled() {
return delegate.isLooseBBOXEnabled();
}
public void postCreateTable(String schemaName,
SimpleFeatureType featureType, Connection cx) throws SQLException {
delegate.postCreateTable(schemaName, featureType, cx);
}
@Override
public void preDropTable(String schemaName, SimpleFeatureType featureType, Connection cx)
throws SQLException {
delegate.preDropTable(schemaName, featureType, cx);
}
@Override
public void postDropTable(String schemaName, SimpleFeatureType featureType, Connection cx)
throws SQLException {
delegate.postDropTable(schemaName, featureType, cx);
}
public void registerClassToSqlMappings(Map<Class<?>, Integer> mappings) {
delegate.registerClassToSqlMappings(mappings);
}
public void registerSqlTypeNameToClassMappings(
Map<String, Class<?>> mappings) {
delegate.registerSqlTypeNameToClassMappings(mappings);
}
@Override
public void registerSqlTypeToSqlTypeNameOverrides(
Map<Integer, String> overrides) {
delegate.registerSqlTypeToSqlTypeNameOverrides(overrides);
}
@Override
public void handleUserDefinedType(ResultSet columnMetaData, ColumnMetadata metadata,
Connection cx) throws SQLException {
delegate.handleUserDefinedType(columnMetaData, metadata, cx);
}
public void setLooseBBOXEnabled(boolean looseBBOXEnabled) {
delegate.setLooseBBOXEnabled(looseBBOXEnabled);
}
@Override
public void prepareGeometryValue(Geometry g, int dimension, int srid, Class binding,
StringBuffer sql) {
if (g != null) {
sql.append("ST_GeomFromWKB(?, " + srid + ")");
} else {
sql.append("?");
}
}
@Override
public void setGeometryValue(Geometry g, int dimension, int srid, Class binding,
PreparedStatement ps, int column) throws SQLException {
if (g != null && !g.isEmpty()) {
if (g instanceof LinearRing ) {
//postgis does not handle linear rings, convert to just a line string
g = g.getFactory().createLineString(((LinearRing) g).getCoordinateSequence());
}
byte[] bytes = new WKBWriter(dimension).write(g);
ps.setBytes(column, bytes);
} else {
ps.setNull(column, Types.OTHER, "Geometry");
}
}
@Override
public PreparedFilterToSQL createPreparedFilterToSQL() {
PostgisPSFilterToSql fts = new PostgisPSFilterToSql(this);
fts.setLooseBBOXEnabled(delegate.isLooseBBOXEnabled());
return fts;
}
@Override
public boolean isLimitOffsetSupported() {
return delegate.isLimitOffsetSupported();
}
@Override
public void applyLimitOffset(StringBuffer sql, int limit, int offset) {
delegate.applyLimitOffset(sql, limit, offset);
}
@Override
public int getGeometryDimension(String schemaName, String tableName, String columnName,
Connection cx) throws SQLException {
return delegate.getGeometryDimension(schemaName, tableName, columnName, cx);
}
}