--- /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.browsing.ui.graph.impl.contribution;\r
+\r
+import org.simantics.browsing.ui.BuiltinKeys;\r
+import org.simantics.browsing.ui.BuiltinKeys.CheckedStateKey;\r
+import org.simantics.browsing.ui.CheckedState;\r
+import org.simantics.browsing.ui.DataSource;\r
+import org.simantics.browsing.ui.NodeContext;\r
+import org.simantics.browsing.ui.PrimitiveQueryUpdater;\r
+import org.simantics.browsing.ui.graph.impl.request.ResourceQuery;\r
+import org.simantics.db.ReadGraph;\r
+import org.simantics.db.exception.DatabaseException;\r
+import org.simantics.db.procedure.Listener;\r
+import org.simantics.db.procedure.Procedure;\r
+import org.simantics.utils.ui.ErrorLogger;\r
+\r
+public abstract class FinalCheckedStateContributionImpl {\r
+\r
+ protected final PrimitiveQueryUpdater updater;\r
+ private final ResourceQuery<CheckedState> labelQuery;\r
+ private final Procedure<CheckedState> procedure;\r
+\r
+ final private NodeContext context;\r
+ final private BuiltinKeys.CheckedStateKey key;\r
+ \r
+ private CheckedState state = null;\r
+\r
+ public Object getIdentity(CheckedStateKey key) {\r
+ return key;\r
+ }\r
+\r
+ public CheckedState getState() {\r
+ \r
+ if (state == null) {\r
+\r
+ final DataSource<ReadGraph> source = updater.getDataSource(ReadGraph.class);\r
+ assert(source != null);\r
+\r
+ source.schedule(graph -> {\r
+ if(procedure instanceof Listener<?>)\r
+ graph.asyncRequest(labelQuery, (Listener<CheckedState>)procedure);\r
+ else\r
+ graph.asyncRequest(labelQuery, procedure);\r
+ });\r
+ \r
+ }\r
+ \r
+ return state;\r
+ \r
+ }\r
+ \r
+ public FinalCheckedStateContributionImpl(final PrimitiveQueryUpdater updater, final NodeContext context, final CheckedStateKey key) {\r
+\r
+ this.updater = updater;\r
+ this.context = context;\r
+ this.key = key;\r
+\r
+ labelQuery = new ResourceQuery<CheckedState>(getIdentity(key), context) {\r
+\r
+ @Override\r
+ public CheckedState perform(ReadGraph graph) throws DatabaseException {\r
+ try {\r
+ return getState(graph, context);\r
+ } catch (DatabaseException e) {\r
+ throw e;\r
+ } catch (Throwable t) {\r
+ ErrorLogger.defaultLogError("LazyGraphLabeler.labelQuery produced unexpected exception.", t);\r
+ return null;\r
+ }\r
+ }\r
+\r
+ @Override\r
+ public String toString() {\r
+ return FinalCheckedStateContributionImpl.this + " with context " + context;\r
+ }\r
+\r
+ };\r
+\r
+ procedure = createProcedure();\r
+\r
+ }\r
+\r
+ protected Procedure<CheckedState> createProcedure() {\r
+\r
+ return new Procedure<CheckedState>() {\r
+\r
+ @Override\r
+ public void execute(CheckedState result) {\r
+ replaceResult(result);\r
+ }\r
+\r
+ @Override\r
+ public void exception(Throwable t) {\r
+ ErrorLogger.defaultLogError("LazyContributionImpl.childQuery failed, see exception for details.", t);\r
+ }\r
+\r
+ };\r
+\r
+ }\r
+\r
+ protected void replaceResult(CheckedState result) {\r
+ state = result;\r
+ updater.scheduleReplace(context, key, state);\r
+ }\r
+\r
+ abstract public CheckedState getState(ReadGraph graph, NodeContext context) throws DatabaseException;\r
+ \r
+\r
+}\r