org.openprovenance.prov.service.core.memory.NonDocumentResourceIndexInMemory Maven / Gradle / Ivy

package org.openprovenance.prov.service.core.memory;

import org.openprovenance.prov.storage.api.DocumentResource;
import org.openprovenance.prov.storage.api.Instantiable;
import org.openprovenance.prov.storage.api.NonDocumentResource;
import org.openprovenance.prov.storage.api.NonDocumentResourceIndex;

import java.util.HashMap;
import java.util.Map;

public class NonDocumentResourceIndexInMemory implements NonDocumentResourceIndex {
    public final Map table;

    public static Instantiable factory =new Instantiable() {
        @Override
        public DocumentResource newResource(DocumentResource dr) {
            return dr;
        }

        @Override
        public DocumentResource newResource() {
            return new DocumentResourceInMemory();
        }
    };

    public NonDocumentResourceIndexInMemory(Map table) {
        this.table=table;
    }

    public NonDocumentResourceIndexInMemory(int limit) {
        this.table=new LRUHashMap<>(limit);
    }

    public NonDocumentResourceIndexInMemory() {
        this.table = new HashMap<>();
    }

    @Override
    public NonDocumentResource get(String key) {
        return table.get(key);
    }

    @Override
    public void put(String key, NonDocumentResource dr) {
        table.put(key,dr);
    }

    @Override
    public void remove(String key) {
        System.out.println("NonDocumentResourceIndexInMemory.remove() " + key);
        table.remove(key);
    }

    @Override
    synchronized public String newId() {
        return "m" + DocumentResourceIndexInMemory.count++;
    }

    @Override
    public NonDocumentResource newResource() {
        String id=newId();
        NonDocumentResource dr=new NonDocumentResourceInMemory();
        dr.setVisibleId(id);
        put(id, dr); // MAYBE, I should not 'put' it by default
        return dr;
    }

    @Override
    public StorageKind kind() {
        return StorageKind.ME;
    }


    public static void register(Map> m) {
        m.put(DocumentResource.getResourceKind(), NonDocumentResourceIndexInMemory.factory);
    }


}