How to get the db handle in multidb mode for Task

[Yihen-Liu]

now i work under the loco version 0.16.3, I want to get the db handle of one multidb, I try, but failed.
my deployment.yaml as follow:

initializers:
  multi_db:
    chains_db:
      uri: mysql://root:123456@localhost:3306/evm_chains
      enable_logging: true
      connect_timeout: 500
      idle_timeout: 500
      min_connections: 1
      max_connections: 1
      auto_migrate: true
      dangerously_truncate: false
      dangerously_recreate: false
    risk_mgr_db:
      uri: mysql://root:123456@localhost:3306/risk_mgr
      enable_logging: true
      connect_timeout: 500
      idle_timeout: 500
      min_connections: 1
      max_connections: 1
      auto_migrate: true
      dangerously_truncate: false
      dangerously_recreate: false

Task implement as follow:

#[async_trait]
impl Task for TelegramNotifier {
    fn task(&self) -> TaskInfo {
        TaskInfo {
            name: "telegram_notifier".to_string(),
            detail: "run telegram notifier task".to_string(),
        }
    }

    async fn run(&self, ctx: &AppContext, _vars: &task::Vars) -> Result<()> {
        

        Ok(())
    }
}

to be honest, ctx only contains db, there is no multi_db, how to get the chain_db set in yaml file?

[Yihen-Liu]

BTW, I have registered the loco_rs::initializers::multi_db::MultiDbInitializer in app.rs

    async fn initializers(_ctx: &AppContext) -> Result<Vec<Box<dyn Initializer>>> {
        let initializers: Vec<Box<dyn Initializer>> = vec![Box::new(
            //loco_rs::initializers::extra_db::ExtraDbInitializer,
            loco_rs::initializers::multi_db::MultiDbInitializer,
        )];

        Ok(initializers)
    }

[kaplanelad]

the controller, you need to define the MultiDb extension as shown in the following example:

use sea_orm::DatabaseConnection;
use axum::{response::IntoResponse, Extension};
use loco_rs::db::MultiDb;

pub async fn list(
    State(ctx): State<AppContext>,
    Extension(multi_db): Extension<MultiDb>,
) -> Result<impl IntoResponse> {
  let third_db = multi_db.get("third_db")?;
  let res = Entity::find().all(third_db).await;
}

docs: https://loco.rs/docs/the-app/models/#multi-db-multi-tenant