Application: gvSIG desktop

/*

 * To change this license header, choose License Headers in Project Properties.

 * To change this template file, choose Tools | Templates

 * and open the template in the editor.

 */

package org.gvsig.fmap.dal.feature.impl.featurereference;

import java.lang.ref.WeakReference;

import java.util.Base64;

import java.util.List;

import java.util.Objects;

import javax.json.JsonObject;

import org.apache.commons.lang3.ArrayUtils;

import org.gvsig.fmap.dal.exception.DataException;

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

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

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

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

import org.gvsig.fmap.dal.feature.impl.DefaultFeatureStore;

import org.gvsig.fmap.dal.feature.spi.FeatureProvider;

import org.gvsig.fmap.dal.feature.spi.FeatureReferenceProviderServices;

import org.gvsig.json.Json;

import org.gvsig.json.JsonArrayBuilder;

import org.gvsig.json.JsonObjectBuilder;

import org.gvsig.tools.persistence.Persistent;

import org.gvsig.tools.persistence.PersistentState;

import org.gvsig.tools.persistence.exception.PersistenceException;

/**

 *

 * @author fdiaz

 */

public class FeatureReferencePK implements FeatureReferenceProviderServices, Persistent {

    public static final String PK_TYPE = "pk";

    private Object[] pk;

    private String[] pkNames;

    private WeakReference storeRef;

    private String featureTypeId;

    private Integer myHashCode;

    /*friend*/

    FeatureReferencePK(FeatureStore store, FeatureProvider data) {

            this.storeRef = new WeakReference(store);

        this.featureTypeId = store.getDefaultFeatureTypeQuietly().getId();

        this.doCalculatePK(data);

    }

    FeatureReferencePK(FeatureStore store, JsonObject code) {

        try {

            this.storeRef = new WeakReference(store);

            this.featureTypeId = store.getDefaultFeatureTypeQuietly().getId();

            this.pk = Json.toArray(code, "values", new Object[code.getJsonArray("values").size()]);

            this.pkNames = (String[]) Json.toArray(code, "names", new String[code.getJsonArray("names").size()]);

        } catch (Exception ex) {

            throw ex;

        }

    }

    public FeatureReferencePK() {

        //For persistence use only

    }

    public void calculatePK(FeatureProvider fdata) {

        doCalculatePK(fdata);

    }

    private void doCalculatePK(FeatureProvider fdata) {

        FeatureAttributeDescriptor[] pkattrs = fdata.getType().getPrimaryKey();

        if (ArrayUtils.isEmpty(pkattrs)) {

            throw new IllegalArgumentException("Feature don't has primary key");

        }

        this.pk = new Object[pkattrs.length];

        this.pkNames = new String[pkattrs.length];

        int n = 0;

        for (FeatureAttributeDescriptor pkattr : pkattrs) {

            this.pk[n] = fdata.get(pkattr.getIndex());

            this.pkNames[n] = pkattr.getName();

            n++;

        }

    }

    @Override

    public Object getOID() {

        return null;

    }

    @Override

    public String[] getKeyNames() {

        return this.pkNames;

    }

    @Override

    public Object getKeyValue(String name) {

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

            if (pkNames[i].equalsIgnoreCase(name)) {

                return pk[i];

            }

        }

        // FIXME exception????

        return null;

    }

    @Override

    public String getFeatureTypeId() {

        return featureTypeId;

    }

    @Override

    public Feature getFeature() throws DataException {

        return this.getStore().getFeatureByReference(this);

    }

    @Override

    public Feature getFeatureQuietly() {

        try {

            return this.getFeature();

        } catch (Exception e) {

            return null;

        }

    }

    @Override

    public Feature getFeature(FeatureType featureType) throws DataException {

        return this.getStore().getFeatureByReference(this, featureType);

    }

    private DefaultFeatureStore getStore() {

        return (DefaultFeatureStore) this.storeRef.get();

    }

    @Override

    public boolean isNewFeature() {

        return false;

    }

    @Override

    public String getCode() {

        String s = toJsonBuilder().toString();

        String r = Base64.getEncoder().encodeToString(s.getBytes());

        return r;

    }

    @Override

    public void saveToState(PersistentState state) throws PersistenceException {

        state.set("type", PK_TYPE);

        state.set("store", (Persistent) storeRef.get());

        state.set("featureTypeId", featureTypeId);

        if (pk == null) {

            state.setNull("pk");

            state.setNull("pkNames");

        } else {

            state.set("pk", pk);

            state.set("pkNames", pkNames);

        }

        state.set("isNewFeature", isNewFeature());

    }

    @Override

    public void loadFromState(PersistentState state)

            throws PersistenceException {

        this.storeRef = new WeakReference(state.get("store"));

        this.featureTypeId = state.getString("featureTypeId");

        List pkList = (List) state.get("pk");

        if (pkList != null) {

            List pkNamesList = (List) state.get("pkNames");

            if (pkNamesList == null || pkList.size() != pkNamesList.size()) {

                throw new PersistenceException("bad pkNames value");

            }

            this.pk = pkList.toArray();

            this.pkNames

                    = (String[]) pkNamesList.toArray(new String[pkList.size()]);

        } else {

            this.pk = null;

            this.pkNames = null;

        }

    }

        @Override

    public boolean equals(Object obj) {

        if (!(obj instanceof FeatureReferencePK)) {

            return false;

        }

        FeatureReferencePK other = (FeatureReferencePK) obj;

        FeatureStore otherStore = (FeatureStore) other.storeRef.get();

        FeatureStore myrStore = (FeatureStore) this.storeRef.get();

        if (otherStore == null || myrStore == null) {

            return false;

        }

        if (!myrStore.equals(otherStore)) {

            return false;

        }

        if (myHashCode != null && other.myHashCode != null && !myHashCode.equals(other.myHashCode)) {

            return false;

        }

        if (this.pk == other.pk) {

            return true;

        }

        if (this.pk == null || other.pk == null) {

            return false;

        }

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

            if (!this.pk[i].equals(other.pk[i])) {

                return false;

            }

        }

        return true;

    }

    @Override

    public int hashCode() {

        if (myHashCode == null) {

            StringBuilder buff = new StringBuilder();

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

                buff.append(Objects.hashCode(this.pk[i]));

                buff.append("##");

            }

            myHashCode = buff.toString().hashCode();

        }

        return myHashCode;

    }

    @Override

    public JsonObjectBuilder toJsonBuilder() {

        JsonObjectBuilder builder = Json.createObjectBuilder();

        builder.add("type", PK_TYPE);

        if (this.pk == null) {

            builder.addNull("names");

            builder.addNull("values");

        } else {

            JsonArrayBuilder arrayNames = Json.createArrayBuilder();

            JsonArrayBuilder arrayValues = Json.createArrayBuilder();

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

                arrayNames.add(pkNames[i]);

                arrayValues.add(pk[i]);

            }

            builder.add("names", arrayNames);

            builder.add("values", arrayValues);

        }

        return builder;

    }

    @Override

    public String toString() {

        if( this.pk == null ) {

            return "NULL";

        }

        boolean needsep = false;

        StringBuilder builder = new StringBuilder();

        for (Object v : pk) {

            if( needsep ) {

                builder.append("/");

            }

            builder.append(Objects.toString(v));

            needsep = true;

        }

        return builder.toString();

    }

}