package org.simantics.db.layer0.util;
+import java.util.ArrayList;
import java.util.Collection;
+import java.util.List;
import java.util.Set;
import org.simantics.db.AsyncReadGraph;
import org.simantics.db.ReadGraph;
import org.simantics.db.Resource;
+import org.simantics.db.ResourceMap;
import org.simantics.db.common.request.ReadRequest;
import org.simantics.db.common.utils.Logger;
import org.simantics.db.exception.DatabaseException;
+import org.simantics.db.layer0.adapter.SubgraphExtent.ExtentStatus;
import org.simantics.db.procedure.AsyncContextMultiProcedure;
-import org.simantics.db.service.CollectionSupport;
+import org.simantics.db.procedure.Procedure;
import org.simantics.db.service.DirectQuerySupport;
import org.simantics.layer0.Layer0;
class ConsistsOfProcess {
- final Set<Resource> result;
- final AsyncContextMultiProcedure<Resource, Resource> structure;
+ final List<InternalEntry> result;
+ final AsyncContextMultiProcedure<InternalEntry, Resource> structure;
+ final AsyncContextMultiProcedure<InternalEntry, Resource> names;
- public static Set<Resource> walk(ReadGraph graph, Collection<Resource> resources, Set<Resource> exclusions, boolean ignoreVirtual) throws DatabaseException {
- ConsistsOfProcess process = new ConsistsOfProcess(graph, resources, exclusions, ignoreVirtual);
+ public static List<InternalEntry> walk(ReadGraph graph, ResourceMap<ExtentStatus> status, Collection<Resource> resources, Set<Resource> exclusions, boolean ignoreVirtual) throws DatabaseException {
+ ConsistsOfProcess process = new ConsistsOfProcess(graph, status, resources, exclusions, ignoreVirtual);
return process.result;
}
+
+ static class InternalEntry {
+ public InternalEntry parent;
+ public Resource resource;
+ public String name;
+ InternalEntry(InternalEntry parent, Resource resource, String name) {
+ this.parent = parent;
+ this.resource = resource;
+ this.name = name;
+ }
+ }
- private ConsistsOfProcess(ReadGraph graph, final Collection<Resource> resources, final Set<Resource> exclusions, final boolean ignoreVirtual) throws DatabaseException {
+ private ConsistsOfProcess(ReadGraph graph, ResourceMap<ExtentStatus> status, final Collection<Resource> resources, final Set<Resource> exclusions, final boolean ignoreVirtual) throws DatabaseException {
final Layer0 L0 = Layer0.getInstance(graph);
final DirectQuerySupport dqs = graph.getService(DirectQuerySupport.class);
- CollectionSupport cs = graph.getService(CollectionSupport.class);
- result = cs.createSet();
+ result = new ArrayList<InternalEntry>();
+
+ names = dqs.compileForEachObject(graph, L0.HasName, new AsyncContextMultiProcedure<InternalEntry, Resource>() {
+
+ @Override
+ public void execute(AsyncReadGraph graph, InternalEntry entry, Resource nameResource) {
+
+ if(status != null)
+ status.put(nameResource, ExtentStatus.EXCLUDED);
+
+ graph.forPossibleValue(nameResource, new Procedure<String>() {
+
+ @Override
+ public void execute(String result) {
+ entry.name = result;
+ }
+
+ @Override
+ public void exception(Throwable t) {
+ Logger.defaultLogError(t);
+ }
+
+ });
+ }
+
+ @Override
+ public void exception(AsyncReadGraph graph, Throwable throwable) {
+ Logger.defaultLogError(throwable);
+ }
+
+ @Override
+ public void finished(AsyncReadGraph graph) {
+ }
+
+ });
- structure = dqs.compileForEachObject(graph, L0.ConsistsOf, new AsyncContextMultiProcedure<Resource, Resource>() {
+ structure = dqs.compileForEachObject(graph, L0.ConsistsOf, new AsyncContextMultiProcedure<InternalEntry, Resource>() {
@Override
- public void execute(AsyncReadGraph graph, Resource parent, Resource child) {
+ public void execute(AsyncReadGraph graph, InternalEntry parent, Resource child) {
if(exclusions.contains(child)) return;
- if(!ignoreVirtual || child.isPersistent())
- if(result.add(child)) {
- dqs.forEachObjectCompiled(graph, child, child, structure);
+ if(!ignoreVirtual || child.isPersistent()) {
+ InternalEntry entry = new InternalEntry(parent, child, null);
+ if(result.add(entry)) {
+ dqs.forEachObjectCompiled(graph, child, entry, structure);
+ dqs.forEachObjectCompiled(graph, child, entry, names);
}
+ }
}
@Override
public void run(ReadGraph graph) throws DatabaseException {
- for(Resource r : resources)
- dqs.forEachObjectCompiled(graph, r, r, structure);
+ for(Resource r : resources) {
+ InternalEntry root = new InternalEntry(null, r, null);
+ dqs.forEachObjectCompiled(graph, r, root, structure);
+ }
}
});