Skip to content

A high-performance, thread-safe key-value embedded data store.

License

Notifications You must be signed in to change notification settings

RPallas92/GrausDB

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

25 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

GrausDb

Rust License

GrausDb is a high-performance, thread-safe key-value embedded data store written in Rust. It is designed for simplicity, efficiency, and reliability.

Features

  • Lock-Free Reads: GrausDb leverages lock-free data structures for high-performance concurrent reads.
  • Persistence: Data is persisted to disk for durability.
  • Log-Based Storage: Key-value pairs are stored in log files.
  • Benchmarks: Benchmarks included to assess the performance.

Installation

To use GrausDb in your Rust project, simply add it as a dependency in your Cargo.toml file:

[dependencies]
graus_db = "0.2.0"

Quick start

Here's a quick example of how to use GrausDb in your Rust application:

use graus_db::{GrausDb, Result};
use bytes::Bytes;

fn main() -> Result<()> {
    let store = GrausDb::open("path")?;

    store.set(Bytes::from_static(b"key"), Bytes::from_static(b"value"))?;

    let val = store.get(&Bytes::from_static(b"key"))?;
    if let Some(value) = val {
        println!("Value: {:?}", value); // Outputs: Value: "value"
    }

    Ok(())
}

It can also be called from multiple threads:

use std::thread;
use graus_db::{GrausDb, Result};
use bytes::Bytes;

fn main() -> Result<()> {
    let store = GrausDb::open("path")?;

    // Calls set method from 8 different threads
    for i in 0..8 {
        let store = store.clone();
        thread::spawn(move || {
            store.set(Bytes::from(format!("key{}", i)), Bytes::from(format!("value{}", i)))
                .unwrap();
        });
    }

    Ok(())
}

API

GrausDb provides a simple and intuitive API for interacting with the key-value store. Below are some of the key functions and methods exposed by GrausDb, along with usage examples.

GrausDb::open

open is used to open a GrausDb instance, creating a new database if it doesn't exist at the specified path.

Example:

use graus_db::{GrausDb, Result};

fn main() -> Result<()> {
    let store = GrausDb::open("my_database")?;
    // Your database is now ready to use.
    Ok(())
}

set

The set method is used to store a key-value pair in the database.

Example:

use graus_db::{GrausDb, Result};
use bytes::Bytes;

fn main() -> Result<()> {
    let store = GrausDb::open("my_database")?;
    store.set(Bytes::from_static(b"key"), Bytes::from_static(b"value"))?;
    // Key "key" now has the value "value" in the database.
    Ok(())
}

get

The get method retrieves the value associated with a given key.

Example:

use graus_db::{GrausDb, Result};
use bytes::Bytes;

fn main() -> Result<()> {
    let store = GrausDb::open("my_database")?;
    store.set(Bytes::from_static(b"key"), Bytes::from_static(b"value"))?;
    
    if let Some(value) = store.get(&Bytes::from_static(b"key"))? {
        println!("Value: {:?}", value); // Outputs: Value: "value"
    } else {
        println!("Key not found");
    }
    Ok(())
}

remove

The remove method deletes a key and its associated value from the database.

Example:

use graus_db::{GrausDb, Result};
use bytes::Bytes;

fn main() -> Result<()> {
    let store = GrausDb::open("my_database")?;
    store.set(Bytes::from_static(b"key"), Bytes::from_static(b"value"))?;
    store.remove(Bytes::from_static(b"key"))?;
    // Key "key" and its value are now removed from the database.
    Ok(())
}

update_if

The update_if method updates the value of an existing key atomically, allowing you to provide a custom update function.

An optional predicate can be passed, the value will only be updated if the predicate is satisfied.

Example:

use graus_db::{GrausDb, Result};
use bytes::{Bytes, BytesMut};

fn main() -> Result<()> {
    let store = GrausDb::open("my_database")?;
    let key = Bytes::from_static(b"key1");

    // Store an initial value of 25 (encoded as u64 in little-endian byte order)
    let initial_value = 25u64.to_le_bytes();
    store.set(key.clone(), Bytes::from_static(&initial_value))?;

    // Update function that decreases the stored value by 1
    let update_fn = |value: &mut BytesMut| {
        let num = u64::from_le_bytes(
            value.as_ref()[..8].try_into().expect("incorrect length"),
        ) - 1;

        value.copy_from_slice(&num.to_le_bytes());
    };

    // Predicate function that only applies the update if the value is greater than 0
    let predicate = |value: &Bytes| {
        let num = u64::from_le_bytes(value[..8].try_into().expect("incorrect length"));
        num > 0
    };

    let result = store.update_if(key.clone(), update_fn, Some(&key), Some(predicate));
    // Key "key1" now has the value "24" in the database.
    // The function was applied because the predicate was met (25 > 0)

    Ok(())
}

For more details on how to use GrausDb, please refer to the tests.

Architecture and Implementation

GrausDb's architecture is built around the principles of log-based storage and lock-free read concurrency:

  • Log-Based Storage: GrausDb stores key-value pairs in log files. Log files are named after monotonically increasing generation numbers with a log extension. This design ensures that data is durably persisted to disk.

  • Lock-Free Concurrency for Reads: GrausDb uses lock-free data structures to provide high-performance concurrent reads to the data. This enables multiple threads to interact with the database efficiently.

  • In-Memory Index: GrausDb maintains an in-memory index that maps keys to their positions in the log. This index allows for fast lookups and efficient data retrieval.

  • Compaction: To maintain efficient storage and reduce disk space usage, GrausDb performs compaction when a threshold is reached. Compaction involves rewriting log files, removing stale data, and reclaiming disk space.

Benchmarks

GrausDb includes built-in benchmarking tools to evaluate its efficiency and to help you make data-driven decisions.

Future Development

Next features:

  • Multithread benchmark
  • Range get

License

GrausDb is licensed under the MIT License.

Happy coding with GrausDb!

Contact

GrausDb is created and maintained by Ricardo Pallas.

Website: https://rpallas92.github.io/

About

A high-performance, thread-safe key-value embedded data store.

Topics

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages