85 lines
2.3 KiB
Rust
85 lines
2.3 KiB
Rust
use serde::{Deserialize, Serialize};
|
|
|
|
use crate::AppConfig;
|
|
|
|
#[derive(Debug, Clone, Serialize, Deserialize)]
|
|
pub struct PoolConfig {
|
|
pub max_concurrent: usize,
|
|
pub cpu_threshold: f32,
|
|
pub redis_list_prefix: String,
|
|
pub redis_log_channel: String,
|
|
pub redis_block_timeout_secs: u64,
|
|
pub redis_max_retries: usize,
|
|
pub worker_id: String,
|
|
}
|
|
|
|
impl PoolConfig {
|
|
pub fn from_env(config: &AppConfig) -> Self {
|
|
let max_concurrent = config
|
|
.env
|
|
.get("HOOK_POOL_MAX_CONCURRENT")
|
|
.and_then(|v| v.parse().ok())
|
|
.unwrap_or_else(num_cpus::get);
|
|
|
|
let cpu_threshold = config
|
|
.env
|
|
.get("HOOK_POOL_CPU_THRESHOLD")
|
|
.and_then(|v| v.parse().ok())
|
|
.unwrap_or(80.0);
|
|
|
|
let redis_list_prefix = config
|
|
.env
|
|
.get("HOOK_POOL_REDIS_LIST_PREFIX")
|
|
.cloned()
|
|
.unwrap_or_else(|| "{hook}".to_string());
|
|
|
|
let redis_log_channel = config
|
|
.env
|
|
.get("HOOK_POOL_REDIS_LOG_CHANNEL")
|
|
.cloned()
|
|
.unwrap_or_else(|| "hook:logs".to_string());
|
|
|
|
let redis_block_timeout_secs = config
|
|
.env
|
|
.get("HOOK_POOL_REDIS_BLOCK_TIMEOUT")
|
|
.and_then(|v| v.parse().ok())
|
|
.unwrap_or(5);
|
|
|
|
let redis_max_retries = config
|
|
.env
|
|
.get("HOOK_POOL_REDIS_MAX_RETRIES")
|
|
.and_then(|v| v.parse().ok())
|
|
.unwrap_or(3);
|
|
|
|
let worker_id = config
|
|
.env
|
|
.get("HOOK_POOL_WORKER_ID")
|
|
.cloned()
|
|
.unwrap_or_else(|| uuid::Uuid::new_v4().to_string());
|
|
|
|
Self {
|
|
max_concurrent,
|
|
cpu_threshold,
|
|
redis_list_prefix,
|
|
redis_log_channel,
|
|
redis_block_timeout_secs,
|
|
redis_max_retries,
|
|
worker_id,
|
|
}
|
|
}
|
|
}
|
|
|
|
impl Default for PoolConfig {
|
|
fn default() -> Self {
|
|
Self {
|
|
max_concurrent: num_cpus::get(),
|
|
cpu_threshold: 80.0,
|
|
redis_list_prefix: "{hook}".to_string(),
|
|
redis_log_channel: "hook:logs".to_string(),
|
|
redis_block_timeout_secs: 5,
|
|
redis_max_retries: 3,
|
|
worker_id: uuid::Uuid::new_v4().to_string(),
|
|
}
|
|
}
|
|
}
|