value async fail safe

nitely · Nov 17, 2024 · c035fdb · c035fdb
1 parent 7107a8c
commit c035fdb
Showing 1 changed file with 5 additions and 4 deletions.
diff --git a/src/hyperx/value.nim b/src/hyperx/value.nim
@@ -34,21 +34,22 @@ proc wakeupSoon(f: Future[void]) {.raises: [].} =
     callSoon wakeup
 
 proc put*[T](vala: ValueAsync[T], val: T) {.async.} =
-  if vala.isClosed:
-    raise newValueAsyncClosedError()
+  check not vala.isClosed, newValueAsyncClosedError()
   while vala.val != nil:
+    check not vala.isClosed, newValueAsyncClosedError()
     vala.putWaiter.clean()
     await vala.putWaiter.fut
   vala.val = val
   wakeupSoon vala.getWaiter.fut
   while vala.val != nil:
+    check not vala.isClosed, newValueAsyncClosedError()
     vala.putWaiter.clean()
     await vala.putWaiter.fut
 
 proc get*[T](vala: ValueAsync[T]): Future[T] {.async.} =
-  if vala.isClosed:
-    raise newValueAsyncClosedError()
+  check not vala.isClosed, newValueAsyncClosedError()
   while vala.val == nil:
+    check not vala.isClosed, newValueAsyncClosedError()
     vala.getWaiter.clean()
     await vala.getWaiter.fut
   result = vala.val