Fixed skipping invalidations

app/src/main/java/androidx/room/RoomTrackingLiveData.java

@@ -29,6 +29,7 @@ import java.util.Set;
 import java.util.concurrent.Callable;
 import java.util.concurrent.Executor;
 import java.util.concurrent.atomic.AtomicBoolean;
+import java.util.concurrent.atomic.AtomicInteger;
 
 /**
  * A LiveData implementation that closely works with {@link InvalidationTracker} to implement
@@ -59,11 +60,8 @@ class RoomTrackingLiveData<T> extends LiveData<T> {
     @SuppressWarnings("WeakerAccess")
     final InvalidationTracker.Observer mObserver;
 
-    @SuppressWarnings("WeakerAccess")
+    final AtomicInteger queued = new AtomicInteger(0);
-    final AtomicBoolean mInvalid = new AtomicBoolean(true);
+    final AtomicInteger running = new AtomicInteger(0);
-
-    @SuppressWarnings("WeakerAccess")
-    final AtomicBoolean mComputing = new AtomicBoolean(false);
 
     @SuppressWarnings("WeakerAccess")
     final AtomicBoolean mRegisteredObserver = new AtomicBoolean(false);
@@ -76,15 +74,12 @@ class RoomTrackingLiveData<T> extends LiveData<T> {
             if (mRegisteredObserver.compareAndSet(false, true)) {
                 mDatabase.getInvalidationTracker().addWeakObserver(mObserver);
             }
-            boolean computed;
-            do {
-                computed = false;
-                // compute can happen only in 1 thread but no reason to lock others.
-                if (mComputing.compareAndSet(false, true)) {
-                    // as long as it is invalid, keep computing.
             try {
+                running.incrementAndGet();
+
                 T value = null;
-                        while (mInvalid.compareAndSet(true, false)) {
+                boolean computed = false;
+                synchronized (mComputeFunction) {
                     int retry = 0;
                     while (!computed) {
                         try {
@@ -107,19 +102,10 @@ class RoomTrackingLiveData<T> extends LiveData<T> {
                     postValue(value);
                 }
             } finally {
-                        // release compute lock
+                queued.decrementAndGet();
-                        mComputing.set(false);
+                running.decrementAndGet();
             }
         }
-                // check invalid after releasing compute lock to avoid the following scenario.
-                // Thread A runs compute()
-                // Thread A checks invalid, it is false
-                // Main thread sets invalid to true
-                // Thread B runs, fails to acquire compute lock and skips
-                // Thread A releases compute lock
-                // We've left invalid in set state. The check below recovers.
-            } while (computed && mInvalid.get());
-        }
     };
 
     @SuppressWarnings("WeakerAccess")
@@ -127,14 +113,19 @@ class RoomTrackingLiveData<T> extends LiveData<T> {
         @MainThread
         @Override
         public void run() {
+            if (running.get() == 0 && queued.get() > 0) {
+                eu.faircode.email.Log.i(mComputeFunction +
+                        " running=" + running + " queued=" + queued);
+                return;
+            }
             boolean isActive = hasActiveObservers();
-            if (mInvalid.compareAndSet(false, true)) {
             if (isActive) {
+                queued.incrementAndGet();
                 getQueryExecutor().execute(mRefreshRunnable);
             }
         }
-        }
     };
+
     @SuppressLint("RestrictedApi")
     RoomTrackingLiveData(
             RoomDatabase database,