li.strolch.persistence.impl.InMemoryStrolchDao Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of agent Show documentation
Strolch Agent which is the runtime for Strolch
The newest version!
package li.strolch.persistence.impl;

import static java.util.stream.Collectors.toList;

import java.util.ArrayList;
import java.util.List;
import java.util.Set;

import li.strolch.model.StrolchRootElement;
import li.strolch.persistence.api.StrolchDao;
import li.strolch.persistence.api.StrolchPersistenceException;
import li.strolch.utils.collections.MapOfLists;

public class InMemoryStrolchDao implements StrolchDao {

	private final MapOfLists elements;

	public InMemoryStrolchDao() {
		this.elements = new MapOfLists<>();
	}

	public MapOfLists getElements() {
		return this.elements;
	}

	@Override
	public boolean supportsPaging() {
		return true;
	}

	@Override
	public long querySize() {
		return this.elements.size();
	}

	@Override
	public long querySize(String... types) {
		long size = 0L;
		for (String type : types) {
			size += this.elements.size(type);
		}
		return size;
	}

	@Override
	public Set queryTypes() throws StrolchPersistenceException {
		return this.elements.keySet();
	}

	@Override
	public List queryAll() throws StrolchPersistenceException {
		return this.elements.values();
	}

	@Override
	public List queryAll(long limit, long offset) throws StrolchPersistenceException {
		return this.elements.values().stream().skip(offset).limit(limit).collect(toList());
	}

	@Override
	public List queryAll(String... types) throws StrolchPersistenceException {
		List values = new ArrayList<>();
		for (String type : types) {
			if (this.elements.containsList(type))
				values.addAll(this.elements.getList(type));
		}
		return values;
	}

	@Override
	public List queryAll(long limit, long offset, String... types) throws StrolchPersistenceException {
		return this.queryAll(types).stream().skip(offset).limit(limit).collect(toList());
	}

	@Override
	public void save(T element) throws StrolchPersistenceException {
		this.elements.addElement(element.getType(), element);
	}

	@Override
	public void saveAll(List elements) throws StrolchPersistenceException {
		elements.forEach(this::save);
	}

	@Override
	public void update(T element) throws StrolchPersistenceException {
		save(element);
	}

	@Override
	public void updateAll(List elements) throws StrolchPersistenceException {
		saveAll(elements);
	}

	@Override
	public void remove(T element) throws StrolchPersistenceException {
		this.elements.removeElement(element.getType(), element);
	}

	@Override
	public void removeAll(List elements) throws StrolchPersistenceException {
		elements.forEach(this::remove);
	}

	@Override
	public long removeAll() throws StrolchPersistenceException {
		int size = this.elements.size();
		this.elements.clear();
		return size;
	}

	@Override
	public long removeAllBy(String type) throws StrolchPersistenceException {
		return this.elements.removeList(type).size();
	}

	@Override
	public T queryBy(String type, String id, int version) throws StrolchPersistenceException {
		throw new UnsupportedOperationException("Versioning is not supported!");
	}

	@Override
	public List queryVersionsFor(String type, String id) throws StrolchPersistenceException {
		throw new UnsupportedOperationException("Versioning is not supported!");
	}

	@Override
	public int queryLatestVersionFor(String type, String id) throws StrolchPersistenceException {
		throw new UnsupportedOperationException("Versioning is not supported!");
	}

	@Override
	public long queryVersionsSizeFor(String type, String id) throws StrolchPersistenceException {
		throw new UnsupportedOperationException("Versioning is not supported!");
	}

	@Override
	public void removeVersion(T element) throws StrolchPersistenceException {
		throw new UnsupportedOperationException("Versioning is not supported!");
	}

	@Override
	public void flush() throws StrolchPersistenceException {
		// do nothing
	}
}