bin+lib steeldb

A simple database built from scratch in Rust

4 releases

0.1.4 Jan 1, 2024
0.1.2 Jan 1, 2024
0.1.1 Jan 1, 2024
0.1.0 Dec 30, 2023

Apache-2.0

58KB
942 lines

SteelDB

This is a study repository. This is mostly for personal use. Building a Database from scratch in Rust. Why not? :)

Source code: https://github.com/paolorechia/steeldb/

Current version documentation: latest

*Database: https://docs.rs/steeldb/latest/steeldb *Parser: https://docs.rs/steeldb-parser/latest/steeldb_parser/

Architecture

image

How to use this version

Should be as simple:

cargo add steeldb
cargo run

This should start a REPL:

------------------------------------------------
|                                               |
|   SteelDB                                     |
|   version: 0.1.0                              |
|                                               |
------------------------------------------------

Type 'exit;' to leave this shell
Current supported commands: [select]

>>

The only implemented clause is select, which selects columns of a previously constructed table. For example:

>> select name, annual_salary;
|---------------------------------|
|    name   |    annual_salary    |
|---------------------------------|
| John Man  |       60000         |
|   Lenon   |       200000        |
|   Mary    |      3012000        |
|---------------------------------|
>>

Commands should always add with a ;.

If you simply try the command above, you will instead see:

>> select name;
"Os { code: 2, kind: NotFound, message: \"No such file or directory\" }"


<------------------ COMMAND FAILED ------------------>
"TableNotFound"

<---------------------------------------------------->

>> 

This is because the table must be pre-created. You can either create one using cargo test and copying it, or copying and pasting this into the file .steeldb/data/test_table.columnar:

TABLE COLUMNAR FORMAT HEADER
Field name: final_grade; Type: f32; Number of elements: 3
4.0
3.2
5
Field name: name; Type: String; Number of elements: 3
John Man
Lenon
Mary
Field name: annual_salary; Type: i32; Number of elements: 3
60000
200000
3012000

Columnar Format

As you can see, the table format is very naive and verbose. It stores data in ASCII. It's not meant to be efficient and will probably be replaced in the future.

More info

  1. https://cstack.github.io/db_tutorial/parts/part1.html
  2. https://www.sqlite.org/arch.html
  3. https://build-your-own.org/database/

Roadmap

This is not a binding roadmap.

What do we need to build a Database like SQLite from scratch?

  1. A REPL shell
  2. A tokenizer
  3. A parser
  4. A code generator
  5. A virtual machine that interprets the generated code
  6. A B+ Tree
  7. Pager
  8. OS Interface

For the first iteration: the bare bones (v0.1.0)

We can simplify some components in the first iteration, so we have first a working end-to-end system. We can then tweak the individual components to have more capabilities.

Here are some simplifications we can do for our first iteration:

  1. Support only a subset of the SQL Syntax, for instance, start with only insert and select operations.
  2. Do not implement a B+ tree in the first iteration. Instead, use a vector or list of structs.
  3. Do not handle Pager in first iteration.
  4. Keep the database persisted into a single file.
  5. Use a single statically defined table.

Our roadmap for the first iteration might end up looking like this:

  1. A REPL shell [x]
  2. A tokenizer [x]
  3. A parser [x]
  4. Add support for the SELECT clause [x]
  5. A code generator [x]
  6. A virtual machine that interprets the generated code [x]
  7. A table struct that stores data in HashMap of Vectors [x]
  8. A hardcoded table for testing [x]
  9. Proper error propagation [x]
  10. Pretty printing of table in REPL [x]
  11. A serialization / desserialization method to write/read data from file [x]
  12. Clean up [x]
    • Handle select columns properly in read method [x]
    • Test that everything is working as expected [x]
    • Tag v1.0 [x]

Second iteration: making it usable (v0.2.0)

  1. Add proper documentation to project []
  2. Add another API besides the REPL to query the database []
    • This can be either a traditional TCP or a HTTP server. It should be as simple as possible, and just receive a string of the SQL command
    • Make REPL support both backends: Standalone process or network server
  3. Adds proper logging strategy to the server []
  4. Add configuration file []
  5. Add create table command []
  6. Add drop table command []
  7. Add alter table command []
  8. Multiple tables query support (add FROM clause support) []
  9. Support filters (add basic WHERE clause support) []
  10. Add proper documentation to project []
  11. Update Cargo.toml

Third iteration: making it scalable (v0.3.0)

  1. Handle concurrency: needs research on approaches to use []
  2. Implement B+ or similar algorithm.
    • Ideally keep support for current columnar support []
    • Support both in-memory and persistent
  3. Adds caching (or pagination) []
  4. Support for transactions []

Fourth iteration: making it useful (v0.4.0)

  1. Implement inner join []
  2. Implement left / right join []
  3. Implement outer join []
  4. Implement nested operations, including WHERE IN (SELECT) []
  5. Implement aggregations []

Fifth iteration: making it time-aware (v0.5.0)

  1. Implement advanced SQL features
  • Window []
  • Having []
  1. Add date and timestamp types []
  2. Implement more SQL functions []

Sixth iteration: making it complete (v0.6.0)

  1. Anything important missing in SQL standards

Seventh iteration: making it compatible (v0.7.0)

  1. Implement the Spark.SQL API in Rust

Eighth iteration: making it attractive (v0.8.0)

  1. Build a SDK in Rust to use it

Nineth iteration: making it snaky (v0.9.0)

  1. Build Python bindings for the Rust SDK

Tenth iteration: placeholder (v1.0.0)

If we reached this point, we actually have an impressive system :)

Knowledge base

How do we build each of these components in Rust?

REPL Shell

This is pobably the simplest component. We just need to implement a CLI shell that reads lines of input until the command end symbol is presented (like ';'). It then forwards the parsed string into the next layers of our system, and returns the result.

A Tokenizer and A Parser

One should be able to generate them using https://github.com/lalrpop/lalrpop. Tutorial: http://lalrpop.github.io/lalrpop/tutorial/001_adding_lalrpop.html

A Code Generator

This should be somehow integrated into the parser. It will become clearer once the first parser is built, how this can be written (if possible, avoiding too much coupling).

Virtual Machine

Probably the easiest way is to implement it as a stack, so it can handled nested commands. For each parsed command, we push it to the stack, and the virtual machine pop it, executing it. This works assuming we parse the commands in a depth-first way, e.g., the most inner command is always parsed first, and it's result is available to the next command execution. This should happen naturally using an auto-generated lexer/parser with lalrpop, as long as the grammar is correctly defined.

To keep it simple, however, we should start not supporting nested commands. We can still have the stack in place and ready for extension in the future.

B+ Tree

The B+ Tree is wide-known algorithm, and is described in several places, for instance: https://en.wikipedia.org/wiki/B%2B_tree This will be skipped until much later! The goal is to first get an usable system/product that supports concurrency and possibly exposes an (HTTP) API to query data. This means things like transactions have to be possibly implemented first.

Dependencies

~2.6–5MB
~83K SLoC