matrix/authentication-service
1
0
Fork 0
mirror of https://github.com/matrix-org/matrix-authentication-service.git synced 2025-11-20 12:02:22 +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

7
crates/tasks/src/lib.rs
View File

@@ -27,6 +27,7 @@ use tracing::debug;
mod database;
mod email;
mod layers;
#[derive(Clone)]
struct State {
@@ -95,11 +96,11 @@ impl JobContextExt for apalis_core::context::JobContext {
}
#[must_use]
pub fn init(pool: &Pool<Postgres>, mailer: &Mailer) -> Monitor<TokioExecutor> {
pub fn init(name: &str, pool: &Pool<Postgres>, mailer: &Mailer) -> Monitor<TokioExecutor> {
let state = State::new(pool.clone(), SystemClock::default(), mailer.clone());
let monitor = Monitor::new();
let monitor = self::database::register(monitor, &state);
let monitor = self::email::register(monitor, &state);
let monitor = self::database::register(name, monitor, &state);
let monitor = self::email::register(name, monitor, &state);
debug!(?monitor, "workers registered");
monitor
}