From beb68b50515dbce588706d7d77e7fb4e88835c68 Mon Sep 17 00:00:00 2001 From: Brandon Thomas Date: Tue, 6 Jul 2021 02:26:58 -0400 Subject: [PATCH] env var controls --- src/tts_inference_job.rs | 16 ++++++++++++---- 1 file changed, 12 insertions(+), 4 deletions(-) diff --git a/src/tts_inference_job.rs b/src/tts_inference_job.rs index 20178b84ad..a1f330fa50 100644 --- a/src/tts_inference_job.rs +++ b/src/tts_inference_job.rs @@ -226,13 +226,21 @@ async fn main() -> AnyhowResult<()> { let cache_miss_strategizers = { let in_memory_strategizer = CacheMissStrategizer::new( - chrono::Duration::milliseconds(5_000), - chrono::Duration::milliseconds(60_000), + chrono::Duration::milliseconds( + easyenv::get_env_num("MEMORY_MAX_COLD_DURATION_MILLIS", 5_000)?, + ), + chrono::Duration::milliseconds( + easyenv::get_env_num("MEMORY_CACHE_FORGET_DURATION_MILLIS", 60_000)?, + ), ); let on_disk_strategizer = CacheMissStrategizer::new( - chrono::Duration::milliseconds(20_000), - chrono::Duration::milliseconds(120_000), + chrono::Duration::milliseconds( + easyenv::get_env_num("DISK_MAX_COLD_DURATION_MILLIS", 20_000)?, + ), + chrono::Duration::milliseconds( + easyenv::get_env_num("DISK_CACHE_FORGET_DURATION_MILLIS", 120_000)?, + ), ); SyncMultiCacheMissStrategizer::new(