Skip to content

Commit

Permalink
init idens
Browse files Browse the repository at this point in the history
  • Loading branch information
Eason0729 committed Dec 7, 2023
1 parent 43dc1c0 commit 074a660
Show file tree
Hide file tree
Showing 6 changed files with 226 additions and 1 deletion.
2 changes: 1 addition & 1 deletion backend/entity/src/test.rs
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ use serde::{Deserialize, Serialize};
use crate::problem;

#[derive(Clone, Debug, PartialEq, Eq, DeriveEntityModel, Deserialize, Serialize)]
#[sea_orm(table_name = "testcase")]
#[sea_orm(table_name = "test")]
pub struct Model {
#[sea_orm(primary_key, auto_increment = true)]
pub id: i32,
Expand Down
22 changes: 22 additions & 0 deletions backend/migration/Cargo.toml
Original file line number Diff line number Diff line change
@@ -0,0 +1,22 @@
[package]
name = "migration"
version = "0.1.0"
edition = "2021"
publish = false

[lib]
name = "migration"
path = "src/lib.rs"

[dependencies]
async-std = { version = "1", features = ["attributes", "tokio1"] }

[dependencies.sea-orm-migration]
version = "0.11.0"
features = [
# Enable at least one `ASYNC_RUNTIME` and `DATABASE_DRIVER` feature if you want to run migration via CLI.
# View the list of supported features at https://www.sea-ql.org/SeaORM/docs/install-and-config/database-and-async-runtime.
# e.g.
# "runtime-tokio-rustls", # `ASYNC_RUNTIME` feature
# "sqlx-postgres", # `DATABASE_DRIVER` feature
]
41 changes: 41 additions & 0 deletions backend/migration/README.md
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
# Running Migrator CLI

- Generate a new migration file
```sh
cargo run -- migrate 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
```
12 changes: 12 additions & 0 deletions backend/migration/src/lib.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,12 @@
pub use sea_orm_migration::prelude::*;

mod m20231207_000001_create_table;

pub struct Migrator;

#[async_trait::async_trait]
impl MigratorTrait for Migrator {
fn migrations() -> Vec<Box<dyn MigrationTrait>> {
vec![Box::new(m20231207_000001_create_table::Migration)]
}
}
144 changes: 144 additions & 0 deletions backend/migration/src/m20231207_000001_create_table.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,144 @@
use sea_orm_migration::prelude::*;

#[derive(Iden)]
enum Announcement{
Id,
Title,
Content,
CreateAt,
UpdateAt,
}
enum Contest {
Id,
Hoster,
Begin,
End,
Title,
Content,
Tags,
Password,
CreateAt,
UpdateAt,
Public,
}
enum Education{
Id,
ProblemId,
UserId,
Tags,
Title,
Content,
}
#[derive(Iden)]
enum Post {
Table,
Id,
Title,
Text,
}
#[derive(Iden)]
enum Problem
{
id,
user_id,
contest_id,
accept_count,
submit_count,
ac_rate,
memory,
time,
difficulty,
public,
tags,
title,
content,
create_at,
UpdateAt,
MatchRule,
}

enum Submit{
id,
user_id,
problem_id,
upload_at,
time,
accuracy,
committed,
lang,
code,
memory,
pass_case,
status,
accept,
score,
}
enum Test{
id,
user_id,
problem_id,
input,
output,
score,
}
enum Token{
id,
user_id,
rand,
permission,
expiry,
}
enum User {
id,
permission,
score,
username,
password,
create_at,
}
enum UserContest{
id,
user_id,
contest_id,
score,
}

#[derive(DeriveMigrationName)]
pub struct Migration;

#[async_trait::async_trait]
impl MigrationTrait for Migration {
async fn up(&self, manager: &SchemaManager) -> Result<(), DbErr> {
// Replace the sample below with your own migration scripts
todo!();

manager
.create_table(
Table::create()
.table(Post::Table)
.if_not_exists()
.col(
ColumnDef::new(Post::Id)
.integer()
.not_null()
.auto_increment()
.primary_key(),
)
.col(ColumnDef::new(Post::Title).string().not_null())
.col(ColumnDef::new(Post::Text).string().not_null())
.to_owned(),
)
.await
}

async fn down(&self, manager: &SchemaManager) -> Result<(), DbErr> {
// Replace the sample below with your own migration scripts
todo!();

manager
.drop_table(Table::drop().table(Post::Table).to_owned())
.await
}
}

/// Learn more at https://docs.rs/sea-query#iden
6 changes: 6 additions & 0 deletions backend/migration/src/main.rs
Original file line number Diff line number Diff line change
@@ -0,0 +1,6 @@
use sea_orm_migration::prelude::*;

#[async_std::main]
async fn main() {
cli::run_cli(migration::Migrator).await;
}

0 comments on commit 074a660

Please sign in to comment.