import org.simantics.db.AsyncReadGraph;
import org.simantics.db.exception.DatabaseException;
-import org.simantics.db.exception.RuntimeDatabaseException;
import org.simantics.db.impl.BlockingAsyncProcedure;
import org.simantics.db.impl.DebugPolicy;
+import org.simantics.db.impl.graph.AsyncBarrierImpl;
+import org.simantics.db.impl.graph.BarrierTracing;
import org.simantics.db.impl.graph.ReadGraphImpl;
+import org.simantics.db.impl.query.QueryProcessor.SessionTask;
import org.simantics.db.procedure.AsyncProcedure;
import org.simantics.db.request.AsyncRead;
import org.slf4j.Logger;
try {
- BlockingAsyncProcedure<T> proc = new BlockingAsyncProcedure<>(graph, new AsyncProcedure<T>() {
+ BlockingAsyncProcedure<T> proc = new BlockingAsyncProcedure<>(graph.asyncBarrier, graph, new AsyncProcedure<T>() {
@Override
public void execute(AsyncReadGraph graph, T result) {
} else {
try {
- proc.execute(graph, (T) getResult());
+ T result = (T) getResult();
+ proc.execute(graph, result);
} catch (Throwable t) {
LOGGER.error("performFromCache proc.execute failed", t);
}
}
- public static <T> void computeForEach(ReadGraphImpl parentGraph, AsyncRead<T> request, AsyncReadEntry<T> entry,
- AsyncProcedure<T> procedure_) throws DatabaseException {
+ public static <T> T computeForEach(ReadGraphImpl graph, AsyncRead<T> request, AsyncReadEntry<T> entry,
+ AsyncProcedure<T> procedure_, boolean needsToBlock) throws DatabaseException {
AsyncProcedure<T> procedure = entry != null ? entry : procedure_;
- ReadGraphImpl queryGraph = parentGraph.withParent(entry);
+ ReadGraphImpl queryGraph = graph.withParent(entry);
+ queryGraph.asyncBarrier.inc();
- BlockingAsyncProcedure<T> proc = new BlockingAsyncProcedure<>(queryGraph, new AsyncProcedure<T>() {
-
- @Override
- public void execute(AsyncReadGraph returnGraph, T result) {
- try {
- procedure.execute(parentGraph, result);
- } catch (Throwable t) {
- LOGGER.error("computeForEach procedure.execute failed", t);
- }
+ BlockingAsyncProcedure<T> proc = new BlockingAsyncProcedure<>(queryGraph.asyncBarrier, graph, null, request);
+
+ class AsyncTask extends SessionTask {
+
+ int counter = 0;
+ T result;
+ DatabaseException exception;
+
+ public AsyncTask(ReadGraphImpl graph) {
+ super(graph);
}
@Override
- public void exception(AsyncReadGraph returnGraph, Throwable t) {
- try {
- procedure.exception(parentGraph, t);
- } catch (Throwable t2) {
- LOGGER.error("computeForEach procedure.exception failed", t2);
+ public void run0(int thread) {
+ if(needsToBlock) proc.waitBarrier();
+ if(proc.isDone()) {
+ ReadGraphImpl executeGraph = graph.withParent(graph.parent);
+ executeGraph.asyncBarrier.inc();
+ try {
+ result = (T)proc.get();
+ if(procedure != null) {
+ procedure.execute(executeGraph, result);
+ }
+ } catch (DatabaseException e) {
+ if(procedure != null) procedure.exception(executeGraph, e);
+ exception = e;
+ } catch (Throwable t) {
+ DatabaseException dbe = new DatabaseException(t);
+ if(procedure != null) procedure.exception(executeGraph, dbe);
+ exception = dbe;
+ } finally {
+ if (entry != null) {
+ // This does not throw
+ entry.performFromCache(executeGraph, procedure_);
+ }
+ executeGraph.asyncBarrier.dec();
+ executeGraph.asyncBarrier.waitBarrier(procedure, executeGraph);
+ }
+ } else {
+ if(counter++ > 10000) {
+ if(BarrierTracing.BOOKKEEPING) {
+ AsyncBarrierImpl.printReverse(queryGraph.asyncBarrier, 2);
+ AsyncBarrierImpl caller = queryGraph.asyncBarrier.caller;
+ while(caller != null) {
+ System.err.println("called by " + AsyncBarrierImpl.report(caller));
+ caller = caller.caller;
+ }
+ for(AsyncBarrierImpl ab : BarrierTracing.debuggerMap.keySet()) {
+ AsyncBarrierImpl.printReverse(ab, 2);
+ }
+ }
+ throw new IllegalStateException("Eternal loop in queries.");
+ }
+ graph.processor.schedule(new AsyncTask(graph));
}
}
+
+ }
- @Override
- public String toString() {
- return procedure.toString();
- }
+ try {
+ request.perform(queryGraph, proc);
+ } finally {
+ queryGraph.asyncBarrier.dec();
+ }
- }, request);
+ AsyncTask task = new AsyncTask(graph);
- request.perform(queryGraph, proc);
-
- proc.get();
+ if(needsToBlock) task.run(0);
+ else if (proc.isDone()) task.run(0);
+ else {
+ graph.processor.schedule(task);
+ return null;
+ }
- if (entry != null)
- entry.performFromCache(parentGraph, procedure_);
+ if(task.exception != null) throw task.exception;
+ else return task.result;
}