rage-iodine 4.2.2 → 4.3.0

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 110642492bc676b7626a1bea040b30677d4030a024ec0449e14bbca2cb7a415f
-  data.tar.gz: 7f69cc726d3c385c2a3cf1c2e56fab00b3d5dcc0c4750cc53cf6bc541284c770
+  metadata.gz: cbc2e4408070fc50bbcb07fe6953c8fba6d48a11d315990b97248c4e15e193a1
+  data.tar.gz: 72d3c1f392ce350e253f9f1958ab2744c4f41bf073e8ba60fad16e91bf00fae6
 SHA512:
-  metadata.gz: 47a1f8db4a5e06730c17a60c8da848fd9a187abed5ae1a9257cd56f99b0d79f08cea30146efb5f0a4d25e83f19cb2ddf75074d9d98020300efe4290921eebb50
-  data.tar.gz: 29fae621181d246bd9f986bfc97d611e1e8dffe53c3fed98e1db8104543f0903bb860493701fc4ff289d9f04eff908274e2a8ebcaf55bc42134937a3e96bf124
+  metadata.gz: f1f9365497d81511cd3482d969101b6d4e6ef59fd08b267d93dae3810e80e02274ccb7d2fa088e8a04aee77a44b482628b3931fc4bcde7fe9b8f92df8adbac8e
+  data.tar.gz: 5acf73720db22d797cec78c27f6c6e7c37cd60f93648f265325a9dbc250c58b8e492050b16a8403d5440e1890693d60a24f01c6f3369aeac2414f7ff33522a84

data/CHANGELOG.md

@@ -6,6 +6,10 @@ Please notice that this change log contains changes for upcoming releases as wel
 
 ## Changes:
 
+#### Change log v.4.3.0 (2025-08-19)
+
+**Update**: Add the `task_inc!`, `task_dec!`, `stopping?` methods.
+
 #### Change log v.4.2.2 (2025-08-10)
 
 **Fix**: Fix `new_websocket` prototype.

data/ext/iodine/fio.c

@@ -390,6 +390,8 @@ typedef struct {
   uint32_t connection_count;
   /* number of currently paused connections */
   uint32_t async_connection_count;
+  uint32_t background_task_count;
+  uint8_t stop_requested;
   /* thread list */
   fio_ls_s thread_ids;
   /* active workers */
@@ -1299,13 +1301,15 @@ static void fio_defer_on_fork(void) {
 }
 
 void fio_graceful_stop(void) {
-  static uint8_t stop_requested;
-
-  if (!fio_data->async_connection_count) {
+  if (!fio_data->async_connection_count && !fio_data->background_task_count) {
     fio_stop();
-  } else if (!stop_requested) {
-    stop_requested = 1;
-    FIO_LOG_INFO("(%d) Waiting for up to 15 seconds to allow active requests to finish...", (int)getpid());
+  } else if (!fio_data->stop_requested) {
+    fio_data->stop_requested = 1;
+    if (fio_data->async_connection_count) {
+      FIO_LOG_INFO("(%d) Waiting for up to 15 seconds to allow active requests to finish...", (int)getpid());
+    } else {
+      FIO_LOG_INFO("(%d) Waiting for up to 15 seconds to allow active background tasks to finish...", (int)getpid());
+    }
     fio_run_every(500, 30, (void (*)(void *))fio_graceful_stop, NULL, (void (*)(void *))fio_stop);
   }
 }
@@ -4316,6 +4320,18 @@ void fio_state_callback_clear_all(void) {
   }
 }
 
+void fio_register_background_task() {
+  fio_atomic_add(&fio_data->background_task_count, 1);
+}
+
+void fio_deregister_background_task() {
+  fio_atomic_sub(&fio_data->background_task_count, 1);
+}
+
+uint8_t fio_is_stop_requested() {
+  return fio_data->stop_requested;
+}
+
 /* *****************************************************************************
 IO bound tasks
 ***************************************************************************** */

data/ext/iodine/fio.h

@@ -1681,6 +1681,15 @@ void fio_state_callback_force(callback_type_e);
 /** Clears all the existing callbacks for the event. */
 void fio_state_callback_clear(callback_type_e);
 
+/** Increments the background tasks counter. */
+void fio_register_background_task();
+
+/** Decrements the background tasks counter. */
+void fio_deregister_background_task();
+
+/** Checks whether the server is about to shut down. */
+uint8_t fio_is_stop_requested();
+
 /* *****************************************************************************
 Lower Level API - for special circumstances, use with care.
 ***************************************************************************** */

data/ext/iodine/iodine_defer.c

@@ -207,6 +207,30 @@ static VALUE iodine_defer_run(VALUE self) {
   (void)self;
 }
 
+/* Increments the background tasks counter. */
+static VALUE iodine_register_background_task(VALUE self) {
+  fio_register_background_task();
+  return Qtrue;
+  (void)self;
+}
+
+/* Decrements the background tasks counter. */
+static VALUE iodine_deregister_background_task(VALUE self) {
+  fio_deregister_background_task();
+  return Qtrue;
+  (void)self;
+}
+
+/* Checks whether the server is about to shut down. */
+static VALUE iodine_is_stop_requested(VALUE self) {
+  if (fio_is_stop_requested()) {
+    return Qtrue;
+  } else {
+    return Qfalse;
+  }
+  (void)self;
+}
+
 /**
 Runs the required block after the specified number of milliseconds have passed.
 Time is counted only once Iodine started running (using {Iodine.start}).
@@ -400,6 +424,13 @@ void iodine_defer_initialize(void) {
                             -1);
   rb_define_module_function(IodineModule, "on_state", iodine_on_state, 1);
 
+  rb_define_module_function(IodineModule, "task_inc!", iodine_register_background_task,
+                            0);
+  rb_define_module_function(IodineModule, "task_dec!", iodine_deregister_background_task,
+                            0);
+  rb_define_module_function(IodineModule, "stopping?", iodine_is_stop_requested,
+                            0);
+
   STATE_PRE_START = rb_intern("pre_start");
   STATE_BEFORE_FORK = rb_intern("before_fork");
   STATE_AFTER_FORK = rb_intern("after_fork");

data/lib/iodine/version.rb

@@ -1,3 +1,3 @@
 module Iodine
-  VERSION = '4.2.2'.freeze
+  VERSION = '4.3.0'.freeze
 end

metadata

@@ -1,13 +1,13 @@
 --- !ruby/object:Gem::Specification
 name: rage-iodine
 version: !ruby/object:Gem::Version
-  version: 4.2.2
+  version: 4.3.0
 platform: ruby
 authors:
 - Boaz Segev
 bindir: exe
 cert_chain: []
-date: 2025-08-10 00:00:00.000000000 Z
+date: 2025-08-19 00:00:00.000000000 Z
 dependencies:
 - !ruby/object:Gem::Dependency
   name: rake