tonbo-io / tonbo

A portable embedded database using Arrow.
https://tonbo.io
Apache License 2.0
606 stars 42 forks source link
arrow big-data database embedded-database htap lsm-tree offline-first parquet rust store-engine

# Tonbo

CI

**[Website](https://tonbo.io/) | [Rust Doc](https://docs.rs/tonbo/latest/tonbo/) | [Blog](https://tonbo.io/blog/introducing-tonbo) | [Community](https://discord.gg/j27XVFVmJM)** ## Introduction Tonbo is an embedded, persistent database offering fast KV-like methods for conveniently writing and scanning type-safe structured data. Tonbo can be used to build data-intensive applications, including other types of databases. Tonbo is implemented with a [Log-Structured Merge Tree](https://en.wikipedia.org/wiki/Log-structured_merge-tree), constructed using [Apache Arrow](https://github.com/apache/arrow-rs) & [Apache Parquet](https://github.com/apache/arrow-rs/tree/master/parquet) data blocks. Leveraging Arrow and Parquet, Tonbo supports: - Pushdown limits, predicates, and projection operators - Zero-copy deserialization - Various storage backends: OPFS, S3, etc. (to be supported in v0.2.0) These features enhance the efficiency of queries on structured data. Tonbo is designed to integrate seamlessly with other Arrow analytical tools, such as DataFusion. For an example, refer to this [preview](examples/datafusion.rs) (official support for DataFusion will be included in v0.2.0). > Note: Tonbo is currently unstable; API and file formats may change in upcoming minor versions. Please avoid using it in production and stay tuned for updates. ## Example ```rust use std::ops::Bound; use futures_util::stream::StreamExt; use tonbo::{executor::tokio::TokioExecutor, Record, Projection, DB}; /// Use macro to define schema of column family just like ORM /// It provides type-safe read & write API #[derive(Record, Debug)] pub struct User { #[record(primary_key)] name: String, email: Option, age: u8, } #[tokio::main] async fn main() { // pluggable async runtime and I/O let db = DB::new("./db_path/users".into(), TokioExecutor::default()) .await .unwrap(); // insert with owned value db.insert(User { name: "Alice".into(), email: Some("alice@gmail.com".into()), age: 22, }) .await .unwrap(); { // tonbo supports transaction let txn = db.transaction().await; // get from primary key let name = "Alice".into(); // get the zero-copy reference of record without any allocations. let user = txn .get( &name, // tonbo supports pushing down projection Projection::All, ) .await .unwrap(); assert!(user.is_some()); assert_eq!(user.unwrap().get().age, Some(22)); { let upper = "Blob".into(); // range scan of users let mut scan = txn .scan((Bound::Included(&name), Bound::Excluded(&upper))) .await // tonbo supports pushing down projection .projection(vec![1]) .take() .await .unwrap(); while let Some(entry) = scan.next().await.transpose().unwrap() { assert_eq!( entry.value(), Some(UserRef { name: "Alice", email: Some("alice@gmail.com"), age: Some(22), }) ); } } // commit transaction txn.commit().await.unwrap(); } } ``` ## Features - [x] Fully asynchronous API. - [x] Zero-copy rusty API ensuring safety with compile-time type and lifetime checks. - [x] Vendor-agnostic: - [ ] Various usage methods, async runtimes, and file systems: - [x] Rust library: - [x] [Customizable async runtime and file system](https://github.com/from-the-basement/tonbo/blob/main/src/executor.rs#L5). - [x] [Tokio and Tokio fs](https://github.com/tokio-rs/tokio). - [ ] [Async-std](https://github.com/async-rs/async-std). - [ ] Python library (via [PyO3](https://github.com/PyO3/pyo3) & [pydantic](https://github.com/pydantic/pydantic)): - [ ] asyncio (via [pyo3-asyncio](https://github.com/awestlake87/pyo3-asyncio)). - [ ] JavaScript library: - [ ] WASM and OPFS. - [ ] Dynamic library with a C interface. - [x] Most lightweight implementation to Arrow / Parquet LSM Trees: - [x] Define schema using just Arrow schema and store data in Parquet files. - [x] (Optimistic) Transactions. - [x] Leveled compaction strategy. - [x] Push down filter, limit and projection. - [ ] Runtime schema definition (*in next release*). - [ ] SQL (via [Apache DataFusion](https://datafusion.apache.org/)). - [ ] Fusion storage across RAM, flash, SSD, and remote Object Storage Service (OSS) for each column-family, balancing performance and cost efficiency per data block: - [ ] Remote storage (via [Arrow object_store](https://github.com/apache/arrow-rs/tree/master/object_store) or [Apache OpenDAL](https://github.com/apache/opendal)). - [ ] Distributed query and compaction. - [ ] Blob storage (like [BlobDB in RocksDB](https://github.com/facebook/rocksdb/wiki/BlobDB)). ## Contributing to Tonbo Follow the Contributing Guide to [contribute](https://github.com/tonbo-io/tonbo/blob/main/CONTRIBUTING.md). Please feel free to ask any question or contact us on Github [Discussions](https://github.com/orgs/tonbo-io/discussions) or [issues](https://github.com/tonbo-io/tonbo/issues).