chore: better code style

redis · Aug 31, 2024 · 1bef7b3 · 1bef7b3
1 parent e9cfc31
commit 1bef7b3
Showing 1 changed file with 5 additions and 7 deletions.
diff --git a/src/main/java/io/lettuce/core/protocol/DefaultAutoBatchFlushEndpoint.java b/src/main/java/io/lettuce/core/protocol/DefaultAutoBatchFlushEndpoint.java
@@ -1266,25 +1266,23 @@ public void execute(Runnable task) {
                 cur = this.owner;
                 if (isOwnerCurrentThreadAndPreemptPrevented(cur)) {
                     // already prevented preemption, safe to skip expensive add/done calls
-                    executeInOwnerWithPreemptPrevention(task, false);
+                    task.run();
                     return;
                 }
             } while (!OWNER.compareAndSet(this, cur, cur.toAdd(1)));
 
             if (cur.isCurrentThread()) {
-                executeInOwnerWithPreemptPrevention(task, true);
+                executeInOwnerWithPreemptPrevention(task);
             } else {
-                cur.thread.execute(() -> executeInOwnerWithPreemptPrevention(task, true));
+                cur.thread.execute(() -> executeInOwnerWithPreemptPrevention(task));
             }
         }
 
-        private void executeInOwnerWithPreemptPrevention(Runnable task, boolean added) {
+        private void executeInOwnerWithPreemptPrevention(Runnable task) {
             try {
                 task.run();
             } finally {
-                if (added) {
-                    done(1);
-                }
+                done(1);
             }
         }