nicholassm / disruptor-rs

Low latency inter-thread communication library in Rust inspired by the LMAX Disruptor.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Crates.io Crates.io Build codecov

Disruptor

This library is a low latency, inter-thread communication library written in Rust.

It's heavily inspired by the brilliant Disruptor library from LMAX.

Contents

Getting Started

Add the following to your Cargo.toml file:

disruptor = "3.1.0"

To read details of how to use the library, check out the documentation on docs.rs/disruptor.

Here's a minimal example demonstrating both single and batch publication. Note, batch publication should be used whenever possible for best latency and throughput (see benchmarks below).

use disruptor::*;

// The event on the ring buffer.
struct Event {
    price: f64
}

fn main() {
    // Factory closure for initializing events in the Ring Buffer.
    let factory = || { Event { price: 0.0 }};

    // Closure for processing events.
    let processor = |e: &Event, sequence: Sequence, end_of_batch: bool| {
        // Your processing logic here.
    };

    let size = 64;
    let mut producer = disruptor::build_single_producer(size, factory, BusySpin)
        .handle_events_with(processor)
        .build();

    // Publish single events into the Disruptor via the `Producer` handle.
    for i in 0..10 {
        producer.publish(|e| {
            e.price = i as f64;
        });
    }

    // Publish a batch of events into the Disruptor.
    producer.batch_publish(5, |iter| {
        for e in iter { // `iter` is guaranteed to yield 5 events.
            e.price = 42.0;
        }
    });
}// At this point, the Producer instance goes out of scope and when the
 // processor is done handling all events then the Disruptor is dropped
 // as well.

The library also supports pinning threads on cores to avoid latency induced by context switching. A more advanced usage demonstrating this and with multiple producers and multiple interdependent consumers could look like this:

use disruptor::*;
use std::thread;

struct Event {
    price: f64
}

fn main() {
    let factory = || { Event { price: 0.0 }};

    // Closure for processing events.
    let h1 = |e: &Event, sequence: Sequence, end_of_batch: bool| {
        // Processing logic here.
    };
    let h2 = |e: &Event, sequence: Sequence, end_of_batch: bool| {
        // Some processing logic here.
    };
    let h3 = |e: &Event, sequence: Sequence, end_of_batch: bool| {
        // More processing logic here.
    };

    let mut producer1 = disruptor::build_multi_producer(64, factory, BusySpin)
        // `h2` handles events concurrently with `h1`.
        .pin_at_core(1).handle_events_with(h1)
        .pin_at_core(2).handle_events_with(h2)
            .and_then()
            // `h3` handles events after `h1` and `h2`.
            .pin_at_core(3).handle_events_with(h3)
        .build();

    // Create another producer.
    let mut producer2 = producer1.clone();

    // Publish into the Disruptor.
    thread::scope(|s| {
        s.spawn(move || {
            for i in 0..10 {
                producer1.publish(|e| {
                    e.price = i as f64;
                });
            }
        });
        s.spawn(move || {
            for i in 10..20 {
                producer2.publish(|e| {
                    e.price = i as f64;
                });
            }
        });
    });
}// At this point, the Producers instances go out of scope and when the
 // processors are done handling all events then the Disruptor is dropped
 // as well.

If you need to store some state in the processor thread which is neither Send nor Sync, e.g. a Rc<RefCell<i32>>, then you can create a closure for initializing that state and pass it along with the processing closure when you build the Disruptor. The Disruptor will then pass a mutable reference to your state on each event. As an example:

use std::{cell::RefCell, rc::Rc};
use disruptor::*;

struct Event {
    price: f64
}

#[derive(Default)]
struct State {
    data: Rc<RefCell<i32>>
}

fn main() {
    let factory = || { Event { price: 0.0 }};
    let initial_state = || { State::default() };

    // Closure for processing events *with* state.
    let processor = |s: &mut State, e: &Event, _: Sequence, _: bool| {
        // Mutate your custom state:
        *s.data.borrow_mut() += 1;
    };

    let size = 64;
    let mut producer = disruptor::build_single_producer(size, factory, BusySpin)
        .handle_events_and_state_with(processor, initial_state)
        .build();

    for i in 0..10 {
        producer.publish(|e| {
            e.price = i as f64;
        });
    }
}

