A lightweight, embedded key-value database for mobile clients (i.e., iOS, Android), written in Rust.
Overview
ThetaDB
is suitable for use on mobile clients with "High-Read, Low-Write" demands, it uses B+ Tree as the foundational layer for index management.
Inspired by Go's BoltDB, ThetaDB uses mmap
, relying on the operating system to keep memory and database files in sync. ThetaDB also implements shadow paging
to guarantee atomicity and durability of transactions, preventing data loss or damage to the internal structures of database.
Installation
ThetaDB provides APIs for these languages: Rust
, Swift
, Kotlin
.
Rust - Cargo
Add Thetadb
as a dependency to your Cargo.toml
:
[dependencies]
thetadb = 0.0.1
Swift - Package Manager
Add ThetaDB
as a dependency to your Package.swift
:
.package(url: "https://github.com/TangentW/ThetaDB.git", from: "0.0.1")
And then specify ThetaDB
as a dependency of the Target in which you wish to use ThetaDB.
Swift - CocoaPods
Add the pod to your Podfile
:
pod 'ThetaDB'
And then execute:
$ pod install
Kotlin
🚧 Yet To Be Developed.
Usage
ThetaDB's APIs are generally similar regardless of the language used.
Open Database
Use following way to open the database at the specified path. If the database file does not exist, ThetaDB will automatically create and initialize it.
Rust Examples
use thetadb::{Options, ThetaDB};
let path = "path/to/db.theta";
// The simplest way to open with default `Options`:
let db = ThetaDB::open(path)?;
// Open with `Options`:
let db = Options::new()
.force_sync(true)
.mempool_capacity(8)
.open(path)?;
Swift Examples
import ThetaDB
let path = "path/to/db.theta"
// The simplest way to open with default `Options`:
let db = try ThetaDB(path: path)
// Open with `Options`:
let db = try ThetaDB(
path: path,
options: .init(
forceSync: true,
mempoolCapacity: 8
)
)
ThetaDB has no specific requirements for filename and extension, but it is recommended to use theta
as extension for easy identification.
ThetaDB will automatically close when the database instance is destroyed.
Get, Insert, Update, Delete
Rust Examples
// Insert a new key-value pair into database.
db.put(b"foo", b"foo")?;
// Check if the database contains a given key.
assert!(db.contains(b"foo")?);
assert!(!db.contains(b"unknown")?);
// Get the value associated with a given key.
assert_eq!(
db.get(b"foo")?,
Some(b"foo".to_vec())
);
assert_eq!(
db.get(b"unknown")?,
None
);
// Update an existing value associated with a given key.
db.put(b"foo", b"bar")?;
assert_eq!(
db.get(b"foo")?,
Some(b"bar".to_vec())
);
// Delete an existing key-value pair from database.
db.delete(b"foo")?;
assert!(!db.contains(b"foo")?);
assert_eq!(
db.get(b"foo")?,
None
);
Swift Examples
// Insert a new key-value pair into database.
try db.put("foo".data(using: .utf8)!, for: "foo")
// Check if the database contains a given key.
assert(try db.contains("foo"))
assert(try !db.contains("unknown"))
// Get the value associated with a given key.
assertEq(
try db.get("foo"),
"foo".data(using: .utf8)
)
assertEq(
try db.get("unknown"),
nil
)
// Update an existing value associated with a given key.
try db.put("bar".data(using: .utf8)!, for: "foo")
assertEq(
try db.get("foo"),
"bar".data(using: .utf8)
)
// Delete an existing key-value pair from database.
try db.delete("foo")
assert(try !db.contains("foo"))
assertEq(
try db.get("foo"),
nil
)
Transaction
ThetaDB has two kinds of transactions: Read-Only Transaction
and Read-Write Transaction
. The read-only transaction allows for read-only access and the read-write transaction allows modification.
ThetaDB allows a number of read-only transactions at a time but allows at most one read-write transaction at a time. When a read-write transaction is committing, it has exclusive access to the database until the commit is completed, at which point other transactions trying to access the database will be blocked. You can think of this situation as shared access
and exclusive access
to reader-writer lock.
Read-Only Transaction
Rust Examples
// Start a read-only transaction.
let tx = db.begin_tx()?;
// Then perform read-only access.
_ = tx.contains(b"foo")?;
_ = tx.get(b"foo")?;
// Or you can perform a read-only transaction using closure,
// with `view` method:
db.view(|tx| {
_ = tx.contains(b"foo")?;
_ = tx.get(b"foo")?;
Ok(())
})?;
Swift Examples
// Start a read-only transaction.
let tx = try db.beginTx()
// Then perform read-only access.
_ = try tx.contains("foo")
_ = try tx.get("foo")
// Or you can perform a read-only transaction using closure,
// with `view` method:
try db.view {
_ = try $0.contains("foo")
_ = try $0.get("foo")
}
Read-Write Transaction
ThetaBD's read-write transactions are designed to automatically rollback, and therefore any changes made to the transaction will be discarded unless you explicity call the commit
method.
Or you can perform a read-write transaction using closure, if no errors occur, then the transaction will be commit automatically after the closure call.
Rust Examples
// Start a read-write transaction.
let mut tx = db.begin_tx_mut()?;
// Then perform read-write access.
tx.put(b"hello", b"world")?;
_ = tx.get(b"hello")?;
// Finally, commit the transaction.
tx.commit()?;
// Or you can perform a read-write transaction using closure,
// with `update` method:
db.update(|tx| {
tx.put(b"hello", b"world")?;
_ = tx.get(b"hello")?;
Ok(())
})?;
Swift Examples
// Start a read-write transaction.
let tx = try db.beginTxMut()
// Then perform read-write access.
try tx.put("world".data(using: .utf8)!, for: "hello")
_ = try tx.get("hello")
// Finally, commit the transaction.
try tx.commit()
// Or you can perform a read-write transaction using closure,
// with `update` method:
try db.update {
try $0.put("world".data(using: .utf8)!, for: "hello")
_ = try $0.get("hello")
}
Attention
❗️ Transaction instances are nonsendable, which means it's not safe to send them to another thread. Rust leverages Ownership
system and the Send
and Sync
traits to enforce requirements automatically, whereas Swift requires us to manually ensure these guarantees.
❗️ Read-only transactions and read-write transaction must not overlap, otherwise a deadlock will be occurred.
😺 So ThetaDB recommends that if you want to use transactions, use the APIs with closure parameter (i.e., view
, update
).
Cursor
We can freely traverse the data in the ThetaDB using Cursor
.
For instance, we can iterate over all the key-value pairs in the ThetaDB like this:
Rust Examples
// Forward traversal.
let mut cursor = db.first_cursor()?;
while let Some((key, value)) = cursor.key_value()? {
println!("{:?} => {:?}", key, value);
cursor.next()?;
}
// Backward traversal.
let mut cursor = db.last_cursor()?;
while let Some((key, value)) = cursor.key_value()? {
println!("{:?} => {:?}", key, value);
cursor.prev()?;
}
Swift Examples
// Forward traversal.
let cursor = try db.firstCursor()
while let (key, value) = try cursor.keyValue() {
print("\(key) => \(value)")
try cursor.next()
}
// Backward traversal.
let cursor2 = try db.lastCursor()
while let (key, value) = try cursor2.keyValue() {
print("\(key) => \(value)")
try cursor2.previous()
}
Or we can perform range queries on ThetaDB in this way:
Rust Examples
let mut cursor = db.cursor_from_key(b"C")?;
// Enable `let_chains` feature, should add `#![feature(let_chains)]`
// to the crate attributes.
while let Some((key, value)) = cursor.key_value()? && key != b"G" {
println!("{:?} => {:?}", key, value);
cursor.next()?;
}
Swift Examples
let cursor = try db.cursor(key: "C")
while let (key, value) = try cursor.keyValue(), key != "G" {
print("\(key) => \(value)")
try cursor.next()
}
Attention
❗️ Cursor is also a transaction (can be understood as a read-only transaction), so it also follows the transaction considerations mentioned above.
Benchmark
🚧 Yet To Be Measured.
License
ThetaDB is released under the MIT license. See LICENSE for more details.