org.vertexium.inmemory.InMemoryTable Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of vertexium-inmemory Show documentation
There is a newer version: 4.10.0
package org.vertexium.inmemory;

import org.vertexium.Authorizations;
import org.vertexium.FetchHints;
import org.vertexium.inmemory.mutations.Mutation;
import org.vertexium.util.StreamUtils;

import java.util.ArrayList;
import java.util.Map;
import java.util.concurrent.ConcurrentSkipListMap;
import java.util.concurrent.locks.ReadWriteLock;
import java.util.concurrent.locks.ReentrantReadWriteLock;
import java.util.stream.Collectors;

public abstract class InMemoryTable {
    private ReadWriteLock rowsLock = new ReentrantReadWriteLock();
    private Map> rows;

    protected InMemoryTable(Map> rows) {
        this.rows = rows;
    }

    protected InMemoryTable() {
        this(new ConcurrentSkipListMap<>());
    }

    public TElement get(InMemoryGraph graph, String id, FetchHints fetchHints, Authorizations authorizations) {
        InMemoryTableElement inMemoryTableElement = getTableElement(id);
        if (inMemoryTableElement == null) {
            return null;
        }
        return inMemoryTableElement.createElement(graph, fetchHints, authorizations);
    }

    public InMemoryTableElement getTableElement(String id) {
        rowsLock.readLock().lock();
        try {
            return rows.get(id);
        } finally {
            rowsLock.readLock().unlock();
        }
    }

    public void append(String id, Mutation... newMutations) {
        rowsLock.writeLock().lock();
        try {
            InMemoryTableElement inMemoryTableElement = rows.get(id);
            if (inMemoryTableElement == null) {
                inMemoryTableElement = createInMemoryTableElement(id);
                rows.put(id, inMemoryTableElement);
            }
            inMemoryTableElement.addAll(newMutations);
        } finally {
            rowsLock.writeLock().unlock();
        }
    }

    protected abstract InMemoryTableElement createInMemoryTableElement(String id);

    public void remove(String id) {
        rowsLock.writeLock().lock();
        try {
            rows.remove(id);
        } finally {
            rowsLock.writeLock().unlock();
        }
    }

    public void clear() {
        rowsLock.writeLock().lock();
        try {
            rows.clear();
        } finally {
            rowsLock.writeLock().unlock();
        }
    }

    public Iterable getAll(
            InMemoryGraph graph,
            FetchHints fetchHints,
            Long endTime,
            Authorizations authorizations
    ) {
        return StreamUtils.stream(getRowValues())
                .filter(element -> graph.isIncludedInTimeSpan(element, fetchHints, endTime, authorizations))
                .map(element -> element.createElement(graph, fetchHints, endTime, authorizations))
                .collect(Collectors.toList());
    }

    public Iterable> getRowValues() {
        rowsLock.readLock().lock();
        try {
            return new ArrayList<>(this.rows.values());
        } finally {
            rowsLock.readLock().unlock();
        }
    }
}