matrix/authentication-service
1
0
Fork 0
mirror of https://github.com/matrix-org/matrix-authentication-service.git synced 2025-08-12 02:22:41 +03:00
Code Activity

Extract the job tracing span logic to a layer

Browse Source
This commit is contained in:
Quentin Gliech
2023-04-03 17:36:15 +02:00
parent f4fff72b22
commit 169d7ce6a2
9 changed files with 174 additions and 87 deletions
Download Patch File Download Diff File Expand all files Collapse all files

14
crates/cli/src/commands/server.rs
View File

@@ -22,6 +22,10 @@ use mas_handlers::{AppState, HttpClientFactory, MatrixHomeserver};
use mas_listener::{server::Server, shutdown::ShutdownStream};
use mas_router::UrlBuilder;
use mas_storage_pg::MIGRATOR;
use rand::{
distributions::{Alphanumeric, DistString},
thread_rng,
};
use tokio::signal::unix::SignalKind;
use tracing::{info, info_span, warn, Instrument};
@@ -52,7 +56,7 @@ impl Options {
let config: RootConfig = root.load_config()?;
// Connect to the database
info!("Conntecting to the database");
info!("Connecting to the database");
let pool = database_from_config(&config.database).await?;
if self.migrate {
@@ -87,8 +91,12 @@ impl Options {
let mailer = mailer_from_config(&config.email, &templates).await?;
mailer.test_connection().await?;
info!("Starting task worker");
let monitor = mas_tasks::init(&pool, &mailer);
#[allow(clippy::disallowed_methods)]
let mut rng = thread_rng();
let worker_name = Alphanumeric.sample_string(&mut rng, 10);
info!(worker_name, "Starting task worker");
let monitor = mas_tasks::init(&worker_name, &pool, &mailer);
// TODO: grab the handle
tokio::spawn(monitor.run());
}