Class: Event::Backend::URing

Inherits:

Object

• Object
• Event::Backend::URing

Defined in:

ext/event/backend/uring.c

Instance Method Summary

• #initialize(loop) ⇒ Object constructor
• #io_read(fiber, io, buffer, offset, length) ⇒ Object
• #io_wait(fiber, io, events) ⇒ Object
• #io_write(fiber, io, buffer, offset, length) ⇒ Object
• #select(duration) ⇒ Object

Constructor Details

#initialize(loop) ⇒ Object

# File 'ext/event/backend/uring.c', line 83

VALUE Event_Backend_URing_initialize(VALUE self, VALUE loop) {
  struct Event_Backend_URing *data = NULL;
  TypedData_Get_Struct(self, struct Event_Backend_URing, &Event_Backend_URing_Type, data);
  
  data->loop = loop;
  data->ring = xmalloc(sizeof(struct io_uring));
  
  int result = io_uring_queue_init(URING_ENTRIES, data->ring, 0);
  
  if (result == -1) {
    rb_sys_fail("io_uring_queue_init");
  }
  
  return self;
}

Instance Method Details

#io_read(fiber, io, buffer, offset, length) ⇒ Object

# File 'ext/event/backend/uring.c', line 197

VALUE Event_Backend_URing_io_read(VALUE self, VALUE fiber, VALUE io, VALUE buffer, VALUE offset, VALUE length) {
  struct Event_Backend_URing *data = NULL;
  TypedData_Get_Struct(self, struct Event_Backend_URing, &Event_Backend_URing_Type, data);

  resize_to_capacity(buffer, NUM2SIZET(offset), NUM2SIZET(length));

  int descriptor = NUM2INT(rb_funcall(io, id_fileno, 0));
  struct io_uring_sqe *sqe = io_uring_get_sqe(data->ring);

  struct iovec iovecs[1];
  iovecs[0].iov_base = RSTRING_PTR(buffer) + NUM2SIZET(offset);
  iovecs[0].iov_len = NUM2SIZET(length);

  io_uring_prep_readv(sqe, descriptor, iovecs, 1, 0);
  io_uring_sqe_set_data(sqe, (void*)fiber);
  io_uring_submit(data->ring);

  // fprintf(stderr, "prep_readv(%p, %d, %ld)\n", sqe, descriptor, iovecs[0].iov_len);

  int result = NUM2INT(rb_funcall(data->loop, id_transfer, 0));

  if (result < 0) {
    rb_syserr_fail(-result, strerror(-result));
  }

  resize_to_fit(buffer, NUM2SIZET(offset), (size_t)result);

  return INT2NUM(result);
}

#io_wait(fiber, io, events) ⇒ Object

# File 'ext/event/backend/uring.c', line 124

VALUE Event_Backend_URing_io_wait(VALUE self, VALUE fiber, VALUE io, VALUE events) {
  struct Event_Backend_URing *data = NULL;
  TypedData_Get_Struct(self, struct Event_Backend_URing, &Event_Backend_URing_Type, data);
  
  int descriptor = NUM2INT(rb_funcall(io, id_fileno, 0));
  struct io_uring_sqe *sqe = io_uring_get_sqe(data->ring);
  
  short flags = poll_flags_from_events(NUM2INT(events));
  
  // fprintf(stderr, "poll_add(%p, %d, %d)\n", sqe, descriptor, flags);
  
  io_uring_prep_poll_add(sqe, descriptor, flags);
  io_uring_sqe_set_data(sqe, (void*)fiber);
  io_uring_submit(data->ring);
  
  VALUE result = rb_funcall(data->loop, id_transfer, 0);
  
  // We explicitly filter the resulting events based on the requested events.
  // In some cases, poll will report events we didn't ask for.
  flags &= NUM2INT(result);
  
  return INT2NUM(events_from_poll_flags(flags));
}

#io_write(fiber, io, buffer, offset, length) ⇒ Object

# File 'ext/event/backend/uring.c', line 227

VALUE Event_Backend_URing_io_write(VALUE self, VALUE fiber, VALUE io, VALUE buffer, VALUE offset, VALUE length) {
  struct Event_Backend_URing *data = NULL;
  TypedData_Get_Struct(self, struct Event_Backend_URing, &Event_Backend_URing_Type, data);

  if ((size_t)RSTRING_LEN(buffer) < NUM2SIZET(offset) + NUM2SIZET(length)) {
    rb_raise(rb_eRuntimeError, "invalid offset/length exceeds bounds of buffer");
  }

  int descriptor = NUM2INT(rb_funcall(io, id_fileno, 0));
  struct io_uring_sqe *sqe = io_uring_get_sqe(data->ring);

  struct iovec iovecs[1];
  iovecs[0].iov_base = RSTRING_PTR(buffer) + NUM2SIZET(offset);
  iovecs[0].iov_len = NUM2SIZET(length);

  io_uring_prep_writev(sqe, descriptor, iovecs, 1, 0);
  io_uring_sqe_set_data(sqe, (void*)fiber);
  io_uring_submit(data->ring);
  
  // fprintf(stderr, "prep_writev(%p, %d, %ld)\n", sqe, descriptor, iovecs[0].iov_len);
  
  int result = NUM2INT(rb_funcall(data->loop, id_transfer, 0));

  if (result < 0) {
    rb_syserr_fail(-result, strerror(-result));
  }

  return INT2NUM(result);
}

#select(duration) ⇒ Object

# File 'ext/event/backend/uring.c', line 257

VALUE Event_Backend_URing_select(VALUE self, VALUE duration) {
  struct Event_Backend_URing *data = NULL;
  TypedData_Get_Struct(self, struct Event_Backend_URing, &Event_Backend_URing_Type, data);
  
  struct io_uring_cqe *cqes[URING_MAX_EVENTS];
  struct __kernel_timespec storage;
  
  int result = io_uring_peek_batch_cqe(data->ring, cqes, URING_MAX_EVENTS);
  
  // fprintf(stderr, "result = %d\n", result);
  
  if (result < 0) {
    rb_syserr_fail(-result, strerror(-result));
  } else if (result == 0) {
    result = io_uring_wait_cqes(data->ring, cqes, 1, make_timeout(duration, &storage), NULL);
    
    // fprintf(stderr, "result (timeout) = %d\n", result);
    
    if (result == -ETIME) {
      result = 0;
    } else if (result < 0) {
      rb_syserr_fail(-result, strerror(-result));
    }
  }
  
  for (int i = 0; i < result; i += 1) {
    VALUE fiber = (VALUE)io_uring_cqe_get_data(cqes[i]);
    VALUE result = INT2NUM(cqes[i]->res);
    
    // fprintf(stderr, "cqes[i]->res = %d\n", cqes[i]->res);
    
    io_uring_cqe_seen(data->ring, cqes[i]);
    
    rb_funcall(fiber, id_transfer, 1, result);
  }
  
  return INT2NUM(result);
}