]> gerrit.simantics Code Review - simantics/platform.git/blob - bundles/org.simantics.db.impl/src/org/simantics/db/impl/procedure/ResultCallWrappedSyncQueryProcedure.java
Add possibility to bookkeep barrier tracing with env var
[simantics/platform.git] / bundles / org.simantics.db.impl / src / org / simantics / db / impl / procedure / ResultCallWrappedSyncQueryProcedure.java
1 /*******************************************************************************
2  * Copyright (c) 2018 Association for Decentralized Information Management
3  * in Industry THTH ry.
4  * All rights reserved. This program and the accompanying materials
5  * are made available under the terms of the Eclipse Public License v1.0
6  * which accompanies this distribution, and is available at
7  * http://www.eclipse.org/legal/epl-v10.html
8  *
9  * Contributors:
10  *     Semantum Oy - initial API and implementation
11  *******************************************************************************/
12 package org.simantics.db.impl.procedure;
13
14 import java.util.ArrayList;
15 import java.util.concurrent.atomic.AtomicBoolean;
16
17 import org.simantics.db.ReadGraph;
18 import org.simantics.db.common.utils.Logger;
19 import org.simantics.db.procedure.SyncMultiProcedure;
20
21 public class ResultCallWrappedSyncQueryProcedure<Result> implements SyncMultiProcedure<Result> {
22
23     private final ArrayList<Result> result;
24     private Throwable exception = null;
25     private final SyncMultiProcedure<Result> procedure;
26     private final AtomicBoolean latch;
27
28     public ResultCallWrappedSyncQueryProcedure(SyncMultiProcedure<Result> procedure) {
29         this.procedure = procedure;
30         latch = new AtomicBoolean(false);
31         result = new ArrayList<Result>();
32     }
33
34     @Override
35     public void execute(ReadGraph graph, Result result) {
36         try {
37             synchronized(this.result) {
38                 this.result.add(result);
39             }
40             procedure.execute(graph, result);
41         } catch (Throwable t) {
42             Logger.defaultLogError("AsyncMultiProcedure.execute failed for " + procedure, t);
43         }
44     }
45
46     @Override
47     public void finished(ReadGraph graph) {
48         if(latch.compareAndSet(false, true)) {
49             try {
50                 procedure.finished(graph);
51             } catch (Throwable t) {
52                 Logger.defaultLogError("AsyncMultiProcedure.exception failed for " + procedure, t);
53             } finally {
54             }
55         } else {
56             Logger.defaultLogError("Finished or exception was called many times (this time is finished)");
57         }
58     }
59
60     @Override
61     public void exception(ReadGraph graph, Throwable t) {
62         if(latch.compareAndSet(false, true)) {
63             try {
64                 this.exception = t;
65                 procedure.exception(graph, t);
66             } catch (Throwable throwable) {
67                 Logger.defaultLogError("AsyncMultiProcedure.exception failed for " + procedure, throwable);
68             } finally {
69             }
70         } else {
71             Logger.defaultLogError("Finished or exception was called many times (this time is exception)");
72         }
73     }
74
75     public ArrayList<Result> get() {
76         return result;
77     }
78
79     public Throwable getException() {
80         return exception;
81     }
82
83     @Override
84     public String toString() {
85         return "ResultCallWrappedQueryProcedure4[" + procedure + "]"; 
86     }
87
88 }