| gvSIG desktop 1

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

import java.lang.ref.WeakReference;

import java.sql.Connection;

import java.sql.ResultSet;

import java.util.Iterator;

import java.util.Locale;

import org.gvsig.data.IDataCollection;

import org.gvsig.data.IDataExplorer;

import org.gvsig.data.IDataStoreParameters;

import org.gvsig.data.ResourceManager;

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

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

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

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

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

import org.gvsig.data.datastores.vectorial.db.jdbc.h2.H2Resource;

import org.gvsig.data.datastores.vectorial.db.jdbc.h2.H2StoreParameters;

import org.gvsig.data.exception.CloseException;

import org.gvsig.data.exception.DataException;

import org.gvsig.data.exception.InitializeException;

import org.gvsig.data.exception.InitializeWriterException;

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.IFeatureAttributeDescriptor;

import org.gvsig.data.vectorial.IFeatureCollection;

import org.gvsig.data.vectorial.IFeatureType;

import org.gvsig.data.vectorial.IsNotFeatureSettingException;

import org.gvsig.metadata.IMetadata;

import org.postgis.PGgeometry;

import com.iver.cit.gvsig.fmap.core.IGeometry;

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

    PostgresqlStoreParameters getParametersPostgresql(){

                return (PostgresqlStoreParameters)this.parameters;

        }

        public void init(IDataStoreParameters parameters) throws InitializeException {

//                super.init(parameters);

//

//

//

//                this.initConnection();

//                this.initFeatureType();

//                this.initSqlProperties();

                PostgresqlResource tmpResource = new PostgresqlResource((PostgresqlStoreParameters)parameters);

                PostgresqlResource theResource;

                ResourceManager resMan = ResourceManager.getResourceManager();

                try {

                        theResource = (PostgresqlResource)resMan.addResource(tmpResource);

                } catch (DataException e1) {

                        throw new InitializeException(this.getName(),e1);

                }

                super.init(parameters,theResource);

                this.initFeatureType();

                this.initSqlProperties();

        }

        private void initFeatureType() throws InitializeException{

                PostgresqlStoreParameters dParams = this.getParametersPostgresql();

                try {

                        this.featureType = PostgresqlStoreUtils.getFeatureType(this.getConnection(), 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"));

                                }

                        }

                }

        }

        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;

        }

        protected Connection getConnection() throws ReadException{

                return super.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;

        }

        protected void doOpen() throws OpenException {

                // FIXME: Resource Manager

        }

        protected void doDispose() throws CloseException{

                super.doDispose();

                // FIXME: Resource Manager

        }

        public String getName() {

                return DATASTORE_NAME;

        }

        protected JDBCFeaturesWriter getFeaturesWriter() throws InitializeWriterException {

                JDBCFeaturesWriter writer = new PostgresqlFeaturesWriter();

                writer.init(this);

                return writer;

        }

        public IDataExplorer getExplorer() {

                // TODO Auto-generated method stub

                return null;

        }

        public IMetadata getMetadata() {

                if (metadata==null){

                        IMetadata tmp=super.getMetadata();

                        return tmp;

                }else{

                        return super.getMetadata();

                }

        }

        protected IFeature createFeatureFromResulset(ResultSet rs, DBFeatureType featureType2) throws ReadException {

                return new PostgresqlFeature(featureType,this,rs);

        }

        /* (non-Javadoc)

         * @see org.gvsig.data.vectorial.FeatureStore#doRefresh()

         */

        protected void doRefresh() throws OpenException, InitializeException {

                this.initFeatureType();

                this.initSqlProperties();

        }

}