ratomic 0.2.0 → 0.2.1

checksums.yaml

@@ -1,7 +1,7 @@
 ---
 SHA256:
-  metadata.gz: 6fbef9f27ee3ebd3839fb95b3e3c4685aaa34be9c22f1314974726c5f38aada8
-  data.tar.gz: 668591405949669d19687a678029ca07f46f155b3740f9a7d5fff599a5d8003c
+  metadata.gz: b4ca3b68838c5f869284756b01f00e56960cbb5dff8e640efc03682d10d28a50
+  data.tar.gz: 959794b19074d1ecfba5bdb08223ba372a6443fbc96fe833d0ce324aa3c52d5e
 SHA512:
-  metadata.gz: e31111fffe0abc54d5e7c945c7bd618dc7b201a0b35c48c438dbe7cf3db414106c5638ac4680f67a018a55545f5f62d09e6efe79887577f9a8516b5d7d6569cc
-  data.tar.gz: b46b875f77412cfbb3edba3ed07fd2087fb70ded07023ad091d32f911d4fa5b9af06fa42a1db770bc24d357683828cc0ec3df5f429ae177c45bc3df4f7ebc9fb
+  metadata.gz: 9697f964c65bdc3a951da84836ebadf947d370d74efd258015fb5401267acd6d78248de238f9da6db60edecccbe0a8bbd83a327e424d6fb6e74066ee60fe16de
+  data.tar.gz: 8bbbba4064050e577b23b48e8363653b5438ea157a68cffc607a53588e2785fdc9c2d5879d846593fa9f2757667df1df77784811c239b4f5b5d9cc684f00ae74

data/CHANGELOG.md

@@ -1,5 +1,10 @@
 ## [Unreleased]
 
+## [0.2.1] - 2026-06-04
+
+- Fix `Ratomic::Queue` slot indexing for non-power-of-two capacities.
+- Fix `Ratomic::Map#fetch_and_modify` to propagate block exceptions instead of panicking.
+
 ## [0.2.0] - 2026-06-04
 
 - Drop Ruby 3.x support and require Ruby 4.

data/README.md

@@ -1,5 +1,12 @@
 # Ratomic
 
