feat: add migration for transaction, category, and tag tables with foreign key constraints

This commit is contained in:
GW_MC
2026-05-28 04:00:01 +00:00
parent 1d5e24bdfb
commit ea16cc2d55
9 changed files with 517 additions and 2 deletions

View File

@@ -1,6 +1,7 @@
pub use sea_orm_migration::prelude::*;
mod m20260526_062833_create_account_table;
mod m20260526_110957_create_transcations_table;
pub struct Migrator;
@@ -8,8 +9,8 @@ pub struct Migrator;
impl MigratorTrait for Migrator {
fn migrations() -> Vec<Box<dyn MigrationTrait>> {
vec![
//
Box::new(m20260526_062833_create_account_table::Migration),
Box::new(m20260526_110957_create_transcations_table::Migration),
]
}
}

View File

@@ -0,0 +1,253 @@
use sea_orm_migration::{prelude::*, schema::*};
#[derive(DeriveMigrationName)]
pub struct Migration;
#[async_trait::async_trait]
impl MigrationTrait for Migration {
async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
// Create tag table
manager
.create_table(
TableCreateStatement::new()
.table("tag")
.if_not_exists()
.col(pk_auto("id").not_null())
.col(string("name").not_null().unique_key())
.col(string("color_code").not_null())
.col(string("created_at").not_null())
.col(string("updated_at").not_null())
.to_owned(),
)
.await?;
// Create indexes
manager
.create_index(
Index::create()
.name("idx-tag-name")
.table("tag")
.col("name")
.to_owned(),
)
.await?;
// Create category table
manager
.create_table(
TableCreateStatement::new()
.table("category")
.if_not_exists()
.col(pk_auto("id").not_null())
.col(
integer("parent_id")
.null()
.comment("the parent category id, can be null for root categories"),
)
.col(string("name").not_null())
.col(string("color_code").not_null())
.col(string("created_at").not_null())
.col(string("updated_at").not_null())
.foreign_key(
ForeignKey::create()
.name("fk-category-parent_id")
.from("category", "parent_id")
.to("category", "id")
.on_delete(ForeignKeyAction::NoAction)
.on_update(ForeignKeyAction::Restrict),
)
.to_owned(),
)
.await?;
manager
.create_index(
Index::create()
.name("idx-category-name_parent_id")
.table("category")
.col("name")
.col("parent_id")
.to_owned(),
)
.await?;
manager
.create_index(
Index::create()
.name("idx-category-name")
.table("category")
.col("name")
.to_owned(),
)
.await?;
manager
.create_index(
Index::create()
.name("idx-category-parent_id")
.table("category")
.col("parent_id")
.to_owned(),
)
.await?;
// Create transaction table
manager
.create_table(
TableCreateStatement::new()
.table("transaction")
.if_not_exists()
.col(pk_auto("id").not_null())
// for debt transaction, amount is negative; for credit transaction, amount is positive. The actual effect on the account balance depends on the account type. For example, a $100 income transaction will have amount = 100, transaction_type = STANDARD, and it will increase the balance of an asset account but decrease the balance of a debt account. A $50 expense transaction will have amount = -50, transaction_type = STANDARD, and it will decrease the balance of an asset account but increase the balance of a debt account. A $200 transfer from account A to account B will have two transactions: one with amount = -200, transaction_type = STANDARD for account A, and another with amount = 200, transaction_type = STANDARD for account B. An opening balance transaction will have transaction_type = OPENING_BALANCE, and its amount can be positive or negative depending on the initial balance of the account.
.col(string("amount").not_null().comment("the amount of this transaction in the original currency. Positive for assets increase or debts decrease, negative for assets decrease or debts increase"))
.col(string("transaction_type").not_null().comment("One of: STANDARD, RECONCILIATION, OPENING_BALANCE"))
.col(string("currency_code").not_null())
.col(string("exchange_rate").null())
.col(string("transaction_date").not_null())
.col(string("metadata").null())
.col(string("created_at").not_null())
.col(string("updated_at").not_null())
// fks
.col(integer("account_id").not_null().comment("the account this transaction belongs to"))
// default null
.col(integer("from_account_id").null().comment("the account this transaction is from, can be null for income and expense transactions"))
// transaction category
.col(integer("category_id").null().comment("the category of this transaction, can be null"))
// Add foreign key constraint to account
.foreign_key(
ForeignKey::create()
.name("fk-transaction-account")
.from("transaction", "account_id")
.to("account", "id")
.on_delete(ForeignKeyAction::NoAction)
.on_update(ForeignKeyAction::Restrict),
)
.foreign_key(
ForeignKey::create()
.name("fk-transaction-from_account")
.from("transaction", "from_account_id")
.to("account", "id")
.on_delete(ForeignKeyAction::NoAction)
.on_update(ForeignKeyAction::Restrict),
)
.foreign_key(
ForeignKey::create()
.name("fk-transaction-category")
.from("transaction", "category_id")
.to("category", "id")
.on_delete(ForeignKeyAction::NoAction)
.on_update(ForeignKeyAction::Restrict),
)
.to_owned(),
)
.await?;
// Create transaction_tag table
manager
.create_table(
TableCreateStatement::new()
.table("transaction_tag")
.if_not_exists()
.col(integer("transaction_id").not_null())
.col(integer("tag_id").not_null())
.primary_key(
Index::create()
.name("pk-transaction_tag")
.col("transaction_id")
.col("tag_id"),
)
.foreign_key(
ForeignKey::create()
.name("fk-transaction_tag-transaction")
.from("transaction_tag", "transaction_id")
.to("transaction", "id")
.on_delete(ForeignKeyAction::NoAction)
.on_update(ForeignKeyAction::Restrict),
)
.foreign_key(
ForeignKey::create()
.name("fk-transaction_tag-tag")
.from("transaction_tag", "tag_id")
.to("tag", "id")
.on_delete(ForeignKeyAction::NoAction)
.on_update(ForeignKeyAction::Restrict),
)
.to_owned(),
)
.await?;
manager
.create_index(
Index::create()
.name("idx-transaction-account_id")
.table("transaction")
.col("account_id")
.to_owned(),
)
.await?;
manager
.create_index(
Index::create()
.name("idx-transaction-from_account_id")
.table("transaction")
.col("from_account_id")
.to_owned(),
)
.await?;
manager
.create_index(
Index::create()
.name("idx-transaction-category_id")
.table("transaction")
.col("category_id")
.to_owned(),
)
.await?;
manager
.create_index(
Index::create()
.name("idx-transaction-transaction_date")
.table("transaction")
.col("transaction_date")
.to_owned(),
)
.await?;
manager
.create_index(
Index::create()
.name("idx-transaction-transaction_type")
.table("transaction")
.col("transaction_type")
.to_owned(),
)
.await?;
Ok(())
}
async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
manager
.drop_table(
TableDropStatement::new()
.table("transaction_tag")
.to_owned(),
)
.await?;
manager
.drop_table(TableDropStatement::new().table("transaction").to_owned())
.await?;
manager
.drop_table(TableDropStatement::new().table("category").to_owned())
.await?;
manager
.drop_table(TableDropStatement::new().table("tag").to_owned())
.await?;
Ok(())
}
}