5 releases
new 0.5.4 | Feb 18, 2025 |
---|---|
0.5.3 | Jan 28, 2025 |
0.5.2 | Dec 5, 2024 |
0.5.1 | Nov 27, 2024 |
0.4.0 |
|
#50 in Database implementations
145 downloads per month
120KB
2K
SLoC
REDSUMER
A lightweight implementation of Redis Streams for Rust, allowing you to manage streaming messages in a simplified way. With redsumer you can:
- Produce new messages in a specific stream.
- Consume messages from specific stream, setting config parameters that allow you a flexible implementation. It also provides an option to minimize the possibility of consuming the same message simultaneously by more than one consumers from the same consumer group.
BASIC OPERATION
Produce a new stream message:
Create a new producer instance and produce a new stream message from a BTreeMap:
use std::collections::BTreeMap;
use redsumer::prelude::*;
use time::OffsetDateTime;
use uuid::Uuid;
#[tokio::main]
async fn main() {
let credentials: Option<ClientCredentials> = None;
let host: &str = "localhost";
let port: u16 = 6379;
let db: i64 = 0;
let stream_name: &str = "my-stream";
let args: ClientArgs = ClientArgs::new(
credentials,
host,
port,
db,
CommunicationProtocol::RESP2,
);
let config: ProducerConfig = ProducerConfig::new(stream_name);
let producer_result: RedsumerResult<Producer> =
Producer::new(
&args,
&config,
);
let producer: Producer = producer_result.unwrap_or_else(|error| {
panic!("Error creating a new RedsumerProducer instance: {:?}", error);
});
let mut message_1: BTreeMap<&str, String> = BTreeMap::new();
message_1.insert("id", Uuid::new_v4().to_string());
message_1.insert("started_at", OffsetDateTime::now_utc().to_string());
let mut message_2: Vec<(String, String)> = Vec::new();
message_2.push(("id".to_string(), Uuid::new_v4().to_string()));
message_2.push(("started_at".to_string(), OffsetDateTime::now_utc().to_string()));
let reply_1: ProduceMessageReply = producer.produce_from_map(message_1).await.unwrap_or_else(|error| {
panic!("Error producing stream message from BTreeMap: {:?}", error.to_string());
});
let reply_2: ProduceMessageReply = producer.produce_from_items(message_2).await.unwrap_or_else(|error| {
panic!("Error producing stream message from Vec: {:?}", error.to_string());
});
println!("Message 1 produced with id: {:?}", reply_1);
println!("Message 2 produced with id: {:?}", reply_2);
}
Similar to the previous example, you can produce a message from a HashMap or a HashSet.
The produce_from_map and produce_from_item methods accepts generic types that implements the ToRedisArgs trait. Take a look at the documentation for more information.
Consume messages from a stream:
Create a new consumer instance and consume messages from stream:
use redsumer::prelude::*;
use redsumer::redis::StreamId;
#[tokio::main]
async fn main() {
let credentials: Option<ClientCredentials> = None;
let host: &str = "localhost";
let port: u16 = 6379;
let db: i64 = 0;
let stream_name: &str = "my-stream";
let group_name: &str = "group-name";
let consumer_name: &str = "consumer";
let initial_stream_id: &str = "0-0";
let min_idle_time_milliseconds: usize = 1000;
let new_messages_count: usize = 3;
let pending_messages_count: usize = 2;
let claimed_messages_count: usize = 1;
let block: usize = 5;
let args: ClientArgs = ClientArgs::new(
credentials,
host,
port,
db,
CommunicationProtocol::RESP2,
);
let config: ConsumerConfig = ConsumerConfig::new(
stream_name,
group_name,
consumer_name,
ReadNewMessagesOptions::new(
new_messages_count,
block
),
ReadPendingMessagesOptions::new(
pending_messages_count
),
ClaimMessagesOptions::new(
claimed_messages_count,
min_idle_time_milliseconds
),
);
let consumer_result: RedsumerResult<Consumer> = Consumer::new(
args,
config,
Some(initial_stream_id.to_string()),
);
let mut consumer: Consumer = consumer_result.unwrap_or_else(|error| {
panic!("Error creating a new RedsumerConsumer instance: {:?}", error);
});
loop {
let consume_reply: ConsumeMessagesReply = consumer.consume().await.unwrap_or_else(|error| {
panic!("Error consuming messages from stream: {:?}", error);
});
for message in consume_reply.get_messages() {
if consumer.is_still_mine(&message.id).unwrap_or_else(|error| {
panic!(
"Error checking if message is still in consumer pending list: {:?}", error
);
}).belongs_to_me() {
// Process message ...
println!("Processing message: {:?}", message);
// ...
let ack_reply: AckMessageReply = consumer.ack(&message.id).await.unwrap_or_else(|error| {
panic!("Error acknowledging message: {:?}", error);
});
if ack_reply.was_acked() {
println!("Message acknowledged: {:?}", message);
}
}
}
}
}
In this example, the consume method is called in a loop to consume messages from the stream. The consume method returns a vector of StreamId instances. Each StreamId instance represents a message in the stream. The is_still_mine method is used to check if the message is still in the consumer pending list. If it is, the message is processed and then acknowledged using the ack method. The ack method returns a boolean indicating if the message was successfully acknowledged.
The main objective of this message consumption strategy is to minimize the possibility that two or more consumers from the same consumer group operating simultaneously consume the same message at the same time. Knowing that it is a complex problem with no definitive solution, including business logic in the message processing instance will always improve results.
Utilities from [redis] crate:
The redis module provides utilities from the redis crate. You can use these utilities to interact with Redis values and errors.
Unwrap Value to a specific type:
The Value enum represents a Redis value. It can be converted to a specific type using the from_redis_value function. This function can be imported from the redis module.
DEVELOPMENT
To try redsumer, follow these recommendations:
Install required tools:
Install the following tools required for the development process.
Cargo LLVM cov:
cargo install cargo-llvm-cov
Cargo Nextest:
cargo install cargo-nextest
Cargo Deny:
cargo install cargo-deny
Taplo Client:
cargo install taplo-cli
Yamlfmt:
go install github.com/google/yamlfmt/cmd/yamlfmt@latest
To format the code:
cargo fmt --all
To verify the code format:
cargo fmt --all --check
To verify lints:
cargo clippy --workspace --all-features
To check the project:
cargo check --workspace --all-features
To format TOML files:
taplo fmt
To verify the TOML files format:
taplo fmt --check --verbose --diff
To check correctness of crate manifest:
cargo verify-project --verbose
To format yaml files:
yamlfmt .
To verify the yaml files format:
yamlfmt --lint .
To generate the documentation:
cargo doc --workspace --all-features --document-private-items --verbose --open
To run doc tests:
cargo test --workspace --all-features --doc
To test using llvm-cov:
cargo llvm-cov nextest --workspace --all-features --show-missing-lines --open
Security analysis:
cargo deny --log-level debug check
CONTRIBUTING
We welcome contributions to redsumer. Here are some ways you can contribute:
- Bug Reports: If you find a bug, please create an issue detailing the problem, the steps to reproduce it, and the expected behavior.
- Feature Requests: If you have an idea for a new feature or an enhancement to an existing one, please create an issue describing your idea.
- Pull Requests: If you've fixed a bug or implemented a new feature, we'd love to see your work! Please submit a pull request. Make sure your code follows the existing style and all tests pass.
Thank you for your interest in improving redsumer!
Dependencies
~6MB
~109K SLoC