Return wait_time and use it when yielding

Shopify · Oct 16, 2024 · 2c47a84 · 2c47a84
1 parent 1ce5f4e
commit 2c47a84
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/ext/semian/resource.c b/ext/semian/resource.c
@@ -68,7 +68,7 @@ semian_resource_acquire_semaphore(int argc, VALUE *argv, VALUE self)
     wait_time = LONG2NUM(res.wait_time);
   }
 
-  return Qnil;
+  return wait_time;
 }
 
 VALUE
@@ -78,7 +78,7 @@ semian_resource_acquire(int argc, VALUE *argv, VALUE self)
     rb_raise(rb_eArgError, "acquire requires a block");
   }
 
-  semian_resource_acquire_semaphore(argc, argv, self);
+  VALUE wait_time = semian_resource_acquire_semaphore(argc, argv, self);
 
   return rb_ensure(rb_yield, wait_time, semian_resource_release_semaphore, self);
 }

diff --git a/ext/semian/resource.h b/ext/semian/resource.h
@@ -41,7 +41,7 @@ semian_resource_acquire(int argc, VALUE *argv, VALUE self);
 
 /*
  * call-seq:
- *    resource.acquire_semaphore(timeout: default_timeout) -> nil
+ *    resource.acquire_semaphore(timeout: default_timeout) -> wait_time
  *
  * Acquires a resource. The call will block for <code>timeout</code> seconds if a ticket
  * is not available. If no ticket is available within the timeout period, Semian::TimeoutError