fix statistic

aiokitchen · May 24, 2023 · 1d66113 · 1d66113
1 parent 65a0c2d
commit 1d66113
Showing 1 changed file with 6 additions and 3 deletions.
diff --git a/aiomisc/process_pool.py b/aiomisc/process_pool.py
@@ -35,14 +35,17 @@ def __init__(self, max_workers: int = DEFAULT_MAX_WORKERS, **kwargs: Any):
         self._statistic.processes = max_workers
 
     def _statistic_callback(
-        self, future: Future, execution_time: float,
+        self,
+        future: Future,
+        start_time: float,
+        loop: asyncio.AbstractEventLoop,
     ) -> None:
         if future.exception():
             self._statistic.error += 1
         else:
             self._statistic.success += 1
         self._statistic.done += 1
-        self._statistic.sum_time += execution_time
+        self._statistic.sum_time += loop.time() - start_time
 
     def submit(self, *args: Any, **kwargs: Any) -> Future:
         """
@@ -53,7 +56,7 @@ def submit(self, *args: Any, **kwargs: Any) -> Future:
         future = super().submit(*args, **kwargs)
         self._statistic.submitted += 1
         future.add_done_callback(
-            lambda f: self._statistic_callback(f, loop.time() - start_time),
+            lambda f: self._statistic_callback(f, start_time, loop),
         )
         return future