+[![Gem Version](https://badge.fury.io/rb/ratomic.svg)](https://badge.fury.io/rb/ratomic)
+[![CI](https://github.com/mperham/ratomic/workflows/CI/badge.svg)](https://github.com/mperham/ratomic/actions)
+[![Coverage Status](https://codecov.io/gh/mperham/ratomic/branch/main/graph/badge.svg)](https://codecov.io/gh/mperham/ratomic)
+[![Ruby Version](https://img.shields.io/badge/ruby-%3E%3D%204.0-ruby.svg)](https://www.ruby-lang.org/en/)
+[![License: MIT](https://img.shields.io/badge/License-MIT-yellow.svg)](https://opensource.org/licenses/MIT)
+
+
 Ratomic provides mutable data structures for use with Ruby's Ractors.
 This allows Ruby code to scale beyond the infamous GVL.
 
@@ -31,8 +38,6 @@ Install the gem and add to the application's Gemfile by executing:
 bundle add ratomic
 ```
 
-TODO: We have not released a gem yet.
-
 ## Usage
 
 Ratomic provides several useful Ractor-safe structures.

data/ext/ratomic/src/lib.rs

@@ -115,11 +115,22 @@ impl HashMap {
         }
 
         let proc = ruby.block_proc()?;
+        let mut error = None;
         rb_self.0.fetch_and_modify(value_to_raw(key), |value| {
-            let result: Value = proc.call((unsafe { value_from_raw(value) },)).unwrap();
-            value_to_raw(result)
+            match proc.call::<_, Value>((unsafe { value_from_raw(value) },)) {
+                Ok(result) => value_to_raw(result),
+                Err(err) => {
+                    error = Some(err);
+                    value
+                }
+            }
         });
-        Ok(())
+
+        if let Some(err) = error {
+            Err(err)
+        } else {
+            Ok(())
+        }
     }
 }
 

data/ext/ratomic/src/mpmc_queue.rs

@@ -15,7 +15,7 @@ unsafe impl Sync for QueueElement {}
 
 pub struct MpmcQueue {
     buffer: Vec<QueueElement>,
-    buffer_mask: usize,
+    buffer_size: usize,
     enqueue_pos: AtomicUsize,
     dequeue_pos: AtomicUsize,
     gc_guard: GcGuard,
@@ -25,7 +25,7 @@ pub struct MpmcQueue {
 
 impl MpmcQueue {
     pub fn new(buffer_size: usize, default: VALUE) -> Self {
-        let mut buffer = Vec::with_capacity(buffer_size.next_power_of_two());
+        let mut buffer = Vec::with_capacity(buffer_size);
         for i in 0..buffer_size {
             buffer.push(QueueElement {
                 sequence: AtomicUsize::new(i),
@@ -42,7 +42,7 @@ impl MpmcQueue {
 
         Self {
             buffer,
-            buffer_mask: buffer_size - 1,
+            buffer_size,
             enqueue_pos: AtomicUsize::new(0),
             dequeue_pos: AtomicUsize::new(0),
             gc_guard,
@@ -55,7 +55,7 @@ impl MpmcQueue {
         let mut cell;
         let mut pos = self.enqueue_pos.load(Ordering::Relaxed);
         loop {
-            cell = &self.buffer[pos & self.buffer_mask];
+            cell = &self.buffer[pos % self.buffer_size];
             let seq = cell.sequence.load(Ordering::Acquire);
             let diff = seq as isize - pos as isize;
             if diff == 0 {
@@ -82,7 +82,7 @@ impl MpmcQueue {
         let mut cell;
         let mut pos = self.dequeue_pos.load(Ordering::Relaxed);
         loop {
-            cell = &self.buffer[pos & self.buffer_mask];
+            cell = &self.buffer[pos % self.buffer_size];
             let seq = cell.sequence.load(Ordering::Acquire);
             let diff = seq as isize - (pos + 1) as isize;
             if diff == 0 {
@@ -102,7 +102,7 @@ impl MpmcQueue {
 
         let data = cell.data.get();
         cell.sequence
-            .store(pos + self.buffer_mask + 1, Ordering::Release);
+            .store(pos + self.buffer_size, Ordering::Release);
         self.write_sem.post();
 
         #[cfg(feature = "simulation")]
@@ -131,7 +131,7 @@ impl MpmcQueue {
 
     pub fn peek(&self) -> Option<VALUE> {
         let pos = self.dequeue_pos.load(Ordering::Relaxed);
-        let cell = &self.buffer[pos & self.buffer_mask];
+        let cell = &self.buffer[pos % self.buffer_size];
         let seq = cell.sequence.load(Ordering::Acquire);
         let diff = seq as isize - (pos + 1) as isize;
         if diff == 0 {

data/lib/ratomic/pool.rb

@@ -73,6 +73,20 @@ module Ratomic
       nil
     end
 
+    # Stop the private coordinator Ractor.
+    #
+    # This is primarily useful for tests and short-lived scripts. A closed pool
+    # should not be used for further checkout/checkin operations.
+    #
+    # @return [nil]
+    def close
+      @control << [:shutdown]
+      @control.value
+      nil
+    rescue Ractor::ClosedError, Ractor::Error
+      nil
+    end
+
     # Checkout an object, yield it, then move it back to the pool.
     #
     # This is the preferred API because it guarantees checkin through an ensure
@@ -101,7 +115,7 @@ module Ratomic
 
       loop do
         command, *args = Ractor.receive
-        handle_command(command, args, available, waiting)
+        break if handle_command(command, args, available, waiting) == :shutdown
       end
     end
     private_class_method :run_control_loop
@@ -114,6 +128,8 @@ module Ratomic
         handle_checkin(args.fetch(0), available, waiting)
       when :cancel
         waiting.delete(args.fetch(0))
+      when :shutdown
+        :shutdown
       end
     end
     private_class_method :handle_command

data/lib/ratomic/version.rb

@@ -1,5 +1,5 @@
 # frozen_string_literal: true
 
 module Ratomic
-  VERSION = "0.2.0"
+  VERSION = "0.2.1"
 end

metadata

@@ -1,7 +1,7 @@
 --- !ruby/object:Gem::Specification
 name: ratomic
 version: !ruby/object:Gem::Version
-  version: 0.2.0
+  version: 0.2.1
 platform: ruby
 authors:
 - Mike Perham