--- /dev/null
+/*******************************************************************************\r
+ * Copyright (c) 2007, 2010 Association for Decentralized Information Management\r
+ * in Industry THTH ry.\r
+ * All rights reserved. This program and the accompanying materials\r
+ * are made available under the terms of the Eclipse Public License v1.0\r
+ * which accompanies this distribution, and is available at\r
+ * http://www.eclipse.org/legal/epl-v10.html\r
+ *\r
+ * Contributors:\r
+ * VTT Technical Research Centre of Finland - initial API and implementation\r
+ *******************************************************************************/\r
+package org.simantics.db.impl.query;\r
+\r
+import gnu.trove.map.hash.TObjectIntHashMap;\r
+\r
+import java.util.concurrent.Semaphore;\r
+\r
+import org.simantics.databoard.binding.Binding;\r
+import org.simantics.databoard.serialization.Serializer;\r
+import org.simantics.databoard.util.URIStringUtils;\r
+import org.simantics.db.common.WriteBindings;\r
+import org.simantics.db.common.exception.DebugException;\r
+import org.simantics.db.impl.graph.ReadGraphImpl;\r
+import org.simantics.db.impl.procedure.InternalProcedure;\r
+import org.simantics.db.procedure.ListenerBase;\r
+\r
+final public class NamespaceIndex extends StringQuery<InternalProcedure<TObjectIntHashMap<String>>> {\r
+ \r
+// public ArrayList<InternalProcedure<TObjectIntHashMap<String>>> procs = null;\r
+ \r
+ private NamespaceIndex(final String id) {\r
+ super(id);\r
+ }\r
+ \r
+ final static void runner(ReadGraphImpl graph, final String id, final QueryProcessor provider, NamespaceIndex cached, final CacheEntry parent, final ListenerBase listener, final InternalProcedure<TObjectIntHashMap<String>> procedure) {\r
+\r
+ NamespaceIndex entry = cached != null ? cached : (NamespaceIndex)provider.namespaceIndexMap22.get(id); \r
+ if(entry == null) {\r
+ \r
+ entry = new NamespaceIndex(id);\r
+ entry.setPending();\r
+ entry.clearResult(provider.querySupport);\r
+ entry.putEntry(provider);\r
+\r
+ provider.performForEach(graph, entry, parent, listener, procedure);\r
+ \r
+ } else {\r
+\r
+ if(entry.isPending()) {\r
+ synchronized(entry) {\r
+ if(entry.isPending()) {\r
+ throw new IllegalStateException();\r
+// if(entry.procs == null) entry.procs = new ArrayList<InternalProcedure<TObjectIntHashMap<String>>>(); \r
+// entry.procs.add(procedure);\r
+// provider.registerDependencies(graph, entry, parent, listener, procedure, false);\r
+// return;\r
+ }\r
+ }\r
+ }\r
+ provider.performForEach(graph, entry, parent, listener, procedure);\r
+ }\r
+ \r
+ }\r
+ \r
+ final public static void queryEach(ReadGraphImpl graph, final String id, final QueryProcessor provider, final CacheEntry parent, final ListenerBase listener, final InternalProcedure<TObjectIntHashMap<String>> procedure) {\r
+\r
+ final NamespaceIndex entry = (NamespaceIndex)provider.namespaceIndexMap22.get(id);\r
+ \r
+ if(parent == null && listener == null && entry != null && entry.isReady()) {\r
+ entry.performFromCache(graph, provider, procedure);\r
+ return;\r
+ }\r
+ \r
+ runner(graph, id, provider, entry, parent, listener, procedure);\r
+ \r
+ }\r
+ \r
+ @Override\r
+ public NamespaceIndex getEntry(QueryProcessor provider) {\r
+ return provider.namespaceIndexMap22.get(id);\r
+ }\r
+ \r
+ @Override\r
+ public void putEntry(QueryProcessor provider) {\r
+ provider.namespaceIndexMap22.put(id, this);\r
+ }\r
+\r
+ @Override\r
+ final public void removeEntry(QueryProcessor provider) {\r
+ provider.namespaceIndexMap22.remove(id);\r
+ }\r
+ \r
+ final private void index(ReadGraphImpl graph, final QueryProcessor provider, int root, final InternalProcedure<TObjectIntHashMap<String>> procedure) {\r
+ \r
+ if(root == 0) {\r
+ add2(graph, null);\r
+ procedure.execute(graph, null);\r
+// System.err.println("NamespaceIndex[" + id + "]->null");\r
+ return;\r
+ }\r
+\r
+ final int consistsOf = provider.getConsistsOf();\r
+ final int hasName = provider.getHasName();\r
+ \r
+ final TObjectIntHashMap<String> result = new TObjectIntHashMap<String>();\r
+ \r
+ Objects.runner(graph, root, consistsOf, graph.parent, null, new SyncIntProcedure() {\r
+ \r
+ @Override\r
+ public void run(ReadGraphImpl graph) {\r
+ \r
+ if(isPending()) { \r
+ add2(graph, result);\r
+ procedure.execute(graph, result);\r
+// System.err.println("NamespaceIndex[" + id + "]->" + result.size());\r
+ } else {\r
+ procedure.exception(graph, (Throwable)statusOrException);\r
+ }\r
+ \r
+ }\r
+\r
+ @Override\r
+ public void finished(ReadGraphImpl graph) {\r
+ \r
+ dec(graph);\r
+ \r
+ }\r
+\r
+ @Override\r
+ public void execute(ReadGraphImpl graph, final int obj) {\r
+ \r
+ //System.out.println(id + " => " + obj);\r
+\r
+ inc();\r
+ \r
+ Objects.runner(graph, obj, hasName, graph.parent, null, new IntProcedure() {\r
+ \r
+ @Override\r
+ public void execute(ReadGraphImpl graph, int i) {\r
+\r
+ inc();\r
+\r
+ ValueQuery.queryEach(graph, i, NamespaceIndex.this, null, new InternalProcedure<byte[]>() {\r
+ \r
+ @Override\r
+ public void execute(ReadGraphImpl graph, byte[] value) {\r
+ \r
+ if(value != null) {\r
+\r
+ try {\r
+\r
+ Binding b = WriteBindings.STRING;\r
+ Serializer serializer = b.serializer();\r
+ final String part = (String)serializer.deserialize(value);\r
+ \r
+ synchronized(result) {\r
+ Object previous = result.put(URIStringUtils.escape(part), obj);\r
+ // TODO: this is not the most elegant solution\r
+ if(previous != null) previous = "";\r
+ }\r
+ \r
+ } catch (Throwable e) {\r
+ if(DebugException.DEBUG) new DebugException(e).printStackTrace();\r
+ }\r
+ \r
+ }\r
+ \r
+ dec(graph);\r
+ \r
+ }\r
+ \r
+ @Override\r
+ public void exception(ReadGraphImpl graph, Throwable t) {\r
+ except(t);\r
+ dec(graph);\r
+ }\r
+\r
+ });\r
+ \r
+ }\r
+ \r
+ @Override\r
+ public void finished(ReadGraphImpl graph) {\r
+ dec(graph);\r
+ }\r
+ \r
+ @Override\r
+ public void exception(ReadGraphImpl graph, Throwable t) {\r
+ except(t);\r
+ dec(graph);\r
+ }\r
+\r
+ });\r
+\r
+ }\r
+ \r
+ });\r
+ \r
+ }\r
+\r
+ @Override\r
+ public void computeForEach(ReadGraphImpl graph, final QueryProcessor processor, final InternalProcedure<TObjectIntHashMap<String>> procedure) {\r
+ \r
+// System.err.println("NamespaceIndex " + id);\r
+ \r
+ if("http://".equals(id) || "http:/".equals(id)) {\r
+ index(graph, processor, processor.getRootLibrary(), procedure);\r
+ } else {\r
+ final String[] parts = URIStringUtils.splitURI(id);\r
+ if(parts != null) {\r
+ NamespaceIndex.queryEach(graph, parts[0], processor, this, null, new InternalProcedure<TObjectIntHashMap<String>>() {\r
+ \r
+ @Override\r
+ public void execute(ReadGraphImpl graph, TObjectIntHashMap<String> index) {\r
+ \r
+ if(index != null) {\r
+ index(graph, processor, index.get(parts[1]), procedure);\r
+ } else {\r
+ add2(graph, null);\r
+ procedure.execute(graph, null);\r
+// System.err.println("NamespaceIndex[" + id + "]->null");\r
+ }\r
+ \r
+ }\r
+ \r
+ @Override\r
+ public void exception(ReadGraphImpl graph, Throwable t) {\r
+ if(DebugException.DEBUG) new DebugException(t).printStackTrace();\r
+ except(t);\r
+ procedure.exception(graph, t);\r
+ }\r
+ \r
+ });\r
+ } else {\r
+ add2(graph, null);\r
+ procedure.execute(graph, null);\r
+// System.err.println("NamespaceIndex[" + id + "]->null");\r
+ }\r
+\r
+ }\r
+ \r
+ }\r
+\r
+ @Override\r
+ public String toString() {\r
+ return "NamespaceIndex[" + id + "]";\r
+ }\r
+\r
+ synchronized private void add(TObjectIntHashMap<String> result) {\r
+ \r
+ throw new Error("Not possible!");\r
+ \r
+ }\r
+\r
+ private void add2(ReadGraphImpl graph, TObjectIntHashMap<String> result) {\r
+ \r
+ if(!isPending()) {\r
+ new Exception(""+hashCode()).printStackTrace();\r
+ }\r
+ \r
+ assert(isPending());\r
+\r
+// ArrayList<InternalProcedure<TObjectIntHashMap<String>>> p = null;\r
+\r
+ synchronized(this) {\r
+\r
+ setResult(result);\r
+ setReady();\r
+// p = procs;\r
+// procs = null; \r
+ \r
+ }\r
+ \r
+// if(p != null) {\r
+// \r
+// for(InternalProcedure<TObjectIntHashMap<String>> proc : p) proc.execute(graph, result);\r
+// \r
+// }\r
+ \r
+ }\r
+ \r
+ @Override\r
+ public void performFromCache(ReadGraphImpl graph, QueryProcessor provider, InternalProcedure<TObjectIntHashMap<String>> procedure) {\r
+ \r
+ assert(isReady());\r
+ \r
+ if(handleException(graph, procedure)) return;\r
+ \r
+ procedure.execute(graph, (TObjectIntHashMap<String>)getResult());\r
+ \r
+ }\r
+ \r
+ @Override\r
+ public synchronized void recompute(ReadGraphImpl graph, QueryProcessor provider) {\r
+ \r
+ final Semaphore s = new Semaphore(0);\r
+ \r
+ computeForEach(graph, provider, new InternalProcedure<TObjectIntHashMap<String>>() {\r
+\r
+ @Override\r
+ public void execute(ReadGraphImpl graph, TObjectIntHashMap<String> result) {\r
+ s.release();\r
+ }\r
+ \r
+ @Override\r
+ public void exception(ReadGraphImpl graph, Throwable t) {\r
+ if(DebugException.DEBUG) new DebugException(t).printStackTrace();\r
+ throw new Error("Error in recompute.", t);\r
+ }\r
+\r
+ });\r
+ \r
+ while(!s.tryAcquire()) {\r
+ provider.resume(graph);\r
+ }\r
+ \r
+ }\r
+ \r
+}\r
+\r