Migrated source code from Simantics SVN

[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
new file mode 100644 (file)
index 0000000..0b461bc
--- /dev/null
+++ b/bundles/org.simantics.db.layer0/src/org/simantics/db/layer0/util/ConsistsOfProcess.java
@@ -0,0 +1,73 @@
+package org.simantics.db.layer0.util;\r
+\r
+import java.util.Collection;\r
+import java.util.Set;\r
+\r
+import org.simantics.db.AsyncReadGraph;\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.Resource;\r
+import org.simantics.db.common.request.ReadRequest;\r
+import org.simantics.db.common.utils.Logger;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.db.procedure.AsyncContextMultiProcedure;\r
+import org.simantics.db.service.CollectionSupport;\r
+import org.simantics.db.service.DirectQuerySupport;\r
+import org.simantics.layer0.Layer0;\r
+\r
+class ConsistsOfProcess {\r
+\r
+       final Set<Resource> result;\r
+       final AsyncContextMultiProcedure<Resource, Resource> structure;\r
+\r
+    public static Set<Resource> walk(ReadGraph graph, Collection<Resource> resources, Set<Resource> exclusions, boolean ignoreVirtual) throws DatabaseException {\r
+       ConsistsOfProcess process = new ConsistsOfProcess(graph, resources, exclusions, ignoreVirtual);\r
+       return process.result;\r
+    }\r
+       \r
+    private ConsistsOfProcess(ReadGraph graph, final Collection<Resource> resources, final Set<Resource> exclusions, final boolean ignoreVirtual) throws DatabaseException {\r
+\r
+               final Layer0 L0 = Layer0.getInstance(graph);\r
+               final DirectQuerySupport dqs = graph.getService(DirectQuerySupport.class);\r
+               \r
+               CollectionSupport cs = graph.getService(CollectionSupport.class);\r
+               result = cs.createSet();\r
+               \r
+               structure = dqs.compileForEachObject(graph, L0.ConsistsOf, new AsyncContextMultiProcedure<Resource, Resource>() {\r
+\r
+                       @Override\r
+                       public void execute(AsyncReadGraph graph, Resource parent, Resource child) {\r
+                               \r
+                               if(exclusions.contains(child)) return;\r
+                               \r
+                               if(!ignoreVirtual || child.isPersistent())\r
+                                       if(result.add(child)) {\r
+                                               dqs.forEachObjectCompiled(graph, child, child, structure);\r
+                                       }\r
+                               \r
+                       }\r
+\r
+                       @Override\r
+                       public void finished(AsyncReadGraph graph) {\r
+                       }\r
+\r
+                       @Override\r
+                       public void exception(AsyncReadGraph graph, Throwable throwable) {\r
+                               Logger.defaultLogError(throwable);\r
+                       }\r
+\r
+               });\r
+               \r
+               graph.syncRequest(new ReadRequest() {\r
+\r
+                       @Override\r
+                       public void run(ReadGraph graph) throws DatabaseException {\r
+                               for(Resource r  : resources)\r
+                                       dqs.forEachObjectCompiled(graph, r, r, structure);\r
+                       }\r
+                       \r
+               });\r
+               \r
+       }\r
+       \r
+    \r
+}\r