uringmachine 0.12.1 → 0.13

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 540e2e4df0b1953f58e36ab4dea2024a793b1bea4fd2358f64733f003b192510
-  data.tar.gz: 463cf782db4604358e1f54270e24e4bb97ee92176cafdb36079fd9d4a5132fda
+  metadata.gz: bce8525371c9f0aa2b279b815d04456573319b3eddaae14b79ddbf20ece1189f
+  data.tar.gz: dc1e70ca4101d607f2c04df639dabc917c379e3765a12c5440682a72cbb24316
 SHA512:
-  metadata.gz: 9e54c0f81a6a6a0523a5210a00e5bd7d3a7c0598d8e15c4434080429ae10328079a06d1075fab88894eb2c6ac827e8c929b4fbc18a2bdc7cf20d7efda0d1bd81
-  data.tar.gz: 9de7ecd460344ec496fc061edb15a9fadd8b9a5b81db0ca8844afd6cc81bb79b7881af54a008b3c43565725f9234ea66728e69c7eaa6a09d090a0977128e0326
+  metadata.gz: 258605dfe54482f5330a7795e18efa3397692b0a306cf7fd1ad6a2408fe8c7e098abca78d66baaddd78faac74aae6beeda9fe6a3d2e39ab4e196b585810bc212
+  data.tar.gz: 6110908bf2c3d8a404f015429835dfb585ae711f14457b785d7e298f6caf2449341ca62ab3fa560054f51c9de6031195e8ead20afabace47cfe24cdf9fd6ae48

data/CHANGELOG.md

@@ -1,3 +1,7 @@
+# 2025-06-07 Version 0.13
+
+- Add `#write_async`
+
 # 2025-06-07 Version 0.12.1
 
 - Improve portability of `UM` constants

data/ext/um/um.c

@@ -72,6 +72,11 @@ static inline void um_process_cqe(struct um *machine, struct io_uring_cqe *cqe)
   //   op, op->kind, op->flags, cqe->res, cqe->flags, machine->pending_count
   // );
 
+  if (op->flags & OP_F_FREE_ON_COMPLETE) {
+    um_op_free(machine, op);
+    return;
+  }
+
   if (unlikely((cqe->res == -ECANCELED) && (op->flags & OP_F_IGNORE_CANCELED))) return;
 
   op->flags |= OP_F_COMPLETED;
@@ -390,6 +395,19 @@ VALUE um_write(struct um *machine, int fd, VALUE str, int len) {
   return raise_if_exception(ret);
 }
 
+VALUE um_write_async(struct um *machine, int fd, VALUE str) {
+  struct um_op *op = um_op_alloc(machine);
+  memset(op, 0, sizeof(struct um_op));
+  op->kind = OP_WRITE_ASYNC;
+  op->flags = OP_F_FREE_ON_COMPLETE;
+  op->fiber = Qnil;
+  RB_OBJ_WRITE(machine->self, &op->value, str);
+
+  struct io_uring_sqe *sqe = um_get_sqe(machine, op);
+  io_uring_prep_write(sqe, fd, RSTRING_PTR(str), RSTRING_LEN(str), -1);
+  return str;
+}
+
 VALUE um_close(struct um *machine, int fd) {
   struct um_op op;
   um_prep_op(machine, &op, OP_CLOSE);

data/ext/um/um.h

@@ -28,6 +28,7 @@ enum op_kind {
   OP_OPEN,
   OP_READ,
   OP_WRITE,
+  OP_WRITE_ASYNC,
   OP_CLOSE,
   OP_STATX,
 
@@ -59,6 +60,7 @@ enum op_kind {
 #define OP_F_ASYNC            (1U << 2) // op belongs to an AsyncOp
 #define OP_F_IGNORE_CANCELED  (1U << 3) // CQE with -ECANCEL should be ignored
 #define OP_F_MULTISHOT        (1U << 4) // op is multishot
+#define OP_F_FREE_ON_COMPLETE (1U << 5) // op should be freed on receiving CQE
 
 struct um_op_result {
   __s32 res;
@@ -230,6 +232,7 @@ VALUE um_close(struct um *machine, int fd);
 VALUE um_open(struct um *machine, VALUE pathname, int flags, int mode);
 VALUE um_waitpid(struct um *machine, int pid, int options);
 VALUE um_statx(struct um *machine, int dirfd, VALUE path, int flags, unsigned int mask);
+VALUE um_write_async(struct um *machine, int fd, VALUE str);
 
 VALUE um_accept(struct um *machine, int fd);
 VALUE um_accept_each(struct um *machine, int fd);

data/ext/um/um_class.c

@@ -131,6 +131,11 @@ VALUE UM_write(int argc, VALUE *argv, VALUE self) {
   return um_write(machine, NUM2INT(fd), buffer, bytes);
 }
 
+VALUE UM_write_async(VALUE self, VALUE fd, VALUE str) {
+  struct um *machine = um_get_machine(self);
+  return um_write_async(machine, NUM2INT(fd), str);
+}
+
 VALUE UM_statx(VALUE self, VALUE dirfd, VALUE path, VALUE flags, VALUE mask) {
   struct um *machine = um_get_machine(self);
   return um_statx(machine, NUM2INT(dirfd), path, NUM2INT(flags), NUM2UINT(mask));
@@ -348,6 +353,7 @@ void Init_UM(void) {
   rb_define_method(cUM, "sleep", UM_sleep, 1);
   rb_define_method(cUM, "periodically", UM_periodically, 1);
   rb_define_method(cUM, "write", UM_write, -1);
+  rb_define_method(cUM, "write_async", UM_write_async, 2);
   rb_define_method(cUM, "statx", UM_statx, 4);
 
   rb_define_method(cUM, "waitpid", UM_waitpid, 2);

data/lib/uringmachine/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 class UringMachine
-  VERSION = '0.12.1'
+  VERSION = '0.13'
 end

data/test/test_um.rb

@@ -532,6 +532,30 @@ class WriteTest < UMBaseTest
   end
 end
 
+class WriteAsyncTest < UMBaseTest
+  def test_write_async
+    r, w = IO.pipe
+
+    assert_equal 0, machine.pending_count
+    machine.write_async(w.fileno, 'foo')
+    assert_equal 1, machine.pending_count
+
+    machine.snooze
+    assert_equal 0, machine.pending_count
+    assert_equal 'foo', r.readpartial(3)
+  end
+
+  def test_write_async_bad_fd
+    r, _w = IO.pipe
+
+    assert_equal 0, machine.pending_count
+    machine.write_async(r.fileno, 'foo')
+    assert_equal 1, machine.pending_count
+    machine.snooze
+    assert_equal 0, machine.pending_count
+  end
+end
+
 class CloseTest < UMBaseTest
   def test_close
     r, w = IO.pipe

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: uringmachine
 version: !ruby/object:Gem::Version
-  version: 0.12.1
+  version: '0.13'
 platform: ruby
 authors:
 - Sharon Rosner