diff --git a/src/lib.rs b/src/lib.rs index 6108be6..9a02fe5 100644 --- a/src/lib.rs +++ b/src/lib.rs @@ -44,6 +44,11 @@ pub mod run { // Debug quotes fall back to the default configuration FsFilterSourceManager::default() }); + let stdout = if !is_debug { + None + } else { + Some(StdoutSink::spawn(StdoutSink)) + }; let use_redis = std::env::var("USE_REDIS").unwrap_or("0".to_string()) == "1"; if use_redis { @@ -58,10 +63,10 @@ pub mod run { let queue = RedisQueueStorage::spawn(RedisQueueStorage::new(con, DEFAULT_KEY.to_string())); - run_cycle(schedule, source, queue, bsky).await + run_cycle(schedule, source, queue, stdout, bsky).await } else { let queue = MemoryQueueStorage::spawn(MemoryQueueStorage::new()); - run_cycle(schedule, source, queue, bsky).await + run_cycle(schedule, source, queue, stdout, bsky).await } } @@ -69,12 +74,10 @@ pub mod run { schedule: &str, source: ActorRef, queue: ActorRef, + stdout: Option>, bsky: Option>, ) -> Result<(), Box> { - let sink = { - let stdout = StdoutSink::spawn(StdoutSink); - SinkManager::spawn(SinkManager::new(Some(stdout), bsky)) - }; + let sink = { SinkManager::spawn(SinkManager::new(stdout, bsky)) }; let cycle = { QuoteCycle::spawn(QuoteCycle::with_thread_rng(source, queue)) };