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 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.tools.persistence.Persistent;

import org.gvsig.tools.persistence.PersistentState;

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

import org.json.JSONArray;

import org.json.JSONObject;

/**

 *

 * @author fdiaz

 */

public class FeatureReferencePK implements FeatureReferenceProviderServices, Persistent {

    private Object[] pk;

    private String[] pkNames;

    private WeakReference storeRef;

    private String featureTypeId;

    /*friend*/

    FeatureReferencePK(FeatureStore store, FeatureProvider data) {

            this.storeRef = new WeakReference(store);

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

        this.calculatePK(data);

    }

    public void calculatePK(FeatureProvider fdata) {

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

        if (ArrayUtils.isEmpty(pkattrs)) {

            this.pk = null;

            this.pkNames = null;

            return;

        }

        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() {

        Object[] data = new Object[7];

        if (this.pk == null) {

            data[0] = -1;

            data[1] = JSONObject.NULL;

            data[2] = JSONObject.NULL;

        } else {

            data[0] = pkNames.length;

            data[1] = new JSONArray(this.pkNames);

            data[2] = new JSONArray(this.pk);

        }

        String s = new JSONArray(data).toString();

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

        return r;

    }

    @Override

    public void saveToState(PersistentState state) throws PersistenceException {

        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);

        }

    }

    @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;

        }

    }

}