Application: gvSIG desktop

package org.gvsig.data.datastores.vectorial.db.jdbc.postgresql;

import java.lang.ref.WeakReference;

import java.sql.Connection;

import java.sql.ResultSet;

import java.sql.Statement;

import java.util.ArrayList;

import java.util.List;

import java.util.Locale;

import org.gvsig.data.IDataCollection;

import org.gvsig.data.IDataExplorer;

import org.gvsig.data.IDataStoreParameters;

import org.gvsig.data.datastores.vectorial.db.DBFeatureType;

import org.gvsig.data.datastores.vectorial.db.jdbc.JDBCFeaturesWriter;

import org.gvsig.data.datastores.vectorial.db.jdbc.JDBCStore;

import org.gvsig.data.exception.CloseException;

import org.gvsig.data.exception.InitializeException;

import org.gvsig.data.exception.OpenException;

import org.gvsig.data.exception.ReadException;

import org.gvsig.data.spatialprovisional.IExtent;

import org.gvsig.data.vectorial.IFeature;

import org.gvsig.data.vectorial.IFeatureCollection;

import org.gvsig.data.vectorial.IFeatureID;

import org.gvsig.data.vectorial.IFeatureType;

import org.gvsig.metadata.IMetadata;

import org.gvsig.metadata.IMetadataManager;

import org.gvsig.metadata.MetadataManager;

public class PostgresqlStore extends JDBCStore{

        public static final String CONNECTION_STRING = "postgresql";

        public static String DATASTORE_NAME = "PostgresqlStore";

        protected static Locale ukLocale = new Locale("en", "UK"); // English, UK version

    private DBFeatureType featureType;

        protected IMetadata metadata;

        PostgresqlStoreParameters getParametersPostgresql(){

                return (PostgresqlStoreParameters)this.parameters;

        }

        public void init(IDataStoreParameters parameters) throws InitializeException {

                super.init(parameters);

                this.initConnection();

                this.initFeatureType();

                this.initSqlProperties();

                        //writer.setCreateTable(false);

                        //writer.setWriteAll(false);

                        //writer.initialize(lyrDef);

        }

        private void initFeatureType() throws InitializeException{

                PostgresqlStoreParameters dParams = this.getParametersPostgresql();

                try {

                        this.featureType = PostgresqlStoreUtils.getFeatureType(this.connection, dParams);

                } catch (ReadException e) {

                        throw new InitializeException(DATASTORE_NAME,e);

                }

                String[] fieldsID = dParams.getFieldsId();

                if (fieldsID == null || fieldsID.length < 1){

                        throw new InitializeException(

                                        DATASTORE_NAME,

                                        new Exception("Field Id not set"));

                } else if (fieldsID.length > 1){

                        //TODO: Falta por implementar soporte para multiples ID

                        throw new InitializeException(

                                        DATASTORE_NAME,

                                        new Exception("Multy fields id not supported yet"));

                } else{

                        for (int i=0;i<fieldsID.length;i++){

                                if (this.featureType.getFieldIndex(fieldsID[i]) < 0) {

                                throw new InitializeException(

                                                DATASTORE_NAME,

                                                new Exception("Field id '"+ fieldsID[i] +"' (id pos "+ (i+1) +") not Found"));

                                }

                        }

                }

        }

        protected void initConnection() throws InitializeException{

                PostgresqlStoreParameters dParams = this.getParametersPostgresql();

                String dburl = dParams.getUrl();

                String dbuser = dParams.getUser();

                String dbpass = dParams.getPassw();

                this.connection = PostgresqlStoreUtils.getConnection(dburl, dbuser, dbpass);

        }

        public IDataCollection getDataCollection(IFeatureType type, String filter, String order) throws ReadException {

                if (useSqlSource ){

                        if (filter != null || order != null){

                                throw new ReadException(DATASTORE_NAME,

                                                new UnsupportedOperationException("Unsuported filter/order in sqlSource mode"));

                        }

                }

                if (type==null){

                        type=getDefaultFeatureType();

                }

                IFeatureCollection coll=null;

                if (featureManager == null){

                        coll=new PostgresqlFeatureCollection(this,type,filter,order);

                }else{

                        if ((order != null && order != "")){

                                coll=new PostgresqlFeatureCollectionWithFeatureID(featureManager,this,type,filter,order);

                        } else{

                                if (filter == null || filter == ""){

                                        coll=new PostgresqlFeatureCollectionEditing(featureManager,this,type);

                                } else {

                                        coll=new PostgresqlFeatureCollectionEditingFiltered(featureManager,this,type,filter);

                                }

                        }

                }

                this.addObserver(new WeakReference(coll));

                return coll;

        }

