1use crate::app::App;
2use crate::utils::{parse_bool_env, parse_env, parse_env_optional};
3use ahash::AHashMap;
4use serde::{Deserialize, Serialize};
5use std::str::FromStr;
6use tracing::{info, warn};
7use url::Url;
8
9fn deserialize_octal_permission<'de, D>(deserializer: D) -> Result<u32, D::Error>
11where
12 D: serde::Deserializer<'de>,
13{
14 use serde::de::{self};
15
16 let s = String::deserialize(deserializer)?;
17
18 if !s.chars().all(|c| c.is_digit(8)) {
20 return Err(de::Error::custom(format!(
21 "invalid octal permission mode '{}': must contain only digits 0-7",
22 s
23 )));
24 }
25
26 let mode = u32::from_str_radix(&s, 8)
28 .map_err(|_| de::Error::custom(format!("invalid octal permission mode: {}", s)))?;
29
30 if mode > 0o777 {
32 return Err(de::Error::custom(format!(
33 "permission mode '{}' exceeds maximum value 777",
34 s
35 )));
36 }
37
38 Ok(mode)
39}
40
41fn parse_driver_enum<T: FromStr + Clone + std::fmt::Debug>(
43 driver_str: String,
44 default_driver: T,
45 driver_name: &str,
46) -> T
47where
48 <T as FromStr>::Err: std::fmt::Debug,
49{
50 match T::from_str(&driver_str.to_lowercase()) {
51 Ok(driver_enum) => driver_enum,
52 Err(e) => {
53 warn!(
54 "Failed to parse {} driver from string '{}': {:?}. Using default: {:?}.",
55 driver_name, driver_str, e, default_driver
56 );
57 default_driver
58 }
59 }
60}
61
62fn override_db_pool_settings(db_conn: &mut DatabaseConnection, prefix: &str) {
63 if let Some(min) = parse_env_optional::<u32>(&format!("{}_POOL_MIN", prefix)) {
64 db_conn.pool_min = Some(min);
65 }
66 if let Some(max) = parse_env_optional::<u32>(&format!("{}_POOL_MAX", prefix)) {
67 db_conn.pool_max = Some(max);
68 }
69}
70
71#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
74#[serde(rename_all = "lowercase")]
75pub enum AdapterDriver {
76 #[default]
77 Local,
78 Redis,
79 #[serde(rename = "redis-cluster")]
80 RedisCluster,
81 Nats,
82 Pulsar,
83 RabbitMq,
84 #[serde(rename = "google-pubsub")]
85 GooglePubSub,
86 Kafka,
87 Iggy,
88}
89
90#[derive(Debug, Clone, Serialize, Deserialize)]
91#[serde(default)]
92pub struct DynamoDbSettings {
93 pub region: String,
94 pub table_name: String,
95 pub endpoint_url: Option<String>,
96 pub aws_access_key_id: Option<String>,
97 pub aws_secret_access_key: Option<String>,
98 pub aws_profile_name: Option<String>,
99}
100
101impl Default for DynamoDbSettings {
102 fn default() -> Self {
103 Self {
104 region: "us-east-1".to_string(),
105 table_name: "sockudo-applications".to_string(),
106 endpoint_url: None,
107 aws_access_key_id: None,
108 aws_secret_access_key: None,
109 aws_profile_name: None,
110 }
111 }
112}
113
114#[derive(Debug, Clone, Serialize, Deserialize)]
115#[serde(default)]
116pub struct ScyllaDbSettings {
117 pub nodes: Vec<String>,
118 pub keyspace: String,
119 pub table_name: String,
120 pub username: Option<String>,
121 pub password: Option<String>,
122 pub replication_class: String,
123 pub replication_factor: u32,
124}
125
126impl Default for ScyllaDbSettings {
127 fn default() -> Self {
128 Self {
129 nodes: vec!["127.0.0.1:9042".to_string()],
130 keyspace: "sockudo".to_string(),
131 table_name: "applications".to_string(),
132 username: None,
133 password: None,
134 replication_class: "SimpleStrategy".to_string(),
135 replication_factor: 3,
136 }
137 }
138}
139
140#[derive(Debug, Clone, Serialize, Deserialize)]
141#[serde(default)]
142pub struct SurrealDbSettings {
143 pub url: String,
144 pub namespace: String,
145 pub database: String,
146 pub username: String,
147 pub password: String,
148 pub table_name: String,
149 pub cache_ttl: u64,
150 pub cache_max_capacity: u64,
151}
152
153impl Default for SurrealDbSettings {
154 fn default() -> Self {
155 Self {
156 url: "ws://127.0.0.1:8000".to_string(),
157 namespace: "sockudo".to_string(),
158 database: "sockudo".to_string(),
159 username: "root".to_string(),
160 password: "root".to_string(),
161 table_name: "applications".to_string(),
162 cache_ttl: 300,
163 cache_max_capacity: 100,
164 }
165 }
166}
167
168impl FromStr for AdapterDriver {
169 type Err = String;
170 fn from_str(s: &str) -> Result<Self, Self::Err> {
171 match s.to_lowercase().as_str() {
172 "local" => Ok(AdapterDriver::Local),
173 "redis" => Ok(AdapterDriver::Redis),
174 "redis-cluster" => Ok(AdapterDriver::RedisCluster),
175 "nats" => Ok(AdapterDriver::Nats),
176 "pulsar" => Ok(AdapterDriver::Pulsar),
177 "rabbitmq" | "rabbit-mq" => Ok(AdapterDriver::RabbitMq),
178 "google-pubsub" | "gcp-pubsub" | "pubsub" => Ok(AdapterDriver::GooglePubSub),
179 "kafka" => Ok(AdapterDriver::Kafka),
180 "iggy" | "apache-iggy" | "apache_iggy" => Ok(AdapterDriver::Iggy),
181 _ => Err(format!("Unknown adapter driver: {s}")),
182 }
183 }
184}
185
186#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
187#[serde(rename_all = "lowercase")]
188pub enum AppManagerDriver {
189 #[default]
190 Memory,
191 Mysql,
192 Dynamodb,
193 PgSql,
194 SurrealDb,
195 ScyllaDb,
196}
197impl FromStr for AppManagerDriver {
198 type Err = String;
199 fn from_str(s: &str) -> Result<Self, Self::Err> {
200 match s.to_lowercase().as_str() {
201 "memory" => Ok(AppManagerDriver::Memory),
202 "mysql" => Ok(AppManagerDriver::Mysql),
203 "dynamodb" => Ok(AppManagerDriver::Dynamodb),
204 "pgsql" | "postgres" | "postgresql" => Ok(AppManagerDriver::PgSql),
205 "surreal" | "surrealdb" => Ok(AppManagerDriver::SurrealDb),
206 "scylladb" | "scylla" => Ok(AppManagerDriver::ScyllaDb),
207 _ => Err(format!("Unknown app manager driver: {s}")),
208 }
209 }
210}
211
212#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
213#[serde(rename_all = "lowercase")]
214pub enum CacheDriver {
215 #[default]
216 Memory,
217 Redis,
218 #[serde(rename = "redis-cluster")]
219 RedisCluster,
220 None,
221}
222
223impl FromStr for CacheDriver {
224 type Err = String;
225 fn from_str(s: &str) -> Result<Self, Self::Err> {
226 match s.to_lowercase().as_str() {
227 "memory" => Ok(CacheDriver::Memory),
228 "redis" => Ok(CacheDriver::Redis),
229 "redis-cluster" => Ok(CacheDriver::RedisCluster),
230 "none" => Ok(CacheDriver::None),
231 _ => Err(format!("Unknown cache driver: {s}")),
232 }
233 }
234}
235
236#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
237#[serde(rename_all = "lowercase")]
238pub enum QueueDriver {
239 Memory,
240 #[default]
241 Redis,
242 #[serde(rename = "redis-cluster")]
243 RedisCluster,
244 Nats,
245 Pulsar,
246 RabbitMq,
247 #[serde(rename = "google-pubsub")]
248 GooglePubSub,
249 Kafka,
250 Iggy,
251 Sqs,
252 Sns,
253 None,
254}
255
256impl FromStr for QueueDriver {
257 type Err = String;
258 fn from_str(s: &str) -> Result<Self, Self::Err> {
259 match s.to_lowercase().as_str() {
260 "memory" => Ok(QueueDriver::Memory),
261 "redis" => Ok(QueueDriver::Redis),
262 "redis-cluster" => Ok(QueueDriver::RedisCluster),
263 "nats" => Ok(QueueDriver::Nats),
264 "pulsar" => Ok(QueueDriver::Pulsar),
265 "rabbitmq" | "rabbit-mq" => Ok(QueueDriver::RabbitMq),
266 "google-pubsub" | "gcp-pubsub" | "pubsub" => Ok(QueueDriver::GooglePubSub),
267 "kafka" => Ok(QueueDriver::Kafka),
268 "iggy" | "apache-iggy" | "apache_iggy" => Ok(QueueDriver::Iggy),
269 "sqs" => Ok(QueueDriver::Sqs),
270 "sns" => Ok(QueueDriver::Sns),
271 "none" => Ok(QueueDriver::None),
272 _ => Err(format!("Unknown queue driver: {s}")),
273 }
274 }
275}
276
277#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
278#[serde(rename_all = "snake_case")]
279pub enum DeltaCoordinationBackend {
280 #[default]
281 Auto,
282 None,
283 Redis,
284 RedisCluster,
285 Nats,
286}
287
288impl FromStr for DeltaCoordinationBackend {
289 type Err = String;
290
291 fn from_str(s: &str) -> Result<Self, Self::Err> {
292 match s.trim().to_ascii_lowercase().as_str() {
293 "auto" => Ok(Self::Auto),
294 "none" => Ok(Self::None),
295 "redis" => Ok(Self::Redis),
296 "redis-cluster" | "redis_cluster" => Ok(Self::RedisCluster),
297 "nats" => Ok(Self::Nats),
298 _ => Err(format!("Unknown delta coordination backend: {s}")),
299 }
300 }
301}
302
303impl AsRef<str> for QueueDriver {
304 fn as_ref(&self) -> &str {
305 match self {
306 QueueDriver::Memory => "memory",
307 QueueDriver::Redis => "redis",
308 QueueDriver::RedisCluster => "redis-cluster",
309 QueueDriver::Nats => "nats",
310 QueueDriver::Pulsar => "pulsar",
311 QueueDriver::RabbitMq => "rabbitmq",
312 QueueDriver::GooglePubSub => "google-pubsub",
313 QueueDriver::Kafka => "kafka",
314 QueueDriver::Iggy => "iggy",
315 QueueDriver::Sqs => "sqs",
316 QueueDriver::Sns => "sns",
317 QueueDriver::None => "none",
318 }
319 }
320}
321
322#[derive(Debug, Clone, Serialize, Deserialize)]
323#[serde(default)]
324pub struct RedisClusterQueueConfig {
325 pub concurrency: u32,
326 pub prefix: Option<String>,
327 pub nodes: Vec<String>,
328 pub request_timeout_ms: u64,
329}
330
331impl Default for RedisClusterQueueConfig {
332 fn default() -> Self {
333 Self {
334 concurrency: 5,
335 prefix: Some("sockudo_queue:".to_string()),
336 nodes: vec!["redis://127.0.0.1:6379".to_string()],
337 request_timeout_ms: 5000,
338 }
339 }
340}
341
342#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
343#[serde(rename_all = "lowercase")]
344pub enum MetricsDriver {
345 #[default]
346 Prometheus,
347}
348
349#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
350#[serde(rename_all = "lowercase")]
351pub enum LogOutputFormat {
352 #[default]
353 Human,
354 Json,
355}
356
357impl FromStr for LogOutputFormat {
358 type Err = String;
359 fn from_str(s: &str) -> Result<Self, Self::Err> {
360 match s.to_lowercase().as_str() {
361 "human" => Ok(LogOutputFormat::Human),
362 "json" => Ok(LogOutputFormat::Json),
363 _ => Err(format!("Unknown log output format: {s}")),
364 }
365 }
366}
367
368impl FromStr for MetricsDriver {
369 type Err = String;
370 fn from_str(s: &str) -> Result<Self, Self::Err> {
371 match s.to_lowercase().as_str() {
372 "prometheus" => Ok(MetricsDriver::Prometheus),
373 _ => Err(format!("Unknown metrics driver: {s}")),
374 }
375 }
376}
377
378impl AsRef<str> for MetricsDriver {
379 fn as_ref(&self) -> &str {
380 match self {
381 MetricsDriver::Prometheus => "prometheus",
382 }
383 }
384}
385
386#[derive(Debug, Clone, Serialize, Deserialize)]
388#[serde(default)]
389pub struct ServerOptions {
390 pub adapter: AdapterConfig,
391 pub app_manager: AppManagerConfig,
392 pub cache: CacheConfig,
393 pub channel_limits: ChannelLimits,
394 pub cors: CorsConfig,
395 pub database: DatabaseConfig,
396 pub database_pooling: DatabasePooling,
397 pub debug: bool,
398 pub event_limits: EventLimits,
399 pub host: String,
400 pub http_api: HttpApiConfig,
401 pub instance: InstanceConfig,
402 pub logging: Option<LoggingConfig>,
403 pub metrics: MetricsConfig,
404 pub mode: String,
405 pub port: u16,
406 pub path_prefix: String,
407 pub presence: PresenceConfig,
408 pub queue: QueueConfig,
409 pub rate_limiter: RateLimiterConfig,
410 pub shutdown_grace_period: u64,
411 pub ssl: SslConfig,
412 pub user_authentication_timeout: u64,
413 pub webhooks: WebhooksConfig,
414 pub websocket_max_payload_kb: u32,
415 pub cleanup: CleanupConfig,
416 pub activity_timeout: u64,
417 pub cluster_health: ClusterHealthConfig,
418 pub unix_socket: UnixSocketConfig,
419 pub delta_compression: DeltaCompressionOptionsConfig,
420 pub tag_filtering: TagFilteringConfig,
421 pub websocket: WebSocketConfig,
422 pub connection_recovery: ConnectionRecoveryConfig,
423 pub history: HistoryConfig,
424 pub presence_history: PresenceHistoryConfig,
425 pub idempotency: IdempotencyConfig,
426 pub ephemeral: EphemeralConfig,
427 pub echo_control: EchoControlConfig,
428 pub event_name_filtering: EventNameFilteringConfig,
429 pub versioned_messages: VersionedMessagesConfig,
430 pub annotations: AnnotationsConfig,
431 pub push: PushConfig,
432 pub health_check_timeout_ms: u64,
435}
436
437#[derive(Debug, Clone, Serialize, Deserialize)]
440#[serde(default)]
441pub struct SqsQueueConfig {
442 pub region: String,
443 pub queue_url_prefix: Option<String>,
444 pub visibility_timeout: i32,
445 pub endpoint_url: Option<String>,
446 pub max_messages: i32,
447 pub wait_time_seconds: i32,
448 pub concurrency: u32,
449 pub fifo: bool,
450 pub message_group_id: Option<String>,
451}
452
453#[derive(Debug, Clone, Serialize, Deserialize)]
454#[serde(default)]
455pub struct SnsQueueConfig {
456 pub region: String,
457 pub topic_arn: String,
458 pub endpoint_url: Option<String>,
459}
460
461#[derive(Debug, Clone, Serialize, Deserialize)]
462#[serde(default)]
463pub struct AdapterConfig {
464 pub driver: AdapterDriver,
465 pub redis: RedisAdapterConfig,
466 pub cluster: RedisClusterAdapterConfig,
467 pub nats: NatsAdapterConfig,
468 pub pulsar: PulsarAdapterConfig,
469 pub rabbitmq: RabbitMqAdapterConfig,
470 pub google_pubsub: GooglePubSubAdapterConfig,
471 pub kafka: KafkaAdapterConfig,
472 pub iggy: IggyConfig,
473 #[serde(default = "default_buffer_multiplier_per_cpu")]
474 pub buffer_multiplier_per_cpu: usize,
475 pub cluster_health: ClusterHealthConfig,
476 #[serde(default = "default_enable_socket_counting")]
477 pub enable_socket_counting: bool,
478 #[serde(default = "default_fallback_to_local")]
479 pub fallback_to_local: bool,
480}
481
482fn default_enable_socket_counting() -> bool {
483 true
484}
485
486fn default_fallback_to_local() -> bool {
487 true
488}
489
490fn default_buffer_multiplier_per_cpu() -> usize {
491 64
492}
493
494impl Default for AdapterConfig {
495 fn default() -> Self {
496 Self {
497 driver: AdapterDriver::default(),
498 redis: RedisAdapterConfig::default(),
499 cluster: RedisClusterAdapterConfig::default(),
500 nats: NatsAdapterConfig::default(),
501 pulsar: PulsarAdapterConfig::default(),
502 rabbitmq: RabbitMqAdapterConfig::default(),
503 google_pubsub: GooglePubSubAdapterConfig::default(),
504 kafka: KafkaAdapterConfig::default(),
505 iggy: IggyConfig::default(),
506 buffer_multiplier_per_cpu: default_buffer_multiplier_per_cpu(),
507 cluster_health: ClusterHealthConfig::default(),
508 enable_socket_counting: default_enable_socket_counting(),
509 fallback_to_local: default_fallback_to_local(),
510 }
511 }
512}
513
514#[derive(Debug, Clone, Serialize, Deserialize)]
515#[serde(default)]
516pub struct RedisAdapterConfig {
517 pub requests_timeout: u64,
518 pub prefix: String,
519 pub redis_pub_options: AHashMap<String, sonic_rs::Value>,
520 pub redis_sub_options: AHashMap<String, sonic_rs::Value>,
521 pub cluster_mode: bool,
522}
523
524#[derive(Debug, Clone, Serialize, Deserialize)]
525#[serde(default)]
526pub struct RedisClusterAdapterConfig {
527 pub nodes: Vec<String>,
528 pub prefix: String,
529 pub request_timeout_ms: u64,
530 pub use_connection_manager: bool,
531 #[serde(default)]
532 pub use_sharded_pubsub: bool,
533}
534
535#[derive(Debug, Clone, Serialize, Deserialize)]
536#[serde(default)]
537pub struct NatsAdapterConfig {
538 pub servers: Vec<String>,
539 pub prefix: String,
540 pub request_timeout_ms: u64,
541 pub username: Option<String>,
542 pub password: Option<String>,
543 pub token: Option<String>,
544 pub connection_timeout_ms: u64,
545 pub nodes_number: Option<u32>,
546 pub discovery_max_wait_ms: u64,
547 pub discovery_idle_wait_ms: u64,
548 pub subscription_capacity: Option<usize>,
549 pub client_capacity: Option<usize>,
550 pub max_reconnects: Option<usize>,
551 pub presence_sync_chunk_size: Option<usize>,
552}
553
554#[derive(Debug, Clone, Serialize, Deserialize)]
555#[serde(default)]
556pub struct PulsarAdapterConfig {
557 pub url: String,
558 pub prefix: String,
559 pub request_timeout_ms: u64,
560 pub token: Option<String>,
561 pub nodes_number: Option<u32>,
562}
563
564#[derive(Debug, Clone, Serialize, Deserialize)]
565#[serde(default)]
566pub struct RabbitMqAdapterConfig {
567 pub url: String,
568 pub prefix: String,
569 pub request_timeout_ms: u64,
570 pub connection_timeout_ms: u64,
571 pub nodes_number: Option<u32>,
572}
573
574#[derive(Debug, Clone, Serialize, Deserialize)]
575#[serde(default)]
576pub struct GooglePubSubAdapterConfig {
577 pub project_id: String,
578 pub prefix: String,
579 pub request_timeout_ms: u64,
580 pub emulator_host: Option<String>,
581 pub nodes_number: Option<u32>,
582}
583
584#[derive(Debug, Clone, Serialize, Deserialize)]
585#[serde(default)]
586pub struct KafkaAdapterConfig {
587 pub brokers: Vec<String>,
588 pub prefix: String,
589 pub request_timeout_ms: u64,
590 pub security_protocol: Option<String>,
591 pub sasl_mechanism: Option<String>,
592 pub sasl_username: Option<String>,
593 pub sasl_password: Option<String>,
594 pub nodes_number: Option<u32>,
595}
596
597#[derive(Debug, Clone, Serialize, Deserialize)]
598#[serde(default)]
599pub struct IggyConfig {
600 pub connection_string: String,
601 pub username: Option<String>,
602 pub password: Option<String>,
603 pub consumer_name: Option<String>,
604 pub stream: String,
605 pub topic_prefix: String,
606 pub queue_topic_prefix: String,
607 pub consumer_group_prefix: String,
608 pub request_timeout_ms: u64,
609 pub poll_interval_ms: u64,
610 pub poll_batch_size: u32,
611 pub partitions_count: u32,
612 pub partition_id: u32,
613 pub auto_create: bool,
614 pub start_from_latest: bool,
615 pub nodes_number: Option<u32>,
616}
617
618#[derive(Debug, Clone, Serialize, Deserialize, Default)]
619#[serde(default)]
620pub struct AppManagerConfig {
621 pub driver: AppManagerDriver,
622 pub array: ArrayConfig,
623 pub cache: CacheSettings,
624 pub scylladb: ScyllaDbSettings,
625}
626
627#[derive(Debug, Clone, Serialize, Deserialize, Default)]
628#[serde(default)]
629pub struct ArrayConfig {
630 pub apps: Vec<App>,
631}
632
633#[derive(Debug, Clone, Serialize, Deserialize)]
634#[serde(default)]
635pub struct CacheSettings {
636 pub enabled: bool,
637 pub ttl: u64,
638}
639
640#[derive(Debug, Clone, Serialize, Deserialize)]
641#[serde(default)]
642pub struct MemoryCacheOptions {
643 pub ttl: u64,
644 pub cleanup_interval: u64,
645 pub max_capacity: u64,
646}
647
648#[derive(Debug, Clone, Serialize, Deserialize)]
649#[serde(default)]
650pub struct CacheConfig {
651 pub driver: CacheDriver,
652 pub redis: RedisConfig,
653 pub memory: MemoryCacheOptions,
654}
655
656#[derive(Debug, Clone, Serialize, Deserialize, Default)]
657#[serde(default)]
658pub struct RedisConfig {
659 pub prefix: Option<String>,
660 pub url_override: Option<String>,
661 pub cluster_mode: bool,
662}
663
664#[derive(Debug, Clone, Serialize, Deserialize)]
665#[serde(default)]
666pub struct ChannelLimits {
667 pub max_name_length: u32,
668 pub cache_ttl: u64,
669}
670
671#[derive(Debug, Clone, Serialize, Deserialize)]
672#[serde(default)]
673pub struct CorsConfig {
674 pub credentials: bool,
675 #[serde(deserialize_with = "deserialize_and_validate_cors_origins")]
676 pub origin: Vec<String>,
677 pub methods: Vec<String>,
678 pub allowed_headers: Vec<String>,
679}
680
681fn deserialize_and_validate_cors_origins<'de, D>(deserializer: D) -> Result<Vec<String>, D::Error>
682where
683 D: serde::Deserializer<'de>,
684{
685 use serde::de::Error;
686 let origins = Vec::<String>::deserialize(deserializer)?;
687
688 if let Err(e) = crate::origin_validation::OriginValidator::validate_patterns(&origins) {
689 return Err(D::Error::custom(format!(
690 "CORS origin pattern validation failed: {}",
691 e
692 )));
693 }
694
695 Ok(origins)
696}
697
698#[derive(Debug, Clone, Serialize, Deserialize, Default)]
699#[serde(default)]
700pub struct DatabaseConfig {
701 pub mysql: DatabaseConnection,
702 pub postgres: DatabaseConnection,
703 pub redis: RedisConnection,
704 pub dynamodb: DynamoDbSettings,
705 pub surrealdb: SurrealDbSettings,
706 pub scylladb: ScyllaDbSettings,
707}
708
709#[derive(Debug, Clone, Serialize, Deserialize)]
710#[serde(default)]
711pub struct DatabaseConnection {
712 pub host: String,
713 pub port: u16,
714 pub username: String,
715 pub password: String,
716 pub database: String,
717 pub table_name: String,
718 pub connection_pool_size: u32,
719 pub pool_min: Option<u32>,
720 pub pool_max: Option<u32>,
721 pub cache_ttl: u64,
722 pub cache_cleanup_interval: u64,
723 pub cache_max_capacity: u64,
724}
725
726#[derive(Debug, Clone, Serialize, Deserialize)]
727#[serde(default)]
728pub struct RedisConnection {
729 pub host: String,
730 pub port: u16,
731 pub db: u32,
732 pub username: Option<String>,
733 pub password: Option<String>,
734 pub key_prefix: String,
735 pub sentinels: Vec<RedisSentinel>,
736 pub sentinel_password: Option<String>,
737 pub name: String,
738 pub cluster: RedisClusterConnection,
739 pub cluster_nodes: Vec<ClusterNode>,
741}
742
743#[derive(Debug, Clone, Serialize, Deserialize)]
744#[serde(default)]
745pub struct RedisSentinel {
746 pub host: String,
747 pub port: u16,
748}
749
750#[derive(Debug, Clone, Serialize, Deserialize, Default)]
751#[serde(default)]
752pub struct RedisClusterConnection {
753 pub nodes: Vec<ClusterNode>,
754 pub username: Option<String>,
755 pub password: Option<String>,
756 #[serde(alias = "useTLS")]
757 pub use_tls: bool,
758}
759
760impl RedisConnection {
761 pub fn is_sentinel_configured(&self) -> bool {
763 !self.sentinels.is_empty()
764 }
765
766 pub fn to_url(&self) -> String {
768 if self.is_sentinel_configured() {
769 self.build_sentinel_url()
770 } else {
771 self.build_standard_url()
772 }
773 }
774
775 fn build_standard_url(&self) -> String {
776 let (scheme, host) = if self.host.starts_with("rediss://") {
778 ("rediss://", self.host.trim_start_matches("rediss://"))
779 } else if self.host.starts_with("redis://") {
780 ("redis://", self.host.trim_start_matches("redis://"))
781 } else {
782 ("redis://", self.host.as_str())
783 };
784
785 let mut url = String::from(scheme);
786
787 if let Some(ref username) = self.username {
788 url.push_str(username);
789 if let Some(ref password) = self.password {
790 url.push(':');
791 url.push_str(&urlencoding::encode(password));
792 }
793 url.push('@');
794 } else if let Some(ref password) = self.password {
795 url.push(':');
796 url.push_str(&urlencoding::encode(password));
797 url.push('@');
798 }
799
800 url.push_str(host);
801 url.push(':');
802 url.push_str(&self.port.to_string());
803 url.push('/');
804 url.push_str(&self.db.to_string());
805
806 url
807 }
808
809 fn build_sentinel_url(&self) -> String {
810 let mut url = String::from("redis+sentinel://");
811
812 if let Some(ref sentinel_password) = self.sentinel_password {
813 url.push(':');
814 url.push_str(&urlencoding::encode(sentinel_password));
815 url.push('@');
816 }
817
818 let sentinel_hosts: Vec<String> = self
819 .sentinels
820 .iter()
821 .map(|s| format!("{}:{}", s.host, s.port))
822 .collect();
823 url.push_str(&sentinel_hosts.join(","));
824
825 url.push('/');
826 url.push_str(&self.name);
827 url.push('/');
828 url.push_str(&self.db.to_string());
829
830 let mut params = Vec::new();
831 if let Some(ref password) = self.password {
832 params.push(format!("password={}", urlencoding::encode(password)));
833 }
834 if let Some(ref username) = self.username {
835 params.push(format!("username={}", urlencoding::encode(username)));
836 }
837
838 if !params.is_empty() {
839 url.push('?');
840 url.push_str(¶ms.join("&"));
841 }
842
843 url
844 }
845
846 pub fn has_cluster_nodes(&self) -> bool {
849 !self.cluster.nodes.is_empty() || !self.cluster_nodes.is_empty()
850 }
851
852 pub fn cluster_node_urls(&self) -> Vec<String> {
855 if !self.cluster.nodes.is_empty() {
856 return self.build_cluster_urls(&self.cluster.nodes);
857 }
858 self.build_cluster_urls(&self.cluster_nodes)
859 }
860
861 pub fn normalize_cluster_seed_urls(&self, seeds: &[String]) -> Vec<String> {
864 self.build_cluster_urls(
865 &seeds
866 .iter()
867 .filter_map(|seed| ClusterNode::from_seed(seed))
868 .collect::<Vec<ClusterNode>>(),
869 )
870 }
871
872 fn build_cluster_urls(&self, nodes: &[ClusterNode]) -> Vec<String> {
873 let username = self
874 .cluster
875 .username
876 .as_deref()
877 .or(self.username.as_deref());
878 let password = self
879 .cluster
880 .password
881 .as_deref()
882 .or(self.password.as_deref());
883 let use_tls = self.cluster.use_tls;
884
885 nodes
886 .iter()
887 .map(|node| node.to_url_with_options(use_tls, username, password))
888 .collect()
889 }
890}
891
892impl RedisSentinel {
893 pub fn to_host_port(&self) -> String {
894 format!("{}:{}", self.host, self.port)
895 }
896}
897
898#[derive(Debug, Clone, Serialize, Deserialize)]
899#[serde(default)]
900pub struct ClusterNode {
901 pub host: String,
902 pub port: u16,
903}
904
905impl ClusterNode {
906 pub fn to_url(&self) -> String {
907 self.to_url_with_options(false, None, None)
908 }
909
910 pub fn to_url_with_options(
911 &self,
912 use_tls: bool,
913 username: Option<&str>,
914 password: Option<&str>,
915 ) -> String {
916 let host = self.host.trim();
917
918 if host.starts_with("redis://") || host.starts_with("rediss://") {
919 if let Ok(parsed) = Url::parse(host)
920 && let Some(host_str) = parsed.host_str()
921 {
922 let scheme = parsed.scheme();
923 let port = parsed.port_or_known_default().unwrap_or(self.port);
924 let parsed_username = (!parsed.username().is_empty()).then_some(parsed.username());
925 let parsed_password = parsed.password();
926 let has_embedded_auth = parsed_username.is_some() || parsed_password.is_some();
927 let (effective_username, effective_password) = if has_embedded_auth {
928 (parsed_username, parsed_password)
929 } else {
930 (username, password)
931 };
932
933 return build_redis_url(
934 scheme,
935 host_str,
936 port,
937 effective_username,
938 effective_password,
939 );
940 }
941
942 let has_port = if let Some(bracket_pos) = host.rfind(']') {
944 host[bracket_pos..].contains(':')
945 } else {
946 host.split(':').count() >= 3
947 };
948 let base = if has_port {
949 host.to_string()
950 } else {
951 format!("{}:{}", host, self.port)
952 };
953
954 if let Ok(parsed) = Url::parse(&base) {
955 let parsed_username = (!parsed.username().is_empty()).then_some(parsed.username());
956 let parsed_password = parsed.password();
957 if let Some(host_str) = parsed.host_str() {
958 let port = parsed.port_or_known_default().unwrap_or(self.port);
959 let has_embedded_auth = parsed_username.is_some() || parsed_password.is_some();
960 let (effective_username, effective_password) = if has_embedded_auth {
961 (parsed_username, parsed_password)
962 } else {
963 (username, password)
964 };
965 return build_redis_url(
966 parsed.scheme(),
967 host_str,
968 port,
969 effective_username,
970 effective_password,
971 );
972 }
973 }
974 return base;
975 }
976
977 let (normalized_host, normalized_port) = split_plain_host_and_port(host, self.port);
978 let scheme = if use_tls { "rediss" } else { "redis" };
979 build_redis_url(
980 scheme,
981 &normalized_host,
982 normalized_port,
983 username,
984 password,
985 )
986 }
987
988 pub fn from_seed(seed: &str) -> Option<Self> {
989 let trimmed = seed.trim();
990 if trimmed.is_empty() {
991 return None;
992 }
993
994 if trimmed.starts_with("redis://") || trimmed.starts_with("rediss://") {
995 let port = Url::parse(trimmed)
996 .ok()
997 .and_then(|parsed| parsed.port_or_known_default())
998 .unwrap_or(6379);
999 return Some(Self {
1000 host: trimmed.to_string(),
1001 port,
1002 });
1003 }
1004
1005 let (host, port) = split_plain_host_and_port(trimmed, 6379);
1006 Some(Self { host, port })
1007 }
1008}
1009
1010fn split_plain_host_and_port(raw_host: &str, default_port: u16) -> (String, u16) {
1011 let host = raw_host.trim();
1012
1013 if host.starts_with('[') {
1015 if let Some(end_bracket) = host.find(']') {
1016 let host_part = host[1..end_bracket].to_string();
1017 let remainder = &host[end_bracket + 1..];
1018 if let Some(port_str) = remainder.strip_prefix(':')
1019 && let Ok(port) = port_str.parse::<u16>()
1020 {
1021 return (host_part, port);
1022 }
1023 return (host_part, default_port);
1024 }
1025 return (host.to_string(), default_port);
1026 }
1027
1028 if host.matches(':').count() == 1
1030 && let Some((host_part, port_part)) = host.rsplit_once(':')
1031 && let Ok(port) = port_part.parse::<u16>()
1032 {
1033 return (host_part.to_string(), port);
1034 }
1035
1036 (host.to_string(), default_port)
1037}
1038
1039fn build_redis_url(
1040 scheme: &str,
1041 host: &str,
1042 port: u16,
1043 username: Option<&str>,
1044 password: Option<&str>,
1045) -> String {
1046 let mut url = format!("{scheme}://");
1047
1048 if let Some(user) = username {
1049 url.push_str(&urlencoding::encode(user));
1050 if let Some(pass) = password {
1051 url.push(':');
1052 url.push_str(&urlencoding::encode(pass));
1053 }
1054 url.push('@');
1055 } else if let Some(pass) = password {
1056 url.push(':');
1057 url.push_str(&urlencoding::encode(pass));
1058 url.push('@');
1059 }
1060
1061 if host.contains(':') && !host.starts_with('[') {
1062 url.push('[');
1063 url.push_str(host);
1064 url.push(']');
1065 } else {
1066 url.push_str(host);
1067 }
1068 url.push(':');
1069 url.push_str(&port.to_string());
1070 url
1071}
1072
1073#[derive(Debug, Clone, Serialize, Deserialize)]
1074#[serde(default)]
1075pub struct DatabasePooling {
1076 pub enabled: bool,
1077 pub min: u32,
1078 pub max: u32,
1079}
1080
1081#[derive(Debug, Clone, Serialize, Deserialize)]
1082#[serde(default)]
1083pub struct EventLimits {
1084 pub max_channels_at_once: u32,
1085 pub max_name_length: u32,
1086 pub max_payload_in_kb: u32,
1087 pub max_batch_size: u32,
1088}
1089
1090#[derive(Debug, Clone, Serialize, Deserialize)]
1091#[serde(default)]
1092pub struct IdempotencyConfig {
1093 pub enabled: bool,
1095 pub ttl_seconds: u64,
1097 pub max_key_length: usize,
1099}
1100
1101#[derive(Debug, Clone, Serialize, Deserialize)]
1102#[serde(default)]
1103pub struct EphemeralConfig {
1104 pub enabled: bool,
1106}
1107
1108#[derive(Debug, Clone, Serialize, Deserialize)]
1109#[serde(default)]
1110pub struct EchoControlConfig {
1111 pub enabled: bool,
1113 pub default_echo_messages: bool,
1115}
1116
1117#[derive(Debug, Clone, Serialize, Deserialize)]
1118#[serde(default)]
1119pub struct EventNameFilteringConfig {
1120 pub enabled: bool,
1122 pub max_events_per_filter: usize,
1124 pub max_event_name_length: usize,
1126}
1127
1128#[derive(Debug, Clone, Serialize, Deserialize)]
1129#[serde(default)]
1130pub struct ConnectionRecoveryConfig {
1131 pub enabled: bool,
1136 pub buffer_ttl_seconds: u64,
1138 pub max_buffer_size: usize,
1140}
1141
1142#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
1143#[serde(rename_all = "lowercase")]
1144pub enum VersionStoreDriver {
1145 #[default]
1146 Memory,
1147 Postgres,
1148 Mysql,
1149 DynamoDb,
1150 ScyllaDb,
1151 SurrealDb,
1152}
1153
1154impl FromStr for VersionStoreDriver {
1155 type Err = String;
1156 fn from_str(s: &str) -> Result<Self, Self::Err> {
1157 match s.to_lowercase().as_str() {
1158 "postgres" | "postgresql" | "pgsql" => Ok(Self::Postgres),
1159 "mysql" => Ok(Self::Mysql),
1160 "dynamodb" => Ok(Self::DynamoDb),
1161 "scylladb" | "scylla" => Ok(Self::ScyllaDb),
1162 "surrealdb" | "surreal" => Ok(Self::SurrealDb),
1163 "memory" => Ok(Self::Memory),
1164 _ => Err(format!("Unknown version store driver: {s}")),
1165 }
1166 }
1167}
1168
1169#[derive(Debug, Clone, Serialize, Deserialize)]
1170#[serde(default)]
1171pub struct VersionedMessagesConfig {
1172 pub enabled: bool,
1174 pub driver: VersionStoreDriver,
1176 pub max_page_size: usize,
1178 pub retention_window_seconds: u64,
1186 pub purge_interval_seconds: u64,
1189 pub purge_batch_size: usize,
1193 pub max_purge_per_tick: usize,
1196}
1197
1198#[derive(Debug, Clone, Serialize, Deserialize, Default)]
1199#[serde(default)]
1200pub struct AnnotationsConfig {
1201 pub enabled: bool,
1204}
1205
1206#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
1207#[serde(rename_all = "lowercase")]
1208pub enum HistoryBackend {
1209 #[default]
1210 Postgres,
1211 Mysql,
1212 DynamoDb,
1213 SurrealDb,
1214 ScyllaDb,
1215 Memory,
1216}
1217
1218impl FromStr for HistoryBackend {
1219 type Err = String;
1220 fn from_str(s: &str) -> Result<Self, Self::Err> {
1221 match s.to_lowercase().as_str() {
1222 "postgres" | "postgresql" | "pgsql" => Ok(Self::Postgres),
1223 "mysql" => Ok(Self::Mysql),
1224 "dynamodb" => Ok(Self::DynamoDb),
1225 "surrealdb" | "surreal" => Ok(Self::SurrealDb),
1226 "scylladb" | "scylla" => Ok(Self::ScyllaDb),
1227 "memory" => Ok(Self::Memory),
1228 _ => Err(format!("Unknown history backend: {s}")),
1229 }
1230 }
1231}
1232
1233#[derive(Debug, Clone, Serialize, Deserialize)]
1234#[serde(default)]
1235pub struct PostgresHistoryConfig {
1236 pub table_prefix: String,
1237 pub write_timeout_ms: u64,
1238}
1239
1240impl Default for PostgresHistoryConfig {
1241 fn default() -> Self {
1242 Self {
1243 table_prefix: "sockudo_history".to_string(),
1244 write_timeout_ms: 5000,
1245 }
1246 }
1247}
1248
1249#[derive(Debug, Clone, Serialize, Deserialize)]
1250#[serde(default)]
1251pub struct MySqlHistoryConfig {
1252 pub table_prefix: String,
1253 pub write_timeout_ms: u64,
1254}
1255
1256impl Default for MySqlHistoryConfig {
1257 fn default() -> Self {
1258 Self {
1259 table_prefix: "sockudo_history".to_string(),
1260 write_timeout_ms: 5000,
1261 }
1262 }
1263}
1264
1265#[derive(Debug, Clone, Serialize, Deserialize)]
1266#[serde(default)]
1267pub struct DynamoDbHistoryConfig {
1268 pub table_prefix: String,
1269 pub write_timeout_ms: u64,
1270}
1271
1272impl Default for DynamoDbHistoryConfig {
1273 fn default() -> Self {
1274 Self {
1275 table_prefix: "sockudo_history".to_string(),
1276 write_timeout_ms: 5000,
1277 }
1278 }
1279}
1280
1281#[derive(Debug, Clone, Serialize, Deserialize)]
1282#[serde(default)]
1283pub struct SurrealDbHistoryConfig {
1284 pub table_prefix: String,
1285 pub write_timeout_ms: u64,
1286}
1287
1288impl Default for SurrealDbHistoryConfig {
1289 fn default() -> Self {
1290 Self {
1291 table_prefix: "sockudo_history".to_string(),
1292 write_timeout_ms: 5000,
1293 }
1294 }
1295}
1296
1297#[derive(Debug, Clone, Serialize, Deserialize)]
1298#[serde(default)]
1299pub struct ScyllaDbHistoryConfig {
1300 pub table_prefix: String,
1301 pub write_timeout_ms: u64,
1302}
1303
1304impl Default for ScyllaDbHistoryConfig {
1305 fn default() -> Self {
1306 Self {
1307 table_prefix: "sockudo_history".to_string(),
1308 write_timeout_ms: 5000,
1309 }
1310 }
1311}
1312
1313#[derive(Debug, Clone, Serialize, Deserialize)]
1314#[serde(default)]
1315pub struct HistoryConfig {
1316 pub enabled: bool,
1317 pub rewind_enabled: bool,
1318 pub backend: HistoryBackend,
1319 pub retention_window_seconds: u64,
1320 pub max_page_size: usize,
1321 pub max_messages_per_channel: Option<usize>,
1322 pub max_bytes_per_channel: Option<u64>,
1323 pub writer_shards: usize,
1324 pub writer_queue_capacity: usize,
1325 pub purge_interval_seconds: u64,
1326 pub purge_batch_size: usize,
1327 pub max_purge_per_tick: usize,
1328 pub postgres: PostgresHistoryConfig,
1329 pub mysql: MySqlHistoryConfig,
1330 pub dynamodb: DynamoDbHistoryConfig,
1331 pub surrealdb: SurrealDbHistoryConfig,
1332 pub scylladb: ScyllaDbHistoryConfig,
1333}
1334
1335impl Default for HistoryConfig {
1336 fn default() -> Self {
1337 Self {
1338 enabled: false,
1339 rewind_enabled: true,
1340 backend: HistoryBackend::Postgres,
1341 retention_window_seconds: 86400,
1342 max_page_size: 100,
1343 max_messages_per_channel: None,
1344 max_bytes_per_channel: None,
1345 writer_shards: 16,
1346 writer_queue_capacity: 4096,
1347 purge_interval_seconds: 300,
1348 purge_batch_size: 1000,
1349 max_purge_per_tick: 100_000,
1350 postgres: PostgresHistoryConfig::default(),
1351 mysql: MySqlHistoryConfig::default(),
1352 dynamodb: DynamoDbHistoryConfig::default(),
1353 surrealdb: SurrealDbHistoryConfig::default(),
1354 scylladb: ScyllaDbHistoryConfig::default(),
1355 }
1356 }
1357}
1358
1359impl Default for VersionedMessagesConfig {
1360 fn default() -> Self {
1361 Self {
1362 enabled: false,
1363 driver: VersionStoreDriver::Memory,
1364 max_page_size: 100,
1365 retention_window_seconds: 0,
1366 purge_interval_seconds: 300,
1367 purge_batch_size: 1000,
1368 max_purge_per_tick: 100_000,
1369 }
1370 }
1371}
1372
1373#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
1374#[serde(rename_all = "lowercase")]
1375pub enum PushStorageDriver {
1376 #[default]
1377 Memory,
1378 Postgres,
1379 Mysql,
1380 DynamoDb,
1381 SurrealDb,
1382 ScyllaDb,
1383}
1384
1385impl FromStr for PushStorageDriver {
1386 type Err = String;
1387 fn from_str(s: &str) -> Result<Self, Self::Err> {
1388 match s.to_lowercase().as_str() {
1389 "memory" => Ok(Self::Memory),
1390 "postgres" | "postgresql" | "pgsql" => Ok(Self::Postgres),
1391 "mysql" => Ok(Self::Mysql),
1392 "dynamodb" => Ok(Self::DynamoDb),
1393 "surrealdb" | "surreal" => Ok(Self::SurrealDb),
1394 "scylladb" | "scylla" => Ok(Self::ScyllaDb),
1395 _ => Err(format!("Unknown push storage driver: {s}")),
1396 }
1397 }
1398}
1399
1400#[derive(Debug, Clone, Serialize, Deserialize, PartialEq, Eq, Default)]
1401#[serde(rename_all = "lowercase")]
1402pub enum PushQueueDriver {
1403 #[default]
1404 Memory,
1405 Redis,
1406 #[serde(rename = "redis-cluster")]
1407 RedisCluster,
1408 Nats,
1409 Pulsar,
1410 RabbitMq,
1411 #[serde(rename = "google-pubsub")]
1412 GooglePubsub,
1413 Kafka,
1414 Iggy,
1415 Sqs,
1416 Sns,
1417}
1418
1419impl FromStr for PushQueueDriver {
1420 type Err = String;
1421 fn from_str(s: &str) -> Result<Self, Self::Err> {
1422 match s.to_lowercase().as_str() {
1423 "memory" => Ok(Self::Memory),
1424 "redis" => Ok(Self::Redis),
1425 "redis-cluster" | "redis_cluster" => Ok(Self::RedisCluster),
1426 "nats" => Ok(Self::Nats),
1427 "pulsar" => Ok(Self::Pulsar),
1428 "rabbitmq" | "rabbit-mq" => Ok(Self::RabbitMq),
1429 "google-pubsub" | "google_pubsub" | "gcp-pubsub" | "pubsub" => Ok(Self::GooglePubsub),
1430 "kafka" => Ok(Self::Kafka),
1431 "iggy" | "apache-iggy" | "apache_iggy" => Ok(Self::Iggy),
1432 "sqs" => Ok(Self::Sqs),
1433 "sns" => Ok(Self::Sns),
1434 _ => Err(format!("Unknown push queue driver: {s}")),
1435 }
1436 }
1437}
1438
1439#[derive(Debug, Clone, Serialize, Deserialize)]
1440#[serde(default)]
1441pub struct PushConfig {
1442 pub storage_driver: PushStorageDriver,
1443 pub queue_driver: PushQueueDriver,
1444 pub fcm_enabled: bool,
1445 pub apns_enabled: bool,
1446 pub webpush_enabled: bool,
1447 pub hms_enabled: bool,
1448 pub wns_enabled: bool,
1449 pub fcm_credential_ref: Option<String>,
1450 pub apns_credential_ref: Option<String>,
1451 pub webpush_credential_ref: Option<String>,
1452 pub hms_credential_ref: Option<String>,
1453 pub wns_credential_ref: Option<String>,
1454 pub accept_worker_count: u32,
1455 pub planner_worker_count: u32,
1456 pub shard_worker_count: u32,
1457 pub dispatch_worker_count: u32,
1458 pub feedback_worker_count: u32,
1459 pub queue_partition_count: u32,
1460 pub channel_shard_count: u32,
1461 pub fanout_fast_threshold: u64,
1462 pub fanout_shard_size: u64,
1463 pub fanout_sync_threshold: u64,
1464 pub backpressure_lag_threshold_secs: u64,
1465 pub publish_status_ttl_days: u64,
1466 pub stale_device_max_age_days: u64,
1467 pub retry: PushRetryConfig,
1468 pub circuit_breaker: PushCircuitBreakerConfig,
1469 pub default_quotas: PushDefaultQuotas,
1470 pub credential_encryption_key: Option<String>,
1471 pub kms_key_ref: Option<String>,
1472 pub vault_secret_ref: Option<String>,
1473 pub dry_run: bool,
1474 pub analytics_enabled: bool,
1475 pub analytics_retention_days: u64,
1476 pub payload_redaction: PushPayloadRedactionConfig,
1477 pub scheduler_interval_secs: u64,
1478}
1479
1480impl Default for PushConfig {
1481 fn default() -> Self {
1482 Self {
1483 storage_driver: PushStorageDriver::Memory,
1484 queue_driver: PushQueueDriver::Memory,
1485 fcm_enabled: false,
1486 apns_enabled: false,
1487 webpush_enabled: false,
1488 hms_enabled: false,
1489 wns_enabled: false,
1490 fcm_credential_ref: None,
1491 apns_credential_ref: None,
1492 webpush_credential_ref: None,
1493 hms_credential_ref: None,
1494 wns_credential_ref: None,
1495 accept_worker_count: 1,
1496 planner_worker_count: 1,
1497 shard_worker_count: 1,
1498 dispatch_worker_count: 1,
1499 feedback_worker_count: 1,
1500 queue_partition_count: 1,
1501 channel_shard_count: 1,
1502 fanout_fast_threshold: 10_000,
1503 fanout_shard_size: 100_000,
1504 fanout_sync_threshold: 0,
1505 backpressure_lag_threshold_secs: 60,
1506 publish_status_ttl_days: 30,
1507 stale_device_max_age_days: 90,
1508 retry: PushRetryConfig::default(),
1509 circuit_breaker: PushCircuitBreakerConfig::default(),
1510 default_quotas: PushDefaultQuotas::default(),
1511 credential_encryption_key: None,
1512 kms_key_ref: None,
1513 vault_secret_ref: None,
1514 dry_run: false,
1515 analytics_enabled: false,
1516 analytics_retention_days: 30,
1517 payload_redaction: PushPayloadRedactionConfig::default(),
1518 scheduler_interval_secs: 5,
1519 }
1520 }
1521}
1522
1523#[derive(Debug, Clone, Serialize, Deserialize)]
1524#[serde(default)]
1525pub struct PushRetryConfig {
1526 pub max_attempts: u32,
1527 pub initial_backoff_ms: u64,
1528 pub max_backoff_ms: u64,
1529 pub max_elapsed_secs: u64,
1530 pub jitter: bool,
1531 pub respect_retry_after: bool,
1532}
1533
1534impl Default for PushRetryConfig {
1535 fn default() -> Self {
1536 Self {
1537 max_attempts: 5,
1538 initial_backoff_ms: 1_000,
1539 max_backoff_ms: 60_000,
1540 max_elapsed_secs: 86_400,
1541 jitter: true,
1542 respect_retry_after: true,
1543 }
1544 }
1545}
1546
1547#[derive(Debug, Clone, Serialize, Deserialize)]
1548#[serde(default)]
1549pub struct PushCircuitBreakerConfig {
1550 pub failure_threshold: u32,
1551 pub cooldown_secs: u64,
1552 pub half_open_max_inflight: u32,
1553}
1554
1555impl Default for PushCircuitBreakerConfig {
1556 fn default() -> Self {
1557 Self {
1558 failure_threshold: 5,
1559 cooldown_secs: 60,
1560 half_open_max_inflight: 10,
1561 }
1562 }
1563}
1564
1565#[derive(Debug, Clone, Serialize, Deserialize)]
1566#[serde(default)]
1567pub struct PushDefaultQuotas {
1568 pub acceptance_rps: u64,
1569 pub delivery_quota_daily: u64,
1570 pub fanout_max: u64,
1571 pub inflight_max: u64,
1572}
1573
1574impl Default for PushDefaultQuotas {
1575 fn default() -> Self {
1576 Self {
1577 acceptance_rps: 100,
1578 delivery_quota_daily: 0,
1579 fanout_max: 0,
1580 inflight_max: 1_000,
1581 }
1582 }
1583}
1584
1585#[derive(Debug, Clone, Serialize, Deserialize)]
1586#[serde(default)]
1587pub struct PushPayloadRedactionConfig {
1588 pub redact_payload: bool,
1589 pub redact_template_data: bool,
1590 pub redact_provider_overrides: bool,
1591 pub allow_debug_payload_logging: bool,
1592}
1593
1594impl Default for PushPayloadRedactionConfig {
1595 fn default() -> Self {
1596 Self {
1597 redact_payload: true,
1598 redact_template_data: true,
1599 redact_provider_overrides: true,
1600 allow_debug_payload_logging: false,
1601 }
1602 }
1603}
1604
1605#[derive(Debug, Clone, Serialize, Deserialize)]
1606#[serde(default)]
1607pub struct PresenceHistoryConfig {
1608 pub enabled: bool,
1609 pub retention_window_seconds: u64,
1610 pub max_page_size: usize,
1611 pub max_events_per_channel: Option<usize>,
1612 pub max_bytes_per_channel: Option<u64>,
1613}
1614
1615impl Default for PresenceHistoryConfig {
1616 fn default() -> Self {
1617 Self {
1618 enabled: false,
1619 retention_window_seconds: 86400,
1620 max_page_size: 100,
1621 max_events_per_channel: None,
1622 max_bytes_per_channel: None,
1623 }
1624 }
1625}
1626
1627#[derive(Debug, Clone, Serialize, Deserialize)]
1628#[serde(default)]
1629pub struct HttpApiConfig {
1630 pub request_limit_in_mb: u32,
1631 pub accept_traffic: AcceptTraffic,
1632 pub usage_enabled: bool,
1633}
1634
1635#[derive(Debug, Clone, Serialize, Deserialize)]
1636#[serde(default)]
1637pub struct AcceptTraffic {
1638 pub memory_threshold: f64,
1639}
1640
1641#[derive(Debug, Clone, Serialize, Deserialize)]
1642#[serde(default)]
1643pub struct InstanceConfig {
1644 pub process_id: String,
1645}
1646
1647#[derive(Debug, Clone, Serialize, Deserialize)]
1648#[serde(default)]
1649pub struct MetricsConfig {
1650 pub enabled: bool,
1651 pub driver: MetricsDriver,
1652 pub host: String,
1653 pub prometheus: PrometheusConfig,
1654 pub port: u16,
1655}
1656
1657#[derive(Debug, Clone, Serialize, Deserialize)]
1658#[serde(default)]
1659pub struct PrometheusConfig {
1660 pub prefix: String,
1661}
1662
1663#[derive(Debug, Clone, Serialize, Deserialize)]
1664#[serde(default)]
1665pub struct LoggingConfig {
1666 pub colors_enabled: bool,
1667 pub include_target: bool,
1668}
1669
1670#[derive(Debug, Clone, Serialize, Deserialize)]
1671#[serde(default)]
1672pub struct PresenceConfig {
1673 pub max_members_per_channel: u32,
1674 pub max_member_size_in_kb: u32,
1675}
1676
1677#[derive(Debug, Clone, Serialize, Deserialize)]
1680#[serde(default)]
1681pub struct WebSocketConfig {
1682 pub max_messages: Option<usize>,
1683 pub max_bytes: Option<usize>,
1684 pub disconnect_on_buffer_full: bool,
1685 pub max_message_size: usize,
1686 pub max_frame_size: usize,
1687 pub write_buffer_size: usize,
1688 pub max_backpressure: usize,
1689 pub auto_ping: bool,
1690 pub ping_interval: u32,
1691 pub idle_timeout: u32,
1692 pub compression: String,
1693}
1694
1695impl Default for WebSocketConfig {
1696 fn default() -> Self {
1697 Self {
1698 max_messages: Some(1000),
1699 max_bytes: None,
1700 disconnect_on_buffer_full: true,
1701 max_message_size: 64 * 1024 * 1024,
1702 max_frame_size: 16 * 1024 * 1024,
1703 write_buffer_size: 16 * 1024,
1704 max_backpressure: 1024 * 1024,
1705 auto_ping: true,
1706 ping_interval: 30,
1707 idle_timeout: 120,
1708 compression: "disabled".to_string(),
1709 }
1710 }
1711}
1712
1713impl WebSocketConfig {
1714 pub fn to_buffer_config(&self) -> crate::websocket::WebSocketBufferConfig {
1716 use crate::websocket::{BufferLimit, WebSocketBufferConfig};
1717
1718 let limit = match (self.max_messages, self.max_bytes) {
1719 (Some(messages), Some(bytes)) => BufferLimit::Both { messages, bytes },
1720 (Some(messages), None) => BufferLimit::Messages(messages),
1721 (None, Some(bytes)) => BufferLimit::Bytes(bytes),
1722 (None, None) => BufferLimit::Messages(1000),
1723 };
1724
1725 WebSocketBufferConfig {
1726 limit,
1727 disconnect_on_full: self.disconnect_on_buffer_full,
1728 }
1729 }
1730
1731 pub fn to_sockudo_ws_config(
1733 &self,
1734 websocket_max_payload_kb: u32,
1735 activity_timeout: u64,
1736 ) -> sockudo_ws::Config {
1737 use sockudo_ws::Compression;
1738
1739 let compression = match self.compression.to_lowercase().as_str() {
1740 "dedicated" => Compression::Dedicated,
1741 "shared" => Compression::Shared,
1742 "window256b" => Compression::Window256B,
1743 "window1kb" => Compression::Window1KB,
1744 "window2kb" => Compression::Window2KB,
1745 "window4kb" => Compression::Window4KB,
1746 "window8kb" => Compression::Window8KB,
1747 "window16kb" => Compression::Window16KB,
1748 "window32kb" => Compression::Window32KB,
1749 _ => Compression::Disabled,
1750 };
1751
1752 sockudo_ws::Config::builder()
1753 .max_payload_length(
1754 self.max_bytes
1755 .unwrap_or(websocket_max_payload_kb as usize * 1024),
1756 )
1757 .max_message_size(self.max_message_size)
1758 .max_frame_size(self.max_frame_size)
1759 .write_buffer_size(self.write_buffer_size)
1760 .max_backpressure(self.max_backpressure)
1761 .idle_timeout(self.idle_timeout)
1762 .auto_ping(self.auto_ping)
1763 .ping_interval(self.ping_interval.max((activity_timeout / 2).max(5) as u32))
1764 .compression(compression)
1765 .build()
1766 }
1767}
1768
1769#[derive(Debug, Clone, Serialize, Deserialize, Default)]
1770#[serde(default)]
1771pub struct QueueConfig {
1772 pub driver: QueueDriver,
1773 pub redis: RedisQueueConfig,
1774 pub redis_cluster: RedisClusterQueueConfig,
1775 pub nats: NatsAdapterConfig,
1776 pub pulsar: PulsarAdapterConfig,
1777 pub rabbitmq: RabbitMqAdapterConfig,
1778 pub google_pubsub: GooglePubSubAdapterConfig,
1779 pub kafka: KafkaAdapterConfig,
1780 pub iggy: IggyConfig,
1781 pub sqs: SqsQueueConfig,
1782 pub sns: SnsQueueConfig,
1783}
1784
1785#[derive(Debug, Clone, Serialize, Deserialize)]
1786#[serde(default)]
1787pub struct RedisQueueConfig {
1788 pub concurrency: u32,
1789 pub prefix: Option<String>,
1790 pub url_override: Option<String>,
1791 pub cluster_mode: bool,
1792}
1793
1794#[derive(Clone, Debug, Serialize, Deserialize)]
1795#[serde(default)]
1796pub struct RateLimit {
1797 pub max_requests: u32,
1798 pub window_seconds: u64,
1799 pub identifier: Option<String>,
1800 pub trust_hops: Option<u32>,
1801}
1802
1803#[derive(Debug, Clone, Serialize, Deserialize)]
1804#[serde(default)]
1805pub struct RateLimiterConfig {
1806 pub enabled: bool,
1807 pub driver: CacheDriver,
1808 pub api_rate_limit: RateLimit,
1809 pub websocket_rate_limit: RateLimit,
1810 pub redis: RedisConfig,
1811}
1812
1813#[derive(Debug, Clone, Serialize, Deserialize)]
1814#[serde(default)]
1815pub struct SslConfig {
1816 pub enabled: bool,
1817 pub cert_path: String,
1818 pub key_path: String,
1819 pub passphrase: Option<String>,
1820 pub ca_path: Option<String>,
1821 pub redirect_http: bool,
1822 pub http_port: Option<u16>,
1823}
1824
1825#[derive(Debug, Clone, Serialize, Deserialize)]
1826#[serde(default)]
1827pub struct WebhooksConfig {
1828 pub batching: BatchingConfig,
1829 pub retry: WebhookRetryConfig,
1830 pub request_timeout_ms: u64,
1831}
1832
1833#[derive(Debug, Clone, Serialize, Deserialize)]
1834#[serde(default)]
1835pub struct BatchingConfig {
1836 pub enabled: bool,
1837 pub duration: u64,
1838 pub size: usize,
1839}
1840
1841#[derive(Debug, Clone, Serialize, Deserialize)]
1842#[serde(default)]
1843pub struct WebhookRetryConfig {
1844 pub enabled: bool,
1845 pub max_attempts: Option<u32>,
1846 pub max_elapsed_time_ms: u64,
1847 pub initial_backoff_ms: u64,
1848 pub max_backoff_ms: u64,
1849}
1850
1851impl Default for WebhooksConfig {
1852 fn default() -> Self {
1853 Self {
1854 batching: BatchingConfig::default(),
1855 retry: WebhookRetryConfig::default(),
1856 request_timeout_ms: 10_000,
1857 }
1858 }
1859}
1860
1861impl Default for WebhookRetryConfig {
1862 fn default() -> Self {
1863 Self {
1864 enabled: true,
1865 max_attempts: None,
1866 max_elapsed_time_ms: 300_000,
1867 initial_backoff_ms: 1_000,
1868 max_backoff_ms: 60_000,
1869 }
1870 }
1871}
1872
1873#[derive(Debug, Clone, Serialize, Deserialize)]
1874#[serde(default)]
1875pub struct ClusterHealthConfig {
1876 pub enabled: bool,
1877 pub heartbeat_interval_ms: u64,
1878 pub node_timeout_ms: u64,
1879 pub cleanup_interval_ms: u64,
1880}
1881
1882#[derive(Debug, Clone, Serialize, Deserialize)]
1883#[serde(default)]
1884pub struct UnixSocketConfig {
1885 pub enabled: bool,
1886 pub path: String,
1887 #[serde(deserialize_with = "deserialize_octal_permission")]
1888 pub permission_mode: u32,
1889}
1890
1891#[derive(Debug, Clone, Serialize, Deserialize)]
1892#[serde(default)]
1893pub struct DeltaCompressionOptionsConfig {
1894 pub enabled: bool,
1895 pub algorithm: String,
1896 pub full_message_interval: u32,
1897 pub min_message_size: usize,
1898 pub max_state_age_secs: u64,
1899 pub max_channel_states_per_socket: usize,
1900 pub max_conflation_states_per_channel: Option<usize>,
1901 pub conflation_key_path: Option<String>,
1902 pub cluster_coordination: bool,
1903 pub coordination_backend: DeltaCoordinationBackend,
1904 pub omit_delta_algorithm: bool,
1905}
1906
1907#[derive(Debug, Clone, Serialize, Deserialize)]
1909#[serde(default)]
1910pub struct CleanupConfig {
1911 pub queue_buffer_size: usize,
1912 pub batch_size: usize,
1913 pub batch_timeout_ms: u64,
1914 pub worker_threads: WorkerThreadsConfig,
1915 pub max_retry_attempts: u32,
1916 pub async_enabled: bool,
1917 pub fallback_to_sync: bool,
1918}
1919
1920#[derive(Debug, Clone)]
1922pub enum WorkerThreadsConfig {
1923 Auto,
1924 Fixed(usize),
1925}
1926
1927impl serde::Serialize for WorkerThreadsConfig {
1928 fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
1929 where
1930 S: serde::Serializer,
1931 {
1932 match self {
1933 WorkerThreadsConfig::Auto => serializer.serialize_str("auto"),
1934 WorkerThreadsConfig::Fixed(n) => serializer.serialize_u64(*n as u64),
1935 }
1936 }
1937}
1938
1939impl<'de> serde::Deserialize<'de> for WorkerThreadsConfig {
1940 fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
1941 where
1942 D: serde::Deserializer<'de>,
1943 {
1944 use serde::de;
1945 struct WorkerThreadsVisitor;
1946 impl<'de> de::Visitor<'de> for WorkerThreadsVisitor {
1947 type Value = WorkerThreadsConfig;
1948
1949 fn expecting(&self, formatter: &mut std::fmt::Formatter) -> std::fmt::Result {
1950 formatter.write_str(r#""auto" or a positive integer"#)
1951 }
1952
1953 fn visit_str<E: de::Error>(self, value: &str) -> Result<Self::Value, E> {
1954 if value.eq_ignore_ascii_case("auto") {
1955 Ok(WorkerThreadsConfig::Auto)
1956 } else if let Ok(n) = value.parse::<usize>() {
1957 Ok(WorkerThreadsConfig::Fixed(n))
1958 } else {
1959 Err(E::custom(format!(
1960 "expected 'auto' or a number, got '{value}'"
1961 )))
1962 }
1963 }
1964
1965 fn visit_u64<E: de::Error>(self, value: u64) -> Result<Self::Value, E> {
1966 Ok(WorkerThreadsConfig::Fixed(value as usize))
1967 }
1968
1969 fn visit_i64<E: de::Error>(self, value: i64) -> Result<Self::Value, E> {
1970 if value >= 0 {
1971 Ok(WorkerThreadsConfig::Fixed(value as usize))
1972 } else {
1973 Err(E::custom("worker_threads must be non-negative"))
1974 }
1975 }
1976 }
1977 deserializer.deserialize_any(WorkerThreadsVisitor)
1978 }
1979}
1980
1981impl Default for CleanupConfig {
1982 fn default() -> Self {
1983 Self {
1984 queue_buffer_size: 1024,
1985 batch_size: 64,
1986 batch_timeout_ms: 100,
1987 worker_threads: WorkerThreadsConfig::Auto,
1988 max_retry_attempts: 3,
1989 async_enabled: true,
1990 fallback_to_sync: true,
1991 }
1992 }
1993}
1994
1995impl CleanupConfig {
1996 pub fn validate(&self) -> Result<(), String> {
1997 if self.queue_buffer_size == 0 {
1998 return Err("queue_buffer_size must be greater than 0".to_string());
1999 }
2000 if self.batch_size == 0 {
2001 return Err("batch_size must be greater than 0".to_string());
2002 }
2003 if self.batch_timeout_ms == 0 {
2004 return Err("batch_timeout_ms must be greater than 0".to_string());
2005 }
2006 if let WorkerThreadsConfig::Fixed(n) = self.worker_threads
2007 && n == 0
2008 {
2009 return Err("worker_threads must be greater than 0 when using fixed count".to_string());
2010 }
2011 Ok(())
2012 }
2013}
2014
2015#[derive(Debug, Clone, Serialize, Deserialize, Default)]
2016#[serde(default)]
2017pub struct TagFilteringConfig {
2018 #[serde(default)]
2019 pub enabled: bool,
2020 #[serde(default = "default_true")]
2021 pub enable_tags: bool,
2022}
2023
2024fn default_true() -> bool {
2025 true
2026}
2027
2028impl Default for ServerOptions {
2031 fn default() -> Self {
2032 Self {
2033 adapter: AdapterConfig::default(),
2034 app_manager: AppManagerConfig::default(),
2035 cache: CacheConfig::default(),
2036 channel_limits: ChannelLimits::default(),
2037 cors: CorsConfig::default(),
2038 database: DatabaseConfig::default(),
2039 database_pooling: DatabasePooling::default(),
2040 debug: false,
2041 tag_filtering: TagFilteringConfig::default(),
2042 event_limits: EventLimits::default(),
2043 host: "0.0.0.0".to_string(),
2044 http_api: HttpApiConfig::default(),
2045 instance: InstanceConfig::default(),
2046 logging: None,
2047 metrics: MetricsConfig::default(),
2048 mode: "production".to_string(),
2049 port: 6001,
2050 path_prefix: "/".to_string(),
2051 presence: PresenceConfig::default(),
2052 queue: QueueConfig::default(),
2053 rate_limiter: RateLimiterConfig::default(),
2054 shutdown_grace_period: 10,
2055 ssl: SslConfig::default(),
2056 user_authentication_timeout: 3600,
2057 webhooks: WebhooksConfig::default(),
2058 websocket_max_payload_kb: 64,
2059 cleanup: CleanupConfig::default(),
2060 activity_timeout: 120,
2061 cluster_health: ClusterHealthConfig::default(),
2062 unix_socket: UnixSocketConfig::default(),
2063 delta_compression: DeltaCompressionOptionsConfig::default(),
2064 websocket: WebSocketConfig::default(),
2065 connection_recovery: ConnectionRecoveryConfig::default(),
2066 history: HistoryConfig::default(),
2067 presence_history: PresenceHistoryConfig::default(),
2068 idempotency: IdempotencyConfig::default(),
2069 ephemeral: EphemeralConfig::default(),
2070 echo_control: EchoControlConfig::default(),
2071 event_name_filtering: EventNameFilteringConfig::default(),
2072 versioned_messages: VersionedMessagesConfig::default(),
2073 annotations: AnnotationsConfig::default(),
2074 push: PushConfig::default(),
2075 health_check_timeout_ms: 400,
2076 }
2077 }
2078}
2079
2080impl Default for SqsQueueConfig {
2081 fn default() -> Self {
2082 Self {
2083 region: "us-east-1".to_string(),
2084 queue_url_prefix: None,
2085 visibility_timeout: 30,
2086 endpoint_url: None,
2087 max_messages: 10,
2088 wait_time_seconds: 5,
2089 concurrency: 5,
2090 fifo: false,
2091 message_group_id: Some("default".to_string()),
2092 }
2093 }
2094}
2095
2096impl Default for SnsQueueConfig {
2097 fn default() -> Self {
2098 Self {
2099 region: "us-east-1".to_string(),
2100 topic_arn: String::new(),
2101 endpoint_url: None,
2102 }
2103 }
2104}
2105
2106impl Default for RedisAdapterConfig {
2107 fn default() -> Self {
2108 Self {
2109 requests_timeout: 5000,
2110 prefix: "sockudo_adapter:".to_string(),
2111 redis_pub_options: AHashMap::new(),
2112 redis_sub_options: AHashMap::new(),
2113 cluster_mode: false,
2114 }
2115 }
2116}
2117
2118impl Default for RedisClusterAdapterConfig {
2119 fn default() -> Self {
2120 Self {
2121 nodes: vec![],
2122 prefix: "sockudo_adapter:".to_string(),
2123 request_timeout_ms: 1000,
2124 use_connection_manager: true,
2125 use_sharded_pubsub: false,
2126 }
2127 }
2128}
2129
2130impl Default for NatsAdapterConfig {
2131 fn default() -> Self {
2132 Self {
2133 servers: vec!["nats://localhost:4222".to_string()],
2134 prefix: "sockudo_adapter:".to_string(),
2135 request_timeout_ms: 5000,
2136 username: None,
2137 password: None,
2138 token: None,
2139 connection_timeout_ms: 5000,
2140 nodes_number: None,
2141 discovery_max_wait_ms: 1000,
2142 discovery_idle_wait_ms: 150,
2143 subscription_capacity: None,
2144 client_capacity: None,
2145 max_reconnects: None,
2146 presence_sync_chunk_size: None,
2147 }
2148 }
2149}
2150
2151impl Default for PulsarAdapterConfig {
2152 fn default() -> Self {
2153 Self {
2154 url: "pulsar://127.0.0.1:6650".to_string(),
2155 prefix: "sockudo-adapter".to_string(),
2156 request_timeout_ms: 5000,
2157 token: None,
2158 nodes_number: None,
2159 }
2160 }
2161}
2162
2163impl Default for RabbitMqAdapterConfig {
2164 fn default() -> Self {
2165 Self {
2166 url: "amqp://guest:guest@127.0.0.1:5672/%2f".to_string(),
2167 prefix: "sockudo_adapter".to_string(),
2168 request_timeout_ms: 5000,
2169 connection_timeout_ms: 5000,
2170 nodes_number: None,
2171 }
2172 }
2173}
2174
2175impl Default for GooglePubSubAdapterConfig {
2176 fn default() -> Self {
2177 Self {
2178 project_id: "".to_string(),
2179 prefix: "sockudo-adapter".to_string(),
2180 request_timeout_ms: 5000,
2181 emulator_host: None,
2182 nodes_number: None,
2183 }
2184 }
2185}
2186
2187impl Default for KafkaAdapterConfig {
2188 fn default() -> Self {
2189 Self {
2190 brokers: vec!["localhost:9092".to_string()],
2191 prefix: "sockudo_adapter".to_string(),
2192 request_timeout_ms: 5000,
2193 security_protocol: None,
2194 sasl_mechanism: None,
2195 sasl_username: None,
2196 sasl_password: None,
2197 nodes_number: None,
2198 }
2199 }
2200}
2201
2202impl Default for IggyConfig {
2203 fn default() -> Self {
2204 Self {
2205 connection_string: "iggy://iggy:iggy@127.0.0.1:8090".to_string(),
2206 username: None,
2207 password: None,
2208 consumer_name: None,
2209 stream: "sockudo".to_string(),
2210 topic_prefix: "sockudo-adapter".to_string(),
2211 queue_topic_prefix: "sockudo-queue".to_string(),
2212 consumer_group_prefix: "sockudo-workers".to_string(),
2213 request_timeout_ms: 5000,
2214 poll_interval_ms: 50,
2215 poll_batch_size: 100,
2216 partitions_count: 1,
2217 partition_id: 0,
2218 auto_create: true,
2219 start_from_latest: true,
2220 nodes_number: None,
2221 }
2222 }
2223}
2224
2225impl Default for CacheSettings {
2226 fn default() -> Self {
2227 Self {
2228 enabled: true,
2229 ttl: 300,
2230 }
2231 }
2232}
2233
2234impl Default for MemoryCacheOptions {
2235 fn default() -> Self {
2236 Self {
2237 ttl: 300,
2238 cleanup_interval: 60,
2239 max_capacity: 10000,
2240 }
2241 }
2242}
2243
2244impl Default for CacheConfig {
2245 fn default() -> Self {
2246 Self {
2247 driver: CacheDriver::default(),
2248 redis: RedisConfig {
2249 prefix: Some("sockudo_cache:".to_string()),
2250 url_override: None,
2251 cluster_mode: false,
2252 },
2253 memory: MemoryCacheOptions::default(),
2254 }
2255 }
2256}
2257
2258impl Default for ChannelLimits {
2259 fn default() -> Self {
2260 Self {
2261 max_name_length: 200,
2262 cache_ttl: 3600,
2263 }
2264 }
2265}
2266impl Default for CorsConfig {
2267 fn default() -> Self {
2268 Self {
2269 credentials: true,
2270 origin: vec!["*".to_string()],
2271 methods: vec!["GET".to_string(), "POST".to_string(), "OPTIONS".to_string()],
2272 allowed_headers: vec![
2273 "Authorization".to_string(),
2274 "Content-Type".to_string(),
2275 "X-Requested-With".to_string(),
2276 "Accept".to_string(),
2277 ],
2278 }
2279 }
2280}
2281
2282impl Default for DatabaseConnection {
2283 fn default() -> Self {
2284 Self {
2285 host: "localhost".to_string(),
2286 port: 3306,
2287 username: "root".to_string(),
2288 password: "".to_string(),
2289 database: "sockudo".to_string(),
2290 table_name: "applications".to_string(),
2291 connection_pool_size: 10,
2292 pool_min: None,
2293 pool_max: None,
2294 cache_ttl: 300,
2295 cache_cleanup_interval: 60,
2296 cache_max_capacity: 100,
2297 }
2298 }
2299}
2300
2301impl Default for RedisConnection {
2302 fn default() -> Self {
2303 Self {
2304 host: "127.0.0.1".to_string(),
2305 port: 6379,
2306 db: 0,
2307 username: None,
2308 password: None,
2309 key_prefix: "sockudo:".to_string(),
2310 sentinels: Vec::new(),
2311 sentinel_password: None,
2312 name: "mymaster".to_string(),
2313 cluster: RedisClusterConnection::default(),
2314 cluster_nodes: Vec::new(),
2315 }
2316 }
2317}
2318
2319impl Default for RedisSentinel {
2320 fn default() -> Self {
2321 Self {
2322 host: "localhost".to_string(),
2323 port: 26379,
2324 }
2325 }
2326}
2327
2328impl Default for ClusterNode {
2329 fn default() -> Self {
2330 Self {
2331 host: "127.0.0.1".to_string(),
2332 port: 7000,
2333 }
2334 }
2335}
2336
2337impl Default for DatabasePooling {
2338 fn default() -> Self {
2339 Self {
2340 enabled: true,
2341 min: 2,
2342 max: 10,
2343 }
2344 }
2345}
2346
2347impl Default for EventLimits {
2348 fn default() -> Self {
2349 Self {
2350 max_channels_at_once: 100,
2351 max_name_length: 200,
2352 max_payload_in_kb: 100,
2353 max_batch_size: 10,
2354 }
2355 }
2356}
2357
2358impl Default for IdempotencyConfig {
2359 fn default() -> Self {
2360 Self {
2361 enabled: true,
2362 ttl_seconds: 120,
2363 max_key_length: 128,
2364 }
2365 }
2366}
2367
2368impl Default for EphemeralConfig {
2369 fn default() -> Self {
2370 Self { enabled: true }
2371 }
2372}
2373
2374impl Default for EchoControlConfig {
2375 fn default() -> Self {
2376 Self {
2377 enabled: true,
2378 default_echo_messages: true,
2379 }
2380 }
2381}
2382
2383impl Default for EventNameFilteringConfig {
2384 fn default() -> Self {
2385 Self {
2386 enabled: true,
2387 max_events_per_filter: 50,
2388 max_event_name_length: 200,
2389 }
2390 }
2391}
2392
2393impl Default for ConnectionRecoveryConfig {
2394 fn default() -> Self {
2395 Self {
2396 enabled: false,
2397 buffer_ttl_seconds: 120,
2398 max_buffer_size: 100,
2399 }
2400 }
2401}
2402
2403impl Default for HttpApiConfig {
2404 fn default() -> Self {
2405 Self {
2406 request_limit_in_mb: 10,
2407 accept_traffic: AcceptTraffic::default(),
2408 usage_enabled: true,
2409 }
2410 }
2411}
2412
2413impl Default for AcceptTraffic {
2414 fn default() -> Self {
2415 Self {
2416 memory_threshold: 0.90,
2417 }
2418 }
2419}
2420
2421impl Default for InstanceConfig {
2422 fn default() -> Self {
2423 Self {
2424 process_id: uuid::Uuid::new_v4().to_string(),
2425 }
2426 }
2427}
2428
2429impl Default for LoggingConfig {
2430 fn default() -> Self {
2431 Self {
2432 colors_enabled: true,
2433 include_target: true,
2434 }
2435 }
2436}
2437
2438impl Default for MetricsConfig {
2439 fn default() -> Self {
2440 Self {
2441 enabled: true,
2442 driver: MetricsDriver::default(),
2443 host: "0.0.0.0".to_string(),
2444 prometheus: PrometheusConfig::default(),
2445 port: 9601,
2446 }
2447 }
2448}
2449
2450impl Default for PrometheusConfig {
2451 fn default() -> Self {
2452 Self {
2453 prefix: "sockudo_".to_string(),
2454 }
2455 }
2456}
2457
2458impl Default for PresenceConfig {
2459 fn default() -> Self {
2460 Self {
2461 max_members_per_channel: 100,
2462 max_member_size_in_kb: 2,
2463 }
2464 }
2465}
2466
2467impl Default for RedisQueueConfig {
2468 fn default() -> Self {
2469 Self {
2470 concurrency: 5,
2471 prefix: Some("sockudo_queue:".to_string()),
2472 url_override: None,
2473 cluster_mode: false,
2474 }
2475 }
2476}
2477
2478impl Default for RateLimit {
2479 fn default() -> Self {
2480 Self {
2481 max_requests: 60,
2482 window_seconds: 60,
2483 identifier: Some("default".to_string()),
2484 trust_hops: Some(0),
2485 }
2486 }
2487}
2488
2489impl Default for RateLimiterConfig {
2490 fn default() -> Self {
2491 Self {
2492 enabled: true,
2493 driver: CacheDriver::Memory,
2494 api_rate_limit: RateLimit {
2495 max_requests: 100,
2496 window_seconds: 60,
2497 identifier: Some("api".to_string()),
2498 trust_hops: Some(0),
2499 },
2500 websocket_rate_limit: RateLimit {
2501 max_requests: 20,
2502 window_seconds: 60,
2503 identifier: Some("websocket_connect".to_string()),
2504 trust_hops: Some(0),
2505 },
2506 redis: RedisConfig {
2507 prefix: Some("sockudo_rl:".to_string()),
2508 url_override: None,
2509 cluster_mode: false,
2510 },
2511 }
2512 }
2513}
2514
2515impl Default for SslConfig {
2516 fn default() -> Self {
2517 Self {
2518 enabled: false,
2519 cert_path: "".to_string(),
2520 key_path: "".to_string(),
2521 passphrase: None,
2522 ca_path: None,
2523 redirect_http: false,
2524 http_port: Some(80),
2525 }
2526 }
2527}
2528
2529impl Default for BatchingConfig {
2530 fn default() -> Self {
2531 Self {
2532 enabled: true,
2533 duration: 50,
2534 size: 100,
2535 }
2536 }
2537}
2538
2539impl Default for ClusterHealthConfig {
2540 fn default() -> Self {
2541 Self {
2542 enabled: true,
2543 heartbeat_interval_ms: 10000,
2544 node_timeout_ms: 30000,
2545 cleanup_interval_ms: 10000,
2546 }
2547 }
2548}
2549
2550impl Default for UnixSocketConfig {
2551 fn default() -> Self {
2552 Self {
2553 enabled: false,
2554 path: "/var/run/sockudo/sockudo.sock".to_string(),
2555 permission_mode: 0o660,
2556 }
2557 }
2558}
2559
2560impl Default for DeltaCompressionOptionsConfig {
2561 fn default() -> Self {
2562 Self {
2563 enabled: true,
2564 algorithm: "fossil".to_string(),
2565 full_message_interval: 10,
2566 min_message_size: 100,
2567 max_state_age_secs: 300,
2568 max_channel_states_per_socket: 100,
2569 max_conflation_states_per_channel: Some(100),
2570 conflation_key_path: None,
2571 cluster_coordination: false,
2572 coordination_backend: DeltaCoordinationBackend::Auto,
2573 omit_delta_algorithm: false,
2574 }
2575 }
2576}
2577
2578#[cfg(test)]
2579mod tests {
2580 use super::{
2581 DeltaCoordinationBackend, PushQueueDriver, PushStorageDriver, QueueDriver, ServerOptions,
2582 VersionStoreDriver,
2583 };
2584 use crate::app::{App, AppPolicy};
2585 use std::str::FromStr;
2586
2587 const APP_BOOTSTRAP_ENV_KEYS: &[&str] = &[
2588 "SOCKUDO_DEFAULT_APP_ID",
2589 "SOCKUDO_DEFAULT_APP_KEY",
2590 "SOCKUDO_DEFAULT_APP_SECRET",
2591 "SOCKUDO_DEFAULT_APP_ENABLED",
2592 "SOCKUDO_SKIP_INLINE_APPS",
2593 "APP_MANAGER_REGISTER_INLINE_APPS",
2594 ];
2595
2596 struct EnvGuard {
2597 previous: Vec<(&'static str, Option<String>)>,
2598 }
2599
2600 impl EnvGuard {
2601 fn app_bootstrap(overrides: &[(&'static str, &'static str)]) -> Self {
2602 let previous = APP_BOOTSTRAP_ENV_KEYS
2603 .iter()
2604 .map(|key| (*key, std::env::var(key).ok()))
2605 .collect();
2606
2607 unsafe {
2610 for key in APP_BOOTSTRAP_ENV_KEYS {
2611 std::env::remove_var(key);
2612 }
2613 for (key, value) in overrides {
2614 std::env::set_var(key, value);
2615 }
2616 }
2617
2618 Self { previous }
2619 }
2620 }
2621
2622 impl Drop for EnvGuard {
2623 fn drop(&mut self) {
2624 unsafe {
2627 for (key, value) in &self.previous {
2628 if let Some(value) = value {
2629 std::env::set_var(key, value);
2630 } else {
2631 std::env::remove_var(key);
2632 }
2633 }
2634 }
2635 }
2636 }
2637
2638 fn inline_test_app() -> App {
2639 App::from_policy(
2640 "app-id".to_string(),
2641 "app-key".to_string(),
2642 "app-secret".to_string(),
2643 true,
2644 AppPolicy::default(),
2645 )
2646 }
2647
2648 #[test]
2649 fn queue_driver_parses_broker_backends() {
2650 assert_eq!(
2651 QueueDriver::from_str("rabbitmq").unwrap(),
2652 QueueDriver::RabbitMq
2653 );
2654 assert_eq!(QueueDriver::from_str("kafka").unwrap(), QueueDriver::Kafka);
2655 assert_eq!(
2656 QueueDriver::from_str("pulsar").unwrap(),
2657 QueueDriver::Pulsar
2658 );
2659 assert_eq!(
2660 QueueDriver::from_str("google-pubsub").unwrap(),
2661 QueueDriver::GooglePubSub
2662 );
2663 }
2664
2665 #[test]
2666 fn delta_coordination_backend_parses_expected_values() {
2667 assert_eq!(
2668 DeltaCoordinationBackend::from_str("auto").unwrap(),
2669 DeltaCoordinationBackend::Auto
2670 );
2671 assert_eq!(
2672 DeltaCoordinationBackend::from_str("redis-cluster").unwrap(),
2673 DeltaCoordinationBackend::RedisCluster
2674 );
2675 assert_eq!(
2676 DeltaCoordinationBackend::from_str("nats").unwrap(),
2677 DeltaCoordinationBackend::Nats
2678 );
2679 }
2680
2681 #[tokio::test]
2682 async fn app_bootstrap_env_overrides_inline_apps() {
2683 {
2684 let _env = EnvGuard::app_bootstrap(&[("SOCKUDO_DEFAULT_APP_ENABLED", "false")]);
2685 let mut options = ServerOptions::default();
2686 options.app_manager.array.apps.push(inline_test_app());
2687
2688 options.override_from_env().await.unwrap();
2689
2690 assert!(options.app_manager.array.apps.is_empty());
2691 }
2692
2693 {
2694 let _env = EnvGuard::app_bootstrap(&[("SOCKUDO_DEFAULT_APP_ENABLED", "true")]);
2695 let mut options = ServerOptions::default();
2696 options.app_manager.array.apps.push(inline_test_app());
2697
2698 options.override_from_env().await.unwrap();
2699
2700 assert_eq!(options.app_manager.array.apps.len(), 1);
2701 assert_eq!(options.app_manager.array.apps[0].id, "app-id");
2702 }
2703
2704 {
2705 let _env = EnvGuard::app_bootstrap(&[
2706 ("SOCKUDO_DEFAULT_APP_ID", "prod-app"),
2707 ("SOCKUDO_DEFAULT_APP_KEY", "prod-key"),
2708 ("SOCKUDO_DEFAULT_APP_SECRET", "prod-secret"),
2709 ("SOCKUDO_DEFAULT_APP_ENABLED", "true"),
2710 ]);
2711 let mut options = ServerOptions::default();
2712 options.app_manager.array.apps.push(inline_test_app());
2713
2714 options.override_from_env().await.unwrap();
2715
2716 assert_eq!(options.app_manager.array.apps.len(), 1);
2717 let app = &options.app_manager.array.apps[0];
2718 assert_eq!(app.id, "prod-app");
2719 assert_eq!(app.key, "prod-key");
2720 assert_eq!(app.secret, "prod-secret");
2721 assert!(app.enabled);
2722 }
2723
2724 {
2725 let _env = EnvGuard::app_bootstrap(&[("APP_MANAGER_REGISTER_INLINE_APPS", "false")]);
2726 let mut options = ServerOptions::default();
2727 options.app_manager.array.apps.push(inline_test_app());
2728
2729 options.override_from_env().await.unwrap();
2730
2731 assert!(options.app_manager.array.apps.is_empty());
2732 }
2733 }
2734
2735 #[tokio::test]
2736 async fn versioned_messages_driver_overrides_from_env() {
2737 let previous = std::env::var("VERSIONED_MESSAGES_DRIVER").ok();
2738 unsafe { std::env::set_var("VERSIONED_MESSAGES_DRIVER", "postgres") };
2741
2742 let mut options = ServerOptions::default();
2743 options.override_from_env().await.unwrap();
2744
2745 if let Some(previous) = previous {
2746 unsafe { std::env::set_var("VERSIONED_MESSAGES_DRIVER", previous) };
2748 } else {
2749 unsafe { std::env::remove_var("VERSIONED_MESSAGES_DRIVER") };
2751 }
2752
2753 assert_eq!(
2754 options.versioned_messages.driver,
2755 VersionStoreDriver::Postgres
2756 );
2757 }
2758
2759 #[tokio::test]
2760 async fn push_storage_driver_overrides_from_env() {
2761 let previous = std::env::var("PUSH_STORAGE_DRIVER").ok();
2762 unsafe { std::env::set_var("PUSH_STORAGE_DRIVER", "mysql") };
2765
2766 let mut options = ServerOptions::default();
2767 options.override_from_env().await.unwrap();
2768
2769 if let Some(previous) = previous {
2770 unsafe { std::env::set_var("PUSH_STORAGE_DRIVER", previous) };
2772 } else {
2773 unsafe { std::env::remove_var("PUSH_STORAGE_DRIVER") };
2775 }
2776
2777 assert_eq!(options.push.storage_driver, PushStorageDriver::Mysql);
2778 }
2779
2780 #[tokio::test]
2781 async fn push_queue_driver_overrides_from_env() {
2782 let previous = std::env::var("PUSH_QUEUE_DRIVER").ok();
2783 unsafe { std::env::set_var("PUSH_QUEUE_DRIVER", "redis-cluster") };
2786
2787 let mut options = ServerOptions::default();
2788 options.override_from_env().await.unwrap();
2789
2790 if let Some(previous) = previous {
2791 unsafe { std::env::set_var("PUSH_QUEUE_DRIVER", previous) };
2793 } else {
2794 unsafe { std::env::remove_var("PUSH_QUEUE_DRIVER") };
2796 }
2797
2798 assert_eq!(options.push.queue_driver, PushQueueDriver::RedisCluster);
2799 }
2800
2801 #[test]
2802 fn push_config_defaults_follow_capacity_model() {
2803 let options = ServerOptions::default();
2804
2805 assert_eq!(options.push.storage_driver, PushStorageDriver::Memory);
2806 assert_eq!(options.push.queue_driver, PushQueueDriver::Memory);
2807 assert!(!options.push.fcm_enabled);
2808 assert!(!options.push.apns_enabled);
2809 assert!(!options.push.webpush_enabled);
2810 assert!(!options.push.hms_enabled);
2811 assert!(!options.push.wns_enabled);
2812 assert_eq!(options.push.fanout_fast_threshold, 10_000);
2813 assert_eq!(options.push.fanout_shard_size, 100_000);
2814 assert_eq!(options.push.publish_status_ttl_days, 30);
2815 assert_eq!(options.push.default_quotas.acceptance_rps, 100);
2816 assert_eq!(options.push.circuit_breaker.failure_threshold, 5);
2817 assert!(options.push.payload_redaction.redact_payload);
2818 }
2819
2820 #[tokio::test]
2821 async fn push_release_env_overrides_are_parsed() {
2822 let keys = [
2823 "PUSH_FCM_ENABLED",
2824 "PUSH_APNS_ENABLED",
2825 "PUSH_WEBPUSH_ENABLED",
2826 "PUSH_HMS_ENABLED",
2827 "PUSH_WNS_ENABLED",
2828 "PUSH_CREDENTIAL_ENCRYPTION_KEY",
2829 "PUSH_FANOUT_FAST_THRESHOLD",
2830 "PUSH_FANOUT_SHARD_SIZE",
2831 "PUSH_FANOUT_SYNC_THRESHOLD",
2832 "PUSH_BACKPRESSURE_LAG_THRESHOLD_SECS",
2833 "PUSH_PUBLISH_STATUS_TTL_DAYS",
2834 "PUSH_FAILURE_THRESHOLD",
2835 "PUSH_SCHEDULER_INTERVAL_SECS",
2836 "PUSH_STALE_DEVICE_MAX_AGE_DAYS",
2837 "PUSH_ANALYTICS_ENABLED",
2838 "PUSH_DEFAULT_ACCEPTANCE_RPS",
2839 "PUSH_DEFAULT_DELIVERY_QUOTA_DAILY",
2840 "PUSH_DEFAULT_FANOUT_MAX",
2841 "PUSH_DEFAULT_INFLIGHT_MAX",
2842 ];
2843 let previous: Vec<_> = keys
2844 .iter()
2845 .map(|key| (*key, std::env::var(key).ok()))
2846 .collect();
2847
2848 unsafe {
2851 std::env::set_var("PUSH_FCM_ENABLED", "true");
2852 std::env::set_var("PUSH_APNS_ENABLED", "true");
2853 std::env::set_var("PUSH_WEBPUSH_ENABLED", "true");
2854 std::env::set_var("PUSH_HMS_ENABLED", "true");
2855 std::env::set_var("PUSH_WNS_ENABLED", "true");
2856 std::env::set_var("PUSH_CREDENTIAL_ENCRYPTION_KEY", "env:key:v1");
2857 std::env::set_var("PUSH_FANOUT_FAST_THRESHOLD", "12345");
2858 std::env::set_var("PUSH_FANOUT_SHARD_SIZE", "54321");
2859 std::env::set_var("PUSH_FANOUT_SYNC_THRESHOLD", "250");
2860 std::env::set_var("PUSH_BACKPRESSURE_LAG_THRESHOLD_SECS", "42");
2861 std::env::set_var("PUSH_PUBLISH_STATUS_TTL_DAYS", "14");
2862 std::env::set_var("PUSH_FAILURE_THRESHOLD", "9");
2863 std::env::set_var("PUSH_SCHEDULER_INTERVAL_SECS", "11");
2864 std::env::set_var("PUSH_STALE_DEVICE_MAX_AGE_DAYS", "120");
2865 std::env::set_var("PUSH_ANALYTICS_ENABLED", "true");
2866 std::env::set_var("PUSH_DEFAULT_ACCEPTANCE_RPS", "700");
2867 std::env::set_var("PUSH_DEFAULT_DELIVERY_QUOTA_DAILY", "8000");
2868 std::env::set_var("PUSH_DEFAULT_FANOUT_MAX", "9000");
2869 std::env::set_var("PUSH_DEFAULT_INFLIGHT_MAX", "1000");
2870 }
2871
2872 let mut options = ServerOptions::default();
2873 options.override_from_env().await.unwrap();
2874
2875 for (key, value) in previous {
2876 unsafe {
2879 if let Some(value) = value {
2880 std::env::set_var(key, value);
2881 } else {
2882 std::env::remove_var(key);
2883 }
2884 }
2885 }
2886
2887 assert!(options.push.fcm_enabled);
2888 assert!(options.push.apns_enabled);
2889 assert!(options.push.webpush_enabled);
2890 assert!(options.push.hms_enabled);
2891 assert!(options.push.wns_enabled);
2892 assert_eq!(
2893 options.push.credential_encryption_key.as_deref(),
2894 Some("env:key:v1")
2895 );
2896 assert_eq!(options.push.fanout_fast_threshold, 12_345);
2897 assert_eq!(options.push.fanout_shard_size, 54_321);
2898 assert_eq!(options.push.fanout_sync_threshold, 250);
2899 assert_eq!(options.push.backpressure_lag_threshold_secs, 42);
2900 assert_eq!(options.push.publish_status_ttl_days, 14);
2901 assert_eq!(options.push.circuit_breaker.failure_threshold, 9);
2902 assert_eq!(options.push.scheduler_interval_secs, 11);
2903 assert_eq!(options.push.stale_device_max_age_days, 120);
2904 assert!(options.push.analytics_enabled);
2905 assert_eq!(options.push.default_quotas.acceptance_rps, 700);
2906 assert_eq!(options.push.default_quotas.delivery_quota_daily, 8_000);
2907 assert_eq!(options.push.default_quotas.fanout_max, 9_000);
2908 assert_eq!(options.push.default_quotas.inflight_max, 1_000);
2909 }
2910}
2911
2912impl ClusterHealthConfig {
2913 pub fn validate(&self) -> Result<(), String> {
2914 if self.heartbeat_interval_ms == 0 {
2915 return Err("heartbeat_interval_ms must be greater than 0".to_string());
2916 }
2917 if self.node_timeout_ms == 0 {
2918 return Err("node_timeout_ms must be greater than 0".to_string());
2919 }
2920 if self.cleanup_interval_ms == 0 {
2921 return Err("cleanup_interval_ms must be greater than 0".to_string());
2922 }
2923
2924 if self.heartbeat_interval_ms > self.node_timeout_ms / 3 {
2925 return Err(format!(
2926 "heartbeat_interval_ms ({}) should be at least 3x smaller than node_timeout_ms ({}) to avoid false positive dead node detection. Recommended: heartbeat_interval_ms <= {}",
2927 self.heartbeat_interval_ms,
2928 self.node_timeout_ms,
2929 self.node_timeout_ms / 3
2930 ));
2931 }
2932
2933 if self.cleanup_interval_ms > self.node_timeout_ms {
2934 return Err(format!(
2935 "cleanup_interval_ms ({}) should not be larger than node_timeout_ms ({}) to ensure timely dead node detection",
2936 self.cleanup_interval_ms, self.node_timeout_ms
2937 ));
2938 }
2939
2940 Ok(())
2941 }
2942}
2943
2944impl ServerOptions {
2945 pub async fn load_from_file(path: &str) -> Result<Self, Box<dyn std::error::Error>> {
2946 let content = tokio::fs::read_to_string(path).await?;
2947 let options: Self = if path.ends_with(".toml") {
2948 toml::from_str(&content)?
2949 } else {
2950 sonic_rs::from_str(&content)?
2952 };
2953 Ok(options)
2954 }
2955
2956 pub async fn override_from_env(&mut self) -> Result<(), Box<dyn std::error::Error>> {
2957 if let Ok(mode) = std::env::var("ENVIRONMENT") {
2959 self.mode = mode;
2960 }
2961 self.debug = parse_bool_env("DEBUG_MODE", self.debug);
2962 if parse_bool_env("DEBUG", false) {
2963 self.debug = true;
2964 info!("DEBUG environment variable forces debug mode ON");
2965 }
2966
2967 self.activity_timeout = parse_env::<u64>("ACTIVITY_TIMEOUT", self.activity_timeout);
2968
2969 if let Ok(host) = std::env::var("HOST") {
2970 self.host = host;
2971 }
2972 self.port = parse_env::<u16>("PORT", self.port);
2973 self.shutdown_grace_period =
2974 parse_env::<u64>("SHUTDOWN_GRACE_PERIOD", self.shutdown_grace_period);
2975 self.user_authentication_timeout = parse_env::<u64>(
2976 "USER_AUTHENTICATION_TIMEOUT",
2977 self.user_authentication_timeout,
2978 );
2979 self.websocket_max_payload_kb =
2980 parse_env::<u32>("WEBSOCKET_MAX_PAYLOAD_KB", self.websocket_max_payload_kb);
2981 if let Ok(id) = std::env::var("INSTANCE_PROCESS_ID") {
2982 self.instance.process_id = id;
2983 }
2984
2985 if let Ok(driver_str) = std::env::var("ADAPTER_DRIVER") {
2987 self.adapter.driver =
2988 parse_driver_enum(driver_str, self.adapter.driver.clone(), "Adapter");
2989 }
2990 self.adapter.buffer_multiplier_per_cpu = parse_env::<usize>(
2991 "ADAPTER_BUFFER_MULTIPLIER_PER_CPU",
2992 self.adapter.buffer_multiplier_per_cpu,
2993 );
2994 self.adapter.enable_socket_counting = parse_env::<bool>(
2995 "ADAPTER_ENABLE_SOCKET_COUNTING",
2996 self.adapter.enable_socket_counting,
2997 );
2998 self.adapter.fallback_to_local =
2999 parse_env::<bool>("ADAPTER_FALLBACK_TO_LOCAL", self.adapter.fallback_to_local);
3000 if let Ok(driver_str) = std::env::var("CACHE_DRIVER") {
3001 self.cache.driver = parse_driver_enum(driver_str, self.cache.driver.clone(), "Cache");
3002 }
3003 if let Ok(driver_str) = std::env::var("QUEUE_DRIVER") {
3004 self.queue.driver = parse_driver_enum(driver_str, self.queue.driver.clone(), "Queue");
3005 }
3006 if let Ok(driver_str) = std::env::var("APP_MANAGER_DRIVER") {
3007 self.app_manager.driver =
3008 parse_driver_enum(driver_str, self.app_manager.driver.clone(), "AppManager");
3009 }
3010 if let Ok(driver_str) = std::env::var("RATE_LIMITER_DRIVER") {
3011 self.rate_limiter.driver = parse_driver_enum(
3012 driver_str,
3013 self.rate_limiter.driver.clone(),
3014 "RateLimiter Backend",
3015 );
3016 }
3017
3018 if let Ok(host) = std::env::var("DATABASE_REDIS_HOST") {
3020 self.database.redis.host = host;
3021 }
3022 self.database.redis.port =
3023 parse_env::<u16>("DATABASE_REDIS_PORT", self.database.redis.port);
3024 if let Ok(username) = std::env::var("DATABASE_REDIS_USERNAME") {
3025 self.database.redis.username = if username.is_empty() {
3026 None
3027 } else {
3028 Some(username)
3029 };
3030 }
3031 if let Ok(password) = std::env::var("DATABASE_REDIS_PASSWORD") {
3032 self.database.redis.password = Some(password);
3033 }
3034 self.database.redis.db = parse_env::<u32>("DATABASE_REDIS_DB", self.database.redis.db);
3035 if let Ok(prefix) = std::env::var("DATABASE_REDIS_KEY_PREFIX") {
3036 self.database.redis.key_prefix = prefix;
3037 }
3038 if let Ok(cluster_username) = std::env::var("DATABASE_REDIS_CLUSTER_USERNAME") {
3039 self.database.redis.cluster.username = if cluster_username.is_empty() {
3040 None
3041 } else {
3042 Some(cluster_username)
3043 };
3044 }
3045 if let Ok(cluster_password) = std::env::var("DATABASE_REDIS_CLUSTER_PASSWORD") {
3046 self.database.redis.cluster.password = Some(cluster_password);
3047 }
3048 self.database.redis.cluster.use_tls = parse_bool_env(
3049 "DATABASE_REDIS_CLUSTER_USE_TLS",
3050 self.database.redis.cluster.use_tls,
3051 );
3052
3053 if let Ok(host) = std::env::var("DATABASE_MYSQL_HOST") {
3055 self.database.mysql.host = host;
3056 }
3057 self.database.mysql.port =
3058 parse_env::<u16>("DATABASE_MYSQL_PORT", self.database.mysql.port);
3059 if let Ok(user) = std::env::var("DATABASE_MYSQL_USERNAME") {
3060 self.database.mysql.username = user;
3061 }
3062 if let Ok(pass) = std::env::var("DATABASE_MYSQL_PASSWORD") {
3063 self.database.mysql.password = pass;
3064 }
3065 if let Ok(db) = std::env::var("DATABASE_MYSQL_DATABASE") {
3066 self.database.mysql.database = db;
3067 }
3068 if let Ok(table) = std::env::var("DATABASE_MYSQL_TABLE_NAME") {
3069 self.database.mysql.table_name = table;
3070 }
3071 override_db_pool_settings(&mut self.database.mysql, "DATABASE_MYSQL");
3072
3073 if let Ok(host) = std::env::var("DATABASE_POSTGRES_HOST") {
3075 self.database.postgres.host = host;
3076 }
3077 self.database.postgres.port =
3078 parse_env::<u16>("DATABASE_POSTGRES_PORT", self.database.postgres.port);
3079 if let Ok(user) = std::env::var("DATABASE_POSTGRES_USERNAME") {
3080 self.database.postgres.username = user;
3081 }
3082 if let Ok(pass) = std::env::var("DATABASE_POSTGRES_PASSWORD") {
3083 self.database.postgres.password = pass;
3084 }
3085 if let Ok(db) = std::env::var("DATABASE_POSTGRES_DATABASE") {
3086 self.database.postgres.database = db;
3087 }
3088 override_db_pool_settings(&mut self.database.postgres, "DATABASE_POSTGRES");
3089
3090 if let Ok(region) = std::env::var("DATABASE_DYNAMODB_REGION") {
3092 self.database.dynamodb.region = region;
3093 }
3094 if let Ok(table) = std::env::var("DATABASE_DYNAMODB_TABLE_NAME") {
3095 self.database.dynamodb.table_name = table;
3096 }
3097 if let Ok(endpoint) = std::env::var("DATABASE_DYNAMODB_ENDPOINT_URL") {
3098 self.database.dynamodb.endpoint_url = Some(endpoint);
3099 }
3100 if let Ok(key_id) = std::env::var("AWS_ACCESS_KEY_ID") {
3101 self.database.dynamodb.aws_access_key_id = Some(key_id);
3102 }
3103 if let Ok(secret) = std::env::var("AWS_SECRET_ACCESS_KEY") {
3104 self.database.dynamodb.aws_secret_access_key = Some(secret);
3105 }
3106
3107 if let Ok(url) = std::env::var("DATABASE_SURREALDB_URL") {
3109 self.database.surrealdb.url = url;
3110 }
3111 if let Ok(namespace) = std::env::var("DATABASE_SURREALDB_NAMESPACE") {
3112 self.database.surrealdb.namespace = namespace;
3113 }
3114 if let Ok(database) = std::env::var("DATABASE_SURREALDB_DATABASE") {
3115 self.database.surrealdb.database = database;
3116 }
3117 if let Ok(username) = std::env::var("DATABASE_SURREALDB_USERNAME") {
3118 self.database.surrealdb.username = username;
3119 }
3120 if let Ok(password) = std::env::var("DATABASE_SURREALDB_PASSWORD") {
3121 self.database.surrealdb.password = password;
3122 }
3123 if let Ok(table) = std::env::var("DATABASE_SURREALDB_TABLE_NAME") {
3124 self.database.surrealdb.table_name = table;
3125 }
3126
3127 let apply_redis_cluster_nodes = |options: &mut Self, nodes: &str| {
3129 let node_list: Vec<String> = nodes
3130 .split(',')
3131 .map(|s| s.trim())
3132 .filter(|s| !s.is_empty())
3133 .map(ToString::to_string)
3134 .collect();
3135
3136 options.adapter.cluster.nodes = node_list.clone();
3137 options.queue.redis_cluster.nodes = node_list.clone();
3138
3139 let parsed_nodes: Vec<ClusterNode> = node_list
3140 .iter()
3141 .filter_map(|seed| ClusterNode::from_seed(seed))
3142 .collect();
3143 options.database.redis.cluster.nodes = parsed_nodes.clone();
3144 options.database.redis.cluster_nodes = parsed_nodes;
3145 };
3146
3147 if let Ok(nodes) = std::env::var("REDIS_CLUSTER_NODES") {
3148 apply_redis_cluster_nodes(self, &nodes);
3149 }
3150 if let Ok(nodes) = std::env::var("DATABASE_REDIS_CLUSTER_NODES") {
3151 apply_redis_cluster_nodes(self, &nodes);
3152 }
3153 self.queue.redis_cluster.concurrency = parse_env::<u32>(
3154 "REDIS_CLUSTER_QUEUE_CONCURRENCY",
3155 self.queue.redis_cluster.concurrency,
3156 );
3157 if let Ok(prefix) = std::env::var("REDIS_CLUSTER_QUEUE_PREFIX") {
3158 self.queue.redis_cluster.prefix = Some(prefix);
3159 }
3160
3161 self.ssl.enabled = parse_bool_env("SSL_ENABLED", self.ssl.enabled);
3163 if let Ok(val) = std::env::var("SSL_CERT_PATH") {
3164 self.ssl.cert_path = val;
3165 }
3166 if let Ok(val) = std::env::var("SSL_KEY_PATH") {
3167 self.ssl.key_path = val;
3168 }
3169 self.ssl.redirect_http = parse_bool_env("SSL_REDIRECT_HTTP", self.ssl.redirect_http);
3170 if let Some(port) = parse_env_optional::<u16>("SSL_HTTP_PORT") {
3171 self.ssl.http_port = Some(port);
3172 }
3173
3174 self.unix_socket.enabled = parse_bool_env("UNIX_SOCKET_ENABLED", self.unix_socket.enabled);
3176 if let Ok(path) = std::env::var("UNIX_SOCKET_PATH") {
3177 self.unix_socket.path = path;
3178 }
3179 if let Ok(mode_str) = std::env::var("UNIX_SOCKET_PERMISSION_MODE") {
3180 if mode_str.chars().all(|c| c.is_digit(8)) {
3181 if let Ok(mode) = u32::from_str_radix(&mode_str, 8) {
3182 if mode <= 0o777 {
3183 self.unix_socket.permission_mode = mode;
3184 } else {
3185 warn!(
3186 "UNIX_SOCKET_PERMISSION_MODE '{}' exceeds maximum value 777. Using default: {:o}",
3187 mode_str, self.unix_socket.permission_mode
3188 );
3189 }
3190 } else {
3191 warn!(
3192 "Failed to parse UNIX_SOCKET_PERMISSION_MODE '{}' as octal. Using default: {:o}",
3193 mode_str, self.unix_socket.permission_mode
3194 );
3195 }
3196 } else {
3197 warn!(
3198 "UNIX_SOCKET_PERMISSION_MODE '{}' must contain only octal digits (0-7). Using default: {:o}",
3199 mode_str, self.unix_socket.permission_mode
3200 );
3201 }
3202 }
3203
3204 if let Ok(driver_str) = std::env::var("METRICS_DRIVER") {
3206 self.metrics.driver =
3207 parse_driver_enum(driver_str, self.metrics.driver.clone(), "Metrics");
3208 }
3209 self.metrics.enabled = parse_bool_env("METRICS_ENABLED", self.metrics.enabled);
3210 if let Ok(val) = std::env::var("METRICS_HOST") {
3211 self.metrics.host = val;
3212 }
3213 self.metrics.port = parse_env::<u16>("METRICS_PORT", self.metrics.port);
3214 if let Ok(val) = std::env::var("METRICS_PROMETHEUS_PREFIX") {
3215 self.metrics.prometheus.prefix = val;
3216 }
3217
3218 self.http_api.usage_enabled =
3220 parse_bool_env("HTTP_API_USAGE_ENABLED", self.http_api.usage_enabled);
3221
3222 self.rate_limiter.enabled =
3224 parse_bool_env("RATE_LIMITER_ENABLED", self.rate_limiter.enabled);
3225 self.rate_limiter.api_rate_limit.max_requests = parse_env::<u32>(
3226 "RATE_LIMITER_API_MAX_REQUESTS",
3227 self.rate_limiter.api_rate_limit.max_requests,
3228 );
3229 self.rate_limiter.api_rate_limit.window_seconds = parse_env::<u64>(
3230 "RATE_LIMITER_API_WINDOW_SECONDS",
3231 self.rate_limiter.api_rate_limit.window_seconds,
3232 );
3233 if let Some(hops) = parse_env_optional::<u32>("RATE_LIMITER_API_TRUST_HOPS") {
3234 self.rate_limiter.api_rate_limit.trust_hops = Some(hops);
3235 }
3236 self.rate_limiter.websocket_rate_limit.max_requests = parse_env::<u32>(
3237 "RATE_LIMITER_WS_MAX_REQUESTS",
3238 self.rate_limiter.websocket_rate_limit.max_requests,
3239 );
3240 self.rate_limiter.websocket_rate_limit.window_seconds = parse_env::<u64>(
3241 "RATE_LIMITER_WS_WINDOW_SECONDS",
3242 self.rate_limiter.websocket_rate_limit.window_seconds,
3243 );
3244 if let Ok(prefix) = std::env::var("RATE_LIMITER_REDIS_PREFIX") {
3245 self.rate_limiter.redis.prefix = Some(prefix);
3246 }
3247
3248 self.queue.redis.concurrency =
3250 parse_env::<u32>("QUEUE_REDIS_CONCURRENCY", self.queue.redis.concurrency);
3251 if let Ok(prefix) = std::env::var("QUEUE_REDIS_PREFIX") {
3252 self.queue.redis.prefix = Some(prefix);
3253 }
3254
3255 if let Ok(region) = std::env::var("QUEUE_SQS_REGION") {
3257 self.queue.sqs.region = region;
3258 }
3259 self.queue.sqs.visibility_timeout = parse_env::<i32>(
3260 "QUEUE_SQS_VISIBILITY_TIMEOUT",
3261 self.queue.sqs.visibility_timeout,
3262 );
3263 self.queue.sqs.max_messages =
3264 parse_env::<i32>("QUEUE_SQS_MAX_MESSAGES", self.queue.sqs.max_messages);
3265 self.queue.sqs.wait_time_seconds = parse_env::<i32>(
3266 "QUEUE_SQS_WAIT_TIME_SECONDS",
3267 self.queue.sqs.wait_time_seconds,
3268 );
3269 self.queue.sqs.concurrency =
3270 parse_env::<u32>("QUEUE_SQS_CONCURRENCY", self.queue.sqs.concurrency);
3271 self.queue.sqs.fifo = parse_bool_env("QUEUE_SQS_FIFO", self.queue.sqs.fifo);
3272 if let Ok(endpoint) = std::env::var("QUEUE_SQS_ENDPOINT_URL") {
3273 self.queue.sqs.endpoint_url = Some(endpoint);
3274 }
3275
3276 if let Ok(region) = std::env::var("QUEUE_SNS_REGION") {
3278 self.queue.sns.region = region;
3279 }
3280 if let Ok(topic_arn) = std::env::var("QUEUE_SNS_TOPIC_ARN") {
3281 self.queue.sns.topic_arn = topic_arn;
3282 }
3283 if let Ok(endpoint) = std::env::var("QUEUE_SNS_ENDPOINT_URL") {
3284 self.queue.sns.endpoint_url = Some(endpoint);
3285 }
3286
3287 self.webhooks.batching.enabled =
3289 parse_bool_env("WEBHOOK_BATCHING_ENABLED", self.webhooks.batching.enabled);
3290 self.webhooks.batching.duration =
3291 parse_env::<u64>("WEBHOOK_BATCHING_DURATION", self.webhooks.batching.duration);
3292 self.webhooks.batching.size =
3293 parse_env::<usize>("WEBHOOK_BATCHING_SIZE", self.webhooks.batching.size);
3294
3295 if let Ok(servers) = std::env::var("NATS_SERVERS") {
3297 self.adapter.nats.servers = servers.split(',').map(|s| s.trim().to_string()).collect();
3298 }
3299 if let Ok(user) = std::env::var("NATS_USERNAME") {
3300 self.adapter.nats.username = Some(user);
3301 }
3302 if let Ok(pass) = std::env::var("NATS_PASSWORD") {
3303 self.adapter.nats.password = Some(pass);
3304 }
3305 if let Ok(token) = std::env::var("NATS_TOKEN") {
3306 self.adapter.nats.token = Some(token);
3307 }
3308 if let Ok(prefix) = std::env::var("NATS_PREFIX") {
3309 self.adapter.nats.prefix = prefix;
3310 }
3311 self.adapter.nats.connection_timeout_ms = parse_env::<u64>(
3312 "NATS_CONNECTION_TIMEOUT_MS",
3313 self.adapter.nats.connection_timeout_ms,
3314 );
3315 self.adapter.nats.request_timeout_ms = parse_env::<u64>(
3316 "NATS_REQUEST_TIMEOUT_MS",
3317 self.adapter.nats.request_timeout_ms,
3318 );
3319 self.adapter.nats.discovery_max_wait_ms = parse_env::<u64>(
3320 "NATS_DISCOVERY_MAX_WAIT_MS",
3321 self.adapter.nats.discovery_max_wait_ms,
3322 );
3323 self.adapter.nats.discovery_idle_wait_ms = parse_env::<u64>(
3324 "NATS_DISCOVERY_IDLE_WAIT_MS",
3325 self.adapter.nats.discovery_idle_wait_ms,
3326 );
3327 if let Some(nodes) = parse_env_optional::<u32>("NATS_NODES_NUMBER") {
3328 self.adapter.nats.nodes_number = Some(nodes);
3329 }
3330 if let Some(v) = parse_env_optional::<usize>("NATS_SUBSCRIPTION_CAPACITY") {
3331 self.adapter.nats.subscription_capacity = Some(v);
3332 }
3333 if let Some(v) = parse_env_optional::<usize>("NATS_CLIENT_CAPACITY") {
3334 self.adapter.nats.client_capacity = Some(v);
3335 }
3336 if let Some(v) = parse_env_optional::<usize>("NATS_MAX_RECONNECTS") {
3337 self.adapter.nats.max_reconnects = Some(v);
3338 }
3339 if let Some(v) = parse_env_optional::<usize>("NATS_PRESENCE_SYNC_CHUNK_SIZE") {
3340 self.adapter.nats.presence_sync_chunk_size = Some(v);
3341 }
3342
3343 if let Ok(url) = std::env::var("PULSAR_URL") {
3345 self.adapter.pulsar.url = url;
3346 }
3347 if let Ok(prefix) = std::env::var("PULSAR_PREFIX") {
3348 self.adapter.pulsar.prefix = prefix;
3349 }
3350 if let Ok(token) = std::env::var("PULSAR_TOKEN") {
3351 self.adapter.pulsar.token = Some(token);
3352 }
3353 self.adapter.pulsar.request_timeout_ms = parse_env::<u64>(
3354 "PULSAR_REQUEST_TIMEOUT_MS",
3355 self.adapter.pulsar.request_timeout_ms,
3356 );
3357 if let Some(nodes) = parse_env_optional::<u32>("PULSAR_NODES_NUMBER") {
3358 self.adapter.pulsar.nodes_number = Some(nodes);
3359 }
3360
3361 if let Ok(url) = std::env::var("RABBITMQ_URL") {
3363 self.adapter.rabbitmq.url = url;
3364 }
3365 if let Ok(prefix) = std::env::var("RABBITMQ_PREFIX") {
3366 self.adapter.rabbitmq.prefix = prefix;
3367 }
3368 self.adapter.rabbitmq.connection_timeout_ms = parse_env::<u64>(
3369 "RABBITMQ_CONNECTION_TIMEOUT_MS",
3370 self.adapter.rabbitmq.connection_timeout_ms,
3371 );
3372 self.adapter.rabbitmq.request_timeout_ms = parse_env::<u64>(
3373 "RABBITMQ_REQUEST_TIMEOUT_MS",
3374 self.adapter.rabbitmq.request_timeout_ms,
3375 );
3376 if let Some(nodes) = parse_env_optional::<u32>("RABBITMQ_NODES_NUMBER") {
3377 self.adapter.rabbitmq.nodes_number = Some(nodes);
3378 }
3379
3380 if let Ok(project_id) = std::env::var("GOOGLE_PUBSUB_PROJECT_ID") {
3382 self.adapter.google_pubsub.project_id = project_id;
3383 }
3384 if let Ok(prefix) = std::env::var("GOOGLE_PUBSUB_PREFIX") {
3385 self.adapter.google_pubsub.prefix = prefix;
3386 }
3387 if let Ok(emulator_host) = std::env::var("PUBSUB_EMULATOR_HOST") {
3388 self.adapter.google_pubsub.emulator_host = Some(emulator_host);
3389 }
3390 self.adapter.google_pubsub.request_timeout_ms = parse_env::<u64>(
3391 "GOOGLE_PUBSUB_REQUEST_TIMEOUT_MS",
3392 self.adapter.google_pubsub.request_timeout_ms,
3393 );
3394 if let Some(nodes) = parse_env_optional::<u32>("GOOGLE_PUBSUB_NODES_NUMBER") {
3395 self.adapter.google_pubsub.nodes_number = Some(nodes);
3396 }
3397
3398 if let Ok(brokers) = std::env::var("KAFKA_BROKERS") {
3400 self.adapter.kafka.brokers = brokers.split(',').map(|s| s.trim().to_string()).collect();
3401 }
3402 if let Ok(prefix) = std::env::var("KAFKA_PREFIX") {
3403 self.adapter.kafka.prefix = prefix;
3404 }
3405 if let Ok(protocol) = std::env::var("KAFKA_SECURITY_PROTOCOL") {
3406 self.adapter.kafka.security_protocol = Some(protocol);
3407 }
3408 if let Ok(mechanism) = std::env::var("KAFKA_SASL_MECHANISM") {
3409 self.adapter.kafka.sasl_mechanism = Some(mechanism);
3410 }
3411 if let Ok(username) = std::env::var("KAFKA_SASL_USERNAME") {
3412 self.adapter.kafka.sasl_username = Some(username);
3413 }
3414 if let Ok(password) = std::env::var("KAFKA_SASL_PASSWORD") {
3415 self.adapter.kafka.sasl_password = Some(password);
3416 }
3417 self.adapter.kafka.request_timeout_ms = parse_env::<u64>(
3418 "KAFKA_REQUEST_TIMEOUT_MS",
3419 self.adapter.kafka.request_timeout_ms,
3420 );
3421 if let Some(nodes) = parse_env_optional::<u32>("KAFKA_NODES_NUMBER") {
3422 self.adapter.kafka.nodes_number = Some(nodes);
3423 }
3424
3425 if let Ok(connection_string) = std::env::var("IGGY_CONNECTION_STRING") {
3427 self.adapter.iggy.connection_string = connection_string.clone();
3428 self.queue.iggy.connection_string = connection_string;
3429 }
3430 if let Ok(username) = std::env::var("IGGY_USERNAME") {
3431 let username = (!username.is_empty()).then_some(username);
3432 self.adapter.iggy.username = username.clone();
3433 self.queue.iggy.username = username;
3434 }
3435 if let Ok(password) = std::env::var("IGGY_PASSWORD") {
3436 let password = (!password.is_empty()).then_some(password);
3437 self.adapter.iggy.password = password.clone();
3438 self.queue.iggy.password = password;
3439 }
3440 if let Ok(consumer_name) = std::env::var("IGGY_CONSUMER_NAME") {
3441 let consumer_name = (!consumer_name.is_empty()).then_some(consumer_name);
3442 self.adapter.iggy.consumer_name = consumer_name.clone();
3443 self.queue.iggy.consumer_name = consumer_name;
3444 } else if let Ok(process_id) = std::env::var("INSTANCE_PROCESS_ID") {
3445 let process_id = (!process_id.is_empty()).then_some(process_id);
3446 self.adapter.iggy.consumer_name = process_id.clone();
3447 self.queue.iggy.consumer_name = process_id;
3448 }
3449 if let Ok(stream) = std::env::var("IGGY_STREAM") {
3450 self.adapter.iggy.stream = stream.clone();
3451 self.queue.iggy.stream = stream;
3452 }
3453 if let Ok(prefix) = std::env::var("IGGY_TOPIC_PREFIX") {
3454 self.adapter.iggy.topic_prefix = prefix;
3455 }
3456 if let Ok(prefix) = std::env::var("IGGY_QUEUE_TOPIC_PREFIX") {
3457 self.queue.iggy.queue_topic_prefix = prefix;
3458 }
3459 if let Ok(prefix) = std::env::var("IGGY_CONSUMER_GROUP_PREFIX") {
3460 self.queue.iggy.consumer_group_prefix = prefix;
3461 }
3462 self.adapter.iggy.request_timeout_ms = parse_env::<u64>(
3463 "IGGY_REQUEST_TIMEOUT_MS",
3464 self.adapter.iggy.request_timeout_ms,
3465 );
3466 self.queue.iggy.request_timeout_ms = parse_env::<u64>(
3467 "IGGY_REQUEST_TIMEOUT_MS",
3468 self.queue.iggy.request_timeout_ms,
3469 );
3470 self.adapter.iggy.poll_interval_ms =
3471 parse_env::<u64>("IGGY_POLL_INTERVAL_MS", self.adapter.iggy.poll_interval_ms);
3472 self.queue.iggy.poll_interval_ms =
3473 parse_env::<u64>("IGGY_POLL_INTERVAL_MS", self.queue.iggy.poll_interval_ms);
3474 self.adapter.iggy.poll_batch_size =
3475 parse_env::<u32>("IGGY_POLL_BATCH_SIZE", self.adapter.iggy.poll_batch_size);
3476 self.queue.iggy.poll_batch_size =
3477 parse_env::<u32>("IGGY_POLL_BATCH_SIZE", self.queue.iggy.poll_batch_size);
3478 self.adapter.iggy.partitions_count = parse_env::<u32>(
3479 "ADAPTER_IGGY_PARTITIONS_COUNT",
3480 parse_env::<u32>("IGGY_PARTITIONS_COUNT", self.adapter.iggy.partitions_count),
3481 );
3482 self.queue.iggy.partitions_count = parse_env::<u32>(
3483 "QUEUE_IGGY_PARTITIONS_COUNT",
3484 parse_env::<u32>("IGGY_PARTITIONS_COUNT", self.queue.iggy.partitions_count),
3485 );
3486 self.adapter.iggy.partition_id = parse_env::<u32>(
3487 "ADAPTER_IGGY_PARTITION_ID",
3488 parse_env::<u32>("IGGY_PARTITION_ID", self.adapter.iggy.partition_id),
3489 );
3490 self.queue.iggy.partition_id = parse_env::<u32>(
3491 "QUEUE_IGGY_PARTITION_ID",
3492 parse_env::<u32>("IGGY_PARTITION_ID", self.queue.iggy.partition_id),
3493 );
3494 self.adapter.iggy.auto_create =
3495 parse_env::<bool>("IGGY_AUTO_CREATE", self.adapter.iggy.auto_create);
3496 self.queue.iggy.auto_create =
3497 parse_env::<bool>("IGGY_AUTO_CREATE", self.queue.iggy.auto_create);
3498 self.adapter.iggy.start_from_latest = parse_env::<bool>(
3499 "IGGY_START_FROM_LATEST",
3500 self.adapter.iggy.start_from_latest,
3501 );
3502 if let Some(nodes) = parse_env_optional::<u32>("IGGY_NODES_NUMBER") {
3503 self.adapter.iggy.nodes_number = Some(nodes);
3504 self.queue.iggy.nodes_number = Some(nodes);
3505 }
3506
3507 if let Ok(origins) = std::env::var("CORS_ORIGINS") {
3509 let parsed: Vec<String> = origins.split(',').map(|s| s.trim().to_string()).collect();
3510 if let Err(e) = crate::origin_validation::OriginValidator::validate_patterns(&parsed) {
3511 warn!(
3512 "CORS_ORIGINS contains invalid patterns: {}. Keeping previous CORS origins.",
3513 e
3514 );
3515 } else {
3516 self.cors.origin = parsed;
3517 }
3518 }
3519 if let Ok(methods) = std::env::var("CORS_METHODS") {
3520 self.cors.methods = methods.split(',').map(|s| s.trim().to_string()).collect();
3521 }
3522 if let Ok(headers) = std::env::var("CORS_HEADERS") {
3523 self.cors.allowed_headers = headers.split(',').map(|s| s.trim().to_string()).collect();
3524 }
3525 self.cors.credentials = parse_bool_env("CORS_CREDENTIALS", self.cors.credentials);
3526
3527 self.database_pooling.enabled =
3529 parse_bool_env("DATABASE_POOLING_ENABLED", self.database_pooling.enabled);
3530 if let Some(min) = parse_env_optional::<u32>("DATABASE_POOL_MIN") {
3531 self.database_pooling.min = min;
3532 }
3533 if let Some(max) = parse_env_optional::<u32>("DATABASE_POOL_MAX") {
3534 self.database_pooling.max = max;
3535 }
3536
3537 if let Some(pool_size) = parse_env_optional::<u32>("DATABASE_CONNECTION_POOL_SIZE") {
3538 self.database.mysql.connection_pool_size = pool_size;
3539 self.database.postgres.connection_pool_size = pool_size;
3540 }
3541 if let Some(cache_ttl) = parse_env_optional::<u64>("CACHE_TTL_SECONDS") {
3542 self.app_manager.cache.ttl = cache_ttl;
3543 self.channel_limits.cache_ttl = cache_ttl;
3544 self.database.mysql.cache_ttl = cache_ttl;
3545 self.database.postgres.cache_ttl = cache_ttl;
3546 self.database.surrealdb.cache_ttl = cache_ttl;
3547 self.cache.memory.ttl = cache_ttl;
3548 }
3549 if let Some(cleanup_interval) = parse_env_optional::<u64>("CACHE_CLEANUP_INTERVAL") {
3550 self.database.mysql.cache_cleanup_interval = cleanup_interval;
3551 self.database.postgres.cache_cleanup_interval = cleanup_interval;
3552 self.cache.memory.cleanup_interval = cleanup_interval;
3553 }
3554 if let Some(max_capacity) = parse_env_optional::<u64>("CACHE_MAX_CAPACITY") {
3555 self.database.mysql.cache_max_capacity = max_capacity;
3556 self.database.postgres.cache_max_capacity = max_capacity;
3557 self.database.surrealdb.cache_max_capacity = max_capacity;
3558 self.cache.memory.max_capacity = max_capacity;
3559 }
3560
3561 let skip_inline_apps = parse_bool_env("SOCKUDO_SKIP_INLINE_APPS", false)
3562 || !parse_bool_env("APP_MANAGER_REGISTER_INLINE_APPS", true);
3563 if skip_inline_apps {
3564 let app_count = self.app_manager.array.apps.len();
3565 self.app_manager.array.apps.clear();
3566 if app_count > 0 {
3567 info!(
3568 "Skipping {} inline app(s) from configuration due to environment override",
3569 app_count
3570 );
3571 }
3572 }
3573
3574 let default_app_id = std::env::var("SOCKUDO_DEFAULT_APP_ID").ok();
3575 let default_app_key = std::env::var("SOCKUDO_DEFAULT_APP_KEY").ok();
3576 let default_app_secret = std::env::var("SOCKUDO_DEFAULT_APP_SECRET").ok();
3577 let default_app_enabled_env = std::env::var("SOCKUDO_DEFAULT_APP_ENABLED").ok();
3578 let default_app_enabled = parse_bool_env("SOCKUDO_DEFAULT_APP_ENABLED", true);
3579 let default_app_credentials_configured =
3580 default_app_id.is_some() || default_app_key.is_some() || default_app_secret.is_some();
3581 let default_app_env_configured =
3582 default_app_credentials_configured || default_app_enabled_env.is_some();
3583 let default_app_should_override_inline = default_app_credentials_configured
3584 || default_app_enabled_env.is_some_and(|_| !default_app_enabled);
3585
3586 if default_app_should_override_inline {
3587 let app_count = self.app_manager.array.apps.len();
3588 self.app_manager.array.apps.clear();
3589 if app_count > 0 {
3590 info!(
3591 "Replacing {} inline app(s) from configuration with SOCKUDO_DEFAULT_APP_* settings",
3592 app_count
3593 );
3594 }
3595 }
3596
3597 if let (Some(app_id), Some(app_key), Some(app_secret)) =
3598 (default_app_id, default_app_key, default_app_secret)
3599 && default_app_enabled
3600 {
3601 let default_app = App::from_policy(
3602 app_id,
3603 app_key,
3604 app_secret,
3605 default_app_enabled,
3606 crate::app::AppPolicy {
3607 limits: crate::app::AppLimitsPolicy {
3608 max_connections: parse_env::<u32>(
3609 "SOCKUDO_DEFAULT_APP_MAX_CONNECTIONS",
3610 100,
3611 ),
3612 max_backend_events_per_second: Some(parse_env::<u32>(
3613 "SOCKUDO_DEFAULT_APP_MAX_BACKEND_EVENTS_PER_SECOND",
3614 100,
3615 )),
3616 max_client_events_per_second: parse_env::<u32>(
3617 "SOCKUDO_DEFAULT_APP_MAX_CLIENT_EVENTS_PER_SECOND",
3618 100,
3619 ),
3620 max_read_requests_per_second: Some(parse_env::<u32>(
3621 "SOCKUDO_DEFAULT_APP_MAX_READ_REQUESTS_PER_SECOND",
3622 100,
3623 )),
3624 max_presence_members_per_channel: Some(parse_env::<u32>(
3625 "SOCKUDO_DEFAULT_APP_MAX_PRESENCE_MEMBERS_PER_CHANNEL",
3626 100,
3627 )),
3628 max_presence_member_size_in_kb: Some(parse_env::<u32>(
3629 "SOCKUDO_DEFAULT_APP_MAX_PRESENCE_MEMBER_SIZE_IN_KB",
3630 100,
3631 )),
3632 max_channel_name_length: Some(parse_env::<u32>(
3633 "SOCKUDO_DEFAULT_APP_MAX_CHANNEL_NAME_LENGTH",
3634 100,
3635 )),
3636 max_event_channels_at_once: Some(parse_env::<u32>(
3637 "SOCKUDO_DEFAULT_APP_MAX_EVENT_CHANNELS_AT_ONCE",
3638 100,
3639 )),
3640 max_event_name_length: Some(parse_env::<u32>(
3641 "SOCKUDO_DEFAULT_APP_MAX_EVENT_NAME_LENGTH",
3642 100,
3643 )),
3644 max_event_payload_in_kb: Some(parse_env::<u32>(
3645 "SOCKUDO_DEFAULT_APP_MAX_EVENT_PAYLOAD_IN_KB",
3646 100,
3647 )),
3648 max_event_batch_size: Some(parse_env::<u32>(
3649 "SOCKUDO_DEFAULT_APP_MAX_EVENT_BATCH_SIZE",
3650 100,
3651 )),
3652 decay_seconds: None,
3653 terminate_on_limit: false,
3654 message_rate_limit: None,
3655 },
3656 features: crate::app::AppFeaturesPolicy {
3657 enable_client_messages: std::env::var(
3658 "SOCKUDO_DEFAULT_APP_ENABLE_CLIENT_MESSAGES",
3659 )
3660 .ok()
3661 .map(|value| {
3662 matches!(
3663 value.trim().to_ascii_lowercase().as_str(),
3664 "true" | "1" | "yes" | "on"
3665 )
3666 })
3667 .unwrap_or_else(|| parse_bool_env("SOCKUDO_ENABLE_CLIENT_MESSAGES", false)),
3668 enable_user_authentication: Some(parse_bool_env(
3669 "SOCKUDO_DEFAULT_APP_ENABLE_USER_AUTHENTICATION",
3670 false,
3671 )),
3672 enable_watchlist_events: Some(parse_bool_env(
3673 "SOCKUDO_DEFAULT_APP_ENABLE_WATCHLIST_EVENTS",
3674 false,
3675 )),
3676 },
3677 channels: crate::app::AppChannelsPolicy {
3678 allowed_origins: {
3679 if let Ok(origins_str) =
3680 std::env::var("SOCKUDO_DEFAULT_APP_ALLOWED_ORIGINS")
3681 {
3682 if !origins_str.is_empty() {
3683 Some(
3684 origins_str
3685 .split(',')
3686 .map(|s| s.trim().to_string())
3687 .collect(),
3688 )
3689 } else {
3690 None
3691 }
3692 } else {
3693 None
3694 }
3695 },
3696 annotations_enabled: Some(parse_bool_env(
3697 "SOCKUDO_DEFAULT_APP_ANNOTATIONS_ENABLED",
3698 false,
3699 )),
3700 channel_delta_compression: None,
3701 channel_namespaces: None,
3702 },
3703 webhooks: None,
3704 idempotency: None,
3705 connection_recovery: None,
3706 history: None,
3707 presence_history: None,
3708 },
3709 );
3710
3711 self.app_manager.array.apps.push(default_app);
3712 info!("Successfully registered default app from env");
3713 } else if default_app_env_configured && !default_app_enabled {
3714 info!("Default app registration disabled by SOCKUDO_DEFAULT_APP_ENABLED=false");
3715 } else if default_app_credentials_configured {
3716 warn!(
3717 "SOCKUDO_DEFAULT_APP_* environment was configured but id, key, and secret were not all provided; no default app registered"
3718 );
3719 }
3720
3721 if let Ok(redis_url_env) = std::env::var("REDIS_URL") {
3723 info!("Applying REDIS_URL environment variable override");
3724
3725 let redis_url_json = sonic_rs::json!(redis_url_env);
3726
3727 self.adapter
3728 .redis
3729 .redis_pub_options
3730 .insert("url".to_string(), redis_url_json.clone());
3731 self.adapter
3732 .redis
3733 .redis_sub_options
3734 .insert("url".to_string(), redis_url_json);
3735
3736 self.cache.redis.url_override = Some(redis_url_env.clone());
3737 self.queue.redis.url_override = Some(redis_url_env.clone());
3738 self.rate_limiter.redis.url_override = Some(redis_url_env);
3739 }
3740
3741 let has_colors_env = std::env::var("LOG_COLORS_ENABLED").is_ok();
3743 let has_target_env = std::env::var("LOG_INCLUDE_TARGET").is_ok();
3744 if has_colors_env || has_target_env {
3745 let logging_config = self.logging.get_or_insert_with(Default::default);
3746 if has_colors_env {
3747 logging_config.colors_enabled =
3748 parse_bool_env("LOG_COLORS_ENABLED", logging_config.colors_enabled);
3749 }
3750 if has_target_env {
3751 logging_config.include_target =
3752 parse_bool_env("LOG_INCLUDE_TARGET", logging_config.include_target);
3753 }
3754 }
3755
3756 self.cleanup.async_enabled =
3758 parse_bool_env("CLEANUP_ASYNC_ENABLED", self.cleanup.async_enabled);
3759 self.cleanup.fallback_to_sync =
3760 parse_bool_env("CLEANUP_FALLBACK_TO_SYNC", self.cleanup.fallback_to_sync);
3761 self.cleanup.queue_buffer_size =
3762 parse_env::<usize>("CLEANUP_QUEUE_BUFFER_SIZE", self.cleanup.queue_buffer_size);
3763 self.cleanup.batch_size = parse_env::<usize>("CLEANUP_BATCH_SIZE", self.cleanup.batch_size);
3764 self.cleanup.batch_timeout_ms =
3765 parse_env::<u64>("CLEANUP_BATCH_TIMEOUT_MS", self.cleanup.batch_timeout_ms);
3766 if let Ok(worker_threads_str) = std::env::var("CLEANUP_WORKER_THREADS") {
3767 self.cleanup.worker_threads = if worker_threads_str.to_lowercase() == "auto" {
3768 WorkerThreadsConfig::Auto
3769 } else if let Ok(n) = worker_threads_str.parse::<usize>() {
3770 WorkerThreadsConfig::Fixed(n)
3771 } else {
3772 warn!(
3773 "Invalid CLEANUP_WORKER_THREADS value '{}', keeping current setting",
3774 worker_threads_str
3775 );
3776 self.cleanup.worker_threads.clone()
3777 };
3778 }
3779 self.cleanup.max_retry_attempts = parse_env::<u32>(
3780 "CLEANUP_MAX_RETRY_ATTEMPTS",
3781 self.cleanup.max_retry_attempts,
3782 );
3783
3784 self.cluster_health.enabled =
3786 parse_bool_env("CLUSTER_HEALTH_ENABLED", self.cluster_health.enabled);
3787 self.cluster_health.heartbeat_interval_ms = parse_env::<u64>(
3788 "CLUSTER_HEALTH_HEARTBEAT_INTERVAL",
3789 self.cluster_health.heartbeat_interval_ms,
3790 );
3791 self.cluster_health.node_timeout_ms = parse_env::<u64>(
3792 "CLUSTER_HEALTH_NODE_TIMEOUT",
3793 self.cluster_health.node_timeout_ms,
3794 );
3795 self.cluster_health.cleanup_interval_ms = parse_env::<u64>(
3796 "CLUSTER_HEALTH_CLEANUP_INTERVAL",
3797 self.cluster_health.cleanup_interval_ms,
3798 );
3799
3800 self.health_check_timeout_ms =
3802 parse_env::<u64>("HEALTH_CHECK_TIMEOUT_MS", self.health_check_timeout_ms);
3803
3804 self.tag_filtering.enabled =
3806 parse_bool_env("TAG_FILTERING_ENABLED", self.tag_filtering.enabled);
3807
3808 if let Ok(val) = std::env::var("WEBSOCKET_MAX_MESSAGES") {
3810 if val.to_lowercase() == "none" || val == "0" {
3811 self.websocket.max_messages = None;
3812 } else if let Ok(n) = val.parse::<usize>() {
3813 self.websocket.max_messages = Some(n);
3814 }
3815 }
3816 if let Ok(val) = std::env::var("WEBSOCKET_MAX_BYTES") {
3817 if val.to_lowercase() == "none" || val == "0" {
3818 self.websocket.max_bytes = None;
3819 } else if let Ok(n) = val.parse::<usize>() {
3820 self.websocket.max_bytes = Some(n);
3821 }
3822 }
3823 self.websocket.disconnect_on_buffer_full = parse_bool_env(
3824 "WEBSOCKET_DISCONNECT_ON_BUFFER_FULL",
3825 self.websocket.disconnect_on_buffer_full,
3826 );
3827 self.websocket.max_message_size = parse_env::<usize>(
3828 "WEBSOCKET_MAX_MESSAGE_SIZE",
3829 self.websocket.max_message_size,
3830 );
3831 self.websocket.max_frame_size =
3832 parse_env::<usize>("WEBSOCKET_MAX_FRAME_SIZE", self.websocket.max_frame_size);
3833 self.websocket.write_buffer_size = parse_env::<usize>(
3834 "WEBSOCKET_WRITE_BUFFER_SIZE",
3835 self.websocket.write_buffer_size,
3836 );
3837 self.websocket.max_backpressure = parse_env::<usize>(
3838 "WEBSOCKET_MAX_BACKPRESSURE",
3839 self.websocket.max_backpressure,
3840 );
3841 self.websocket.auto_ping = parse_bool_env("WEBSOCKET_AUTO_PING", self.websocket.auto_ping);
3842 self.websocket.ping_interval =
3843 parse_env::<u32>("WEBSOCKET_PING_INTERVAL", self.websocket.ping_interval);
3844 self.websocket.idle_timeout =
3845 parse_env::<u32>("WEBSOCKET_IDLE_TIMEOUT", self.websocket.idle_timeout);
3846 if let Ok(mode) = std::env::var("WEBSOCKET_COMPRESSION") {
3847 self.websocket.compression = mode;
3848 }
3849
3850 self.connection_recovery.enabled = parse_bool_env(
3852 "CONNECTION_RECOVERY_ENABLED",
3853 self.connection_recovery.enabled,
3854 );
3855 self.connection_recovery.buffer_ttl_seconds = parse_env::<u64>(
3856 "CONNECTION_RECOVERY_BUFFER_TTL",
3857 self.connection_recovery.buffer_ttl_seconds,
3858 );
3859 self.connection_recovery.max_buffer_size = parse_env::<usize>(
3860 "CONNECTION_RECOVERY_MAX_BUFFER_SIZE",
3861 self.connection_recovery.max_buffer_size,
3862 );
3863
3864 self.history.enabled = parse_bool_env("HISTORY_ENABLED", self.history.enabled);
3865 self.history.rewind_enabled =
3866 parse_bool_env("HISTORY_REWIND_ENABLED", self.history.rewind_enabled);
3867 self.history.retention_window_seconds = parse_env::<u64>(
3868 "HISTORY_RETENTION_WINDOW_SECONDS",
3869 self.history.retention_window_seconds,
3870 );
3871 self.history.max_page_size =
3872 parse_env::<usize>("HISTORY_MAX_PAGE_SIZE", self.history.max_page_size);
3873 self.history.writer_shards =
3874 parse_env::<usize>("HISTORY_WRITER_SHARDS", self.history.writer_shards);
3875 self.history.writer_queue_capacity = parse_env::<usize>(
3876 "HISTORY_WRITER_QUEUE_CAPACITY",
3877 self.history.writer_queue_capacity,
3878 );
3879 if let Ok(backend) = std::env::var("HISTORY_BACKEND") {
3880 self.history.backend = HistoryBackend::from_str(&backend)?;
3881 }
3882 if let Ok(max_messages) = std::env::var("HISTORY_MAX_MESSAGES_PER_CHANNEL") {
3883 self.history.max_messages_per_channel = Some(
3884 max_messages
3885 .parse::<usize>()
3886 .map_err(|e| format!("Invalid HISTORY_MAX_MESSAGES_PER_CHANNEL: {e}"))?,
3887 );
3888 }
3889 if let Ok(max_bytes) = std::env::var("HISTORY_MAX_BYTES_PER_CHANNEL") {
3890 self.history.max_bytes_per_channel = Some(
3891 max_bytes
3892 .parse::<u64>()
3893 .map_err(|e| format!("Invalid HISTORY_MAX_BYTES_PER_CHANNEL: {e}"))?,
3894 );
3895 }
3896 if let Ok(table_prefix) = std::env::var("HISTORY_POSTGRES_TABLE_PREFIX") {
3897 self.history.postgres.table_prefix = table_prefix;
3898 }
3899 self.history.postgres.write_timeout_ms = parse_env::<u64>(
3900 "HISTORY_POSTGRES_WRITE_TIMEOUT_MS",
3901 self.history.postgres.write_timeout_ms,
3902 );
3903 self.history.purge_interval_seconds = parse_env::<u64>(
3904 "HISTORY_PURGE_INTERVAL_SECONDS",
3905 self.history.purge_interval_seconds,
3906 );
3907 self.history.purge_batch_size =
3908 parse_env::<usize>("HISTORY_PURGE_BATCH_SIZE", self.history.purge_batch_size);
3909 self.history.max_purge_per_tick = parse_env::<usize>(
3910 "HISTORY_MAX_PURGE_PER_TICK",
3911 self.history.max_purge_per_tick,
3912 );
3913
3914 self.presence_history.enabled =
3915 parse_bool_env("PRESENCE_HISTORY_ENABLED", self.presence_history.enabled);
3916 self.presence_history.retention_window_seconds = parse_env::<u64>(
3917 "PRESENCE_HISTORY_RETENTION_WINDOW_SECONDS",
3918 self.presence_history.retention_window_seconds,
3919 );
3920 self.presence_history.max_page_size = parse_env::<usize>(
3921 "PRESENCE_HISTORY_MAX_PAGE_SIZE",
3922 self.presence_history.max_page_size,
3923 );
3924 if let Ok(max_events) = std::env::var("PRESENCE_HISTORY_MAX_EVENTS_PER_CHANNEL") {
3925 self.presence_history.max_events_per_channel =
3926 Some(max_events.parse::<usize>().map_err(|e| {
3927 format!("Invalid PRESENCE_HISTORY_MAX_EVENTS_PER_CHANNEL: {e}")
3928 })?);
3929 }
3930 if let Ok(max_bytes) = std::env::var("PRESENCE_HISTORY_MAX_BYTES_PER_CHANNEL") {
3931 self.presence_history.max_bytes_per_channel = Some(
3932 max_bytes
3933 .parse::<u64>()
3934 .map_err(|e| format!("Invalid PRESENCE_HISTORY_MAX_BYTES_PER_CHANNEL: {e}"))?,
3935 );
3936 }
3937 self.idempotency.enabled = parse_bool_env("IDEMPOTENCY_ENABLED", self.idempotency.enabled);
3938 self.idempotency.ttl_seconds =
3939 parse_env::<u64>("IDEMPOTENCY_TTL_SECONDS", self.idempotency.ttl_seconds);
3940 self.idempotency.max_key_length = parse_env::<usize>(
3941 "IDEMPOTENCY_MAX_KEY_LENGTH",
3942 self.idempotency.max_key_length,
3943 );
3944
3945 self.ephemeral.enabled = parse_bool_env("EPHEMERAL_ENABLED", self.ephemeral.enabled);
3946
3947 self.echo_control.enabled =
3948 parse_bool_env("ECHO_CONTROL_ENABLED", self.echo_control.enabled);
3949 self.echo_control.default_echo_messages = parse_bool_env(
3950 "ECHO_CONTROL_DEFAULT_ECHO_MESSAGES",
3951 self.echo_control.default_echo_messages,
3952 );
3953
3954 self.event_name_filtering.enabled = parse_bool_env(
3955 "EVENT_NAME_FILTERING_ENABLED",
3956 self.event_name_filtering.enabled,
3957 );
3958 self.event_name_filtering.max_events_per_filter = parse_env::<usize>(
3959 "EVENT_NAME_FILTERING_MAX_EVENTS_PER_FILTER",
3960 self.event_name_filtering.max_events_per_filter,
3961 );
3962 self.event_name_filtering.max_event_name_length = parse_env::<usize>(
3963 "EVENT_NAME_FILTERING_MAX_EVENT_NAME_LENGTH",
3964 self.event_name_filtering.max_event_name_length,
3965 );
3966 self.versioned_messages.enabled = parse_bool_env(
3967 "VERSIONED_MESSAGES_ENABLED",
3968 self.versioned_messages.enabled,
3969 );
3970 if let Ok(driver_str) = std::env::var("VERSIONED_MESSAGES_DRIVER") {
3971 self.versioned_messages.driver = parse_driver_enum(
3972 driver_str,
3973 self.versioned_messages.driver.clone(),
3974 "VersionedMessages Backend",
3975 );
3976 }
3977 if let Ok(driver_str) = std::env::var("PUSH_STORAGE_DRIVER") {
3978 self.push.storage_driver = parse_driver_enum(
3979 driver_str,
3980 self.push.storage_driver.clone(),
3981 "Push Storage Backend",
3982 );
3983 }
3984 if let Ok(driver_str) = std::env::var("PUSH_QUEUE_DRIVER") {
3985 self.push.queue_driver = parse_driver_enum(
3986 driver_str,
3987 self.push.queue_driver.clone(),
3988 "Push Queue Backend",
3989 );
3990 }
3991 self.push.fcm_enabled = parse_bool_env("PUSH_FCM_ENABLED", self.push.fcm_enabled);
3992 self.push.apns_enabled = parse_bool_env("PUSH_APNS_ENABLED", self.push.apns_enabled);
3993 self.push.webpush_enabled =
3994 parse_bool_env("PUSH_WEBPUSH_ENABLED", self.push.webpush_enabled);
3995 self.push.hms_enabled = parse_bool_env("PUSH_HMS_ENABLED", self.push.hms_enabled);
3996 self.push.wns_enabled = parse_bool_env("PUSH_WNS_ENABLED", self.push.wns_enabled);
3997 if let Some(key) = parse_env_optional::<String>("PUSH_CREDENTIAL_ENCRYPTION_KEY") {
3998 self.push.credential_encryption_key = Some(key);
3999 }
4000 self.push.fanout_fast_threshold = parse_env::<u64>(
4001 "PUSH_FANOUT_FAST_THRESHOLD",
4002 self.push.fanout_fast_threshold,
4003 );
4004 self.push.fanout_shard_size =
4005 parse_env::<u64>("PUSH_FANOUT_SHARD_SIZE", self.push.fanout_shard_size);
4006 self.push.fanout_sync_threshold = parse_env::<u64>(
4007 "PUSH_FANOUT_SYNC_THRESHOLD",
4008 self.push.fanout_sync_threshold,
4009 );
4010 self.push.backpressure_lag_threshold_secs = parse_env::<u64>(
4011 "PUSH_BACKPRESSURE_LAG_THRESHOLD_SECS",
4012 self.push.backpressure_lag_threshold_secs,
4013 );
4014 self.push.publish_status_ttl_days = parse_env::<u64>(
4015 "PUSH_PUBLISH_STATUS_TTL_DAYS",
4016 self.push.publish_status_ttl_days,
4017 );
4018 self.push.circuit_breaker.failure_threshold = parse_env::<u32>(
4019 "PUSH_FAILURE_THRESHOLD",
4020 self.push.circuit_breaker.failure_threshold,
4021 );
4022 self.push.scheduler_interval_secs = parse_env::<u64>(
4023 "PUSH_SCHEDULER_INTERVAL_SECS",
4024 self.push.scheduler_interval_secs,
4025 );
4026 self.push.stale_device_max_age_days = parse_env::<u64>(
4027 "PUSH_STALE_DEVICE_MAX_AGE_DAYS",
4028 self.push.stale_device_max_age_days,
4029 );
4030 self.push.analytics_enabled =
4031 parse_bool_env("PUSH_ANALYTICS_ENABLED", self.push.analytics_enabled);
4032 self.push.default_quotas.acceptance_rps = parse_env::<u64>(
4033 "PUSH_DEFAULT_ACCEPTANCE_RPS",
4034 self.push.default_quotas.acceptance_rps,
4035 );
4036 self.push.default_quotas.delivery_quota_daily = parse_env::<u64>(
4037 "PUSH_DEFAULT_DELIVERY_QUOTA_DAILY",
4038 self.push.default_quotas.delivery_quota_daily,
4039 );
4040 self.push.default_quotas.fanout_max = parse_env::<u64>(
4041 "PUSH_DEFAULT_FANOUT_MAX",
4042 self.push.default_quotas.fanout_max,
4043 );
4044 self.push.default_quotas.inflight_max = parse_env::<u64>(
4045 "PUSH_DEFAULT_INFLIGHT_MAX",
4046 self.push.default_quotas.inflight_max,
4047 );
4048 self.versioned_messages.max_page_size = parse_env::<usize>(
4049 "VERSIONED_MESSAGES_MAX_PAGE_SIZE",
4050 self.versioned_messages.max_page_size,
4051 );
4052 self.versioned_messages.retention_window_seconds = parse_env::<u64>(
4053 "VERSIONED_MESSAGES_RETENTION_WINDOW_SECONDS",
4054 self.versioned_messages.retention_window_seconds,
4055 );
4056 self.versioned_messages.purge_interval_seconds = parse_env::<u64>(
4057 "VERSIONED_MESSAGES_PURGE_INTERVAL_SECONDS",
4058 self.versioned_messages.purge_interval_seconds,
4059 );
4060 self.versioned_messages.purge_batch_size = parse_env::<usize>(
4061 "VERSIONED_MESSAGES_PURGE_BATCH_SIZE",
4062 self.versioned_messages.purge_batch_size,
4063 );
4064 self.versioned_messages.max_purge_per_tick = parse_env::<usize>(
4065 "VERSIONED_MESSAGES_MAX_PURGE_PER_TICK",
4066 self.versioned_messages.max_purge_per_tick,
4067 );
4068 self.annotations.enabled = parse_bool_env("ANNOTATIONS_ENABLED", self.annotations.enabled);
4069
4070 Ok(())
4071 }
4072
4073 pub fn validate(&self) -> Result<(), String> {
4074 if self.unix_socket.enabled {
4075 if self.unix_socket.path.is_empty() {
4076 return Err(
4077 "Unix socket path cannot be empty when Unix socket is enabled".to_string(),
4078 );
4079 }
4080
4081 self.validate_unix_socket_security()?;
4082
4083 if self.ssl.enabled {
4084 tracing::warn!(
4085 "Both Unix socket and SSL are enabled. This is unusual as Unix sockets are typically used behind reverse proxies that handle SSL termination."
4086 );
4087 }
4088
4089 if self.unix_socket.permission_mode > 0o777 {
4090 return Err(format!(
4091 "Unix socket permission_mode ({:o}) is invalid. Must be a valid octal mode (0o000 to 0o777)",
4092 self.unix_socket.permission_mode
4093 ));
4094 }
4095 }
4096
4097 if let Err(e) = self.cleanup.validate() {
4098 return Err(format!("Invalid cleanup configuration: {}", e));
4099 }
4100
4101 if self.history.enabled {
4102 if self.history.max_page_size == 0 {
4103 return Err("history.max_page_size must be greater than 0".to_string());
4104 }
4105 if self.history.writer_shards == 0 {
4106 return Err("history.writer_shards must be greater than 0".to_string());
4107 }
4108 if self.history.writer_queue_capacity == 0 {
4109 return Err("history.writer_queue_capacity must be greater than 0".to_string());
4110 }
4111 if self.history.retention_window_seconds == 0 {
4112 return Err("history.retention_window_seconds must be greater than 0".to_string());
4113 }
4114 if self.history.postgres.table_prefix.trim().is_empty() {
4115 return Err("history.postgres.table_prefix must not be empty".to_string());
4116 }
4117 }
4118
4119 if self.presence_history.enabled {
4120 if self.presence_history.max_page_size == 0 {
4121 return Err("presence_history.max_page_size must be greater than 0".to_string());
4122 }
4123 if self.presence_history.retention_window_seconds == 0 {
4124 return Err(
4125 "presence_history.retention_window_seconds must be greater than 0".to_string(),
4126 );
4127 }
4128 }
4129
4130 if self.versioned_messages.enabled && self.versioned_messages.max_page_size == 0 {
4131 return Err("versioned_messages.max_page_size must be greater than 0".to_string());
4132 }
4133 if self.annotations.enabled && !self.versioned_messages.enabled {
4134 return Err("annotations require versioned_messages.enabled".to_string());
4135 }
4136
4137 if self.adapter.nats.presence_sync_chunk_size == Some(0) {
4138 return Err("nats.presence_sync_chunk_size must be > 0 when set".to_string());
4139 }
4140
4141 Ok(())
4142 }
4143
4144 fn validate_unix_socket_security(&self) -> Result<(), String> {
4145 let path = &self.unix_socket.path;
4146
4147 if path.contains("../") || path.contains("..\\") {
4148 return Err(
4149 "Unix socket path contains directory traversal sequences (../). This is not allowed for security reasons.".to_string()
4150 );
4151 }
4152
4153 if self.unix_socket.permission_mode & 0o002 != 0 {
4154 warn!(
4155 "Unix socket permission mode ({:o}) allows world write access. This may be a security risk. Consider using more restrictive permissions like 0o660 or 0o750.",
4156 self.unix_socket.permission_mode
4157 );
4158 }
4159
4160 if self.unix_socket.permission_mode & 0o007 > 0o005 {
4161 warn!(
4162 "Unix socket permission mode ({:o}) grants write permissions to others. Consider using more restrictive permissions.",
4163 self.unix_socket.permission_mode
4164 );
4165 }
4166
4167 if self.mode == "production" && path.starts_with("/tmp/") {
4168 warn!(
4169 "Unix socket path '{}' is in /tmp directory. In production, consider using a more permanent location like /var/run/sockudo/ for better security and persistence.",
4170 path
4171 );
4172 }
4173
4174 if !path.starts_with('/') {
4175 return Err(
4176 "Unix socket path must be absolute (start with /) for security and reliability."
4177 .to_string(),
4178 );
4179 }
4180
4181 Ok(())
4182 }
4183}
4184
4185#[cfg(test)]
4186mod redis_connection_tests {
4187 use super::{ClusterNode, RedisClusterConnection, RedisConnection, RedisSentinel};
4188
4189 #[test]
4190 fn test_standard_url_basic() {
4191 let conn = RedisConnection {
4192 host: "127.0.0.1".to_string(),
4193 port: 6379,
4194 db: 0,
4195 username: None,
4196 password: None,
4197 key_prefix: "sockudo:".to_string(),
4198 sentinels: Vec::new(),
4199 sentinel_password: None,
4200 name: "mymaster".to_string(),
4201 cluster: RedisClusterConnection::default(),
4202 cluster_nodes: Vec::new(),
4203 };
4204 assert_eq!(conn.to_url(), "redis://127.0.0.1:6379/0");
4205 }
4206
4207 #[test]
4208 fn test_standard_url_with_password() {
4209 let conn = RedisConnection {
4210 host: "127.0.0.1".to_string(),
4211 port: 6379,
4212 db: 2,
4213 username: None,
4214 password: Some("secret".to_string()),
4215 key_prefix: "sockudo:".to_string(),
4216 sentinels: Vec::new(),
4217 sentinel_password: None,
4218 name: "mymaster".to_string(),
4219 cluster: RedisClusterConnection::default(),
4220 cluster_nodes: Vec::new(),
4221 };
4222 assert_eq!(conn.to_url(), "redis://:secret@127.0.0.1:6379/2");
4223 }
4224
4225 #[test]
4226 fn test_standard_url_with_username_and_password() {
4227 let conn = RedisConnection {
4228 host: "redis.example.com".to_string(),
4229 port: 6380,
4230 db: 1,
4231 username: Some("admin".to_string()),
4232 password: Some("pass123".to_string()),
4233 key_prefix: "sockudo:".to_string(),
4234 sentinels: Vec::new(),
4235 sentinel_password: None,
4236 name: "mymaster".to_string(),
4237 cluster: RedisClusterConnection::default(),
4238 cluster_nodes: Vec::new(),
4239 };
4240 assert_eq!(
4241 conn.to_url(),
4242 "redis://admin:pass123@redis.example.com:6380/1"
4243 );
4244 }
4245
4246 #[test]
4247 fn test_standard_url_with_special_chars_in_password() {
4248 let conn = RedisConnection {
4249 host: "127.0.0.1".to_string(),
4250 port: 6379,
4251 db: 0,
4252 username: None,
4253 password: Some("pass@word#123".to_string()),
4254 key_prefix: "sockudo:".to_string(),
4255 sentinels: Vec::new(),
4256 sentinel_password: None,
4257 name: "mymaster".to_string(),
4258 cluster: RedisClusterConnection::default(),
4259 cluster_nodes: Vec::new(),
4260 };
4261 assert_eq!(conn.to_url(), "redis://:pass%40word%23123@127.0.0.1:6379/0");
4262 }
4263
4264 #[test]
4265 fn test_is_sentinel_configured_false() {
4266 let conn = RedisConnection::default();
4267 assert!(!conn.is_sentinel_configured());
4268 }
4269
4270 #[test]
4271 fn test_is_sentinel_configured_true() {
4272 let conn = RedisConnection {
4273 sentinels: vec![RedisSentinel {
4274 host: "sentinel1".to_string(),
4275 port: 26379,
4276 }],
4277 ..Default::default()
4278 };
4279 assert!(conn.is_sentinel_configured());
4280 }
4281
4282 #[test]
4283 fn test_sentinel_url_basic() {
4284 let conn = RedisConnection {
4285 host: "127.0.0.1".to_string(),
4286 port: 6379,
4287 db: 0,
4288 username: None,
4289 password: None,
4290 key_prefix: "sockudo:".to_string(),
4291 sentinels: vec![
4292 RedisSentinel {
4293 host: "sentinel1".to_string(),
4294 port: 26379,
4295 },
4296 RedisSentinel {
4297 host: "sentinel2".to_string(),
4298 port: 26379,
4299 },
4300 ],
4301 sentinel_password: None,
4302 name: "mymaster".to_string(),
4303 cluster: RedisClusterConnection::default(),
4304 cluster_nodes: Vec::new(),
4305 };
4306 assert_eq!(
4307 conn.to_url(),
4308 "redis+sentinel://sentinel1:26379,sentinel2:26379/mymaster/0"
4309 );
4310 }
4311
4312 #[test]
4313 fn test_sentinel_url_with_sentinel_password() {
4314 let conn = RedisConnection {
4315 host: "127.0.0.1".to_string(),
4316 port: 6379,
4317 db: 0,
4318 username: None,
4319 password: None,
4320 key_prefix: "sockudo:".to_string(),
4321 sentinels: vec![RedisSentinel {
4322 host: "sentinel1".to_string(),
4323 port: 26379,
4324 }],
4325 sentinel_password: Some("sentinelpass".to_string()),
4326 name: "mymaster".to_string(),
4327 cluster: RedisClusterConnection::default(),
4328 cluster_nodes: Vec::new(),
4329 };
4330 assert_eq!(
4331 conn.to_url(),
4332 "redis+sentinel://:sentinelpass@sentinel1:26379/mymaster/0"
4333 );
4334 }
4335
4336 #[test]
4337 fn test_sentinel_url_with_master_password() {
4338 let conn = RedisConnection {
4339 host: "127.0.0.1".to_string(),
4340 port: 6379,
4341 db: 1,
4342 username: None,
4343 password: Some("masterpass".to_string()),
4344 key_prefix: "sockudo:".to_string(),
4345 sentinels: vec![RedisSentinel {
4346 host: "sentinel1".to_string(),
4347 port: 26379,
4348 }],
4349 sentinel_password: None,
4350 name: "mymaster".to_string(),
4351 cluster: RedisClusterConnection::default(),
4352 cluster_nodes: Vec::new(),
4353 };
4354 assert_eq!(
4355 conn.to_url(),
4356 "redis+sentinel://sentinel1:26379/mymaster/1?password=masterpass"
4357 );
4358 }
4359
4360 #[test]
4361 fn test_sentinel_url_with_all_auth() {
4362 let conn = RedisConnection {
4363 host: "127.0.0.1".to_string(),
4364 port: 6379,
4365 db: 2,
4366 username: Some("redisuser".to_string()),
4367 password: Some("redispass".to_string()),
4368 key_prefix: "sockudo:".to_string(),
4369 sentinels: vec![
4370 RedisSentinel {
4371 host: "sentinel1".to_string(),
4372 port: 26379,
4373 },
4374 RedisSentinel {
4375 host: "sentinel2".to_string(),
4376 port: 26380,
4377 },
4378 ],
4379 sentinel_password: Some("sentinelauth".to_string()),
4380 name: "production-master".to_string(),
4381 cluster: RedisClusterConnection::default(),
4382 cluster_nodes: Vec::new(),
4383 };
4384 assert_eq!(
4385 conn.to_url(),
4386 "redis+sentinel://:sentinelauth@sentinel1:26379,sentinel2:26380/production-master/2?password=redispass&username=redisuser"
4387 );
4388 }
4389
4390 #[test]
4391 fn test_sentinel_to_host_port() {
4392 let sentinel = RedisSentinel {
4393 host: "sentinel.example.com".to_string(),
4394 port: 26379,
4395 };
4396 assert_eq!(sentinel.to_host_port(), "sentinel.example.com:26379");
4397 }
4398
4399 #[test]
4400 fn test_cluster_node_urls_with_shared_cluster_auth_and_tls() {
4401 let conn = RedisConnection {
4402 cluster: RedisClusterConnection {
4403 nodes: vec![
4404 ClusterNode {
4405 host: "node1.secure-cluster.com".to_string(),
4406 port: 7000,
4407 },
4408 ClusterNode {
4409 host: "redis://node2.secure-cluster.com:7001".to_string(),
4410 port: 7001,
4411 },
4412 ClusterNode {
4413 host: "rediss://node3.secure-cluster.com".to_string(),
4414 port: 7002,
4415 },
4416 ],
4417 username: None,
4418 password: Some("cluster-secret".to_string()),
4419 use_tls: true,
4420 },
4421 ..Default::default()
4422 };
4423
4424 assert_eq!(
4425 conn.cluster_node_urls(),
4426 vec![
4427 "rediss://:cluster-secret@node1.secure-cluster.com:7000",
4428 "redis://:cluster-secret@node2.secure-cluster.com:7001",
4429 "rediss://:cluster-secret@node3.secure-cluster.com:7002",
4430 ]
4431 );
4432 }
4433
4434 #[test]
4435 fn test_cluster_node_urls_fallback_to_legacy_nodes() {
4436 let conn = RedisConnection {
4437 password: Some("fallback-secret".to_string()),
4438 cluster_nodes: vec![ClusterNode {
4439 host: "legacy-node.example.com".to_string(),
4440 port: 7000,
4441 }],
4442 ..Default::default()
4443 };
4444
4445 assert_eq!(
4446 conn.cluster_node_urls(),
4447 vec!["redis://:fallback-secret@legacy-node.example.com:7000"]
4448 );
4449 }
4450
4451 #[test]
4452 fn test_normalize_cluster_seed_urls() {
4453 let conn = RedisConnection {
4454 cluster: RedisClusterConnection {
4455 nodes: Vec::new(),
4456 username: Some("svc-user".to_string()),
4457 password: Some("svc-pass".to_string()),
4458 use_tls: true,
4459 },
4460 ..Default::default()
4461 };
4462
4463 let seeds = vec![
4464 "node1.example.com:7000".to_string(),
4465 "redis://node2.example.com:7001".to_string(),
4466 "rediss://node3.example.com".to_string(),
4467 ];
4468
4469 assert_eq!(
4470 conn.normalize_cluster_seed_urls(&seeds),
4471 vec![
4472 "rediss://svc-user:svc-pass@node1.example.com:7000",
4473 "redis://svc-user:svc-pass@node2.example.com:7001",
4474 "rediss://svc-user:svc-pass@node3.example.com:6379",
4475 ]
4476 );
4477 }
4478}
4479
4480#[cfg(test)]
4481mod cluster_node_tests {
4482 use super::ClusterNode;
4483
4484 #[test]
4485 fn test_to_url_basic_host() {
4486 let node = ClusterNode {
4487 host: "localhost".to_string(),
4488 port: 6379,
4489 };
4490 assert_eq!(node.to_url(), "redis://localhost:6379");
4491 }
4492
4493 #[test]
4494 fn test_to_url_ip_address() {
4495 let node = ClusterNode {
4496 host: "127.0.0.1".to_string(),
4497 port: 6379,
4498 };
4499 assert_eq!(node.to_url(), "redis://127.0.0.1:6379");
4500 }
4501
4502 #[test]
4503 fn test_to_url_with_redis_protocol() {
4504 let node = ClusterNode {
4505 host: "redis://example.com".to_string(),
4506 port: 6379,
4507 };
4508 assert_eq!(node.to_url(), "redis://example.com:6379");
4509 }
4510
4511 #[test]
4512 fn test_to_url_with_rediss_protocol() {
4513 let node = ClusterNode {
4514 host: "rediss://secure.example.com".to_string(),
4515 port: 6379,
4516 };
4517 assert_eq!(node.to_url(), "rediss://secure.example.com:6379");
4518 }
4519
4520 #[test]
4521 fn test_to_url_with_rediss_protocol_and_port_in_url() {
4522 let node = ClusterNode {
4523 host: "rediss://secure.example.com:7000".to_string(),
4524 port: 6379,
4525 };
4526 assert_eq!(node.to_url(), "rediss://secure.example.com:7000");
4527 }
4528
4529 #[test]
4530 fn test_to_url_with_redis_protocol_and_port_in_url() {
4531 let node = ClusterNode {
4532 host: "redis://example.com:7001".to_string(),
4533 port: 6379,
4534 };
4535 assert_eq!(node.to_url(), "redis://example.com:7001");
4536 }
4537
4538 #[test]
4539 fn test_to_url_with_trailing_whitespace() {
4540 let node = ClusterNode {
4541 host: " rediss://secure.example.com ".to_string(),
4542 port: 6379,
4543 };
4544 assert_eq!(node.to_url(), "rediss://secure.example.com:6379");
4545 }
4546
4547 #[test]
4548 fn test_to_url_custom_port() {
4549 let node = ClusterNode {
4550 host: "redis-cluster.example.com".to_string(),
4551 port: 7000,
4552 };
4553 assert_eq!(node.to_url(), "redis://redis-cluster.example.com:7000");
4554 }
4555
4556 #[test]
4557 fn test_to_url_plain_host_with_port_in_host_field() {
4558 let node = ClusterNode {
4559 host: "redis-cluster.example.com:7010".to_string(),
4560 port: 7000,
4561 };
4562 assert_eq!(node.to_url(), "redis://redis-cluster.example.com:7010");
4563 }
4564
4565 #[test]
4566 fn test_to_url_with_options_adds_auth_and_tls() {
4567 let node = ClusterNode {
4568 host: "node.example.com".to_string(),
4569 port: 7000,
4570 };
4571 assert_eq!(
4572 node.to_url_with_options(true, Some("svc-user"), Some("secret")),
4573 "rediss://svc-user:secret@node.example.com:7000"
4574 );
4575 }
4576
4577 #[test]
4578 fn test_to_url_with_options_keeps_embedded_auth() {
4579 let node = ClusterNode {
4580 host: "rediss://:node-secret@node.example.com:7000".to_string(),
4581 port: 7000,
4582 };
4583 assert_eq!(
4584 node.to_url_with_options(true, Some("global-user"), Some("global-secret")),
4585 "rediss://:node-secret@node.example.com:7000"
4586 );
4587 }
4588
4589 #[test]
4590 fn test_from_seed_parses_plain_host_port() {
4591 let node = ClusterNode::from_seed("cluster-node-1:7005").expect("node should parse");
4592 assert_eq!(node.host, "cluster-node-1");
4593 assert_eq!(node.port, 7005);
4594 }
4595
4596 #[test]
4597 fn test_from_seed_keeps_scheme_urls() {
4598 let node =
4599 ClusterNode::from_seed("rediss://secure.example.com:7005").expect("node should parse");
4600 assert_eq!(node.host, "rediss://secure.example.com:7005");
4601 assert_eq!(node.port, 7005);
4602 }
4603
4604 #[test]
4605 fn test_to_url_aws_elasticache_hostname() {
4606 let node = ClusterNode {
4607 host: "rediss://my-cluster.use1.cache.amazonaws.com".to_string(),
4608 port: 6379,
4609 };
4610 assert_eq!(
4611 node.to_url(),
4612 "rediss://my-cluster.use1.cache.amazonaws.com:6379"
4613 );
4614 }
4615
4616 #[test]
4617 fn test_to_url_with_ipv6_no_port() {
4618 let node = ClusterNode {
4619 host: "rediss://[::1]".to_string(),
4620 port: 6379,
4621 };
4622 assert_eq!(node.to_url(), "rediss://[::1]:6379");
4623 }
4624
4625 #[test]
4626 fn test_to_url_with_ipv6_and_port_in_url() {
4627 let node = ClusterNode {
4628 host: "rediss://[::1]:7000".to_string(),
4629 port: 6379,
4630 };
4631 assert_eq!(node.to_url(), "rediss://[::1]:7000");
4632 }
4633
4634 #[test]
4635 fn test_to_url_with_ipv6_full_address_no_port() {
4636 let node = ClusterNode {
4637 host: "rediss://[2001:db8::1]".to_string(),
4638 port: 6379,
4639 };
4640 assert_eq!(node.to_url(), "rediss://[2001:db8::1]:6379");
4641 }
4642
4643 #[test]
4644 fn test_to_url_with_ipv6_full_address_with_port() {
4645 let node = ClusterNode {
4646 host: "rediss://[2001:db8::1]:7000".to_string(),
4647 port: 6379,
4648 };
4649 assert_eq!(node.to_url(), "rediss://[2001:db8::1]:7000");
4650 }
4651
4652 #[test]
4653 fn test_to_url_with_redis_protocol_ipv6() {
4654 let node = ClusterNode {
4655 host: "redis://[::1]".to_string(),
4656 port: 6379,
4657 };
4658 assert_eq!(node.to_url(), "redis://[::1]:6379");
4659 }
4660}
4661
4662#[cfg(test)]
4663mod cors_config_tests {
4664 use super::CorsConfig;
4665
4666 fn cors_from_json(json: &str) -> sonic_rs::Result<CorsConfig> {
4667 sonic_rs::from_str(json)
4668 }
4669
4670 #[test]
4671 fn test_deserialize_valid_exact_origins() {
4672 let config =
4673 cors_from_json(r#"{"origin": ["https://example.com", "https://other.com"]}"#).unwrap();
4674 assert_eq!(config.origin.len(), 2);
4675 }
4676
4677 #[test]
4678 fn test_deserialize_valid_wildcard_patterns() {
4679 let config =
4680 cors_from_json(r#"{"origin": ["*.example.com", "https://*.staging.example.com"]}"#)
4681 .unwrap();
4682 assert_eq!(config.origin.len(), 2);
4683 }
4684
4685 #[test]
4686 fn test_deserialize_allows_special_markers() {
4687 assert!(cors_from_json(r#"{"origin": ["*"]}"#).is_ok());
4688 assert!(cors_from_json(r#"{"origin": ["Any"]}"#).is_ok());
4689 assert!(cors_from_json(r#"{"origin": ["any"]}"#).is_ok());
4690 assert!(cors_from_json(r#"{"origin": ["*", "https://example.com"]}"#).is_ok());
4691 }
4692
4693 #[test]
4694 fn test_deserialize_rejects_invalid_patterns() {
4695 assert!(cors_from_json(r#"{"origin": ["*.*bad"]}"#).is_err());
4696 assert!(cors_from_json(r#"{"origin": ["*."]}"#).is_err());
4697 assert!(cors_from_json(r#"{"origin": [""]}"#).is_err());
4698 assert!(cors_from_json(r#"{"origin": ["https://"]}"#).is_err());
4699 }
4700
4701 #[test]
4702 fn test_deserialize_rejects_mixed_valid_and_invalid() {
4703 assert!(cors_from_json(r#"{"origin": ["https://good.com", "*.*bad"]}"#).is_err());
4704 }
4705}