gvSIG projects pool

package org.gvsig.vcsgis.lib.repository.localdb.tables;

import java.sql.Timestamp;

import java.text.SimpleDateFormat;

import javax.json.JsonObject;

import org.gvsig.expressionevaluator.spi.JsonUtils;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataTypes;

import org.gvsig.fmap.dal.feature.EditableFeatureType;

import org.gvsig.fmap.dal.feature.Feature;

import org.gvsig.fmap.dal.feature.FeatureSet;

import org.gvsig.fmap.dal.feature.FeatureStore;

import org.gvsig.fmap.dal.feature.FeatureType;

import org.gvsig.fmap.geom.Geometry;

import org.gvsig.json.Json;

import org.gvsig.json.JsonObjectBuilder;

import org.gvsig.tools.dataTypes.CoercionException;

import org.gvsig.tools.dispose.DisposableIterable;

import org.gvsig.tools.dispose.DisposeUtils;

import org.gvsig.tools.dynform.spi.DynFormSPIManager;

import org.gvsig.tools.dynobject.DynObjectValueItem;

import static org.gvsig.vcsgis.lib.VCSGisManager.OP_DELETE;

import static org.gvsig.vcsgis.lib.VCSGisManager.OP_INSERT;

import static org.gvsig.vcsgis.lib.VCSGisManager.OP_UPDATE;

import static org.gvsig.vcsgis.lib.VCSGisManager.VCSGISCODELEN;

import org.gvsig.vcsgis.lib.VCSGisUtils;

import org.gvsig.vcsgis.lib.repository.VCSGisRepositoryData;

import org.gvsig.vcsgis.lib.repository.localdb.VCSGisRepositoryLocaldb;

/**

 *

 * @author gvSIG Team

 */

@SuppressWarnings("UseSpecificCatch")

public class DataRepoTable extends AbstractRepoTable {

    public static final String TABLE_NAME = "VCSGISREPO_DATA";

    private static final String COD_DATA = "COD_DATA";

    private static final String COD_ENTITY = "COD_ENTITY";

    private static final String COD_REVISION = "COD_REVISION";

    private static final String DATA_REVNUMBER = "DAT_REVNUMBER";

    private static final String DATA_EFECTIVEDATE = "DAT_EFECTIVEDATE";

    private static final String DATA_OPERATION = "DAT_OPERATION";

    private static final String DATA_FEATURERELATEDCODE = "DAT_FEATURERELATEDCODE";

    private static final String DATA_GEOM = "DAT_GEOM";

    private static final String DATA_DATA = "DAT_DATA";

    public static class DataRepoRow extends AbstractRow implements VCSGisRepositoryData {

        public DataRepoRow(VCSGisRepositoryLocaldb repository) {

            super(repository, TABLE_NAME, COD_DATA, null);

        }

        public DataRepoRow(VCSGisRepositoryLocaldb repository, Feature feature) {

            super(repository, TABLE_NAME, COD_DATA, feature);

        }

        @Override

        public String getDataCode() {

            return this.getCode();

        }

        @Override

        public String getRevisionCode() {

            return this.feature.getString(COD_REVISION);

        }

        @Override

        public String getEntityCode() {

            return this.feature.getString(COD_ENTITY);

        }

        @Override

        public int getOperation() {

            return this.feature.getInt(DATA_OPERATION);

        }

        @Override

        public int getRevisionNumber() {

            return this.feature.getInt(DATA_REVNUMBER);

        }

        @Override

        public Timestamp getEfectiveDate() {

            return this.feature.getTimestamp(DATA_EFECTIVEDATE);

        }

        @Override

        public String getFeatureRelatedCode() {

            return this.feature.getString(DATA_FEATURERELATEDCODE);

        }

        public Geometry getGeometry() {

            return this.feature.getGeometry(DATA_GEOM);

        }

        @Override

        public String getData() {

            return this.feature.getString(DATA_DATA);

        }

        public void setRevisionCode(String code) {

            this.feature.set(COD_REVISION, code);

        }

        public void setEntityCode(String code) {

            this.feature.set(COD_ENTITY, code);

        }

        public void setOperation(int op) {

            this.feature.set(DATA_OPERATION, op);

        }

        public void setRevisionNumber(long revnum) {

            this.feature.set(DATA_REVNUMBER, revnum);

        }

