Make CancellationToken.wait only wait for a limited time (#300)

`Event` didn't have this problem, since it assumed any notify implied
that the event had been set (frankly a slightly insane assumption, but
whatever).

CHANGELOG.md

@@ -12,6 +12,12 @@ Changes are grouped as follows
 - `Fixed` for any bug fixes.
 - `Security` in case of vulnerabilities.
 
+## 7.0.1
+
+### Fixed
+
+ * Ensure that `CancellationToken.wait(timeout)` only waits for at most `timeout`, even if it is notified in that time.
+
 ## 7.0.0
 
 ### Changed

cognite/extractorutils/threading.py

@@ -1,6 +1,7 @@
 import logging
 import signal
 from threading import Condition
+from time import time
 from typing import Any, Optional
 
 
@@ -59,11 +60,21 @@ def set(self) -> None:
         self.cancel()
 
     def wait(self, timeout: Optional[float] = None) -> bool:
+        endtime = None
+        if timeout is not None:
+            endtime = time() + timeout
+
         while not self.is_cancelled:
             with self._cv:
-                did_not_time_out = self._cv.wait(timeout)
-                if not did_not_time_out:
-                    return False
+                if endtime is not None:
+                    remaining_time = endtime - time()
+                    if remaining_time <= 0.0:
+                        return True
+                    timed_out = not self._cv.wait(remaining_time)
+                    if timed_out:
+                        return False
+                else:
+                    self._cv.wait()
         return True
 
     def create_child_token(self) -> "CancellationToken":

pyproject.toml

@@ -1,6 +1,6 @@
 [tool.poetry]
 name = "cognite-extractor-utils"
-version = "7.0.0"
+version = "7.0.1"
 description = "Utilities for easier development of extractors for CDF"
 authors = ["Mathias Lohne <[email protected]>"]
 license = "Apache-2.0"