PostgreSQLGeometryFieldAdder.java example

Explorer
com.revolsys.open-master
package com.revolsys.gis.postgresql.type;

import java.util.HashMap;
import java.util.Map;

import com.revolsys.datatype.DataType;
import com.revolsys.datatype.DataTypes;
import com.revolsys.geometry.model.GeometryFactory;
import com.revolsys.gis.postgresql.PostgreSQLRecordStore;
import com.revolsys.io.PathName;
import com.revolsys.jdbc.JdbcUtils;
import com.revolsys.jdbc.field.JdbcFieldAdder;
import com.revolsys.jdbc.io.AbstractJdbcRecordStore;
import com.revolsys.logging.Logs;
import com.revolsys.record.property.FieldProperties;
import com.revolsys.record.schema.FieldDefinition;
import com.revolsys.record.schema.RecordDefinitionImpl;
import com.revolsys.util.Property;

public class PostgreSQLGeometryFieldAdder extends JdbcFieldAdder {
  private static final Map<String, DataType> DATA_TYPE_MAP = new HashMap<>();

  static {
    DATA_TYPE_MAP.put("GEOMETRY", DataTypes.GEOMETRY);
    DATA_TYPE_MAP.put("POINT", DataTypes.POINT);
    DATA_TYPE_MAP.put("LINESTRING", DataTypes.LINE_STRING);
    DATA_TYPE_MAP.put("POLYGON", DataTypes.POLYGON);
    DATA_TYPE_MAP.put("MULTIPOINT", DataTypes.MULTI_POINT);
    DATA_TYPE_MAP.put("MULTILINESTRING", DataTypes.MULTI_LINE_STRING);
    DATA_TYPE_MAP.put("MULTIPOLYGON", DataTypes.MULTI_POLYGON);
  }

  private final PostgreSQLRecordStore recordStore;

  public PostgreSQLGeometryFieldAdder(final PostgreSQLRecordStore recordStore) {
    this.recordStore = recordStore;
  }

  @Override
  public FieldDefinition addField(final AbstractJdbcRecordStore recordStore,
    final RecordDefinitionImpl recordDefinition, final String dbName, final String name,
    final String dataTypeName, final int sqlType, final int length, final int scale,
    final boolean required, final String description) {
    final PathName typePath = recordDefinition.getPathName();
    final PathName schemaPath = typePath.getParent();
    String dbSchemaName = this.recordStore.getDatabaseSchemaName(schemaPath);
    if (!Property.hasValue(dbSchemaName)) {
      dbSchemaName = "public";
    }
    final String tableName = this.recordStore.getDatabaseTableName(typePath);
    final String columnName = name.toLowerCase();
    try {
      int srid = 0;
      String type = "geometry";
      int axisCount = 3;
      try {
        final String sql = "select SRID, TYPE, COORD_DIMENSION from GEOMETRY_COLUMNS where UPPER(F_TABLE_SCHEMA) = UPPER(?) AND UPPER(F_TABLE_NAME) = UPPER(?) AND UPPER(F_GEOMETRY_COLUMN) = UPPER(?)";
        final Map<String, Object> values = JdbcUtils.selectMap(this.recordStore, sql, dbSchemaName,
          tableName, columnName);
        srid = (Integer)values.get("srid");
        type = (String)values.get("type");
        axisCount = (Integer)values.get("coord_dimension");
      } catch (final IllegalArgumentException e) {
        Logs.warn(this, "Cannot get geometry column metadata for " + typePath + "." + columnName);
      }

      final DataType dataType = DATA_TYPE_MAP.get(type);
      final GeometryFactory storeGeometryFactory = this.recordStore.getGeometryFactory();
      final GeometryFactory geometryFactory;
      if (storeGeometryFactory == null) {
        geometryFactory = GeometryFactory.floating(srid, axisCount);
      } else {
        geometryFactory = GeometryFactory.fixed(srid, axisCount, storeGeometryFactory.getScaleX(),
          storeGeometryFactory.getScaleY(), storeGeometryFactory.getScaleZ());
      }
      final FieldDefinition field = new PostgreSQLGeometryJdbcFieldDefinition(dbName, name,
        dataType, required, description, null, srid, axisCount, geometryFactory);
      recordDefinition.addField(field);
      field.setProperty(FieldProperties.GEOMETRY_FACTORY, geometryFactory);
      return field;
    } catch (final Throwable e) {
      Logs.error(this, "Attribute not registered in GEOMETRY_COLUMN table " + dbSchemaName + "."
        + tableName + "." + name, e);
      return null;
    }
  }
}