uringmachine 0.30.0 → 0.31.0

data/ext/um/um_stream_class.c

@@ -1,317 +0,0 @@
-#include "um.h"
-
-VALUE cStream;
-VALUE eStreamRESPError;
-
-VALUE SYM_bp_read;
-VALUE SYM_bp_recv;
-VALUE SYM_ssl;
-
-inline int stream_has_target_obj_p(struct um_stream *stream) {
-  switch (stream->mode) {
-    case STREAM_SSL:
-    case STREAM_STRING:
-    case STREAM_IO_BUFFER:
-      return true;
-    default:
-      return false;
-  }
-}
-
-inline void stream_mark_segments(struct um_stream *stream) {
-  struct um_segment *curr = stream->head;
-  while (curr) {
-    // rb_gc_mark_movable(curr->obj);
-    curr = curr->next;
-  }
-}
-
-inline void stream_compact_segments(struct um_stream *stream) {
-  struct um_segment *curr = stream->head;
-  while (curr) {
-    // curr->obj = rb_gc_location(curr->obj);
-    curr = curr->next;
-  }
-}
-
-static void Stream_mark(void *ptr) {
-  struct um_stream *stream = ptr;
-  rb_gc_mark_movable(stream->self);
-  rb_gc_mark_movable(stream->machine->self);
-
-  if (stream_has_target_obj_p(stream)) {
-    rb_gc_mark_movable(stream->target);
-    stream_mark_segments(stream);
-  }
-}
-
-static void Stream_compact(void *ptr) {
-  struct um_stream *stream = ptr;
-  stream->self = rb_gc_location(stream->self);
-
-  if (stream_has_target_obj_p(stream)) {
-    stream->target = rb_gc_location(stream->target);
-    stream_compact_segments(stream);
-  }
-}
-
-static void Stream_free(void *ptr) {
-  struct um_stream *stream = ptr;
-  stream_clear(stream);
-}
-
-static const rb_data_type_t Stream_type = {
-  .wrap_struct_name = "UringMachine::Stream",
-  .function = {
-    .dmark = Stream_mark,
-    .dfree = Stream_free,
-    .dsize = NULL,
-    .dcompact = Stream_compact
-  },
-  .flags = RUBY_TYPED_FREE_IMMEDIATELY | RUBY_TYPED_WB_PROTECTED | RUBY_TYPED_EMBEDDABLE
-};
-
-static VALUE Stream_allocate(VALUE klass) {
-  struct um_stream *stream;
-  VALUE self = TypedData_Make_Struct(klass, struct um_stream, &Stream_type, stream);
-  return self;
-}
-
-static inline struct um_stream *um_get_stream(VALUE self) {
-  struct um_stream *stream;
-  TypedData_Get_Struct(self, struct um_stream, &Stream_type, stream);
-  return stream;
-}
-
-static inline void stream_setup(struct um_stream *stream, VALUE target, VALUE mode) {
-  stream->working_buffer = NULL;
-  if (mode == SYM_bp_read || mode == Qnil) {
-    stream->mode = STREAM_BP_READ;
-    stream->fd = NUM2INT(target);
-  }
-  else if (mode == SYM_bp_recv) {
-    stream->mode = STREAM_BP_RECV;
-    stream->fd = NUM2INT(target);
-  }
-  else if (mode == SYM_ssl) {
-    stream->mode = STREAM_SSL;
-    stream->target = target;
-    um_ssl_set_bio(stream->machine, target);
-  }
-  else
-    rb_raise(eUMError, "Invalid stream mode");
-}
-
-/* call-seq:
- *   UM::Stream.new(machine, fd, mode = nil) -> stream
- *   machine.stream(fd, mode = nil) -> stream
- *   machine.stream(fd, mode = nil) { |stream| ... }
- *
- * Initializes a new stream with the given UringMachine instance, target and
- * optional mode. The target maybe a file descriptor, or an instance of
- * OpenSSL::SSL::SSLSocket. In case of an SSL socket, the mode should be :ssl.
- *
- * @param machine [UringMachine] UringMachine instance
- * @param target [integer, OpenSSL::SSL::SSLSocket] stream target: file descriptor or SSL socket
- * @param mode [Symbol] optional stream mode: :bp_read, :bp_recv, :ssl
- * @return [void]
- */
-VALUE Stream_initialize(int argc, VALUE *argv, VALUE self) {
-  VALUE machine;
-  VALUE target;
-  VALUE mode;
-  rb_scan_args(argc, argv, "21", &machine, &target, &mode);
-
-  struct um_stream *stream = um_get_stream(self);
-  memset(stream, 0, sizeof(struct um_stream));
-
-  RB_OBJ_WRITE(self, &stream->self, self);
-  stream->machine = um_get_machine(machine);
-  stream_setup(stream, target, mode);
-
-  return self;
-}
-
-/* call-seq:
- *   stream.mode -> mode
- *
- * Returns the stream mode.
- *
- * @return [Symbol] stream mode
- */
-VALUE Stream_mode(VALUE self) {
-  struct um_stream *stream = um_get_stream(self);
-  switch (stream->mode) {
-    case STREAM_BP_READ:  return SYM_bp_read;
-    case STREAM_BP_RECV:  return SYM_bp_recv;
-    case STREAM_SSL:      return SYM_ssl;
-    default:              return Qnil;
-  }
-  return Qnil;
-}
-
-/* call-seq:
- *   stream.get_line(limit) -> str
- *
- * Reads from the string until a newline character is encountered. Returns the
- * line without the newline delimiter. If limit is 0, the line length is not
- * limited. If no newline delimiter is found before EOF, returns nil.
- *
- * @param limit [integer] maximum line length (0 means no limit)
- * @return [String, nil] read data or nil
- */
-VALUE Stream_get_line(VALUE self, VALUE limit) {
-  struct um_stream *stream = um_get_stream(self);
-  return stream_get_line(stream, Qnil, NUM2ULONG(limit));
-}
-
-/* call-seq:
- *   stream.get_string(len) -> str
- *
- * Reads len bytes from the stream. If len is 0, reads all available bytes. If
- * len is negative, reads up to -len available bytes. If len is positive and eof
- * is encountered before len bytes are read, returns nil.
- *
- * @param len [integer] number of bytes to read
- * @return [String, nil] read data or nil
- */
-VALUE Stream_get_string(VALUE self, VALUE len) {
-  struct um_stream *stream = um_get_stream(self);
-  return stream_get_string(stream, Qnil, NUM2LONG(len), 0, false);
-}
-
-/* call-seq:
- *   stream.skip(len) -> len
- *
- * Skips len bytes in the stream.
- *
- * @param len [integer] number of bytes to skip
- * @return [Integer] len
- */
-VALUE Stream_skip(VALUE self, VALUE len) {
-  struct um_stream *stream = um_get_stream(self);
-  stream_skip(stream, NUM2LONG(len), true);
-  return len;
-}
-
-/* call-seq:
- *   stream.each { |data| } -> stream
- *
- * Reads from the target, passing each chunk to the given block.
- *
- * @return [UringMachine::Stream] stream
- */
-VALUE Stream_each(VALUE self) {
-  struct um_stream *stream = um_get_stream(self);
-  stream_each(stream);
-  return self;
-}
-
-/* call-seq:
- *   stream.resp_decode -> obj
- *
- * Decodes an object from a RESP (Redis protocol) message.
- *
- * @return [any] decoded object
- */
-VALUE Stream_resp_decode(VALUE self) {
-  struct um_stream *stream = um_get_stream(self);
-  VALUE out_buffer = rb_utf8_str_new_literal("");
-  VALUE obj = resp_decode(stream, out_buffer);
-  RB_GC_GUARD(out_buffer);
-  return obj;
-}
-
-/* call-seq:
- *   stream.resp_encode(obj) -> string
- *
- * Encodes an object into a RESP (Redis protocol) message.
- *
- * @param str [String] string buffer
- * @param obj [any] object to be encoded
- * @return [String] str
- */
-VALUE Stream_resp_encode(VALUE self, VALUE str, VALUE obj) {
-  struct um_write_buffer buf;
-  write_buffer_init(&buf, str);
-  rb_str_modify(str);
-  resp_encode(&buf, obj);
-  write_buffer_update_len(&buf);
-  return str;
-}
-
-/* call-seq:
- *   stream.eof? -> bool
- *
- * Returns true if stream has reached EOF.
- *
- * @return [bool] EOF reached
- */
-VALUE Stream_eof_p(VALUE self) {
-  struct um_stream *stream = um_get_stream(self);
-  return stream->eof ? Qtrue : Qfalse;
-}
-
-/* call-seq:
- *   stream.consumed -> int
- *
- * Returns the total number of bytes consumed from the stream.
- *
- * @return [Integer] total bytes consumed
- */
-VALUE Stream_consumed(VALUE self) {
-  struct um_stream *stream = um_get_stream(self);
-  return LONG2NUM(stream->consumed_bytes);
-}
-
-/* call-seq:
- *   stream.pending -> int
- *
- * Returns the number of bytes available for reading.
- *
- * @return [Integer] bytes available
- */
-VALUE Stream_pending(VALUE self) {
-  struct um_stream *stream = um_get_stream(self);
-  return LONG2NUM(stream->pending_bytes);
-}
-
-/* call-seq:
- *   stream.clear -> stream
- *
- * Clears all available bytes and stops any ongoing read operation.
- *
- * @return [UM::Stream] self
- */
-VALUE Stream_clear(VALUE self) {
-  struct um_stream *stream = um_get_stream(self);
-  stream_clear(stream);
-  return self;
-}
-
-void Init_Stream(void) {
-  cStream = rb_define_class_under(cUM, "Stream", rb_cObject);
-  rb_define_alloc_func(cStream, Stream_allocate);
-
-  rb_define_method(cStream, "initialize", Stream_initialize, -1);
-  rb_define_method(cStream, "mode", Stream_mode, 0);
-
-  rb_define_method(cStream, "get_line", Stream_get_line, 1);
-  rb_define_method(cStream, "get_string", Stream_get_string, 1);
-  rb_define_method(cStream, "skip", Stream_skip, 1);
-  rb_define_method(cStream, "each", Stream_each, 0);
-
-  rb_define_method(cStream, "resp_decode", Stream_resp_decode, 0);
-  rb_define_singleton_method(cStream, "resp_encode", Stream_resp_encode, 2);
-
-  rb_define_method(cStream, "eof?", Stream_eof_p, 0);
-  rb_define_method(cStream, "consumed", Stream_consumed, 0);
-  rb_define_method(cStream, "pending", Stream_pending, 0);
-  rb_define_method(cStream, "clear", Stream_clear, 0);
-
-  eStreamRESPError = rb_define_class_under(cStream, "RESPError", rb_eStandardError);
-
-  SYM_bp_read = ID2SYM(rb_intern("bp_read"));
-  SYM_bp_recv = ID2SYM(rb_intern("bp_recv"));
-  SYM_ssl     = ID2SYM(rb_intern("ssl"));
-}