brokerage-db 0.2.4

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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52

use crate::account::BrokerageAccount;
use anyhow::Result;
use async_trait::async_trait;
use bson::doc;
use mongodb::{IndexModel, options::IndexOptions};
use tfiala_mongodb_migrator::migrator::Env;

pub struct Migration001 {}

const BROKERAGE_ACCOUNT_UNIQUE_INDEX_NAME: &str = "brokerage_account_unique_idx";

#[async_trait]
impl tfiala_mongodb_migrator::migration::Migration for Migration001 {
    async fn up(&self, env: Env) -> Result<()> {
        let db = env.db.unwrap();

        //
        // Create initial brokerage-accounts indexes.
        //
        db.create_collection(BrokerageAccount::COLLECTION_NAME)
            .await?;
        let collection = db.collection::<BrokerageAccount>(BrokerageAccount::COLLECTION_NAME);
        let indexes = vec![
            IndexModel::builder()
                .keys(doc! { "brokerage_id": 1, "account_id": 1 })
                .options(
                    IndexOptions::builder()
                        .name(Some(BROKERAGE_ACCOUNT_UNIQUE_INDEX_NAME.to_owned()))
                        .unique(true)
                        .build(),
                )
                .build(),
        ];

        collection.create_indexes(indexes).await?;

        Ok(())
    }

    async fn down(&self, env: Env) -> Result<()> {
        let db = env.db.unwrap();
        let collection = db.collection::<BrokerageAccount>(BrokerageAccount::COLLECTION_NAME);

        collection
            .drop_index(BROKERAGE_ACCOUNT_UNIQUE_INDEX_NAME)
            .await?;

        collection.drop().await?;

        Ok(())
    }
}