        public IFeature getFeatureByID(IFeatureID id) throws ReadException {

                return getFeatureByID(featureType,((PostgresqlFeatureID)id).getKey());

        }

        public IFeature getFeatureByID(IFeatureType featureType2, Object[] featureKey) throws ReadException{

                if (useSqlSource){

                        throw new ReadException(this.getName(),

                                        new UnsupportedOperationException("Unsuported featureByID in sqlSource mode"));

                }

                ResultSet rs=null;

                try{

                        this.open();

                        Statement st=this.getConnection().createStatement(ResultSet.TYPE_SCROLL_INSENSITIVE, ResultSet.CONCUR_READ_ONLY);

                        String sql = this.getSqlSelectPart() + " WHERE "+ PostgresqlStoreUtils.getFliterForID(this.featureType, featureKey);

                        rs=st.executeQuery(sql);

                        if (rs.isLast()) {

                                return null;

                        }else{

                                if(rs.next()){

                                        return PostgresqlStoreUtils.createFeature(this, rs, this.featureType);

                                }

                        }

                } catch (java.sql.SQLException e) {

                        e.printStackTrace();

                        throw new ReadException(this.getName(), e);

                } finally{

                        if (rs != null)

                                try {

                                        rs.close();

                                } catch (java.sql.SQLException e) {

                                        // TODO ?????

                                        e.printStackTrace();

                                }

                }

                return null;

        }

        protected Connection getConnection(){

                // FIXME: OJO REsource manager

                return this.connection;

        }

        Connection getCurrentConnection(){

                return this.getConnection();

        }

        private void initSqlProperties() throws InitializeException{

                PostgresqlStoreParameters dParams = (PostgresqlStoreParameters)this.getParameters();

                if (dParams.getSqlSoure() != null){

                        this.sqlSource = dParams.getSqlSoure();

                        this.useSqlSource = true;

                        this.sqlSelectPart = null;

                        this.baseWhereClause = null;

                        this.baseOrder = null;

                } else {

                        this.sqlSelectPart = "SELECT " + dParams.getFieldsString() +" FROM "+ dParams.tableID();

                        this.baseWhereClause = dParams.getBaseFilter();

                        if (dParams.getWorkingArea() != null){

                                String waWhere = getWorkingAreaWhere(dParams.getWorkingArea(), dParams.getSRISD());

                                if (waWhere != null){

                                        this.baseWhereClause = "(("+this.baseWhereClause+") and "+waWhere +")";

                                }

                        }

                        this.baseOrder = dParams.getBaseOrder();

                }

        }

        private String getWorkingAreaWhere(IExtent r, String strEPSG) {

                //TODO????

                if (r==null){

                        return null;

                }

                return null;

        }

        public boolean canAlterFeatureType() {

                return true;

        }

        public void open() throws OpenException {

                // FIXME: Resource Manager

        }

        public void close() throws CloseException {

                // FIXME: Resource Manager

                try {

                        connection.close();

                } catch (java.sql.SQLException e) {

                        throw new CloseException("H2",e);

                }

        }

        public void dispose() {

                // FIXME: Resource Manager

        }

        public String getName() {

                return DATASTORE_NAME;

        }

        public JDBCFeaturesWriter getFeaturesWriter() {

//                IFeaturesWriter writer = new H2FeaturesWriter();

//                writer.init(this);

//                return writer;

                return null;

        }

        public IMetadata getMetadata() {

                if (metadata==null){

                        IMetadataManager manager=MetadataManager.getManager();

                        metadata=manager.create(DATASTORE_NAME);

                        //TODO: Apadir los meteadatos

                }

                return metadata;

        }

        public IFeatureType getDefaultFeatureType() {

                // TODO Auto-generated method stub

                return this.featureType;

        }

        public Object getDefaultLabelingStrategy() {

                // TODO Auto-generated method stub

                return null;

        }

        public Object getDefaultLegend() {

                // TODO Auto-generated method stub

                return null;

        }

        public List getFeatureTypes() {

                ArrayList list = new ArrayList();

                list.add(this.featureType);

                return list;

        }

        public boolean isWithDefaultLegend() {

                // TODO Auto-generated method stub

                return false;

        }

        public IDataExplorer getExplorer() {

                // TODO Auto-generated method stub

                return null;

        }

}