Import/export changes. A load.

[simantics/platform.git] / bundles / org.simantics.db.layer0 / src / org / simantics / db / layer0 / util / ConsistsOfProcess.java

diff --git a/bundles/org.simantics.db.layer0/src/org/simantics/db/layer0/util/ConsistsOfProcess.java b/bundles/org.simantics.db.layer0/src/org/simantics/db/layer0/util/ConsistsOfProcess.java
index 39c661e4be2d6649b8bb85074b59fcc8e04a9eb0..d910f13ff946a77e40d9b2dce830c2020c678f40 100644 (file)
--- a/bundles/org.simantics.db.layer0/src/org/simantics/db/layer0/util/ConsistsOfProcess.java
+++ b/bundles/org.simantics.db.layer0/src/org/simantics/db/layer0/util/ConsistsOfProcess.java
@@ -1,48 +1,100 @@
 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);
                                        }
+                               }
                                
                        }
 
@@ -61,8 +113,10 @@ class ConsistsOfProcess {
 
                        @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);
+                               }
                        }
                        
                });