perf: reduce deadline creation

x42005e1f · Nov 8, 2024 · a8b49e1 · a8b49e1
1 parent 276e92d
commit a8b49e1
Show file tree

Hide file tree

Showing 2 changed files with 12 additions and 12 deletions.
diff --git a/src/aiologic/locks/condition.py b/src/aiologic/locks/condition.py
@@ -187,9 +187,6 @@ def wait_for(self, /, predicate, timeout=None):
     def notify(self, /, count=1, *, deadline=None):
         waiters = self.__waiters
 
-        if deadline is None:
-            deadline = self.__timer()
-
         notified = 0
 
         while waiters and notified != count:
@@ -200,6 +197,9 @@ def notify(self, /, count=1, *, deadline=None):
             else:
                 event, time = token
 
+                if deadline is None:
+                    deadline = self.__timer()
+
                 if time <= deadline:
                     if event.set():
                         notified += 1
@@ -216,9 +216,6 @@ def notify(self, /, count=1, *, deadline=None):
     def notify_all(self, /, *, deadline=None):
         waiters = self.__waiters
 
-        if deadline is None:
-            deadline = self.__timer()
-
         notified = 0
 
         while waiters:
@@ -229,6 +226,9 @@ def notify_all(self, /, *, deadline=None):
             else:
                 event, time = token
 
+                if deadline is None:
+                    deadline = self.__timer()
+
                 if time <= deadline:
                     if event.set():
                         notified += 1

diff --git a/src/aiologic/locks/event.py b/src/aiologic/locks/event.py
@@ -279,9 +279,6 @@ def __wakeup(self, /, deadline=None):
         waiters = self.__waiters
         is_unset = self.__is_unset
 
-        if deadline is None:
-            deadline = self.__timer()
-
         while waiters:
             try:
                 token = waiters[0]
@@ -290,6 +287,9 @@ def __wakeup(self, /, deadline=None):
             else:
                 event, marker, time, _ = token
 
+                if deadline is None:
+                    deadline = self.__timer()
+
                 if time <= deadline and marker is not is_unset.get(None):
                     token[3] = deadline
 
@@ -468,9 +468,6 @@ def __get(self, /):
     def __wakeup(self, /, deadline=None):
         waiters = self.__waiters
 
-        if deadline is None:
-            deadline = self.__timer()
-
         while waiters:
             try:
                 token = waiters[0]
@@ -479,6 +476,9 @@ def __wakeup(self, /, deadline=None):
             else:
                 event, marker, time, _ = token
 
+                if deadline is None:
+                    deadline = self.__timer()
+
                 if time <= deadline and marker is not self.__get():
                     token[3] = deadline