docbox-management 0.12.0

Management core library for managing docbox used by the cli and other tools
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
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115

use crate::{
    database::DatabaseProvider,
    tenant::{
        MigrateTenantsOutcome, TenantTarget,
        get_tenants::get_tenants,
        migrate_tenant_storage::{MigrateTenantStorageError, migrate_tenant_storage},
    },
};
use docbox_core::{
    database::{
        DbErr,
        models::tenant::{Tenant, TenantId},
    },
    storage::StorageLayerFactory,
};
use serde::{Deserialize, Serialize};
use thiserror::Error;

#[derive(Debug, Error)]
pub enum MigrateTenantsStorageError {
    #[error("failed to get tenants: {0}")]
    GetTenants(DbErr),

    #[error(transparent)]
    MigrateTenant(#[from] MigrateTenantStorageError),
}

#[derive(Debug, Clone, Default, Serialize, Deserialize)]
pub struct MigrateTenantsStorageConfig {
    /// Filter to a specific environment
    pub env: Option<String>,
    /// Filter to a specific tenant
    pub tenant_id: Option<TenantId>,
    /// Filter to skip failed migrations and continue
    pub skip_failed: bool,
    /// Specific migrations to run
    pub target_migration_name: Option<String>,
}

#[tracing::instrument(skip(db_provider, storage_factory))]
pub async fn migrate_tenants_storage(
    db_provider: &impl DatabaseProvider,
    storage_factory: &StorageLayerFactory,
    config: MigrateTenantsStorageConfig,
) -> Result<MigrateTenantsOutcome, MigrateTenantsStorageError> {
    let tenants = get_tenants(db_provider)
        .await
        .map_err(MigrateTenantsStorageError::GetTenants)?;

    // Filter to our desired tenants
    let tenants: Vec<Tenant> = tenants
        .into_iter()
        .filter(|tenant| {
            if config.env.as_ref().is_some_and(|env| tenant.env.ne(env)) {
                return false;
            }

            if config
                .tenant_id
                .as_ref()
                .is_some_and(|schema| tenant.id.ne(schema))
            {
                return false;
            }

            true
        })
        .collect();

    let mut applied_tenants = Vec::new();
    let mut failed_tenants = Vec::new();

    for tenant in tenants {
        let result = migrate_tenant_storage(
            db_provider,
            storage_factory,
            &tenant,
            config.target_migration_name.as_deref(),
        )
        .await;
        match result {
            Ok(_) => {
                applied_tenants.push(TenantTarget {
                    env: tenant.env,
                    name: tenant.name,
                    tenant_id: tenant.id,
                });
            }
            Err(error) => {
                failed_tenants.push((
                    error.to_string(),
                    TenantTarget {
                        env: tenant.env,
                        name: tenant.name,
                        tenant_id: tenant.id,
                    },
                ));

                tracing::error!(?error, "failed to apply tenant migration");

                if !config.skip_failed {
                    tracing::debug!(?applied_tenants, "completed migrations");
                    break;
                }

                return Err(MigrateTenantsStorageError::MigrateTenant(error));
            }
        }
    }

    Ok(MigrateTenantsOutcome {
        applied_tenants,
        failed_tenants,
    })
}