ef61b193c4
- pool/mod.rs: pass shared http_client Arc to HookWorker - worker.rs: remove double-locking (sync() manages its own lock), await all webhook handles before returning, share http_client, hoist namespace query out of loop - redis.rs: atomic NAK via Lua script (LREM + LPUSH in one eval) - sync/lock.rs: increase LOCK_TTL from 60s to 300s for large repos - sync/mod.rs: split sync/sync_work, fsck_only/fsck_work, gc_only/gc_work so callers can choose locked vs lock-free path; run_gc + sync_skills outside the DB transaction - hook/mod.rs: remove unused http field from HookService - ssh/mod.rs, http/mod.rs: remove unused HookService/http imports
44 lines
1.1 KiB
Rust
44 lines
1.1 KiB
Rust
pub mod redis;
|
|
pub mod types;
|
|
pub mod worker;
|
|
|
|
pub use redis::RedisConsumer;
|
|
pub use types::{HookTask, PoolConfig, TaskType};
|
|
pub use worker::HookWorker;
|
|
|
|
use db::cache::AppCache;
|
|
use db::database::AppDatabase;
|
|
use deadpool_redis::cluster::Pool as RedisPool;
|
|
use slog::Logger;
|
|
use std::sync::Arc;
|
|
use tokio_util::sync::CancellationToken;
|
|
|
|
/// Start the hook worker background task.
|
|
/// Returns a handle to the cancellation token so the caller can shut it down.
|
|
pub fn start_worker(
|
|
db: AppDatabase,
|
|
cache: AppCache,
|
|
redis_pool: RedisPool,
|
|
logger: Logger,
|
|
config: PoolConfig,
|
|
) -> CancellationToken {
|
|
let consumer = RedisConsumer::new(
|
|
redis_pool.clone(),
|
|
config.redis_list_prefix.clone(),
|
|
config.redis_block_timeout_secs,
|
|
logger.clone(),
|
|
);
|
|
|
|
let http_client = Arc::new(reqwest::Client::new());
|
|
let worker = HookWorker::new(db, cache, logger, consumer, http_client);
|
|
|
|
let cancel = CancellationToken::new();
|
|
let cancel_clone = cancel.clone();
|
|
|
|
tokio::spawn(async move {
|
|
worker.run(cancel_clone).await;
|
|
});
|
|
|
|
cancel
|
|
}
|