Features

  • Single Producer Single Consumer (SPSC).
  • Multi Producer Single Consumer (MPSC).
  • Multi Producer Multi Consumer (MPMC) with consumer interdependencies.
  • Busy-spin wait strategies.
  • Batch publication of events.
  • Batch consumption of events.
  • Thread affinity can be set for the event processor thread(s).
  • Set thread name of each event processor thread.

Design Choices

Everything in the library is about low-latency and this heavily influences all choices made in this library. As an example, you cannot allocate an event and move that into the ringbuffer. Instead, events are allocated on startup to ensure they are co-located in memory to increase cache coherency. However, you can still allocate a struct on the heap and move ownership to a field in the event on the Ringbuffer. As long as you realize that this can add latency, because the struct is allocated by one thread and dropped by another. Hence, there's synchronization happening in the allocator.

There's also no use of dynamic dispatch - everything is monomorphed.

Correctness

This library needs to use Unsafe to achieve low latency. Although the absence of bugs cannot be guaranteed, these approaches have been used to eliminate bugs:

  • Minimal usage of Unsafe blocks.
  • High test coverage.
  • All tests are run on Miri in CI/CD.
  • Verification in TLA+ (see the verification/ folder).

Performance

The SPSC and MPSC Disruptor variants have been benchmarked and compared to Crossbeam. See the code in the benches/spsc.rs and benches/mpsc.rs files.

The results below of the SPSC benchmark are gathered from running the benchmarks on a 2016 Macbook Pro running a 2,6 GHz Quad-Core Intel Core i7. So on a modern Intel Xeon the numbers should be even better. Furthermore, it's not possible to isolate cores on Mac and pin threads which would produce even more stable results. This is future work.

If you have any suggestions to improving the benchmarks, please feel free to open an issue.

To provide a somewhat realistic benchmark not only burst of different sizes are considered but also variable pauses between bursts: 0 ms, 1 ms and 10 ms.

The latencies below are the mean latency per element with 95% confidence interval (standard criterion settings). Capturing all latencies and calculating misc. percentiles (and in particular the max latency) is future work. However, I expect the below measurements to be representative for the actual performance you can achieve in a real application.

No Pause Between Bursts

Latency:

Burst Size Crossbeam Disruptor Improvement
1 65 ns 32 ns 51%
10 68 ns 9 ns 87%
100 29 ns 8 ns 72%

Throughput:

Burst Size Crossbeam Disruptor Improvement
1 15.2M / s 31.7M / s 109%
10 14.5M / s 117.3M / s 709%
100 34.3M / s 119.7M / s 249%

1 ms Pause Between Bursts

Latency:

Burst Size Crossbeam Disruptor Improvement
1 63 ns 33 ns 48%
10 67 ns 8 ns 88%
100 30 ns 9 ns 70%

Throughput:

Burst Size Crossbeam Disruptor Improvement
1 15.9M / s 30.7M / s 93%
10 14.9M / s 117.7M / s 690%
100 33.8M / s 105.0M / s 211%

10 ms Pause Between Bursts

Latency:

Burst Size Crossbeam Disruptor Improvement
1 51 ns 32 ns 37%
10 67 ns 9 ns 87%
100 30 ns 10 ns 67%

Throughput:

Burst Size Crossbeam Disruptor Improvement
1 19.5M / s 31.6M / s 62%
10 14.9M / s 114.5M / s 668%
100 33.6M / s 105.0M / s 213%

Conclusion

There's clearly a difference between the Disruptor and the Crossbeam libs. However, this is not because the Crossbeam library is not a great piece of software. It is. The Disruptor trades CPU and memory resources for lower latency and higher throughput and that is why it's able to achieve these results. The Disruptor also excels if you can publish batches of events as demonstrated in the benchmarks with bursts of 10 and 100 events.

Both libraries greatly improves as the burst size goes up but the Disruptor's performance is more resilient to the pauses between bursts which is one of the design goals.

Related Work

There are multiple other Rust projects that mimic the LMAX Disruptor library:

  1. Turbine
  2. Disrustor

A key feature that this library supports is multiple producers from different threads that neither of the above libraries support (at the time of writing).

Contributions

You are welcome to create a Pull-Request or open an issue with suggestions for improvements.

Changes are accepted solely at my discretion and I will focus on whether the changes are a good fit for the purpose and design of this crate.

Roadmap

Empty! All the items have been implemented.

About

Low latency inter-thread communication library in Rust inspired by the LMAX Disruptor.

License:MIT License


Languages

Language:Rust 87.2%Language:TLA 12.8%