Migrate from rusqlite to seaorm

Author: Swatinem

Cargo.lock

@@ -1,6 +1,13 @@
 [workspace]
 resolver = "3"
-members = ["client", "proto-codegen", "server", "service", "stresstest"]
+members = [
+    "client",
+    "proto-codegen",
+    "server",
+    "service",
+    "stresstest",
+    "migration",
+]
 default-members = ["server"]
 
 [profile.dev]

Cargo.toml

@@ -0,0 +1,12 @@
+[package]
+name = "migration"
+version = "0.1.0"
+edition = "2024"
+publish = false
+
+[dependencies]
+async-std = { version = "1", features = ["attributes", "tokio1"] }
+sea-orm-migration = { version = "1.1.0", features = [
+  "sqlx-postgres",
+  "runtime-tokio-rustls",
+] }

migration/Cargo.toml

@@ -0,0 +1,41 @@
+# Running Migrator CLI
+
+- Generate a new migration file
+    ```sh
+    cargo run -- generate MIGRATION_NAME
+    ```
+- Apply all pending migrations
+    ```sh
+    cargo run
+    ```
+    ```sh
+    cargo run -- up
+    ```
+- Apply first 10 pending migrations
+    ```sh
+    cargo run -- up -n 10
+    ```
+- Rollback last applied migrations
+    ```sh
+    cargo run -- down
+    ```
+- Rollback last 10 applied migrations
+    ```sh
+    cargo run -- down -n 10
+    ```
+- Drop all tables from the database, then reapply all migrations
+    ```sh
+    cargo run -- fresh
+    ```
+- Rollback all applied migrations, then reapply all migrations
+    ```sh
+    cargo run -- refresh
+    ```
+- Rollback all applied migrations
+    ```sh
+    cargo run -- reset
+    ```
+- Check the status of all migrations
+    ```sh
+    cargo run -- status
+    ```

migration/README.md

@@ -0,0 +1,12 @@
+pub use sea_orm_migration::prelude::*;
+
+mod m20220101_000001_create_table;
+
+pub struct Migrator;
+
+#[async_trait::async_trait]
+impl MigratorTrait for Migrator {
+    fn migrations() -> Vec<Box<dyn MigrationTrait>> {
+        vec![Box::new(m20220101_000001_create_table::Migration)]
+    }
+}

migration/src/lib.rs

@@ -0,0 +1,64 @@
+use sea_orm_migration::{prelude::*, schema::*};
+
+#[derive(DeriveMigrationName)]
+pub struct Migration;
+
+#[derive(DeriveIden)]
+enum Part {
+    Table,
+    Id,
+    Compression,
+    #[allow(clippy::enum_variant_names)]
+    PartSize,
+    CompressedSize,
+    SegmentId,
+    SegmentOffset,
+}
+
+#[derive(DeriveIden)]
+enum Blob {
+    Table,
+    Id,
+    Parts,
+}
+
+#[async_trait::async_trait]
+impl MigrationTrait for Migration {
+    async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
+        manager
+            .create_table(
+                Table::create()
+                    .table(Part::Table)
+                    .if_not_exists()
+                    .col(big_integer(Part::Id).auto_increment().primary_key())
+                    .col(unsigned(Part::PartSize))
+                    .col(small_unsigned(Part::Compression))
+                    .col(unsigned(Part::CompressedSize))
+                    .col(uuid(Part::SegmentId))
+                    .col(unsigned(Part::SegmentOffset))
+                    .to_owned(),
+            )
+            .await?;
+
+        manager
+            .create_table(
+                Table::create()
+                    .table(Blob::Table)
+                    .if_not_exists()
+                    .col(string(Blob::Id).primary_key())
+                    .col(array(Blob::Parts, ColumnType::BigUnsigned))
+                    .to_owned(),
+            )
+            .await
+    }
+
+    async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
+        manager
+            .drop_table(Table::drop().table(Part::Table).to_owned())
+            .await?;
+
+        manager
+            .drop_table(Table::drop().table(Blob::Table).to_owned())
+            .await
+    }
+}

migration/src/m20220101_000001_create_table.rs

@@ -0,0 +1,6 @@
+use sea_orm_migration::prelude::*;
+
+#[async_std::main]
+async fn main() {
+    cli::run_cli(migration::Migrator).await;
+}

migration/src/main.rs

@@ -4,9 +4,16 @@ version = "0.1.0"
 edition = "2024"
 
 [dependencies]
+anyhow = "1.0.98"
 proto-codegen = { path = "../proto-codegen" }
-rusqlite = "0.36.0"
+migration = { path = "../migration" }
+sea-orm = { version = "1.1.0", features = [
+    "sqlx-postgres",
+    "runtime-tokio-rustls",
+    "macros",
+] }
 uuid = { version = "1.17.0", features = ["v4"] }
+tokio = { version = "1.46.0", features = ["full"] }
 
 [dev-dependencies]
 tempfile = "3.20.0"

service/Cargo.toml

@@ -0,0 +1,16 @@
+//! `SeaORM` Entity, @generated by sea-orm-codegen 1.1.13
+
+use sea_orm::entity::prelude::*;
+
+#[derive(Clone, Debug, PartialEq, DeriveEntityModel, Eq)]
+#[sea_orm(table_name = "blob")]
+pub struct Model {
+    #[sea_orm(primary_key, auto_increment = false)]
+    pub id: String,
+    pub parts: Vec<i64>,
+}
+
+#[derive(Copy, Clone, Debug, EnumIter, DeriveRelation)]
+pub enum Relation {}
+
+impl ActiveModelBehavior for ActiveModel {}