        public void setEfectiveDate(Timestamp date) {

            this.feature.set(DATA_EFECTIVEDATE, date);

        }

        public void setIdentifier(String code) {

            this.feature.set(DATA_FEATURERELATEDCODE, code);

        }

        public void setGeometry(Geometry geom) {

            this.feature.set(DATA_GEOM, geom);

        }

        public void setValues(String jsonValues) {

            this.feature.set(DATA_DATA, jsonValues);

        }

        @Override

        public JsonObject getDataAsJson() {

            try {

                String s = this.getData();

                if( s == null ) {

                    return null;

                }

                JsonObject json = JsonUtils.toJsonObject(s);

                return json;

            } catch (CoercionException ex) {

                return null;

            }

        }

        @Override

        public JsonObject toJson() {

            return this.toJsonBuilder().build();

        }

        @Override

        public JsonObjectBuilder toJsonBuilder() {

            JsonObjectBuilder builder = Json.createObjectBuilder();

            builder.add("DataCode", this.getDataCode());

            builder.add("EntityCode", this.getEntityCode());

            builder.add("RevisionCode", this.getRevisionCode());

            builder.add("RevisionNumber", this.getRevisionNumber());

            builder.add("EfectiveDate", this.getEfectiveDate().toString());

            builder.add("Operation", this.getOperation());

            builder.add("FeatureRelatedCode", this.getFeatureRelatedCode());

            builder.add("Data", this.getData());

            return builder;

        }

    }

    public DataRepoTable() {

        super(TABLE_NAME, featureType());

    }

    public DisposableIterable<Feature> getDataOfEntityFromRevisionNumber(VCSGisRepositoryLocaldb repository, String entityCode, long revnumber) {

        // if revnumber < 0 return all features

        FeatureStore store = null;

        DisposableIterable<Feature> features = null;

        try {

            String sql = String.format(

                    VCSGisUtils.getSqlTemplate(

                            repository.getServerExplorer().getProviderName(), 

                            "getDataOfEntityFromRevisionNumber"),

                    revnumber,

                    entityCode

            );

            store = repository.openFeatureStore(TABLE_NAME,sql);

            FeatureSet fset = store.getFeatureSet();

//            dumpData("Data", fset);

            features = fset.iterable();

            return features;

        } catch (Exception ex) {

            DisposeUtils.disposeQuietly(features);

            throw new RuntimeException("Can't retrieve data of 'ENTITY["+entityCode+"] from revision "+revnumber+".", ex);

        } finally {

            if( store!=null ) {

                DisposeUtils.dispose(store);

            }

        }

    }

    public DisposableIterable<Feature> getDataOfEntityToRevisionNumber(VCSGisRepositoryLocaldb repository, String entityCode, long revnumber) {

        // if revnumber < 0 return all features

        FeatureStore store = null;

        DisposableIterable<Feature> features = null;

        try {

            String sql = String.format(

                    VCSGisUtils.getSqlTemplate(

                            repository.getServerExplorer().getProviderName(), 

                            "getDataOfEntityToRevisionNumber"),

                    revnumber,

                    entityCode

            );

            store = repository.openFeatureStore(TABLE_NAME,sql);

            FeatureSet fset = store.getFeatureSet();

//            dumpData("Data", fset);

            features = fset.iterable();

            return features;

        } catch (Exception ex) {

            DisposeUtils.disposeQuietly(features);

            throw new RuntimeException("Can't retrieve data of 'ENTITY["+entityCode+"] to revision "+revnumber+".", ex);

        } finally {

            if( store!=null ) {

                DisposeUtils.dispose(store);

            }

        }

    }

    public DisposableIterable<Feature> getDataOfEntityFromEfectiveDate(VCSGisRepositoryLocaldb repository, String entityCode, Timestamp efectiveDate) {

        FeatureStore store = null;

        DisposableIterable<Feature> features = null;

        SimpleDateFormat dateFormat = new SimpleDateFormat(

                VCSGisUtils.getSqlTemplate(

                        repository.getServerExplorer().getProviderName(),

                        "dateFormat")

        );

        try {

            String sql = String.format(

                    VCSGisUtils.getSqlTemplate(

                            repository.getServerExplorer().getProviderName(), 

                            "getDataOfEntityFromEfectiveDate"

                    ),

                    entityCode,

                    dateFormat.format(efectiveDate)

            );

            store = repository.openFeatureStore(TABLE_NAME,sql);

            FeatureSet fset = store.getFeatureSet();

            features = fset.iterable();

            return features;

        } catch (Exception ex) {

            DisposeUtils.disposeQuietly(features);

            throw new RuntimeException("Can't data of 'ENTITY["+entityCode+"] from efective date "+efectiveDate+".", ex);

        } finally {

            if( store!=null ) {

                DisposeUtils.dispose(store);

            }

        }

    }

