prompto.intrinsic.PromptoDeleteAndStoreQuery Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of Core Show documentation
The newest version!
package prompto.intrinsic;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Iterator;
import java.util.Map;
import java.util.Set;

import prompto.error.PromptoError;
import prompto.grammar.Identifier;
import prompto.runtime.Context;
import prompto.store.DataStore;
import prompto.store.IAuditMetadata;
import prompto.store.IStorable;
import prompto.store.IStore;
import prompto.value.DocumentValue;
import prompto.value.IInstance;
import prompto.value.IIterable;
import prompto.value.IValue;
import prompto.value.NullValue;

public class PromptoDeleteAndStoreQuery {

	Set deletables = new HashSet<>();
	Map storables = new HashMap<>();
	IAuditMetadata metadata;
	
	public void execute() {
		if(deletables.isEmpty())
			deletables = null;
		if(storables.isEmpty())
			storables = null;
		if(deletables!=null || storables!=null) {
			IStore store = DataStore.getInstance();
			try {
				store.deleteAndStore(deletables, storables==null ? null : storables.values(), metadata);
				if(storables!=null)
					storables.values().forEach(IStorable::clear);
			} catch(PromptoError e) {
				throw new RuntimeException(e); // TODO for now
			} 
		}
	}

	@SuppressWarnings("unchecked")
	public void delete(Context context, IValue value) {
		if(value==NullValue.instance())
			return;
		else if(value instanceof IInstance) try {
			IValue dbId = ((IInstance)value).getMember(context, new Identifier(IStore.dbIdName), false);
			if(dbId!=null)
				deletables.add((PromptoDbId)dbId.getStorableData());
		} catch(PromptoError e) {
			throw new RuntimeException(e); // TODO for now
		} else if(value instanceof IIterable)
			((IIterable)value).getIterable(context).forEach((item)->
				delete(context, item));
		else if(value instanceof Iterator) {
			Iterator iter = (Iterator)value;
			while(iter.hasNext())
				delete(context, iter.next());
		} else
			throw new UnsupportedOperationException("Can't delete " + value.getClass());
	}

	@SuppressWarnings("unchecked")
	public void store(Context context, IValue value) {
		if(value==NullValue.instance())
			return;
		else if(value instanceof IInstance) try {
			((IInstance)value).collectStorables((s)->
				storables.put(s.getOrCreateDbId(), s));
		} catch(PromptoError e) {
			throw new RuntimeException(e);
		} else if(value instanceof IIterable) {
			((IIterable)value).getIterable(context).forEach((item)->
				store(context, item));
		} else if(value instanceof Iterator) {
			Iterator iter = (Iterator)value;
			while(iter.hasNext())
				store(iter.next());
		} else
			throw new UnsupportedOperationException("Can't store " + value.getClass());
	}

	@SuppressWarnings("unchecked")
	public void delete(Object value) {
		if(value==null)
			return;
		else if(value instanceof PromptoRoot) try {
			PromptoDbId dbId = ((PromptoRoot)value).getDbId();
			if(dbId!=null)
				deletables.add(dbId);
		} catch(PromptoError e) {
			throw new RuntimeException(e); // TODO for now
		} else if(value instanceof Iterable)
			((Iterable