Application: gvSIG desktop

package org.gvsig.fmap.dal.impl;

import org.gvsig.fmap.dal.DatabaseWorkspaceManager;

import java.util.HashMap;

import java.util.Map;

import org.gvsig.fmap.dal.AbstractStoresRepository;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.fmap.dal.DataServerExplorer;

import org.gvsig.fmap.dal.DataServerExplorerParameters;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.StoresRepository;

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

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

import static org.gvsig.fmap.dal.DatabaseWorkspaceManager.FIELD_REPOSITORY_NAME;

import org.gvsig.tools.exception.BaseException;

import org.gvsig.tools.util.UnmodifiableBasicSet;

import org.gvsig.tools.util.UnmodifiableBasicSetAdapter;

import org.gvsig.tools.visitor.VisitCanceledException;

import org.gvsig.tools.visitor.Visitor;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import static org.gvsig.fmap.dal.DatabaseWorkspaceManager.FIELD_REPOSITORY_PARAMETERS;

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public class DatabaseWorkspaceStoresRepository 

        extends AbstractStoresRepository

        implements StoresRepository 

    {

    private static final Logger LOGGER = LoggerFactory.getLogger(DefaultDatabaseWorkspaceManager.class);

    private final DataServerExplorerParameters connection;

    private Map<String,DataStoreParameters> repository;

    public DatabaseWorkspaceStoresRepository(String name, String label, DataServerExplorerParameters connection) {

        super(name, label);

        this.connection = connection;

        this.repository = null;

    }

    private FeatureStore getTable() {

        String tableName = DatabaseWorkspaceManager.TABLE_REPOSITORY_NAME;

        try {

            DataManager dataManager = DALLocator.getDataManager();

            DataServerExplorer server = dataManager.openServerExplorer(

                    this.connection.getExplorerName(),

                    this.connection

            );

            DataStoreParameters params = server.get(tableName);

            if( params!=null ) {

                FeatureStore store = (FeatureStore) dataManager.openStore(

                        params.getDataStoreName(), 

                        params

                );

                return store;

            }

        } catch (Exception ex) {

            LOGGER.warn("Can't open table '"+tableName+"'.",ex);

        }

        return null;

    }

    private void load() {

        try {

            this.repository = new HashMap<>();

            final DataManager dataManager = DALLocator.getDataManager();

            FeatureStore store = this.getTable();

            store.accept(new Visitor() {

                @Override

                public void visit(Object obj) throws VisitCanceledException, BaseException {

                    Feature f = (Feature) obj;

                    String name = f.getString(FIELD_REPOSITORY_NAME);

                    byte[] data = f.getByteArray(FIELD_REPOSITORY_PARAMETERS);

                    DataStoreParameters params = dataManager.createStoreParameters(data);

                    if( params==null ) {

                        LOGGER.warn("Can't restore parameters from repository entry '"+name+"'.");

                        return;

                    }

                    repository.put(name, params);

                }

            });

        } catch (Exception ex) {

            LOGGER.warn("Can't load repository information", ex);

        }

    }

    @Override

    protected DataStoreParameters getMyParameters(String name) {

        if( this.repository==null ) {

            this.load();

        }

        return this.repository.get(name);

    }

    @Override

    protected boolean isEmptyMyRepository() {

        if( this.repository==null ) {

            this.load();

        }

        return this.repository.isEmpty();

    }

    @Override

    protected int getMySize() {

        if( this.repository==null ) {

            this.load();

        }

        return this.repository.size();

    }

    @Override

    protected UnmodifiableBasicSet<String> getMyKeySet() {

        if( this.repository==null ) {

            this.load();

        }

        return new UnmodifiableBasicSetAdapter<>(this.repository.keySet());

    }

}