brokerage-db 0.2.7

Database management for trader brokerage data with a MongoDB backend.
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36

use anyhow::Result;
use mongodb::Database;
use tfiala_mongodb_migrator::{migration::Migration, migrator::default::DefaultMigrator};

mod v001_add_accounts;
mod v002_add_security;
mod v003_add_trade_executions;
mod v004_add_eod_summary;

fn get_migrations() -> Vec<Box<dyn Migration>> {
    vec![
        Box::new(v001_add_accounts::Migration001 {}),
        Box::new(v002_add_security::Migration002 {}),
        Box::new(v003_add_trade_executions::Migration003 {}),
        Box::new(v004_add_eod_summary::Migration004 {}),
    ]
}

pub async fn run_migrations(db: &Database) -> Result<()> {
    DefaultMigrator::new()
        .with_conn(db.clone())
        .with_migrations_vec(get_migrations())
        .up()
        .await?;
    Ok(())
}

pub async fn run_down_migrations(db: &Database) -> Result<()> {
    DefaultMigrator::new()
        .with_conn(db.clone())
        .with_migrations_vec(get_migrations())
        .down()
        .await?;

    Ok(())
}