X-Git-Url: https://gerrit.simantics.org/r/gitweb?a=blobdiff_plain;f=bundles%2Forg.simantics.db.common%2Fsrc%2Forg%2Fsimantics%2Fdb%2Fcommon%2Fservice%2FServiceActivityMonitorImpl.java;fp=bundles%2Forg.simantics.db.common%2Fsrc%2Forg%2Fsimantics%2Fdb%2Fcommon%2Fservice%2FServiceActivityMonitorImpl.java;h=1f6f69e12317ade1da034776f4f39fceb42d12b5;hb=969bd23cab98a79ca9101af33334000879fb60c5;hp=0000000000000000000000000000000000000000;hpb=866dba5cd5a3929bbeae85991796acb212338a08;p=simantics%2Fplatform.git

diff --git a/bundles/org.simantics.db.common/src/org/simantics/db/common/service/ServiceActivityMonitorImpl.java b/bundles/org.simantics.db.common/src/org/simantics/db/common/service/ServiceActivityMonitorImpl.java
new file mode 100644
index 000000000..1f6f69e12
--- /dev/null
+++ b/bundles/org.simantics.db.common/src/org/simantics/db/common/service/ServiceActivityMonitorImpl.java
@@ -0,0 +1,64 @@
+package org.simantics.db.common.service;
+
+import java.util.concurrent.Semaphore;
+import java.util.concurrent.TimeUnit;
+import java.util.concurrent.atomic.AtomicInteger;
+
+import org.simantics.db.service.ServiceActivityMonitor;
+
+import gnu.trove.map.TObjectIntMap;
+import gnu.trove.map.hash.TObjectIntHashMap;
+
+public class ServiceActivityMonitorImpl implements ServiceActivityMonitor {
+
+    public static final long REPORTING_PERIOD = 5L;
+    
+    Semaphore sem = new Semaphore(1);
+    AtomicInteger ref = new AtomicInteger();
+    TObjectIntMap<Object> ids = new TObjectIntHashMap<>();
+
+    @Override
+    public void registerActivity(Object id) {
+        if (ref.getAndIncrement() == 0)
+            sem.acquireUninterruptibly(1);
+        synchronized (ids) {
+            ids.adjustOrPutValue(id, 1, 1);
+        }
+    }
+
+    @Override
+    public void unregisterActivity(Object id) {
+        synchronized (ids) {
+            int refs = ids.get(id);
+            if (refs == 1) {
+                ids.remove(id);
+            } else {
+                ids.put(id, refs-1);
+            }
+        }
+        if (ref.decrementAndGet() == 0)
+            sem.release();
+    }
+
+    @Override
+    public void waitForCompletion() throws InterruptedException {
+        while(true) {
+            if(waitForCompletion(REPORTING_PERIOD, TimeUnit.SECONDS))
+                return;
+            synchronized (ids) {
+                System.err.println("ServiceActivityMonitorImpl.waitForCompletion: " + ids);
+            }    
+        }
+    }
+
+    @Override
+    public boolean waitForCompletion(long timeout, TimeUnit unit) throws InterruptedException {
+        if(sem.tryAcquire(timeout, unit)) {
+            sem.release();
+            return true;
+        }
+        else
+            return false;
+    }
+
+}
\ No newline at end of file