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.impl;

import java.io.File;

import java.util.Collection;

import java.util.Iterator;

import java.util.Set;

import org.gvsig.fmap.dal.DALLocator;

import org.gvsig.fmap.dal.DataManager;

import org.gvsig.tools.ToolsLocator;

import org.gvsig.tools.dispose.Disposable;

import org.gvsig.tools.dispose.impl.AbstractDisposable;

import org.gvsig.tools.exception.BaseException;

import org.gvsig.tools.folders.FoldersManager;

import org.gvsig.tools.util.Size;

import org.gvsig.tools.util.Size64;

import org.h2.mvstore.MVMap;

import org.h2.mvstore.MVStore;

import org.slf4j.Logger;

import org.slf4j.LoggerFactory;

import org.gvsig.tools.util.Factory;

import org.gvsig.tools.util.IsEmpty;

/**

 *

 * @author fdiaz

 * @param <K>

 */

public class LargeSetImpl<K> extends AbstractDisposable implements Set<K>, Size64, Size, IsEmpty, Disposable {

    protected static final Logger LOGGER = LoggerFactory.getLogger(LargeSetImpl.class);

    private MVStore mvstore;

    private MVMap<K, Boolean> mvMap;

    public LargeSetImpl() {

        super();

        FoldersManager folderManager = ToolsLocator.getFoldersManager();

        File f = folderManager.getUniqueTemporaryFile("MinLargeSet.mv");

        this.mvstore = MVStore.open(f.getAbsolutePath());

        this.mvMap = mvstore.openMap("DefaultMap");

    }

    @Override

    protected void doDispose() throws BaseException {

        this.mvstore.close();

        this.mvMap = null;

        this.mvstore = null;

    }

    @Override

    public int size() {

        return this.mvMap.size();

    }

    @Override

    public boolean isEmpty() {

        return this.mvMap.isEmpty();

    }

    @Override

    public boolean contains(Object o) {

        return this.mvMap.containsKey((K)o);

    }

    @Override

    public Iterator<K> iterator() {

        return this.mvMap.keySet().iterator();

    }

    @Override

    public Object[] toArray() {

        //FIXME: Ojo, si el set es grande, esto puede puede provocar un desbordamiento de memoria.

        //duda ?implementarlo o lanzar UnsupportedOperationException?

//        LOGGER.warn("Access to LargeSet's toArray() method.");

//        return this.mvMap.entrySet().toArray();

        throw new UnsupportedOperationException("Not supported yet.");

    }

    @Override

    public <T> T[] toArray(T[] a) {

        //FIXME: Ojo, si el set es grande, esto puede puede provocar un desbordamiento de memoria.

        //duda ?implementarlo o lanzar UnsupportedOperationException?

//        LOGGER.warn("Access to LargeSet's toArray(T[] a) method.");

//        return this.mvMap.keySet().toArray((T[]) new K[this.size()]);

        throw new UnsupportedOperationException("Not supported yet.");

    }

    @Override

    public boolean add(K e) {

        return this.mvMap.put(e, Boolean.TRUE)!=null;

    }

    @Override

    public boolean remove(Object o) {

        return this.mvMap.remove((K)o)!=null;

    }

    @Override

    public boolean containsAll(Collection<?> c) {

        //FIXME: Comprobar que esto no lo haga en memoria

        return this.mvMap.keySet().containsAll(c);

        //Si lo hace en memoria, usar lo siguiente;

//        if (!c.stream().noneMatch(object -> (!contains((K)object)))) {

//            return false;

//        }

//        return true;

    }

    @Override

    public boolean addAll(Collection<? extends K> c) {

        boolean result = false;

        for (K k : c) {

            result |= this.add(k);

        }

        return result;

    }

    @Override

    public boolean retainAll(Collection<?> c) {

        boolean result = false;

        for (Object next : this) {

            if(!c.contains(next)){

                result |= remove((K)next);

            }

        }

        return result;

    }

    @Override

    public boolean removeAll(Collection<?> c) {

        boolean result = false;

        for (Object object : c) {

            result |= remove((K)object);

        }

        return result;

    }

    @Override

    public void clear() {

        this.mvMap.clear();

    }

    @Override

    public long size64() {

        return this.mvMap.sizeAsLong();

    }

    public static void selfRegister() {

        DataManager dataManager = DALLocator.getDataManager();

        dataManager.registerLargeSet(new Factory() {

            @Override

            public String getName() {

                return "MVStoreSetFactory";

            }

            @Override

            public Object create(Object... parameters) {

                return new LargeSetImpl();

            }

        });        

    }

}