Application: gvSIG desktop

package org.gvsig.fmap.dal.impl;

import java.util.HashMap;

import java.util.Iterator;

import java.util.Map;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataStore;

import org.gvsig.fmap.dal.DataStoreParameters;

import org.gvsig.fmap.dal.StoresRepository;

import org.gvsig.tools.util.GetItemByKey;

import org.gvsig.tools.util.UnmodifiableBasicMap;

import org.gvsig.tools.util.UnmodifiableBasicSet;

import org.gvsig.tools.util.UnmodifiableBasicSetAdapter;

import org.gvsig.tools.util.UnmodifiableBasicSetChained;

/**

 *

 * @author jjdelcerro

 */

@SuppressWarnings("UseSpecificCatch")

public class DefaultStoresRepository implements StoresRepository {

    private final Map<String,DataStoreParameters> repository;

    private final Map<String,UnmodifiableBasicMap<String,DataStoreParameters>> delegateds;

    public DefaultStoresRepository() {

        this.repository = new HashMap<>();

        this.delegateds = new HashMap<>();

    }

    public void add(String name, UnmodifiableBasicMap<String,DataStoreParameters> repository) {

        this.delegateds.put(name, repository);

    }

    @Override

    public void add(String name, DataStoreParameters parameters) {

        this.repository.put(name, parameters);

    }

    @Override

    public void remove(String name) {

        this.repository.remove(name);

    }

    @Override

    public DataStore get(String name) {

        DataStoreParameters parameters = this.repository.get(name);

        if( parameters==null ) {

            for (GetItemByKey<String,DataStoreParameters> theRepository : this.delegateds.values()) {

                parameters = theRepository.get(name);

                if( parameters!=null ) {

                    break;

                }

            }

            if( parameters==null ) {

                throw new IllegalArgumentException("Can't exists store '"+name+"' in the repository.");

            }

        }

        DataStore store;

        try {

            store = DALLocator.getDataManager().openStore(parameters.getDataStoreName(), parameters);

        } catch (Exception ex) {

            throw new RuntimeException("Can't open store '"+name+"'.", ex);

        }

        return store;

    }

    @Override

    public boolean isEmpty() {

        if( !this.repository.isEmpty() ) {

            return false;

        }

        for (UnmodifiableBasicMap<String,DataStoreParameters> theRepository : this.delegateds.values()) {

            if( !theRepository.isEmpty() ) {

                return false;

            }

        }

        return true;

    }

    @Override

    public boolean containsKey(String name) {

        if( this.repository.containsKey(name) ) {

            return true;

        }

        for (GetItemByKey<String,DataStoreParameters> theRepository : this.delegateds.values()) {

            DataStoreParameters parameters = theRepository.get(name);

            if( parameters!=null ) {

                return true;

            }

        }

        return false;

    }

    @Override

    public Map<String, DataStore> toMap() {

        throw new UnsupportedOperationException();

    }

    @Override

    public int size() {

        int sz = this.repository.size();

        for (UnmodifiableBasicMap<String,DataStoreParameters> theRepository : this.delegateds.values()) {

            sz += theRepository.size();

        }

        return sz;

    }

    @Override

    public Iterator<DataStore> iterator() {

        final Iterator<String> it = this.keySet().iterator();

        return new Iterator<DataStore>() {

            @Override

            public boolean hasNext() {

                return it.hasNext();

            }

            @Override

            public DataStore next() {

                String name = it.next();

                return get(name);

            }

        };

    }

    @Override

    public UnmodifiableBasicSet<String> keySet() {

        UnmodifiableBasicSet[] sets = new UnmodifiableBasicSet[this.delegateds.size()+1];

        int i=0;

        sets[i++] = new UnmodifiableBasicSetAdapter(this.repository.keySet());

        for (UnmodifiableBasicMap<String,DataStoreParameters> theRepository : this.delegateds.values()) {

            sets[i++] = theRepository.keySet();

        }

        return new UnmodifiableBasicSetChained(sets);

    }

}