Application: gvSIG desktop

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

import java.lang.ref.WeakReference;

import java.sql.Connection;

import java.sql.ResultSet;

import java.util.Locale;

import org.gvsig.data.CloseException;

import org.gvsig.data.DataManager;

import org.gvsig.data.IDataCollection;

import org.gvsig.data.IDataExplorer;

import org.gvsig.data.InitializeException;

import org.gvsig.data.OpenException;

import org.gvsig.data.ReadException;

import org.gvsig.data.datastores.vectorial.InitializeWriterException;

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

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

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

import org.gvsig.data.spatialprovisional.IExtent;

import org.gvsig.data.vectorial.IFeature;

import org.gvsig.data.vectorial.IFeatureCollection;

import org.gvsig.data.vectorial.IFeatureType;

import org.gvsig.metadata.IMetadata;

public class H2Store extends JDBCStore{

        public static final String CONNECTION_STRING = "h2";

        public static String DATASTORE_NAME = "H2Store";

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

    public H2StoreParameters getParametersH2(){

                return (H2StoreParameters)this.parameters;

        }

        /* (non-Javadoc)

         * @see org.gvsig.data.datastores.vectorial.db.jdbc.JDBCStore#createResource(org.gvsig.data.datastores.vectorial.file.dbf.DBFStoreParameters)

         */

        protected JDBCResource createResource(JDBCStoreParameters params) {

                return new H2Resource((H2StoreParameters)params);

        }

        protected void initFeatureType() throws InitializeException{

                H2StoreParameters dParams = this.getParametersH2();

                try {

                        this.defaultFeatureType = H2Utils.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("Field Id not set",

                                        this.getName());

                } else if (fieldsID.length > 1){

                        //TODO: Falta por implementar soporte para multiples ID

                        throw new InitializeException("Multy fields id not supported yet",

                                        this.getName());

                } else{

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

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

                                throw new InitializeException("Field id '"+ fieldsID[i] +"' (id pos "+ (i+1) +") not Found",

                                                this.getName());

                                }

                        }

                }

        }

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

                if (useSqlSource ){

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

                                throw new ReadException("Unsuported filter/order in sqlSource mode",this.getName());

                        }

                }

//                if (type!=null && type != this.getDefaultFeatureType()){

//                        throw new UnsupportedOperationException("Type != null");

//                }else{

                        type = this.getDefaultFeatureType();

//                }

                IFeatureCollection coll;

                if (featureManager == null){

                        coll=new H2FeatureCollection(this,(DBFeatureType)type,filter,order);

                }else{

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

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

                        } else{

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

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

                                } else {

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

                                }

                        }

                }

                this.addObserver(new WeakReference(coll));

                return coll;

        }

        protected Connection getConnection() throws ReadException{

                return ((H2Resource)this.resource).getConnection();

        }

        protected void initSqlProperties() throws InitializeException{

                H2StoreParameters dParams = (H2StoreParameters)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 doDispose() throws CloseException {

                super.doDispose();

        }

        public String getName() {

                return DATASTORE_NAME;

        }

        protected JDBCFeaturesWriter getFeaturesWriter() throws InitializeWriterException {

                H2FeaturesWriter writer = new H2FeaturesWriter();

                writer.init(this);

                return writer;

        }

        public IDataExplorer getExplorer() throws ReadException {

                DataManager dm = DataManager.getManager();

                H2ExplorerParameters explorerParams = (H2ExplorerParameters)dm.createDataExplorerParameters(H2Explorer.DATAEXPLORER_NAME);

                explorerParams.loadFromStoreParameters(this.getParametersH2());

                return dm.createDataExplorer(explorerParams);

        }

        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 H2Feature(featureType2,this,rs);

        }

        /* (non-Javadoc)

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

         */

        protected void doRefresh() throws OpenException, InitializeException {

                this.initFeatureType();

                this.initSqlProperties();

        }

}