Rid TypicalPropertyTester of database read transactions

[simantics/platform.git] / bundles / org.simantics.modeling.ui / src / org / simantics / modeling / ui / diagramEditor / HasDiagramSourceListener.java

diff --git a/bundles/org.simantics.modeling.ui/src/org/simantics/modeling/ui/diagramEditor/HasDiagramSourceListener.java b/bundles/org.simantics.modeling.ui/src/org/simantics/modeling/ui/diagramEditor/HasDiagramSourceListener.java
new file mode 100644 (file)
index 0000000..c787367
--- /dev/null
+++ b/bundles/org.simantics.modeling.ui/src/org/simantics/modeling/ui/diagramEditor/HasDiagramSourceListener.java
@@ -0,0 +1,58 @@
+/*******************************************************************************
+ * Copyright (c) 2019 Association for Decentralized Information Management
+ * in Industry THTH ry.
+ * All rights reserved. This program and the accompanying materials
+ * are made available under the terms of the Eclipse Public License v1.0
+ * which accompanies this distribution, and is available at
+ * http://www.eclipse.org/legal/epl-v10.html
+ *
+ * Contributors:
+ *     Semantum Oy - initial API and implementation, gitlab #399
+ *******************************************************************************/
+package org.simantics.modeling.ui.diagramEditor;
+
+import org.simantics.db.Resource;
+import org.simantics.db.common.procedure.adapter.DisposableListener;
+import org.simantics.diagram.ui.DiagramModelHints;
+import org.simantics.g2d.diagram.IDiagram;
+import org.simantics.g2d.element.ElementUtils;
+import org.simantics.utils.DataContainer;
+import org.slf4j.Logger;
+import org.slf4j.LoggerFactory;
+
+/**
+ * @author Tuukka Lehtonen
+ * @since 1.41.0, 1.35.2
+ * @see gitlab #399
+ */
+class HasDiagramSourceListener extends DisposableListener<Resource> {
+
+    private static final Logger LOGGER = LoggerFactory.getLogger(HasDiagramSourceListener.class);
+
+    private DataContainer<IDiagram> diagramContainer;
+
+    public HasDiagramSourceListener(DataContainer<IDiagram> diagramContainer) {
+        this.diagramContainer = diagramContainer;
+    }
+
+    @Override
+    public void execute(Resource source) {
+        DataContainer<IDiagram> c = diagramContainer;
+        IDiagram d = c != null ? diagramContainer.get() : null;
+        if (d != null) {
+            ElementUtils.setOrRemoveHint(d, DiagramModelHints.KEY_HAS_DIAGRAM_SOURCE, source);
+        }
+    }
+
+    @Override
+    public void exception(Throwable t) {
+        LOGGER.error("Received unexpected exception", t);
+    }
+
+    @Override
+    public void dispose() {
+        super.dispose();
+        diagramContainer = null;
+    }
+
+}