brokerage-db 0.2.1

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
53
54
55
56
57
58
59
60
61
62
63

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

pub struct Migration002 {}

const SECURITIES_UNIQUE_INDEX_NAME: &str = "securities_unique_idx";
const SECURITIES_IBKR_CONID_INDEX_NAME: &str = "securities_conid_idx";

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

        //
        // Create initial security (stock, bond, option, etc.) collection
        //
        db.create_collection(Security::COLLECTION_NAME).await?;

        let collection = db.collection::<Security>(Security::COLLECTION_NAME);
        let indexes = vec![
            IndexModel::builder()
                .keys(doc! { "ticker": 1, "listing_exchange": 1 })
                .options(
                    IndexOptions::builder()
                        .name(Some(SECURITIES_UNIQUE_INDEX_NAME.to_owned()))
                        .unique(true)
                        .build(),
                )
                .build(),
            IndexModel::builder()
                .keys(doc! { "ibkr_conid": 1 })
                .options(
                    IndexOptions::builder()
                        .name(Some(SECURITIES_IBKR_CONID_INDEX_NAME.to_owned()))
                        .build(),
                )
                .build(),
        ];

        collection.create_indexes(indexes).await?;

        Ok(())
    }

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

        collection
            .drop_index(SECURITIES_IBKR_CONID_INDEX_NAME)
            .await?;

        collection.drop_index(SECURITIES_UNIQUE_INDEX_NAME).await?;

        collection.drop().await?;

        Ok(())
    }
}