    private void dumpData(String s, FeatureSet fset) {

        LOGGER.debug("BEGIN "+s);

        DisposableIterable<Feature> features = fset.iterable();

        for (Feature feature : features) {

            LOGGER.debug("DATA: "+ feature.toJson().toString());

        }

        LOGGER.debug("END "+s);

    }

    public static final FeatureType featureType() {

        DataManager dataManager = DALLocator.getDataManager();

        EditableFeatureType ft = dataManager.createFeatureType();

        ft.setLabel("VCSGIS Data");

        ft.getTags().set("ID", TABLE_NAME);

        ft.getTags().set(DynFormSPIManager.TAG_DYNFORM_HEIGHT, 400);

        ft.getTags().set(DynFormSPIManager.TAG_DYNFORM_WIDTH, 500);

        ft.getTags().set(DynFormSPIManager.TAG_DYNFORM_LAYOUTMODE, DynFormSPIManager.TAG_DYNFORM_LAYOUTMODE_VALUE_TABS);

        ft.add(COD_DATA, DataTypes.STRING)

                .setSize(VCSGISCODELEN)

                .setIsPrimaryKey(true)

                .setLabel("Code");

        ft.add(COD_REVISION, DataTypes.STRING)

                .setForeingkey(

                        true, 

                        false, 

                        RevisionsRepoTable.TABLE_NAME, 

                        RevisionsRepoTable.COD_REVISION, 

                        "FORMAT('r%d %s', REV_NUMBER, REV_OPERATIONDATE)"

                )

//                .setIsIndexed(true)

//                .setAllowIndexDuplicateds(true)

                .setSize(VCSGISCODELEN)

                .setLabel("Revision");

        ft.add(COD_ENTITY, DataTypes.STRING)

                .setForeingkey(

                        true, 

                        false, 

                        EntitiesRepoTable.TABLE_NAME, 

                        EntitiesRepoTable.COD_ENTITY, 

                        EntitiesRepoTable.ENTITY_NAME

                )

//                .setIsIndexed(true)

//                .setAllowIndexDuplicateds(true)

                .setSize(VCSGISCODELEN)

                .setLabel("Entity");

        ft.add(DATA_OPERATION, DataTypes.INTEGER)

//                .setIsIndexed(true)

//                .setAllowIndexDuplicateds(true)

                .setLabel("Operation")

                .setAvailableValues(new DynObjectValueItem[] {

                    new DynObjectValueItem(OP_INSERT, "Insert"),

                    new DynObjectValueItem(OP_DELETE, "Delete"),

                    new DynObjectValueItem(OP_UPDATE, "Update")

                });

        ft.add(DATA_REVNUMBER, DataTypes.LONG)

//                .setIsIndexed(true)

//                .setAllowIndexDuplicateds(true)

                .setLabel("Rev. number");

        ft.add(DATA_EFECTIVEDATE, DataTypes.TIMESTAMP)

//                .setIsIndexed(true)

//                .setAllowIndexDuplicateds(true)

                .setLabel("User date");

        ft.add(DATA_FEATURERELATEDCODE, DataTypes.STRING)

//                .setIsIndexed(true)

//                .setAllowIndexDuplicateds(true)

                .setSize(VCSGISCODELEN)

                .setLabel("Identifier")

                .setDescription("Identifier if the feature that represent.");

        ft.add(DATA_GEOM, DataTypes.GEOMETRY)

                .setGeometryType(Geometry.TYPES.GEOMETRY, Geometry.SUBTYPES.GEOM3D)

//                .setIsIndexed(true)

                .setLabel("Geometry");

        ft.add(DATA_DATA, DataTypes.STRING)

                .setSize(10240)

                .setLabel("Feature data serialized in json.")

                .setGroup("Data");

        return ft.getNotEditableCopy();

    }

    public boolean canUpdate(String relatedFeatureCode) {

        return true;

    }

    public boolean canDelete(String relatedFeatureCode) {

        return true;

    }

}