relay_config/
config.rs

1use std::collections::{BTreeMap, HashMap};
2use std::error::Error;
3use std::io::Write;
4use std::net::{IpAddr, SocketAddr, ToSocketAddrs};
5use std::num::NonZeroU8;
6use std::path::{Path, PathBuf};
7use std::str::FromStr;
8use std::time::Duration;
9use std::{env, fmt, fs, io};
10
11use anyhow::Context;
12use relay_auth::{PublicKey, RelayId, SecretKey, generate_key_pair, generate_relay_id};
13use relay_common::Dsn;
14use relay_kafka::{
15    ConfigError as KafkaConfigError, KafkaConfigParam, KafkaTopic, KafkaTopicConfig,
16    TopicAssignments,
17};
18use relay_metrics::MetricNamespace;
19use serde::de::{DeserializeOwned, Unexpected, Visitor};
20use serde::{Deserialize, Deserializer, Serialize, Serializer};
21use uuid::Uuid;
22
23use crate::aggregator::{AggregatorServiceConfig, ScopedAggregatorConfig};
24use crate::byte_size::ByteSize;
25use crate::upstream::UpstreamDescriptor;
26use crate::{RedisConfig, RedisConfigs, RedisConfigsRef, build_redis_configs};
27
28const DEFAULT_NETWORK_OUTAGE_GRACE_PERIOD: u64 = 10;
29
30static CONFIG_YAML_HEADER: &str = r###"# Please see the relevant documentation.
31# Performance tuning: https://docs.sentry.io/product/relay/operating-guidelines/
32# All config options: https://docs.sentry.io/product/relay/options/
33"###;
34
35/// Indicates config related errors.
36#[derive(Clone, Copy, Debug, Eq, Hash, PartialEq)]
37#[non_exhaustive]
38pub enum ConfigErrorKind {
39    /// Failed to open the file.
40    CouldNotOpenFile,
41    /// Failed to save a file.
42    CouldNotWriteFile,
43    /// Parsing YAML failed.
44    BadYaml,
45    /// Parsing JSON failed.
46    BadJson,
47    /// Invalid config value
48    InvalidValue,
49    /// The user attempted to run Relay with processing enabled, but uses a binary that was
50    /// compiled without the processing feature.
51    ProcessingNotAvailable,
52}
53
54impl fmt::Display for ConfigErrorKind {
55    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
56        match self {
57            Self::CouldNotOpenFile => write!(f, "could not open config file"),
58            Self::CouldNotWriteFile => write!(f, "could not write config file"),
59            Self::BadYaml => write!(f, "could not parse yaml config file"),
60            Self::BadJson => write!(f, "could not parse json config file"),
61            Self::InvalidValue => write!(f, "invalid config value"),
62            Self::ProcessingNotAvailable => write!(
63                f,
64                "was not compiled with processing, cannot enable processing"
65            ),
66        }
67    }
68}
69
70/// Defines the source of a config error
71#[derive(Debug, Default)]
72enum ConfigErrorSource {
73    /// An error occurring independently.
74    #[default]
75    None,
76    /// An error originating from a configuration file.
77    File(PathBuf),
78    /// An error originating in a field override (an env var, or a CLI parameter).
79    FieldOverride(String),
80}
81
82impl fmt::Display for ConfigErrorSource {
83    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
84        match self {
85            ConfigErrorSource::None => Ok(()),
86            ConfigErrorSource::File(file_name) => {
87                write!(f, " (file {})", file_name.display())
88            }
89            ConfigErrorSource::FieldOverride(name) => write!(f, " (field {name})"),
90        }
91    }
92}
93
94/// Indicates config related errors.
95#[derive(Debug)]
96pub struct ConfigError {
97    source: ConfigErrorSource,
98    kind: ConfigErrorKind,
99}
100
101impl ConfigError {
102    #[inline]
103    fn new(kind: ConfigErrorKind) -> Self {
104        Self {
105            source: ConfigErrorSource::None,
106            kind,
107        }
108    }
109
110    #[inline]
111    fn field(field: &'static str) -> Self {
112        Self {
113            source: ConfigErrorSource::FieldOverride(field.to_owned()),
114            kind: ConfigErrorKind::InvalidValue,
115        }
116    }
117
118    #[inline]
119    fn file(kind: ConfigErrorKind, p: impl AsRef<Path>) -> Self {
120        Self {
121            source: ConfigErrorSource::File(p.as_ref().to_path_buf()),
122            kind,
123        }
124    }
125
126    /// Returns the error kind of the error.
127    pub fn kind(&self) -> ConfigErrorKind {
128        self.kind
129    }
130}
131
132impl fmt::Display for ConfigError {
133    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
134        write!(f, "{}{}", self.kind(), self.source)
135    }
136}
137
138impl Error for ConfigError {}
139
140enum ConfigFormat {
141    Yaml,
142    Json,
143}
144
145impl ConfigFormat {
146    pub fn extension(&self) -> &'static str {
147        match self {
148            ConfigFormat::Yaml => "yml",
149            ConfigFormat::Json => "json",
150        }
151    }
152}
153
154trait ConfigObject: DeserializeOwned + Serialize {
155    /// The format in which to serialize this configuration.
156    fn format() -> ConfigFormat;
157
158    /// The basename of the config file.
159    fn name() -> &'static str;
160
161    /// The full filename of the config file, including the file extension.
162    fn path(base: &Path) -> PathBuf {
163        base.join(format!("{}.{}", Self::name(), Self::format().extension()))
164    }
165
166    /// Loads the config file from a file within the given directory location.
167    fn load(base: &Path) -> anyhow::Result<Self> {
168        let path = Self::path(base);
169
170        let f = fs::File::open(&path)
171            .with_context(|| ConfigError::file(ConfigErrorKind::CouldNotOpenFile, &path))?;
172        let f = io::BufReader::new(f);
173
174        let mut source = serde_vars::EnvSource::default();
175        match Self::format() {
176            ConfigFormat::Yaml => {
177                serde_vars::deserialize(serde_yaml::Deserializer::from_reader(f), &mut source)
178                    .with_context(|| ConfigError::file(ConfigErrorKind::BadYaml, &path))
179            }
180            ConfigFormat::Json => {
181                serde_vars::deserialize(&mut serde_json::Deserializer::from_reader(f), &mut source)
182                    .with_context(|| ConfigError::file(ConfigErrorKind::BadJson, &path))
183            }
184        }
185    }
186
187    /// Writes the configuration to a file within the given directory location.
188    fn save(&self, base: &Path) -> anyhow::Result<()> {
189        let path = Self::path(base);
190        let mut options = fs::OpenOptions::new();
191        options.write(true).truncate(true).create(true);
192
193        // Remove all non-user permissions for the newly created file
194        #[cfg(unix)]
195        {
196            use std::os::unix::fs::OpenOptionsExt;
197            options.mode(0o600);
198        }
199
200        let mut f = options
201            .open(&path)
202            .with_context(|| ConfigError::file(ConfigErrorKind::CouldNotWriteFile, &path))?;
203
204        match Self::format() {
205            ConfigFormat::Yaml => {
206                f.write_all(CONFIG_YAML_HEADER.as_bytes())?;
207                serde_yaml::to_writer(&mut f, self)
208                    .with_context(|| ConfigError::file(ConfigErrorKind::CouldNotWriteFile, &path))?
209            }
210            ConfigFormat::Json => serde_json::to_writer_pretty(&mut f, self)
211                .with_context(|| ConfigError::file(ConfigErrorKind::CouldNotWriteFile, &path))?,
212        }
213
214        f.write_all(b"\n").ok();
215
216        Ok(())
217    }
218}
219
220/// Structure used to hold information about configuration overrides via
221/// CLI parameters or environment variables
222#[derive(Debug, Default)]
223pub struct OverridableConfig {
224    /// The operation mode of this relay.
225    pub mode: Option<String>,
226    /// The instance type of this relay.
227    pub instance: Option<String>,
228    /// The log level of this relay.
229    pub log_level: Option<String>,
230    /// The log format of this relay.
231    pub log_format: Option<String>,
232    /// The upstream relay or sentry instance.
233    pub upstream: Option<String>,
234    /// Alternate upstream provided through a Sentry DSN. Key and project will be ignored.
235    pub upstream_dsn: Option<String>,
236    /// The host the relay should bind to (network interface).
237    pub host: Option<String>,
238    /// The port to bind for the unencrypted relay HTTP server.
239    pub port: Option<String>,
240    /// "true" if processing is enabled "false" otherwise
241    pub processing: Option<String>,
242    /// the kafka bootstrap.servers configuration string
243    pub kafka_url: Option<String>,
244    /// the redis server url
245    pub redis_url: Option<String>,
246    /// The globally unique ID of the relay.
247    pub id: Option<String>,
248    /// The secret key of the relay
249    pub secret_key: Option<String>,
250    /// The public key of the relay
251    pub public_key: Option<String>,
252    /// Outcome source
253    pub outcome_source: Option<String>,
254    /// shutdown timeout
255    pub shutdown_timeout: Option<String>,
256    /// Server name reported in the Sentry SDK.
257    pub server_name: Option<String>,
258}
259
260/// The relay credentials
261#[derive(Serialize, Deserialize, Debug, Clone, PartialEq, Eq)]
262pub struct Credentials {
263    /// The secret key of the relay
264    pub secret_key: SecretKey,
265    /// The public key of the relay
266    pub public_key: PublicKey,
267    /// The globally unique ID of the relay.
268    pub id: RelayId,
269}
270
271impl Credentials {
272    /// Generates new random credentials.
273    pub fn generate() -> Self {
274        relay_log::info!("generating new relay credentials");
275        let (sk, pk) = generate_key_pair();
276        Self {
277            secret_key: sk,
278            public_key: pk,
279            id: generate_relay_id(),
280        }
281    }
282
283    /// Serializes this configuration to JSON.
284    pub fn to_json_string(&self) -> anyhow::Result<String> {
285        serde_json::to_string(self)
286            .with_context(|| ConfigError::new(ConfigErrorKind::CouldNotWriteFile))
287    }
288}
289
290impl ConfigObject for Credentials {
291    fn format() -> ConfigFormat {
292        ConfigFormat::Json
293    }
294    fn name() -> &'static str {
295        "credentials"
296    }
297}
298
299/// Information on a downstream Relay.
300#[derive(Debug, Clone, PartialEq, Eq, Serialize, Deserialize)]
301#[serde(rename_all = "camelCase")]
302pub struct RelayInfo {
303    /// The public key that this Relay uses to authenticate and sign requests.
304    pub public_key: PublicKey,
305
306    /// Marks an internal relay that has privileged access to more project configuration.
307    #[serde(default)]
308    pub internal: bool,
309}
310
311impl RelayInfo {
312    /// Creates a new RelayInfo
313    pub fn new(public_key: PublicKey) -> Self {
314        Self {
315            public_key,
316            internal: false,
317        }
318    }
319}
320
321/// The operation mode of a relay.
322#[derive(Clone, Copy, Debug, Eq, PartialEq, Serialize)]
323#[serde(rename_all = "camelCase")]
324pub enum RelayMode {
325    /// This relay acts as a proxy for all requests and events.
326    ///
327    /// Events are normalized and rate limits from the upstream are enforced, but the relay will not
328    /// fetch project configurations from the upstream or perform PII stripping. All events are
329    /// accepted unless overridden on the file system.
330    Proxy,
331
332    /// Project configurations are managed by the upstream.
333    ///
334    /// Project configurations are always fetched from the upstream, unless they are statically
335    /// overridden in the file system. This relay must be allowed in the upstream Sentry. This is
336    /// only possible, if the upstream is Sentry directly, or another managed Relay.
337    Managed,
338}
339
340impl<'de> Deserialize<'de> for RelayMode {
341    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
342    where
343        D: Deserializer<'de>,
344    {
345        let s = String::deserialize(deserializer)?;
346        match s.as_str() {
347            "proxy" => Ok(RelayMode::Proxy),
348            "managed" => Ok(RelayMode::Managed),
349            "static" => Err(serde::de::Error::custom(
350                "Relay mode 'static' has been removed. Please use 'managed' or 'proxy' instead.",
351            )),
352            other => Err(serde::de::Error::unknown_variant(
353                other,
354                &["proxy", "managed"],
355            )),
356        }
357    }
358}
359
360impl fmt::Display for RelayMode {
361    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
362        match self {
363            RelayMode::Proxy => write!(f, "proxy"),
364            RelayMode::Managed => write!(f, "managed"),
365        }
366    }
367}
368
369/// The instance type of Relay.
370#[derive(Clone, Copy, Debug, Eq, PartialEq, Deserialize, Serialize)]
371#[serde(rename_all = "camelCase")]
372pub enum RelayInstance {
373    /// This Relay is run as a default instance.
374    Default,
375
376    /// This Relay is run as a canary instance where experiments can be run.
377    Canary,
378}
379
380impl RelayInstance {
381    /// Returns `true` if the [`RelayInstance`] is of type [`RelayInstance::Canary`].
382    pub fn is_canary(&self) -> bool {
383        matches!(self, RelayInstance::Canary)
384    }
385}
386
387impl fmt::Display for RelayInstance {
388    fn fmt(&self, f: &mut fmt::Formatter) -> fmt::Result {
389        match self {
390            RelayInstance::Default => write!(f, "default"),
391            RelayInstance::Canary => write!(f, "canary"),
392        }
393    }
394}
395
396impl FromStr for RelayInstance {
397    type Err = fmt::Error;
398
399    fn from_str(s: &str) -> Result<Self, Self::Err> {
400        match s {
401            "canary" => Ok(RelayInstance::Canary),
402            _ => Ok(RelayInstance::Default),
403        }
404    }
405}
406
407/// Error returned when parsing an invalid [`RelayMode`].
408#[derive(Clone, Copy, Debug, Eq, PartialEq)]
409pub struct ParseRelayModeError;
410
411impl fmt::Display for ParseRelayModeError {
412    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
413        write!(f, "Relay mode must be one of: managed or proxy")
414    }
415}
416
417impl Error for ParseRelayModeError {}
418
419impl FromStr for RelayMode {
420    type Err = ParseRelayModeError;
421
422    fn from_str(s: &str) -> Result<Self, Self::Err> {
423        match s {
424            "proxy" => Ok(RelayMode::Proxy),
425            "managed" => Ok(RelayMode::Managed),
426            _ => Err(ParseRelayModeError),
427        }
428    }
429}
430
431/// Returns `true` if this value is equal to `Default::default()`.
432fn is_default<T: Default + PartialEq>(t: &T) -> bool {
433    *t == T::default()
434}
435
436/// Checks if we are running in docker.
437fn is_docker() -> bool {
438    if fs::metadata("/.dockerenv").is_ok() {
439        return true;
440    }
441
442    fs::read_to_string("/proc/self/cgroup").is_ok_and(|s| s.contains("/docker"))
443}
444
445/// Default value for the "bind" configuration.
446fn default_host() -> IpAddr {
447    if is_docker() {
448        // Docker images rely on this service being exposed
449        "0.0.0.0".parse().unwrap()
450    } else {
451        "127.0.0.1".parse().unwrap()
452    }
453}
454
455/// Controls responses from the readiness health check endpoint based on authentication.
456///
457/// Independent of the the readiness condition, shutdown always switches Relay into unready state.
458#[derive(Clone, Copy, Debug, Eq, PartialEq, Deserialize, Serialize)]
459#[serde(rename_all = "lowercase")]
460#[derive(Default)]
461pub enum ReadinessCondition {
462    /// (default) Relay is ready when authenticated and connected to the upstream.
463    ///
464    /// Before authentication has succeeded and during network outages, Relay responds as not ready.
465    /// Relay reauthenticates based on the `http.auth_interval` parameter. During reauthentication,
466    /// Relay remains ready until authentication fails.
467    ///
468    /// Authentication is only required for Relays in managed mode. Other Relays will only check for
469    /// network outages.
470    #[default]
471    Authenticated,
472    /// Relay reports readiness regardless of the authentication and networking state.
473    Always,
474}
475
476/// Relay specific configuration values.
477#[derive(Serialize, Deserialize, Debug)]
478#[serde(default)]
479pub struct Relay {
480    /// The operation mode of this relay.
481    pub mode: RelayMode,
482    /// The instance type of this relay.
483    pub instance: RelayInstance,
484    /// The upstream relay or sentry instance.
485    pub upstream: UpstreamDescriptor<'static>,
486    /// The host the relay should bind to (network interface).
487    pub host: IpAddr,
488    /// The port to bind for the unencrypted relay HTTP server.
489    pub port: u16,
490    /// The host the relay should bind to (network interface) for internally exposed APIs, like
491    /// health checks.
492    ///
493    /// If not configured, internal routes are exposed on the main HTTP server.
494    ///
495    /// Note: configuring the internal http server on an address which overlaps with the main
496    /// server (e.g. main on `0.0.0.0:3000` and internal on `127.0.0.1:3000`) is a misconfiguration
497    /// resulting in approximately half of the requests sent to `127.0.0.1:3000` to fail, as the handling
498    /// http server is chosen by the operating system 'at random'.
499    ///
500    /// As a best practice you should always choose different ports to avoid this issue.
501    ///
502    /// Defaults to [`Self::host`].
503    pub internal_host: Option<IpAddr>,
504    /// The port to bind for internally exposed APIs.
505    ///
506    /// Defaults to [`Self::port`].
507    pub internal_port: Option<u16>,
508    /// Optional port to bind for the encrypted relay HTTPS server.
509    #[serde(skip_serializing)]
510    pub tls_port: Option<u16>,
511    /// The path to the identity (DER-encoded PKCS12) to use for TLS.
512    #[serde(skip_serializing)]
513    pub tls_identity_path: Option<PathBuf>,
514    /// Password for the PKCS12 archive.
515    #[serde(skip_serializing)]
516    pub tls_identity_password: Option<String>,
517    /// Always override project IDs from the URL and DSN with the identifier used at the upstream.
518    ///
519    /// Enable this setting for Relays used to redirect traffic to a migrated Sentry instance.
520    /// Validation of project identifiers can be safely skipped in these cases.
521    #[serde(skip_serializing_if = "is_default")]
522    pub override_project_ids: bool,
523}
524
525impl Default for Relay {
526    fn default() -> Self {
527        Relay {
528            mode: RelayMode::Managed,
529            instance: RelayInstance::Default,
530            upstream: "https://sentry.io/".parse().unwrap(),
531            host: default_host(),
532            port: 3000,
533            internal_host: None,
534            internal_port: None,
535            tls_port: None,
536            tls_identity_path: None,
537            tls_identity_password: None,
538            override_project_ids: false,
539        }
540    }
541}
542
543/// Control the metrics.
544#[derive(Serialize, Deserialize, Debug)]
545#[serde(default)]
546pub struct Metrics {
547    /// Hostname and port of the statsd server.
548    ///
549    /// Defaults to `None`.
550    pub statsd: Option<String>,
551    /// Common prefix that should be added to all metrics.
552    ///
553    /// Defaults to `"sentry.relay"`.
554    pub prefix: String,
555    /// Default tags to apply to all metrics.
556    pub default_tags: BTreeMap<String, String>,
557    /// Tag name to report the hostname to for each metric. Defaults to not sending such a tag.
558    pub hostname_tag: Option<String>,
559    /// Global sample rate for all emitted metrics between `0.0` and `1.0`.
560    ///
561    /// For example, a value of `0.3` means that only 30% of the emitted metrics will be sent.
562    /// Defaults to `1.0` (100%).
563    pub sample_rate: f32,
564    /// Interval for periodic metrics emitted from Relay.
565    ///
566    /// Setting it to `0` seconds disables the periodic metrics.
567    /// Defaults to 5 seconds.
568    pub periodic_secs: u64,
569    /// Whether local metric aggregation using statdsproxy should be enabled.
570    ///
571    /// Defaults to `true`.
572    pub aggregate: bool,
573    /// Allows emission of metrics with high cardinality tags.
574    ///
575    /// High cardinality tags are dynamic values attached to metrics,
576    /// such as project IDs. When enabled, these tags will be included
577    /// in the emitted metrics. When disabled, the tags will be omitted.
578    ///
579    /// Defaults to `false`.
580    pub allow_high_cardinality_tags: bool,
581}
582
583impl Default for Metrics {
584    fn default() -> Self {
585        Metrics {
586            statsd: None,
587            prefix: "sentry.relay".into(),
588            default_tags: BTreeMap::new(),
589            hostname_tag: None,
590            sample_rate: 1.0,
591            periodic_secs: 5,
592            aggregate: true,
593            allow_high_cardinality_tags: false,
594        }
595    }
596}
597
598/// Controls various limits
599#[derive(Serialize, Deserialize, Debug)]
600#[serde(default)]
601pub struct Limits {
602    /// How many requests can be sent concurrently from Relay to the upstream before Relay starts
603    /// buffering.
604    pub max_concurrent_requests: usize,
605    /// How many queries can be sent concurrently from Relay to the upstream before Relay starts
606    /// buffering.
607    ///
608    /// The concurrency of queries is additionally constrained by `max_concurrent_requests`.
609    pub max_concurrent_queries: usize,
610    /// The maximum payload size for events.
611    pub max_event_size: ByteSize,
612    /// The maximum size for each attachment.
613    pub max_attachment_size: ByteSize,
614    /// The maximum combined size for all attachments in an envelope or request.
615    pub max_attachments_size: ByteSize,
616    /// The maximum combined size for all client reports in an envelope or request.
617    pub max_client_reports_size: ByteSize,
618    /// The maximum payload size for a monitor check-in.
619    pub max_check_in_size: ByteSize,
620    /// The maximum payload size for an entire envelopes. Individual limits still apply.
621    pub max_envelope_size: ByteSize,
622    /// The maximum number of session items per envelope.
623    pub max_session_count: usize,
624    /// The maximum payload size for general API requests.
625    pub max_api_payload_size: ByteSize,
626    /// The maximum payload size for file uploads and chunks.
627    pub max_api_file_upload_size: ByteSize,
628    /// The maximum payload size for chunks
629    pub max_api_chunk_upload_size: ByteSize,
630    /// The maximum payload size for a profile
631    pub max_profile_size: ByteSize,
632    /// The maximum payload size for a trace metric.
633    pub max_trace_metric_size: ByteSize,
634    /// The maximum payload size for a log.
635    pub max_log_size: ByteSize,
636    /// The maximum payload size for a span.
637    pub max_span_size: ByteSize,
638    /// The maximum payload size for an item container.
639    pub max_container_size: ByteSize,
640    /// The maximum payload size for a statsd metric.
641    pub max_statsd_size: ByteSize,
642    /// The maximum payload size for metric buckets.
643    pub max_metric_buckets_size: ByteSize,
644    /// The maximum payload size for a compressed replay.
645    pub max_replay_compressed_size: ByteSize,
646    /// The maximum payload size for an uncompressed replay.
647    #[serde(alias = "max_replay_size")]
648    max_replay_uncompressed_size: ByteSize,
649    /// The maximum size for a replay recording Kafka message.
650    pub max_replay_message_size: ByteSize,
651    /// The maximum number of threads to spawn for CPU and web work, each.
652    ///
653    /// The total number of threads spawned will roughly be `2 * max_thread_count`. Defaults to
654    /// the number of logical CPU cores on the host.
655    pub max_thread_count: usize,
656    /// Controls the maximum concurrency of each worker thread.
657    ///
658    /// Increasing the concurrency, can lead to a better utilization of worker threads by
659    /// increasing the amount of I/O done concurrently.
660    //
661    /// Currently has no effect on defaults to `1`.
662    pub max_pool_concurrency: usize,
663    /// The maximum number of seconds a query is allowed to take across retries. Individual requests
664    /// have lower timeouts. Defaults to 30 seconds.
665    pub query_timeout: u64,
666    /// The maximum number of seconds to wait for pending envelopes after receiving a shutdown
667    /// signal.
668    pub shutdown_timeout: u64,
669    /// Server keep-alive timeout in seconds.
670    ///
671    /// By default, keep-alive is set to 5 seconds.
672    pub keepalive_timeout: u64,
673    /// Server idle timeout in seconds.
674    ///
675    /// The idle timeout limits the amount of time a connection is kept open without activity.
676    /// Setting this too short may abort connections before Relay is able to send a response.
677    ///
678    /// By default there is no idle timeout.
679    pub idle_timeout: Option<u64>,
680    /// Sets the maximum number of concurrent connections.
681    ///
682    /// Upon reaching the limit, the server will stop accepting connections.
683    ///
684    /// By default there is no limit.
685    pub max_connections: Option<usize>,
686    /// The TCP listen backlog.
687    ///
688    /// Configures the TCP listen backlog for the listening socket of Relay.
689    /// See [`man listen(2)`](https://man7.org/linux/man-pages/man2/listen.2.html)
690    /// for a more detailed description of the listen backlog.
691    ///
692    /// Defaults to `1024`, a value [google has been using for a long time](https://git.kernel.org/pub/scm/linux/kernel/git/torvalds/linux.git/commit/?id=19f92a030ca6d772ab44b22ee6a01378a8cb32d4).
693    pub tcp_listen_backlog: u32,
694}
695
696impl Default for Limits {
697    fn default() -> Self {
698        Limits {
699            max_concurrent_requests: 100,
700            max_concurrent_queries: 5,
701            max_event_size: ByteSize::mebibytes(1),
702            max_attachment_size: ByteSize::mebibytes(200),
703            max_attachments_size: ByteSize::mebibytes(200),
704            max_client_reports_size: ByteSize::kibibytes(4),
705            max_check_in_size: ByteSize::kibibytes(100),
706            max_envelope_size: ByteSize::mebibytes(200),
707            max_session_count: 100,
708            max_api_payload_size: ByteSize::mebibytes(20),
709            max_api_file_upload_size: ByteSize::mebibytes(40),
710            max_api_chunk_upload_size: ByteSize::mebibytes(100),
711            max_profile_size: ByteSize::mebibytes(50),
712            max_trace_metric_size: ByteSize::kibibytes(2),
713            max_log_size: ByteSize::mebibytes(1),
714            max_span_size: ByteSize::mebibytes(1),
715            max_container_size: ByteSize::mebibytes(12),
716            max_statsd_size: ByteSize::mebibytes(1),
717            max_metric_buckets_size: ByteSize::mebibytes(1),
718            max_replay_compressed_size: ByteSize::mebibytes(10),
719            max_replay_uncompressed_size: ByteSize::mebibytes(100),
720            max_replay_message_size: ByteSize::mebibytes(15),
721            max_thread_count: num_cpus::get(),
722            max_pool_concurrency: 1,
723            query_timeout: 30,
724            shutdown_timeout: 10,
725            keepalive_timeout: 5,
726            idle_timeout: None,
727            max_connections: None,
728            tcp_listen_backlog: 1024,
729        }
730    }
731}
732
733/// Controls traffic steering.
734#[derive(Debug, Default, Deserialize, Serialize)]
735#[serde(default)]
736pub struct Routing {
737    /// Accept and forward unknown Envelope items to the upstream.
738    ///
739    /// Forwarding unknown items should be enabled in most cases to allow proxying traffic for newer
740    /// SDK versions. The upstream in Sentry makes the final decision on which items are valid. If
741    /// this is disabled, just the unknown items are removed from Envelopes, and the rest is
742    /// processed as usual.
743    ///
744    /// Defaults to `true` for all Relay modes other than processing mode. In processing mode, this
745    /// is disabled by default since the item cannot be handled.
746    pub accept_unknown_items: Option<bool>,
747}
748
749/// Http content encoding for both incoming and outgoing web requests.
750#[derive(Clone, Copy, Debug, Default, Deserialize, Serialize)]
751#[serde(rename_all = "lowercase")]
752pub enum HttpEncoding {
753    /// Identity function without no compression.
754    ///
755    /// This is the default encoding and does not require the presence of the `content-encoding`
756    /// HTTP header.
757    #[default]
758    Identity,
759    /// Compression using a [zlib](https://en.wikipedia.org/wiki/Zlib) structure with
760    /// [deflate](https://en.wikipedia.org/wiki/DEFLATE) encoding.
761    ///
762    /// These structures are defined in [RFC 1950](https://datatracker.ietf.org/doc/html/rfc1950)
763    /// and [RFC 1951](https://datatracker.ietf.org/doc/html/rfc1951).
764    Deflate,
765    /// A format using the [Lempel-Ziv coding](https://en.wikipedia.org/wiki/LZ77_and_LZ78#LZ77)
766    /// (LZ77), with a 32-bit CRC.
767    ///
768    /// This is the original format of the UNIX gzip program. The HTTP/1.1 standard also recommends
769    /// that the servers supporting this content-encoding should recognize `x-gzip` as an alias, for
770    /// compatibility purposes.
771    Gzip,
772    /// A format using the [Brotli](https://en.wikipedia.org/wiki/Brotli) algorithm.
773    Br,
774    /// A format using the [Zstd](https://en.wikipedia.org/wiki/Zstd) compression algorithm.
775    Zstd,
776}
777
778impl HttpEncoding {
779    /// Parses a [`HttpEncoding`] from its `content-encoding` header value.
780    pub fn parse(str: &str) -> Self {
781        let str = str.trim();
782        if str.eq_ignore_ascii_case("zstd") {
783            Self::Zstd
784        } else if str.eq_ignore_ascii_case("br") {
785            Self::Br
786        } else if str.eq_ignore_ascii_case("gzip") || str.eq_ignore_ascii_case("x-gzip") {
787            Self::Gzip
788        } else if str.eq_ignore_ascii_case("deflate") {
789            Self::Deflate
790        } else {
791            Self::Identity
792        }
793    }
794
795    /// Returns the value for the `content-encoding` HTTP header.
796    ///
797    /// Returns `None` for [`Identity`](Self::Identity), and `Some` for other encodings.
798    pub fn name(&self) -> Option<&'static str> {
799        match self {
800            Self::Identity => None,
801            Self::Deflate => Some("deflate"),
802            Self::Gzip => Some("gzip"),
803            Self::Br => Some("br"),
804            Self::Zstd => Some("zstd"),
805        }
806    }
807}
808
809/// Controls authentication with upstream.
810#[derive(Serialize, Deserialize, Debug)]
811#[serde(default)]
812pub struct Http {
813    /// Timeout for upstream requests in seconds.
814    ///
815    /// This timeout covers the time from sending the request until receiving response headers.
816    /// Neither the connection process and handshakes, nor reading the response body is covered in
817    /// this timeout.
818    pub timeout: u32,
819    /// Timeout for establishing connections with the upstream in seconds.
820    ///
821    /// This includes SSL handshakes. Relay reuses connections when the upstream supports connection
822    /// keep-alive. Connections are retained for a maximum 75 seconds, or 15 seconds of inactivity.
823    pub connection_timeout: u32,
824    /// Maximum interval between failed request retries in seconds.
825    pub max_retry_interval: u32,
826    /// The custom HTTP Host header to send to the upstream.
827    pub host_header: Option<String>,
828    /// The interval in seconds at which Relay attempts to reauthenticate with the upstream server.
829    ///
830    /// Re-authentication happens even when Relay is idle. If authentication fails, Relay reverts
831    /// back into startup mode and tries to establish a connection. During this time, incoming
832    /// envelopes will be buffered.
833    ///
834    /// Defaults to `600` (10 minutes).
835    pub auth_interval: Option<u64>,
836    /// The maximum time of experiencing uninterrupted network failures until Relay considers that
837    /// it has encountered a network outage in seconds.
838    ///
839    /// During a network outage relay will try to reconnect and will buffer all upstream messages
840    /// until it manages to reconnect.
841    pub outage_grace_period: u64,
842    /// The time Relay waits before retrying an upstream request, in seconds.
843    ///
844    /// This time is only used before going into a network outage mode.
845    pub retry_delay: u64,
846    /// The interval in seconds for continued failed project fetches at which Relay will error.
847    ///
848    /// A successful fetch resets this interval. Relay does nothing during long
849    /// times without emitting requests.
850    pub project_failure_interval: u64,
851    /// Content encoding to apply to upstream store requests.
852    ///
853    /// By default, Relay applies `zstd` content encoding to compress upstream requests. Compression
854    /// can be disabled to reduce CPU consumption, but at the expense of increased network traffic.
855    ///
856    /// This setting applies to all store requests of SDK data, including events, transactions,
857    /// envelopes and sessions. At the moment, this does not apply to Relay's internal queries.
858    ///
859    /// Available options are:
860    ///
861    ///  - `identity`: Disables compression.
862    ///  - `deflate`: Compression using a zlib header with deflate encoding.
863    ///  - `gzip` (default): Compression using gzip.
864    ///  - `br`: Compression using the brotli algorithm.
865    ///  - `zstd`: Compression using the zstd algorithm.
866    pub encoding: HttpEncoding,
867    /// Submit metrics globally through a shared endpoint.
868    ///
869    /// As opposed to regular envelopes which are sent to an endpoint inferred from the project's
870    /// DSN, this submits metrics to the global endpoint with Relay authentication.
871    ///
872    /// This option does not have any effect on processing mode.
873    pub global_metrics: bool,
874}
875
876impl Default for Http {
877    fn default() -> Self {
878        Http {
879            timeout: 5,
880            connection_timeout: 3,
881            max_retry_interval: 60, // 1 minute
882            host_header: None,
883            auth_interval: Some(600), // 10 minutes
884            outage_grace_period: DEFAULT_NETWORK_OUTAGE_GRACE_PERIOD,
885            retry_delay: default_retry_delay(),
886            project_failure_interval: default_project_failure_interval(),
887            encoding: HttpEncoding::Zstd,
888            global_metrics: false,
889        }
890    }
891}
892
893/// Default for unavailable upstream retry period, 1s.
894fn default_retry_delay() -> u64 {
895    1
896}
897
898/// Default for project failure interval, 90s.
899fn default_project_failure_interval() -> u64 {
900    90
901}
902
903/// Default for max disk size, 500 MB.
904fn spool_envelopes_max_disk_size() -> ByteSize {
905    ByteSize::mebibytes(500)
906}
907
908/// Default number of encoded envelope bytes to cache before writing to disk.
909fn spool_envelopes_batch_size_bytes() -> ByteSize {
910    ByteSize::kibibytes(10)
911}
912
913fn spool_envelopes_max_envelope_delay_secs() -> u64 {
914    24 * 60 * 60
915}
916
917/// Default refresh frequency in ms for the disk usage monitoring.
918fn spool_disk_usage_refresh_frequency_ms() -> u64 {
919    100
920}
921
922/// Default bounded buffer size for handling backpressure.
923fn spool_max_backpressure_envelopes() -> usize {
924    500
925}
926
927/// Default max memory usage for unspooling.
928fn spool_max_backpressure_memory_percent() -> f32 {
929    0.9
930}
931
932/// Default number of partitions for the buffer.
933fn spool_envelopes_partitions() -> NonZeroU8 {
934    NonZeroU8::new(1).unwrap()
935}
936
937/// Persistent buffering configuration for incoming envelopes.
938#[derive(Debug, Serialize, Deserialize)]
939pub struct EnvelopeSpool {
940    /// The path of the SQLite database file(s) which persist the data.
941    ///
942    /// Based on the number of partitions, more database files will be created within the same path.
943    ///
944    /// If not set, the envelopes will be buffered in memory.
945    pub path: Option<PathBuf>,
946    /// The maximum size of the buffer to keep, in bytes.
947    ///
948    /// When the on-disk buffer reaches this size, new envelopes will be dropped.
949    ///
950    /// Defaults to 500MB.
951    #[serde(default = "spool_envelopes_max_disk_size")]
952    pub max_disk_size: ByteSize,
953    /// Size of the batch of compressed envelopes that are spooled to disk at once.
954    ///
955    /// Note that this is the size after which spooling will be triggered but it does not guarantee
956    /// that exactly this size will be spooled, it can be greater or equal.
957    ///
958    /// Defaults to 10 KiB.
959    #[serde(default = "spool_envelopes_batch_size_bytes")]
960    pub batch_size_bytes: ByteSize,
961    /// Maximum time between receiving the envelope and processing it.
962    ///
963    /// When envelopes spend too much time in the buffer (e.g. because their project cannot be loaded),
964    /// they are dropped.
965    ///
966    /// Defaults to 24h.
967    #[serde(default = "spool_envelopes_max_envelope_delay_secs")]
968    pub max_envelope_delay_secs: u64,
969    /// The refresh frequency in ms of how frequently disk usage is updated by querying SQLite
970    /// internal page stats.
971    ///
972    /// Defaults to 100ms.
973    #[serde(default = "spool_disk_usage_refresh_frequency_ms")]
974    pub disk_usage_refresh_frequency_ms: u64,
975    /// The amount of envelopes that the envelope buffer can push to its output queue.
976    ///
977    /// Defaults to 500.
978    #[serde(default = "spool_max_backpressure_envelopes")]
979    pub max_backpressure_envelopes: usize,
980    /// The relative memory usage above which the buffer service will stop dequeueing envelopes.
981    ///
982    /// Only applies when [`Self::path`] is set.
983    ///
984    /// This value should be lower than [`Health::max_memory_percent`] to prevent flip-flopping.
985    ///
986    /// Warning: This threshold can cause the buffer service to deadlock when the buffer consumes
987    /// excessive memory (as influenced by [`Self::batch_size_bytes`]).
988    ///
989    /// This scenario arises when the buffer stops spooling due to reaching the
990    /// [`Self::max_backpressure_memory_percent`] limit, but the batch threshold for spooling
991    /// ([`Self::batch_size_bytes`]) is never reached. As a result, no data is spooled, memory usage
992    /// continues to grow, and the system becomes deadlocked.
993    ///
994    /// ### Example
995    /// Suppose the system has 1GB of available memory and is configured to spool only after
996    /// accumulating 10GB worth of envelopes. If Relay consumes 900MB of memory, it will stop
997    /// unspooling due to reaching the [`Self::max_backpressure_memory_percent`] threshold.
998    ///
999    /// However, because the buffer hasn't accumulated the 10GB needed to trigger spooling,
1000    /// no data will be offloaded. Memory usage keeps increasing until it hits the
1001    /// [`Health::max_memory_percent`] threshold, e.g., at 950MB. At this point:
1002    ///
1003    /// - No more envelopes are accepted.
1004    /// - The buffer remains stuck, as unspooling won’t resume until memory drops below 900MB which
1005    ///   will not happen.
1006    /// - A deadlock occurs, with the system unable to recover without manual intervention.
1007    ///
1008    /// Defaults to 90% (5% less than max memory).
1009    #[serde(default = "spool_max_backpressure_memory_percent")]
1010    pub max_backpressure_memory_percent: f32,
1011    /// Number of partitions of the buffer.
1012    ///
1013    /// A partition is a separate instance of the buffer which has its own isolated queue, stacks
1014    /// and other resources.
1015    ///
1016    /// Defaults to 1.
1017    #[serde(default = "spool_envelopes_partitions")]
1018    pub partitions: NonZeroU8,
1019}
1020
1021impl Default for EnvelopeSpool {
1022    fn default() -> Self {
1023        Self {
1024            path: None,
1025            max_disk_size: spool_envelopes_max_disk_size(),
1026            batch_size_bytes: spool_envelopes_batch_size_bytes(),
1027            max_envelope_delay_secs: spool_envelopes_max_envelope_delay_secs(),
1028            disk_usage_refresh_frequency_ms: spool_disk_usage_refresh_frequency_ms(),
1029            max_backpressure_envelopes: spool_max_backpressure_envelopes(),
1030            max_backpressure_memory_percent: spool_max_backpressure_memory_percent(),
1031            partitions: spool_envelopes_partitions(),
1032        }
1033    }
1034}
1035
1036/// Persistent buffering configuration.
1037#[derive(Debug, Serialize, Deserialize, Default)]
1038pub struct Spool {
1039    /// Configuration for envelope spooling.
1040    #[serde(default)]
1041    pub envelopes: EnvelopeSpool,
1042}
1043
1044/// Controls internal caching behavior.
1045#[derive(Serialize, Deserialize, Debug)]
1046#[serde(default)]
1047pub struct Cache {
1048    /// The full project state will be requested by this Relay if set to `true`.
1049    pub project_request_full_config: bool,
1050    /// The cache timeout for project configurations in seconds.
1051    pub project_expiry: u32,
1052    /// Continue using project state this many seconds after cache expiry while a new state is
1053    /// being fetched. This is added on top of `project_expiry`.
1054    ///
1055    /// Default is 2 minutes.
1056    pub project_grace_period: u32,
1057    /// Refresh a project after the specified seconds.
1058    ///
1059    /// The time must be between expiry time and the grace period.
1060    ///
1061    /// By default there are no refreshes enabled.
1062    pub project_refresh_interval: Option<u32>,
1063    /// The cache timeout for downstream relay info (public keys) in seconds.
1064    pub relay_expiry: u32,
1065    /// Unused cache timeout for envelopes.
1066    ///
1067    /// The envelope buffer is instead controlled by `envelope_buffer_size`, which controls the
1068    /// maximum number of envelopes in the buffer. A time based configuration may be re-introduced
1069    /// at a later point.
1070    #[serde(alias = "event_expiry")]
1071    envelope_expiry: u32,
1072    /// The maximum amount of envelopes to queue before dropping them.
1073    #[serde(alias = "event_buffer_size")]
1074    envelope_buffer_size: u32,
1075    /// The cache timeout for non-existing entries.
1076    pub miss_expiry: u32,
1077    /// The buffer timeout for batched project config queries before sending them upstream in ms.
1078    pub batch_interval: u32,
1079    /// The buffer timeout for batched queries of downstream relays in ms. Defaults to 100ms.
1080    pub downstream_relays_batch_interval: u32,
1081    /// The maximum number of project configs to fetch from Sentry at once. Defaults to 500.
1082    ///
1083    /// `cache.batch_interval` controls how quickly batches are sent, this controls the batch size.
1084    pub batch_size: usize,
1085    /// Interval for watching local cache override files in seconds.
1086    pub file_interval: u32,
1087    /// Interval for fetching new global configs from the upstream, in seconds.
1088    pub global_config_fetch_interval: u32,
1089}
1090
1091impl Default for Cache {
1092    fn default() -> Self {
1093        Cache {
1094            project_request_full_config: false,
1095            project_expiry: 300,       // 5 minutes
1096            project_grace_period: 120, // 2 minutes
1097            project_refresh_interval: None,
1098            relay_expiry: 3600,   // 1 hour
1099            envelope_expiry: 600, // 10 minutes
1100            envelope_buffer_size: 1000,
1101            miss_expiry: 60,                       // 1 minute
1102            batch_interval: 100,                   // 100ms
1103            downstream_relays_batch_interval: 100, // 100ms
1104            batch_size: 500,
1105            file_interval: 10,                // 10 seconds
1106            global_config_fetch_interval: 10, // 10 seconds
1107        }
1108    }
1109}
1110
1111fn default_max_secs_in_future() -> u32 {
1112    60 // 1 minute
1113}
1114
1115fn default_max_session_secs_in_past() -> u32 {
1116    5 * 24 * 3600 // 5 days
1117}
1118
1119fn default_chunk_size() -> ByteSize {
1120    ByteSize::mebibytes(1)
1121}
1122
1123fn default_projectconfig_cache_prefix() -> String {
1124    "relayconfig".to_owned()
1125}
1126
1127#[allow(clippy::unnecessary_wraps)]
1128fn default_max_rate_limit() -> Option<u32> {
1129    Some(300) // 5 minutes
1130}
1131
1132/// Controls Sentry-internal event processing.
1133#[derive(Serialize, Deserialize, Debug)]
1134pub struct Processing {
1135    /// True if the Relay should do processing. Defaults to `false`.
1136    pub enabled: bool,
1137    /// GeoIp DB file source.
1138    #[serde(default)]
1139    pub geoip_path: Option<PathBuf>,
1140    /// Maximum future timestamp of ingested events.
1141    #[serde(default = "default_max_secs_in_future")]
1142    pub max_secs_in_future: u32,
1143    /// Maximum age of ingested sessions. Older sessions will be dropped.
1144    #[serde(default = "default_max_session_secs_in_past")]
1145    pub max_session_secs_in_past: u32,
1146    /// Kafka producer configurations.
1147    pub kafka_config: Vec<KafkaConfigParam>,
1148    /// Additional kafka producer configurations.
1149    ///
1150    /// The `kafka_config` is the default producer configuration used for all topics. A secondary
1151    /// kafka config can be referenced in `topics:` like this:
1152    ///
1153    /// ```yaml
1154    /// secondary_kafka_configs:
1155    ///   mycustomcluster:
1156    ///     - name: 'bootstrap.servers'
1157    ///       value: 'sentry_kafka_metrics:9093'
1158    ///
1159    /// topics:
1160    ///   transactions: ingest-transactions
1161    ///   metrics:
1162    ///     name: ingest-metrics
1163    ///     config: mycustomcluster
1164    /// ```
1165    ///
1166    /// Then metrics will be produced to an entirely different Kafka cluster.
1167    #[serde(default)]
1168    pub secondary_kafka_configs: BTreeMap<String, Vec<KafkaConfigParam>>,
1169    /// Kafka topic names.
1170    #[serde(default)]
1171    pub topics: TopicAssignments,
1172    /// Whether to validate the supplied topics by calling Kafka's metadata endpoints.
1173    #[serde(default)]
1174    pub kafka_validate_topics: bool,
1175    /// Redis hosts to connect to for storing state for rate limits.
1176    #[serde(default)]
1177    pub redis: Option<RedisConfigs>,
1178    /// Maximum chunk size of attachments for Kafka.
1179    #[serde(default = "default_chunk_size")]
1180    pub attachment_chunk_size: ByteSize,
1181    /// Prefix to use when looking up project configs in Redis. Defaults to "relayconfig".
1182    #[serde(default = "default_projectconfig_cache_prefix")]
1183    pub projectconfig_cache_prefix: String,
1184    /// Maximum rate limit to report to clients.
1185    #[serde(default = "default_max_rate_limit")]
1186    pub max_rate_limit: Option<u32>,
1187    /// Configures the quota cache ratio between `0.0` and `1.0`.
1188    ///
1189    /// The quota cache, caches the specified ratio of remaining quota in memory to reduce the
1190    /// amount of synchronizations required with Redis.
1191    ///
1192    /// The ratio is applied to the (per second) rate of the quota, not the total limit.
1193    /// For example a quota with limit 100 with a 10 second window is treated equally to a quota of
1194    /// 10 with a 1 second window.
1195    ///
1196    /// By default quota caching is disabled.
1197    pub quota_cache_ratio: Option<f32>,
1198    /// Relative amount of the total quota limit to which quota caching is applied.
1199    ///
1200    /// If exceeded, the rate limiter will no longer cache the quota and sync with Redis on every call instead.
1201    /// Lowering this value reduces the probability of incorrectly over-accepting.
1202    ///
1203    /// Must be between `0.0` and `1.0`, by default there is no limit configured.
1204    pub quota_cache_max: Option<f32>,
1205    /// Configuration for attachment uploads.
1206    #[serde(default)]
1207    pub upload: UploadServiceConfig,
1208}
1209
1210impl Default for Processing {
1211    /// Constructs a disabled processing configuration.
1212    fn default() -> Self {
1213        Self {
1214            enabled: false,
1215            geoip_path: None,
1216            max_secs_in_future: default_max_secs_in_future(),
1217            max_session_secs_in_past: default_max_session_secs_in_past(),
1218            kafka_config: Vec::new(),
1219            secondary_kafka_configs: BTreeMap::new(),
1220            topics: TopicAssignments::default(),
1221            kafka_validate_topics: false,
1222            redis: None,
1223            attachment_chunk_size: default_chunk_size(),
1224            projectconfig_cache_prefix: default_projectconfig_cache_prefix(),
1225            max_rate_limit: default_max_rate_limit(),
1226            quota_cache_ratio: None,
1227            quota_cache_max: None,
1228            upload: UploadServiceConfig::default(),
1229        }
1230    }
1231}
1232
1233/// Configuration for normalization in this Relay.
1234#[derive(Debug, Default, Serialize, Deserialize)]
1235#[serde(default)]
1236pub struct Normalization {
1237    /// Level of normalization for Relay to apply to incoming data.
1238    #[serde(default)]
1239    pub level: NormalizationLevel,
1240}
1241
1242/// Configuration for the level of normalization this Relay should do.
1243#[derive(Copy, Clone, Debug, Default, Serialize, Deserialize, Eq, PartialEq)]
1244#[serde(rename_all = "lowercase")]
1245pub enum NormalizationLevel {
1246    /// Runs normalization, excluding steps that break future compatibility.
1247    ///
1248    /// Processing Relays run [`NormalizationLevel::Full`] if this option is set.
1249    #[default]
1250    Default,
1251    /// Run full normalization.
1252    ///
1253    /// It includes steps that break future compatibility and should only run in
1254    /// the last layer of relays.
1255    Full,
1256}
1257
1258/// Configuration values for the outcome aggregator
1259#[derive(Serialize, Deserialize, Debug)]
1260#[serde(default)]
1261pub struct OutcomeAggregatorConfig {
1262    /// Defines the width of the buckets into which outcomes are aggregated, in seconds.
1263    pub bucket_interval: u64,
1264    /// Defines how often all buckets are flushed, in seconds.
1265    pub flush_interval: u64,
1266}
1267
1268impl Default for OutcomeAggregatorConfig {
1269    fn default() -> Self {
1270        Self {
1271            bucket_interval: 60,
1272            flush_interval: 120,
1273        }
1274    }
1275}
1276
1277/// Configuration values for attachment uploads.
1278#[derive(Serialize, Deserialize, Debug)]
1279#[serde(default)]
1280pub struct UploadServiceConfig {
1281    /// The base URL for the objectstore service.
1282    ///
1283    /// This defaults to [`None`], which means that the service will be disabled,
1284    /// unless a proper configuration is provided.
1285    pub objectstore_url: Option<String>,
1286
1287    /// Maximum concurrency of uploads.
1288    pub max_concurrent_requests: usize,
1289
1290    /// Maximum duration of an attachment upload in seconds. Uploads that take longer are discarded.
1291    pub timeout: u64,
1292}
1293
1294impl Default for UploadServiceConfig {
1295    fn default() -> Self {
1296        Self {
1297            objectstore_url: None,
1298            max_concurrent_requests: 10,
1299            timeout: 60,
1300        }
1301    }
1302}
1303
1304/// Determines how to emit outcomes.
1305/// For compatibility reasons, this can either be true, false or AsClientReports
1306#[derive(Copy, Clone, Debug, PartialEq, Eq)]
1307
1308pub enum EmitOutcomes {
1309    /// Do not emit any outcomes
1310    None,
1311    /// Emit outcomes as client reports
1312    AsClientReports,
1313    /// Emit outcomes as outcomes
1314    AsOutcomes,
1315}
1316
1317impl EmitOutcomes {
1318    /// Returns true of outcomes are emitted via http, kafka, or client reports.
1319    pub fn any(&self) -> bool {
1320        !matches!(self, EmitOutcomes::None)
1321    }
1322}
1323
1324impl Serialize for EmitOutcomes {
1325    fn serialize<S>(&self, serializer: S) -> Result<S::Ok, S::Error>
1326    where
1327        S: Serializer,
1328    {
1329        // For compatibility, serialize None and AsOutcomes as booleans.
1330        match self {
1331            Self::None => serializer.serialize_bool(false),
1332            Self::AsClientReports => serializer.serialize_str("as_client_reports"),
1333            Self::AsOutcomes => serializer.serialize_bool(true),
1334        }
1335    }
1336}
1337
1338struct EmitOutcomesVisitor;
1339
1340impl Visitor<'_> for EmitOutcomesVisitor {
1341    type Value = EmitOutcomes;
1342
1343    fn expecting(&self, formatter: &mut fmt::Formatter) -> fmt::Result {
1344        formatter.write_str("true, false, or 'as_client_reports'")
1345    }
1346
1347    fn visit_bool<E>(self, v: bool) -> Result<Self::Value, E>
1348    where
1349        E: serde::de::Error,
1350    {
1351        Ok(if v {
1352            EmitOutcomes::AsOutcomes
1353        } else {
1354            EmitOutcomes::None
1355        })
1356    }
1357
1358    fn visit_str<E>(self, v: &str) -> Result<Self::Value, E>
1359    where
1360        E: serde::de::Error,
1361    {
1362        if v == "as_client_reports" {
1363            Ok(EmitOutcomes::AsClientReports)
1364        } else {
1365            Err(E::invalid_value(Unexpected::Str(v), &"as_client_reports"))
1366        }
1367    }
1368}
1369
1370impl<'de> Deserialize<'de> for EmitOutcomes {
1371    fn deserialize<D>(deserializer: D) -> Result<Self, D::Error>
1372    where
1373        D: Deserializer<'de>,
1374    {
1375        deserializer.deserialize_any(EmitOutcomesVisitor)
1376    }
1377}
1378
1379/// Outcome generation specific configuration values.
1380#[derive(Serialize, Deserialize, Debug)]
1381#[serde(default)]
1382pub struct Outcomes {
1383    /// Controls whether outcomes will be emitted when processing is disabled.
1384    /// Processing relays always emit outcomes (for backwards compatibility).
1385    /// Can take the following values: false, "as_client_reports", true
1386    pub emit_outcomes: EmitOutcomes,
1387    /// Controls wheather client reported outcomes should be emitted.
1388    pub emit_client_outcomes: bool,
1389    /// The maximum number of outcomes that are batched before being sent
1390    /// via http to the upstream (only applies to non processing relays).
1391    pub batch_size: usize,
1392    /// The maximum time interval (in milliseconds) that an outcome may be batched
1393    /// via http to the upstream (only applies to non processing relays).
1394    pub batch_interval: u64,
1395    /// Defines the source string registered in the outcomes originating from
1396    /// this Relay (typically something like the region or the layer).
1397    pub source: Option<String>,
1398    /// Configures the outcome aggregator.
1399    pub aggregator: OutcomeAggregatorConfig,
1400}
1401
1402impl Default for Outcomes {
1403    fn default() -> Self {
1404        Outcomes {
1405            emit_outcomes: EmitOutcomes::AsClientReports,
1406            emit_client_outcomes: true,
1407            batch_size: 1000,
1408            batch_interval: 500,
1409            source: None,
1410            aggregator: OutcomeAggregatorConfig::default(),
1411        }
1412    }
1413}
1414
1415/// Minimal version of a config for dumping out.
1416#[derive(Serialize, Deserialize, Debug, Default)]
1417pub struct MinimalConfig {
1418    /// The relay part of the config.
1419    pub relay: Relay,
1420}
1421
1422impl MinimalConfig {
1423    /// Saves the config in the given config folder as config.yml
1424    pub fn save_in_folder<P: AsRef<Path>>(&self, p: P) -> anyhow::Result<()> {
1425        let path = p.as_ref();
1426        if fs::metadata(path).is_err() {
1427            fs::create_dir_all(path)
1428                .with_context(|| ConfigError::file(ConfigErrorKind::CouldNotOpenFile, path))?;
1429        }
1430        self.save(path)
1431    }
1432}
1433
1434impl ConfigObject for MinimalConfig {
1435    fn format() -> ConfigFormat {
1436        ConfigFormat::Yaml
1437    }
1438
1439    fn name() -> &'static str {
1440        "config"
1441    }
1442}
1443
1444/// Alternative serialization of RelayInfo for config file using snake case.
1445mod config_relay_info {
1446    use serde::ser::SerializeMap;
1447
1448    use super::*;
1449
1450    // Uses snake_case as opposed to camelCase.
1451    #[derive(Debug, Serialize, Deserialize, Clone)]
1452    struct RelayInfoConfig {
1453        public_key: PublicKey,
1454        #[serde(default)]
1455        internal: bool,
1456    }
1457
1458    impl From<RelayInfoConfig> for RelayInfo {
1459        fn from(v: RelayInfoConfig) -> Self {
1460            RelayInfo {
1461                public_key: v.public_key,
1462                internal: v.internal,
1463            }
1464        }
1465    }
1466
1467    impl From<RelayInfo> for RelayInfoConfig {
1468        fn from(v: RelayInfo) -> Self {
1469            RelayInfoConfig {
1470                public_key: v.public_key,
1471                internal: v.internal,
1472            }
1473        }
1474    }
1475
1476    pub(super) fn deserialize<'de, D>(des: D) -> Result<HashMap<RelayId, RelayInfo>, D::Error>
1477    where
1478        D: Deserializer<'de>,
1479    {
1480        let map = HashMap::<RelayId, RelayInfoConfig>::deserialize(des)?;
1481        Ok(map.into_iter().map(|(k, v)| (k, v.into())).collect())
1482    }
1483
1484    pub(super) fn serialize<S>(elm: &HashMap<RelayId, RelayInfo>, ser: S) -> Result<S::Ok, S::Error>
1485    where
1486        S: Serializer,
1487    {
1488        let mut map = ser.serialize_map(Some(elm.len()))?;
1489
1490        for (k, v) in elm {
1491            map.serialize_entry(k, &RelayInfoConfig::from(v.clone()))?;
1492        }
1493
1494        map.end()
1495    }
1496}
1497
1498/// Authentication options.
1499#[derive(Serialize, Deserialize, Debug, Default)]
1500pub struct AuthConfig {
1501    /// Controls responses from the readiness health check endpoint based on authentication.
1502    #[serde(default, skip_serializing_if = "is_default")]
1503    pub ready: ReadinessCondition,
1504
1505    /// Statically authenticated downstream relays.
1506    #[serde(default, with = "config_relay_info")]
1507    pub static_relays: HashMap<RelayId, RelayInfo>,
1508
1509    /// How old a signature can be before it is considered invalid, in seconds.
1510    ///
1511    /// Defaults to 5 minutes.
1512    #[serde(default = "default_max_age")]
1513    pub signature_max_age: u64,
1514}
1515
1516fn default_max_age() -> u64 {
1517    300
1518}
1519
1520/// GeoIp database configuration options.
1521#[derive(Serialize, Deserialize, Debug, Default)]
1522pub struct GeoIpConfig {
1523    /// The path to GeoIP database.
1524    pub path: Option<PathBuf>,
1525}
1526
1527/// Cardinality Limiter configuration options.
1528#[derive(Serialize, Deserialize, Debug)]
1529#[serde(default)]
1530pub struct CardinalityLimiter {
1531    /// Cache vacuum interval in seconds for the in memory cache.
1532    ///
1533    /// The cache will scan for expired values based on this interval.
1534    ///
1535    /// Defaults to 180 seconds, 3 minutes.
1536    pub cache_vacuum_interval: u64,
1537}
1538
1539impl Default for CardinalityLimiter {
1540    fn default() -> Self {
1541        Self {
1542            cache_vacuum_interval: 180,
1543        }
1544    }
1545}
1546
1547/// Settings to control Relay's health checks.
1548///
1549/// After breaching one of the configured thresholds, Relay will
1550/// return an `unhealthy` status from its health endpoint.
1551#[derive(Serialize, Deserialize, Debug)]
1552#[serde(default)]
1553pub struct Health {
1554    /// Interval to refresh internal health checks.
1555    ///
1556    /// Shorter intervals will decrease the time it takes the health check endpoint to report
1557    /// issues, but can also increase sporadic unhealthy responses.
1558    ///
1559    /// Defaults to `3000`` (3 seconds).
1560    pub refresh_interval_ms: u64,
1561    /// Maximum memory watermark in bytes.
1562    ///
1563    /// By default, there is no absolute limit set and the watermark
1564    /// is only controlled by setting [`Self::max_memory_percent`].
1565    pub max_memory_bytes: Option<ByteSize>,
1566    /// Maximum memory watermark as a percentage of maximum system memory.
1567    ///
1568    /// Defaults to `0.95` (95%).
1569    pub max_memory_percent: f32,
1570    /// Health check probe timeout in milliseconds.
1571    ///
1572    /// Any probe exceeding the timeout will be considered failed.
1573    /// This limits the max execution time of Relay health checks.
1574    ///
1575    /// Defaults to 900 milliseconds.
1576    pub probe_timeout_ms: u64,
1577    /// The refresh frequency of memory stats which are used to poll memory
1578    /// usage of Relay.
1579    ///
1580    /// The implementation of memory stats guarantees that the refresh will happen at
1581    /// least every `x` ms since memory readings are lazy and are updated only if needed.
1582    pub memory_stat_refresh_frequency_ms: u64,
1583}
1584
1585impl Default for Health {
1586    fn default() -> Self {
1587        Self {
1588            refresh_interval_ms: 3000,
1589            max_memory_bytes: None,
1590            max_memory_percent: 0.95,
1591            probe_timeout_ms: 900,
1592            memory_stat_refresh_frequency_ms: 100,
1593        }
1594    }
1595}
1596
1597/// COGS configuration.
1598#[derive(Serialize, Deserialize, Debug)]
1599#[serde(default)]
1600pub struct Cogs {
1601    /// Maximium amount of COGS measurements allowed to backlog.
1602    ///
1603    /// Any additional COGS measurements recorded will be dropped.
1604    ///
1605    /// Defaults to `10_000`.
1606    pub max_queue_size: u64,
1607    /// Relay COGS resource id.
1608    ///
1609    /// All Relay related COGS measurements are emitted with this resource id.
1610    ///
1611    /// Defaults to `relay_service`.
1612    pub relay_resource_id: String,
1613}
1614
1615impl Default for Cogs {
1616    fn default() -> Self {
1617        Self {
1618            max_queue_size: 10_000,
1619            relay_resource_id: "relay_service".to_owned(),
1620        }
1621    }
1622}
1623
1624#[derive(Serialize, Deserialize, Debug, Default)]
1625struct ConfigValues {
1626    #[serde(default)]
1627    relay: Relay,
1628    #[serde(default)]
1629    http: Http,
1630    #[serde(default)]
1631    cache: Cache,
1632    #[serde(default)]
1633    spool: Spool,
1634    #[serde(default)]
1635    limits: Limits,
1636    #[serde(default)]
1637    logging: relay_log::LogConfig,
1638    #[serde(default)]
1639    routing: Routing,
1640    #[serde(default)]
1641    metrics: Metrics,
1642    #[serde(default)]
1643    sentry: relay_log::SentryConfig,
1644    #[serde(default)]
1645    processing: Processing,
1646    #[serde(default)]
1647    outcomes: Outcomes,
1648    #[serde(default)]
1649    aggregator: AggregatorServiceConfig,
1650    #[serde(default)]
1651    secondary_aggregators: Vec<ScopedAggregatorConfig>,
1652    #[serde(default)]
1653    auth: AuthConfig,
1654    #[serde(default)]
1655    geoip: GeoIpConfig,
1656    #[serde(default)]
1657    normalization: Normalization,
1658    #[serde(default)]
1659    cardinality_limiter: CardinalityLimiter,
1660    #[serde(default)]
1661    health: Health,
1662    #[serde(default)]
1663    cogs: Cogs,
1664}
1665
1666impl ConfigObject for ConfigValues {
1667    fn format() -> ConfigFormat {
1668        ConfigFormat::Yaml
1669    }
1670
1671    fn name() -> &'static str {
1672        "config"
1673    }
1674}
1675
1676/// Config struct.
1677pub struct Config {
1678    values: ConfigValues,
1679    credentials: Option<Credentials>,
1680    path: PathBuf,
1681}
1682
1683impl fmt::Debug for Config {
1684    fn fmt(&self, f: &mut fmt::Formatter<'_>) -> fmt::Result {
1685        f.debug_struct("Config")
1686            .field("path", &self.path)
1687            .field("values", &self.values)
1688            .finish()
1689    }
1690}
1691
1692impl Config {
1693    /// Loads a config from a given config folder.
1694    pub fn from_path<P: AsRef<Path>>(path: P) -> anyhow::Result<Config> {
1695        let path = env::current_dir()
1696            .map(|x| x.join(path.as_ref()))
1697            .unwrap_or_else(|_| path.as_ref().to_path_buf());
1698
1699        let config = Config {
1700            values: ConfigValues::load(&path)?,
1701            credentials: if Credentials::path(&path).exists() {
1702                Some(Credentials::load(&path)?)
1703            } else {
1704                None
1705            },
1706            path: path.clone(),
1707        };
1708
1709        if cfg!(not(feature = "processing")) && config.processing_enabled() {
1710            return Err(ConfigError::file(ConfigErrorKind::ProcessingNotAvailable, &path).into());
1711        }
1712
1713        Ok(config)
1714    }
1715
1716    /// Creates a config from a JSON value.
1717    ///
1718    /// This is mostly useful for tests.
1719    pub fn from_json_value(value: serde_json::Value) -> anyhow::Result<Config> {
1720        Ok(Config {
1721            values: serde_json::from_value(value)
1722                .with_context(|| ConfigError::new(ConfigErrorKind::BadJson))?,
1723            credentials: None,
1724            path: PathBuf::new(),
1725        })
1726    }
1727
1728    /// Override configuration with values coming from other sources (e.g. env variables or
1729    /// command line parameters)
1730    pub fn apply_override(
1731        &mut self,
1732        mut overrides: OverridableConfig,
1733    ) -> anyhow::Result<&mut Self> {
1734        let relay = &mut self.values.relay;
1735
1736        if let Some(mode) = overrides.mode {
1737            relay.mode = mode
1738                .parse::<RelayMode>()
1739                .with_context(|| ConfigError::field("mode"))?;
1740        }
1741
1742        if let Some(deployment) = overrides.instance {
1743            relay.instance = deployment
1744                .parse::<RelayInstance>()
1745                .with_context(|| ConfigError::field("deployment"))?;
1746        }
1747
1748        if let Some(log_level) = overrides.log_level {
1749            self.values.logging.level = log_level.parse()?;
1750        }
1751
1752        if let Some(log_format) = overrides.log_format {
1753            self.values.logging.format = log_format.parse()?;
1754        }
1755
1756        if let Some(upstream) = overrides.upstream {
1757            relay.upstream = upstream
1758                .parse::<UpstreamDescriptor>()
1759                .with_context(|| ConfigError::field("upstream"))?;
1760        } else if let Some(upstream_dsn) = overrides.upstream_dsn {
1761            relay.upstream = upstream_dsn
1762                .parse::<Dsn>()
1763                .map(|dsn| UpstreamDescriptor::from_dsn(&dsn).into_owned())
1764                .with_context(|| ConfigError::field("upstream_dsn"))?;
1765        }
1766
1767        if let Some(host) = overrides.host {
1768            relay.host = host
1769                .parse::<IpAddr>()
1770                .with_context(|| ConfigError::field("host"))?;
1771        }
1772
1773        if let Some(port) = overrides.port {
1774            relay.port = port
1775                .as_str()
1776                .parse()
1777                .with_context(|| ConfigError::field("port"))?;
1778        }
1779
1780        let processing = &mut self.values.processing;
1781        if let Some(enabled) = overrides.processing {
1782            match enabled.to_lowercase().as_str() {
1783                "true" | "1" => processing.enabled = true,
1784                "false" | "0" | "" => processing.enabled = false,
1785                _ => return Err(ConfigError::field("processing").into()),
1786            }
1787        }
1788
1789        if let Some(redis) = overrides.redis_url {
1790            processing.redis = Some(RedisConfigs::Unified(RedisConfig::single(redis)))
1791        }
1792
1793        if let Some(kafka_url) = overrides.kafka_url {
1794            let existing = processing
1795                .kafka_config
1796                .iter_mut()
1797                .find(|e| e.name == "bootstrap.servers");
1798
1799            if let Some(config_param) = existing {
1800                config_param.value = kafka_url;
1801            } else {
1802                processing.kafka_config.push(KafkaConfigParam {
1803                    name: "bootstrap.servers".to_owned(),
1804                    value: kafka_url,
1805                })
1806            }
1807        }
1808        // credentials overrides
1809        let id = if let Some(id) = overrides.id {
1810            let id = Uuid::parse_str(&id).with_context(|| ConfigError::field("id"))?;
1811            Some(id)
1812        } else {
1813            None
1814        };
1815        let public_key = if let Some(public_key) = overrides.public_key {
1816            let public_key = public_key
1817                .parse::<PublicKey>()
1818                .with_context(|| ConfigError::field("public_key"))?;
1819            Some(public_key)
1820        } else {
1821            None
1822        };
1823
1824        let secret_key = if let Some(secret_key) = overrides.secret_key {
1825            let secret_key = secret_key
1826                .parse::<SecretKey>()
1827                .with_context(|| ConfigError::field("secret_key"))?;
1828            Some(secret_key)
1829        } else {
1830            None
1831        };
1832        let outcomes = &mut self.values.outcomes;
1833        if overrides.outcome_source.is_some() {
1834            outcomes.source = overrides.outcome_source.take();
1835        }
1836
1837        if let Some(credentials) = &mut self.credentials {
1838            //we have existing credentials we may override some entries
1839            if let Some(id) = id {
1840                credentials.id = id;
1841            }
1842            if let Some(public_key) = public_key {
1843                credentials.public_key = public_key;
1844            }
1845            if let Some(secret_key) = secret_key {
1846                credentials.secret_key = secret_key
1847            }
1848        } else {
1849            //no existing credentials we may only create the full credentials
1850            match (id, public_key, secret_key) {
1851                (Some(id), Some(public_key), Some(secret_key)) => {
1852                    self.credentials = Some(Credentials {
1853                        secret_key,
1854                        public_key,
1855                        id,
1856                    })
1857                }
1858                (None, None, None) => {
1859                    // nothing provided, we'll just leave the credentials None, maybe we
1860                    // don't need them in the current command or we'll override them later
1861                }
1862                _ => {
1863                    return Err(ConfigError::field("incomplete credentials").into());
1864                }
1865            }
1866        }
1867
1868        let limits = &mut self.values.limits;
1869        if let Some(shutdown_timeout) = overrides.shutdown_timeout
1870            && let Ok(shutdown_timeout) = shutdown_timeout.parse::<u64>()
1871        {
1872            limits.shutdown_timeout = shutdown_timeout;
1873        }
1874
1875        if let Some(server_name) = overrides.server_name {
1876            self.values.sentry.server_name = Some(server_name.into());
1877        }
1878
1879        Ok(self)
1880    }
1881
1882    /// Checks if the config is already initialized.
1883    pub fn config_exists<P: AsRef<Path>>(path: P) -> bool {
1884        fs::metadata(ConfigValues::path(path.as_ref())).is_ok()
1885    }
1886
1887    /// Returns the filename of the config file.
1888    pub fn path(&self) -> &Path {
1889        &self.path
1890    }
1891
1892    /// Dumps out a YAML string of the values.
1893    pub fn to_yaml_string(&self) -> anyhow::Result<String> {
1894        serde_yaml::to_string(&self.values)
1895            .with_context(|| ConfigError::new(ConfigErrorKind::CouldNotWriteFile))
1896    }
1897
1898    /// Regenerates the relay credentials.
1899    ///
1900    /// This also writes the credentials back to the file.
1901    pub fn regenerate_credentials(&mut self, save: bool) -> anyhow::Result<()> {
1902        let creds = Credentials::generate();
1903        if save {
1904            creds.save(&self.path)?;
1905        }
1906        self.credentials = Some(creds);
1907        Ok(())
1908    }
1909
1910    /// Return the current credentials
1911    pub fn credentials(&self) -> Option<&Credentials> {
1912        self.credentials.as_ref()
1913    }
1914
1915    /// Set new credentials.
1916    ///
1917    /// This also writes the credentials back to the file.
1918    pub fn replace_credentials(
1919        &mut self,
1920        credentials: Option<Credentials>,
1921    ) -> anyhow::Result<bool> {
1922        if self.credentials == credentials {
1923            return Ok(false);
1924        }
1925
1926        match credentials {
1927            Some(ref creds) => {
1928                creds.save(&self.path)?;
1929            }
1930            None => {
1931                let path = Credentials::path(&self.path);
1932                if fs::metadata(&path).is_ok() {
1933                    fs::remove_file(&path).with_context(|| {
1934                        ConfigError::file(ConfigErrorKind::CouldNotWriteFile, &path)
1935                    })?;
1936                }
1937            }
1938        }
1939
1940        self.credentials = credentials;
1941        Ok(true)
1942    }
1943
1944    /// Returns `true` if the config is ready to use.
1945    pub fn has_credentials(&self) -> bool {
1946        self.credentials.is_some()
1947    }
1948
1949    /// Returns the secret key if set.
1950    pub fn secret_key(&self) -> Option<&SecretKey> {
1951        self.credentials.as_ref().map(|x| &x.secret_key)
1952    }
1953
1954    /// Returns the public key if set.
1955    pub fn public_key(&self) -> Option<&PublicKey> {
1956        self.credentials.as_ref().map(|x| &x.public_key)
1957    }
1958
1959    /// Returns the relay ID.
1960    pub fn relay_id(&self) -> Option<&RelayId> {
1961        self.credentials.as_ref().map(|x| &x.id)
1962    }
1963
1964    /// Returns the relay mode.
1965    pub fn relay_mode(&self) -> RelayMode {
1966        self.values.relay.mode
1967    }
1968
1969    /// Returns the instance type of relay.
1970    pub fn relay_instance(&self) -> RelayInstance {
1971        self.values.relay.instance
1972    }
1973
1974    /// Returns the upstream target as descriptor.
1975    pub fn upstream_descriptor(&self) -> &UpstreamDescriptor<'_> {
1976        &self.values.relay.upstream
1977    }
1978
1979    /// Returns the custom HTTP "Host" header.
1980    pub fn http_host_header(&self) -> Option<&str> {
1981        self.values.http.host_header.as_deref()
1982    }
1983
1984    /// Returns the listen address.
1985    pub fn listen_addr(&self) -> SocketAddr {
1986        (self.values.relay.host, self.values.relay.port).into()
1987    }
1988
1989    /// Returns the listen address for internal APIs.
1990    ///
1991    /// Internal APIs are APIs which do not need to be publicly exposed,
1992    /// like health checks.
1993    ///
1994    /// Returns `None` when there is no explicit address configured for internal APIs,
1995    /// and they should instead be exposed on the main [`Self::listen_addr`].
1996    pub fn listen_addr_internal(&self) -> Option<SocketAddr> {
1997        match (
1998            self.values.relay.internal_host,
1999            self.values.relay.internal_port,
2000        ) {
2001            (Some(host), None) => Some((host, self.values.relay.port).into()),
2002            (None, Some(port)) => Some((self.values.relay.host, port).into()),
2003            (Some(host), Some(port)) => Some((host, port).into()),
2004            (None, None) => None,
2005        }
2006    }
2007
2008    /// Returns the TLS listen address.
2009    pub fn tls_listen_addr(&self) -> Option<SocketAddr> {
2010        if self.values.relay.tls_identity_path.is_some() {
2011            let port = self.values.relay.tls_port.unwrap_or(3443);
2012            Some((self.values.relay.host, port).into())
2013        } else {
2014            None
2015        }
2016    }
2017
2018    /// Returns the path to the identity bundle
2019    pub fn tls_identity_path(&self) -> Option<&Path> {
2020        self.values.relay.tls_identity_path.as_deref()
2021    }
2022
2023    /// Returns the password for the identity bundle
2024    pub fn tls_identity_password(&self) -> Option<&str> {
2025        self.values.relay.tls_identity_password.as_deref()
2026    }
2027
2028    /// Returns `true` when project IDs should be overriden rather than validated.
2029    ///
2030    /// Defaults to `false`, which requires project ID validation.
2031    pub fn override_project_ids(&self) -> bool {
2032        self.values.relay.override_project_ids
2033    }
2034
2035    /// Returns `true` if Relay requires authentication for readiness.
2036    ///
2037    /// See [`ReadinessCondition`] for more information.
2038    pub fn requires_auth(&self) -> bool {
2039        match self.values.auth.ready {
2040            ReadinessCondition::Authenticated => self.relay_mode() == RelayMode::Managed,
2041            ReadinessCondition::Always => false,
2042        }
2043    }
2044
2045    /// Returns the interval at which Realy should try to re-authenticate with the upstream.
2046    ///
2047    /// Always disabled in processing mode.
2048    pub fn http_auth_interval(&self) -> Option<Duration> {
2049        if self.processing_enabled() {
2050            return None;
2051        }
2052
2053        match self.values.http.auth_interval {
2054            None | Some(0) => None,
2055            Some(secs) => Some(Duration::from_secs(secs)),
2056        }
2057    }
2058
2059    /// The maximum time of experiencing uninterrupted network failures until Relay considers that
2060    /// it has encountered a network outage.
2061    pub fn http_outage_grace_period(&self) -> Duration {
2062        Duration::from_secs(self.values.http.outage_grace_period)
2063    }
2064
2065    /// Time Relay waits before retrying an upstream request.
2066    ///
2067    /// Before going into a network outage, Relay may fail to make upstream
2068    /// requests. This is the time Relay waits before retrying the same request.
2069    pub fn http_retry_delay(&self) -> Duration {
2070        Duration::from_secs(self.values.http.retry_delay)
2071    }
2072
2073    /// Time of continued project request failures before Relay emits an error.
2074    pub fn http_project_failure_interval(&self) -> Duration {
2075        Duration::from_secs(self.values.http.project_failure_interval)
2076    }
2077
2078    /// Content encoding of upstream requests.
2079    pub fn http_encoding(&self) -> HttpEncoding {
2080        self.values.http.encoding
2081    }
2082
2083    /// Returns whether metrics should be sent globally through a shared endpoint.
2084    pub fn http_global_metrics(&self) -> bool {
2085        self.values.http.global_metrics
2086    }
2087
2088    /// Returns whether this Relay should emit outcomes.
2089    ///
2090    /// This is `true` either if `outcomes.emit_outcomes` is explicitly enabled, or if this Relay is
2091    /// in processing mode.
2092    pub fn emit_outcomes(&self) -> EmitOutcomes {
2093        if self.processing_enabled() {
2094            return EmitOutcomes::AsOutcomes;
2095        }
2096        self.values.outcomes.emit_outcomes
2097    }
2098
2099    /// Returns whether this Relay should emit client outcomes
2100    ///
2101    /// Relays that do not emit client outcomes will forward client recieved outcomes
2102    /// directly to the next relay in the chain as client report envelope.  This is only done
2103    /// if this relay emits outcomes at all. A relay that will not emit outcomes
2104    /// will forward the envelope unchanged.
2105    ///
2106    /// This flag can be explicitly disabled on processing relays as well to prevent the
2107    /// emitting of client outcomes to the kafka topic.
2108    pub fn emit_client_outcomes(&self) -> bool {
2109        self.values.outcomes.emit_client_outcomes
2110    }
2111
2112    /// Returns the maximum number of outcomes that are batched before being sent
2113    pub fn outcome_batch_size(&self) -> usize {
2114        self.values.outcomes.batch_size
2115    }
2116
2117    /// Returns the maximum interval that an outcome may be batched
2118    pub fn outcome_batch_interval(&self) -> Duration {
2119        Duration::from_millis(self.values.outcomes.batch_interval)
2120    }
2121
2122    /// The originating source of the outcome
2123    pub fn outcome_source(&self) -> Option<&str> {
2124        self.values.outcomes.source.as_deref()
2125    }
2126
2127    /// Returns the width of the buckets into which outcomes are aggregated, in seconds.
2128    pub fn outcome_aggregator(&self) -> &OutcomeAggregatorConfig {
2129        &self.values.outcomes.aggregator
2130    }
2131
2132    /// Returns logging configuration.
2133    pub fn logging(&self) -> &relay_log::LogConfig {
2134        &self.values.logging
2135    }
2136
2137    /// Returns logging configuration.
2138    pub fn sentry(&self) -> &relay_log::SentryConfig {
2139        &self.values.sentry
2140    }
2141
2142    /// Returns the socket addresses for statsd.
2143    ///
2144    /// If stats is disabled an empty vector is returned.
2145    pub fn statsd_addrs(&self) -> anyhow::Result<Vec<SocketAddr>> {
2146        if let Some(ref addr) = self.values.metrics.statsd {
2147            let addrs = addr
2148                .as_str()
2149                .to_socket_addrs()
2150                .with_context(|| ConfigError::file(ConfigErrorKind::InvalidValue, &self.path))?
2151                .collect();
2152            Ok(addrs)
2153        } else {
2154            Ok(vec![])
2155        }
2156    }
2157
2158    /// Return the prefix for statsd metrics.
2159    pub fn metrics_prefix(&self) -> &str {
2160        &self.values.metrics.prefix
2161    }
2162
2163    /// Returns the default tags for statsd metrics.
2164    pub fn metrics_default_tags(&self) -> &BTreeMap<String, String> {
2165        &self.values.metrics.default_tags
2166    }
2167
2168    /// Returns the name of the hostname tag that should be attached to each outgoing metric.
2169    pub fn metrics_hostname_tag(&self) -> Option<&str> {
2170        self.values.metrics.hostname_tag.as_deref()
2171    }
2172
2173    /// Returns the global sample rate for all metrics.
2174    pub fn metrics_sample_rate(&self) -> f32 {
2175        self.values.metrics.sample_rate
2176    }
2177
2178    /// Returns whether local metric aggregation should be enabled.
2179    pub fn metrics_aggregate(&self) -> bool {
2180        self.values.metrics.aggregate
2181    }
2182
2183    /// Returns whether high cardinality tags should be removed before sending metrics.
2184    pub fn metrics_allow_high_cardinality_tags(&self) -> bool {
2185        self.values.metrics.allow_high_cardinality_tags
2186    }
2187
2188    /// Returns the interval for periodic metrics emitted from Relay.
2189    ///
2190    /// `None` if periodic metrics are disabled.
2191    pub fn metrics_periodic_interval(&self) -> Option<Duration> {
2192        match self.values.metrics.periodic_secs {
2193            0 => None,
2194            secs => Some(Duration::from_secs(secs)),
2195        }
2196    }
2197
2198    /// Returns the default timeout for all upstream HTTP requests.
2199    pub fn http_timeout(&self) -> Duration {
2200        Duration::from_secs(self.values.http.timeout.into())
2201    }
2202
2203    /// Returns the connection timeout for all upstream HTTP requests.
2204    pub fn http_connection_timeout(&self) -> Duration {
2205        Duration::from_secs(self.values.http.connection_timeout.into())
2206    }
2207
2208    /// Returns the failed upstream request retry interval.
2209    pub fn http_max_retry_interval(&self) -> Duration {
2210        Duration::from_secs(self.values.http.max_retry_interval.into())
2211    }
2212
2213    /// Returns the expiry timeout for cached projects.
2214    pub fn project_cache_expiry(&self) -> Duration {
2215        Duration::from_secs(self.values.cache.project_expiry.into())
2216    }
2217
2218    /// Returns `true` if the full project state should be requested from upstream.
2219    pub fn request_full_project_config(&self) -> bool {
2220        self.values.cache.project_request_full_config
2221    }
2222
2223    /// Returns the expiry timeout for cached relay infos (public keys).
2224    pub fn relay_cache_expiry(&self) -> Duration {
2225        Duration::from_secs(self.values.cache.relay_expiry.into())
2226    }
2227
2228    /// Returns the maximum number of buffered envelopes
2229    pub fn envelope_buffer_size(&self) -> usize {
2230        self.values
2231            .cache
2232            .envelope_buffer_size
2233            .try_into()
2234            .unwrap_or(usize::MAX)
2235    }
2236
2237    /// Returns the expiry timeout for cached misses before trying to refetch.
2238    pub fn cache_miss_expiry(&self) -> Duration {
2239        Duration::from_secs(self.values.cache.miss_expiry.into())
2240    }
2241
2242    /// Returns the grace period for project caches.
2243    pub fn project_grace_period(&self) -> Duration {
2244        Duration::from_secs(self.values.cache.project_grace_period.into())
2245    }
2246
2247    /// Returns the refresh interval for a project.
2248    ///
2249    /// Validates the refresh time to be between the grace period and expiry.
2250    pub fn project_refresh_interval(&self) -> Option<Duration> {
2251        self.values
2252            .cache
2253            .project_refresh_interval
2254            .map(Into::into)
2255            .map(Duration::from_secs)
2256    }
2257
2258    /// Returns the duration in which batchable project config queries are
2259    /// collected before sending them in a single request.
2260    pub fn query_batch_interval(&self) -> Duration {
2261        Duration::from_millis(self.values.cache.batch_interval.into())
2262    }
2263
2264    /// Returns the duration in which downstream relays are requested from upstream.
2265    pub fn downstream_relays_batch_interval(&self) -> Duration {
2266        Duration::from_millis(self.values.cache.downstream_relays_batch_interval.into())
2267    }
2268
2269    /// Returns the interval in seconds in which local project configurations should be reloaded.
2270    pub fn local_cache_interval(&self) -> Duration {
2271        Duration::from_secs(self.values.cache.file_interval.into())
2272    }
2273
2274    /// Returns the interval in seconds in which fresh global configs should be
2275    /// fetched from  upstream.
2276    pub fn global_config_fetch_interval(&self) -> Duration {
2277        Duration::from_secs(self.values.cache.global_config_fetch_interval.into())
2278    }
2279
2280    /// Returns the path of the buffer file if the `cache.persistent_envelope_buffer.path` is configured.
2281    ///
2282    /// In case a partition with id > 0 is supplied, the filename of the envelopes path will be
2283    /// suffixed with `.{partition_id}`.
2284    pub fn spool_envelopes_path(&self, partition_id: u8) -> Option<PathBuf> {
2285        let mut path = self
2286            .values
2287            .spool
2288            .envelopes
2289            .path
2290            .as_ref()
2291            .map(|path| path.to_owned())?;
2292
2293        if partition_id == 0 {
2294            return Some(path);
2295        }
2296
2297        let file_name = path.file_name().and_then(|f| f.to_str())?;
2298        let new_file_name = format!("{file_name}.{partition_id}");
2299        path.set_file_name(new_file_name);
2300
2301        Some(path)
2302    }
2303
2304    /// The maximum size of the buffer, in bytes.
2305    pub fn spool_envelopes_max_disk_size(&self) -> usize {
2306        self.values.spool.envelopes.max_disk_size.as_bytes()
2307    }
2308
2309    /// Number of encoded envelope bytes that need to be accumulated before
2310    /// flushing one batch to disk.
2311    pub fn spool_envelopes_batch_size_bytes(&self) -> usize {
2312        self.values.spool.envelopes.batch_size_bytes.as_bytes()
2313    }
2314
2315    /// Returns the time after which we drop envelopes as a [`Duration`] object.
2316    pub fn spool_envelopes_max_age(&self) -> Duration {
2317        Duration::from_secs(self.values.spool.envelopes.max_envelope_delay_secs)
2318    }
2319
2320    /// Returns the refresh frequency for disk usage monitoring as a [`Duration`] object.
2321    pub fn spool_disk_usage_refresh_frequency_ms(&self) -> Duration {
2322        Duration::from_millis(self.values.spool.envelopes.disk_usage_refresh_frequency_ms)
2323    }
2324
2325    /// Returns the maximum number of envelopes that can be put in the bounded buffer.
2326    pub fn spool_max_backpressure_envelopes(&self) -> usize {
2327        self.values.spool.envelopes.max_backpressure_envelopes
2328    }
2329
2330    /// Returns the relative memory usage up to which the disk buffer will unspool envelopes.
2331    pub fn spool_max_backpressure_memory_percent(&self) -> f32 {
2332        self.values.spool.envelopes.max_backpressure_memory_percent
2333    }
2334
2335    /// Returns the number of partitions for the buffer.
2336    pub fn spool_partitions(&self) -> NonZeroU8 {
2337        self.values.spool.envelopes.partitions
2338    }
2339
2340    /// Returns the maximum size of an event payload in bytes.
2341    pub fn max_event_size(&self) -> usize {
2342        self.values.limits.max_event_size.as_bytes()
2343    }
2344
2345    /// Returns the maximum size of each attachment.
2346    pub fn max_attachment_size(&self) -> usize {
2347        self.values.limits.max_attachment_size.as_bytes()
2348    }
2349
2350    /// Returns the maximum combined size of attachments or payloads containing attachments
2351    /// (minidump, unreal, standalone attachments) in bytes.
2352    pub fn max_attachments_size(&self) -> usize {
2353        self.values.limits.max_attachments_size.as_bytes()
2354    }
2355
2356    /// Returns the maximum combined size of client reports in bytes.
2357    pub fn max_client_reports_size(&self) -> usize {
2358        self.values.limits.max_client_reports_size.as_bytes()
2359    }
2360
2361    /// Returns the maximum payload size of a monitor check-in in bytes.
2362    pub fn max_check_in_size(&self) -> usize {
2363        self.values.limits.max_check_in_size.as_bytes()
2364    }
2365
2366    /// Returns the maximum payload size of a log in bytes.
2367    pub fn max_log_size(&self) -> usize {
2368        self.values.limits.max_log_size.as_bytes()
2369    }
2370
2371    /// Returns the maximum payload size of a span in bytes.
2372    pub fn max_span_size(&self) -> usize {
2373        self.values.limits.max_span_size.as_bytes()
2374    }
2375
2376    /// Returns the maximum payload size of an item container in bytes.
2377    pub fn max_container_size(&self) -> usize {
2378        self.values.limits.max_container_size.as_bytes()
2379    }
2380
2381    /// Returns the maximum payload size for logs integration items in bytes.
2382    pub fn max_logs_integration_size(&self) -> usize {
2383        // Not explicitly configured, inherited from the maximum size of a log container.
2384        self.max_container_size()
2385    }
2386
2387    /// Returns the maximum payload size for spans integration items in bytes.
2388    pub fn max_spans_integration_size(&self) -> usize {
2389        // Not explicitly configured, inherited from the maximum size of a span container.
2390        self.max_container_size()
2391    }
2392
2393    /// Returns the maximum size of an envelope payload in bytes.
2394    ///
2395    /// Individual item size limits still apply.
2396    pub fn max_envelope_size(&self) -> usize {
2397        self.values.limits.max_envelope_size.as_bytes()
2398    }
2399
2400    /// Returns the maximum number of sessions per envelope.
2401    pub fn max_session_count(&self) -> usize {
2402        self.values.limits.max_session_count
2403    }
2404
2405    /// Returns the maximum payload size of a statsd metric in bytes.
2406    pub fn max_statsd_size(&self) -> usize {
2407        self.values.limits.max_statsd_size.as_bytes()
2408    }
2409
2410    /// Returns the maximum payload size of metric buckets in bytes.
2411    pub fn max_metric_buckets_size(&self) -> usize {
2412        self.values.limits.max_metric_buckets_size.as_bytes()
2413    }
2414
2415    /// Returns the maximum payload size for general API requests.
2416    pub fn max_api_payload_size(&self) -> usize {
2417        self.values.limits.max_api_payload_size.as_bytes()
2418    }
2419
2420    /// Returns the maximum payload size for file uploads and chunks.
2421    pub fn max_api_file_upload_size(&self) -> usize {
2422        self.values.limits.max_api_file_upload_size.as_bytes()
2423    }
2424
2425    /// Returns the maximum payload size for chunks
2426    pub fn max_api_chunk_upload_size(&self) -> usize {
2427        self.values.limits.max_api_chunk_upload_size.as_bytes()
2428    }
2429
2430    /// Returns the maximum payload size for a profile
2431    pub fn max_profile_size(&self) -> usize {
2432        self.values.limits.max_profile_size.as_bytes()
2433    }
2434
2435    /// Returns the maximum payload size for a trace metric.
2436    pub fn max_trace_metric_size(&self) -> usize {
2437        self.values.limits.max_trace_metric_size.as_bytes()
2438    }
2439
2440    /// Returns the maximum payload size for a compressed replay.
2441    pub fn max_replay_compressed_size(&self) -> usize {
2442        self.values.limits.max_replay_compressed_size.as_bytes()
2443    }
2444
2445    /// Returns the maximum payload size for an uncompressed replay.
2446    pub fn max_replay_uncompressed_size(&self) -> usize {
2447        self.values.limits.max_replay_uncompressed_size.as_bytes()
2448    }
2449
2450    /// Returns the maximum message size for an uncompressed replay.
2451    ///
2452    /// This is greater than max_replay_compressed_size because
2453    /// it can include additional metadata about the replay in
2454    /// addition to the recording.
2455    pub fn max_replay_message_size(&self) -> usize {
2456        self.values.limits.max_replay_message_size.as_bytes()
2457    }
2458
2459    /// Returns the maximum number of active requests
2460    pub fn max_concurrent_requests(&self) -> usize {
2461        self.values.limits.max_concurrent_requests
2462    }
2463
2464    /// Returns the maximum number of active queries
2465    pub fn max_concurrent_queries(&self) -> usize {
2466        self.values.limits.max_concurrent_queries
2467    }
2468
2469    /// The maximum number of seconds a query is allowed to take across retries.
2470    pub fn query_timeout(&self) -> Duration {
2471        Duration::from_secs(self.values.limits.query_timeout)
2472    }
2473
2474    /// The maximum number of seconds to wait for pending envelopes after receiving a shutdown
2475    /// signal.
2476    pub fn shutdown_timeout(&self) -> Duration {
2477        Duration::from_secs(self.values.limits.shutdown_timeout)
2478    }
2479
2480    /// Returns the server keep-alive timeout in seconds.
2481    ///
2482    /// By default keep alive is set to a 5 seconds.
2483    pub fn keepalive_timeout(&self) -> Duration {
2484        Duration::from_secs(self.values.limits.keepalive_timeout)
2485    }
2486
2487    /// Returns the server idle timeout in seconds.
2488    pub fn idle_timeout(&self) -> Option<Duration> {
2489        self.values.limits.idle_timeout.map(Duration::from_secs)
2490    }
2491
2492    /// Returns the maximum connections.
2493    pub fn max_connections(&self) -> Option<usize> {
2494        self.values.limits.max_connections
2495    }
2496
2497    /// TCP listen backlog to configure on Relay's listening socket.
2498    pub fn tcp_listen_backlog(&self) -> u32 {
2499        self.values.limits.tcp_listen_backlog
2500    }
2501
2502    /// Returns the number of cores to use for thread pools.
2503    pub fn cpu_concurrency(&self) -> usize {
2504        self.values.limits.max_thread_count
2505    }
2506
2507    /// Returns the number of tasks that can run concurrently in the worker pool.
2508    pub fn pool_concurrency(&self) -> usize {
2509        self.values.limits.max_pool_concurrency
2510    }
2511
2512    /// Returns the maximum size of a project config query.
2513    pub fn query_batch_size(&self) -> usize {
2514        self.values.cache.batch_size
2515    }
2516
2517    /// Get filename for static project config.
2518    pub fn project_configs_path(&self) -> PathBuf {
2519        self.path.join("projects")
2520    }
2521
2522    /// True if the Relay should do processing.
2523    pub fn processing_enabled(&self) -> bool {
2524        self.values.processing.enabled
2525    }
2526
2527    /// Level of normalization for Relay to apply to incoming data.
2528    pub fn normalization_level(&self) -> NormalizationLevel {
2529        self.values.normalization.level
2530    }
2531
2532    /// The path to the GeoIp database required for event processing.
2533    pub fn geoip_path(&self) -> Option<&Path> {
2534        self.values
2535            .geoip
2536            .path
2537            .as_deref()
2538            .or(self.values.processing.geoip_path.as_deref())
2539    }
2540
2541    /// Maximum future timestamp of ingested data.
2542    ///
2543    /// Events past this timestamp will be adjusted to `now()`. Sessions will be dropped.
2544    pub fn max_secs_in_future(&self) -> i64 {
2545        self.values.processing.max_secs_in_future.into()
2546    }
2547
2548    /// Maximum age of ingested sessions. Older sessions will be dropped.
2549    pub fn max_session_secs_in_past(&self) -> i64 {
2550        self.values.processing.max_session_secs_in_past.into()
2551    }
2552
2553    /// Configuration name and list of Kafka configuration parameters for a given topic.
2554    pub fn kafka_configs(
2555        &self,
2556        topic: KafkaTopic,
2557    ) -> Result<KafkaTopicConfig<'_>, KafkaConfigError> {
2558        self.values.processing.topics.get(topic).kafka_configs(
2559            &self.values.processing.kafka_config,
2560            &self.values.processing.secondary_kafka_configs,
2561        )
2562    }
2563
2564    /// Whether to validate the topics against Kafka.
2565    pub fn kafka_validate_topics(&self) -> bool {
2566        self.values.processing.kafka_validate_topics
2567    }
2568
2569    /// All unused but configured topic assignments.
2570    pub fn unused_topic_assignments(&self) -> &relay_kafka::Unused {
2571        &self.values.processing.topics.unused
2572    }
2573
2574    /// Configuration of the attachment upload service.
2575    pub fn upload(&self) -> &UploadServiceConfig {
2576        &self.values.processing.upload
2577    }
2578
2579    /// Redis servers to connect to for project configs, cardinality limits,
2580    /// rate limiting, and metrics metadata.
2581    pub fn redis(&self) -> Option<RedisConfigsRef<'_>> {
2582        let redis_configs = self.values.processing.redis.as_ref()?;
2583
2584        Some(build_redis_configs(
2585            redis_configs,
2586            self.cpu_concurrency() as u32,
2587            self.pool_concurrency() as u32,
2588        ))
2589    }
2590
2591    /// Chunk size of attachments in bytes.
2592    pub fn attachment_chunk_size(&self) -> usize {
2593        self.values.processing.attachment_chunk_size.as_bytes()
2594    }
2595
2596    /// Maximum metrics batch size in bytes.
2597    pub fn metrics_max_batch_size_bytes(&self) -> usize {
2598        self.values.aggregator.max_flush_bytes
2599    }
2600
2601    /// Default prefix to use when looking up project configs in Redis. This is only done when
2602    /// Relay is in processing mode.
2603    pub fn projectconfig_cache_prefix(&self) -> &str {
2604        &self.values.processing.projectconfig_cache_prefix
2605    }
2606
2607    /// Maximum rate limit to report to clients in seconds.
2608    pub fn max_rate_limit(&self) -> Option<u64> {
2609        self.values.processing.max_rate_limit.map(u32::into)
2610    }
2611
2612    /// Amount of remaining quota which is cached in memory.
2613    pub fn quota_cache_ratio(&self) -> Option<f32> {
2614        self.values.processing.quota_cache_ratio
2615    }
2616
2617    /// Maximum limit (ratio) for the in memory quota cache.
2618    pub fn quota_cache_max(&self) -> Option<f32> {
2619        self.values.processing.quota_cache_max
2620    }
2621
2622    /// Cache vacuum interval for the cardinality limiter in memory cache.
2623    ///
2624    /// The cache will scan for expired values based on this interval.
2625    pub fn cardinality_limiter_cache_vacuum_interval(&self) -> Duration {
2626        Duration::from_secs(self.values.cardinality_limiter.cache_vacuum_interval)
2627    }
2628
2629    /// Interval to refresh internal health checks.
2630    pub fn health_refresh_interval(&self) -> Duration {
2631        Duration::from_millis(self.values.health.refresh_interval_ms)
2632    }
2633
2634    /// Maximum memory watermark in bytes.
2635    pub fn health_max_memory_watermark_bytes(&self) -> u64 {
2636        self.values
2637            .health
2638            .max_memory_bytes
2639            .as_ref()
2640            .map_or(u64::MAX, |b| b.as_bytes() as u64)
2641    }
2642
2643    /// Maximum memory watermark as a percentage of maximum system memory.
2644    pub fn health_max_memory_watermark_percent(&self) -> f32 {
2645        self.values.health.max_memory_percent
2646    }
2647
2648    /// Health check probe timeout.
2649    pub fn health_probe_timeout(&self) -> Duration {
2650        Duration::from_millis(self.values.health.probe_timeout_ms)
2651    }
2652
2653    /// Refresh frequency for polling new memory stats.
2654    pub fn memory_stat_refresh_frequency_ms(&self) -> u64 {
2655        self.values.health.memory_stat_refresh_frequency_ms
2656    }
2657
2658    /// Maximum amount of COGS measurements buffered in memory.
2659    pub fn cogs_max_queue_size(&self) -> u64 {
2660        self.values.cogs.max_queue_size
2661    }
2662
2663    /// Resource ID to use for Relay COGS measurements.
2664    pub fn cogs_relay_resource_id(&self) -> &str {
2665        &self.values.cogs.relay_resource_id
2666    }
2667
2668    /// Returns configuration for the default metrics aggregator.
2669    pub fn default_aggregator_config(&self) -> &AggregatorServiceConfig {
2670        &self.values.aggregator
2671    }
2672
2673    /// Returns configuration for non-default metrics aggregator.
2674    pub fn secondary_aggregator_configs(&self) -> &Vec<ScopedAggregatorConfig> {
2675        &self.values.secondary_aggregators
2676    }
2677
2678    /// Returns aggregator config for a given metrics namespace.
2679    pub fn aggregator_config_for(&self, namespace: MetricNamespace) -> &AggregatorServiceConfig {
2680        for entry in &self.values.secondary_aggregators {
2681            if entry.condition.matches(Some(namespace)) {
2682                return &entry.config;
2683            }
2684        }
2685        &self.values.aggregator
2686    }
2687
2688    /// Return the statically configured Relays.
2689    pub fn static_relays(&self) -> &HashMap<RelayId, RelayInfo> {
2690        &self.values.auth.static_relays
2691    }
2692
2693    /// Returns the max age a signature is considered valid, in seconds.
2694    pub fn signature_max_age(&self) -> Duration {
2695        Duration::from_secs(self.values.auth.signature_max_age)
2696    }
2697
2698    /// Returns `true` if unknown items should be accepted and forwarded.
2699    pub fn accept_unknown_items(&self) -> bool {
2700        let forward = self.values.routing.accept_unknown_items;
2701        forward.unwrap_or_else(|| !self.processing_enabled())
2702    }
2703}
2704
2705impl Default for Config {
2706    fn default() -> Self {
2707        Self {
2708            values: ConfigValues::default(),
2709            credentials: None,
2710            path: PathBuf::new(),
2711        }
2712    }
2713}
2714
2715#[cfg(test)]
2716mod tests {
2717
2718    use super::*;
2719
2720    /// Regression test for renaming the envelope buffer flags.
2721    #[test]
2722    fn test_event_buffer_size() {
2723        let yaml = r###"
2724cache:
2725    event_buffer_size: 1000000
2726    event_expiry: 1800
2727"###;
2728
2729        let values: ConfigValues = serde_yaml::from_str(yaml).unwrap();
2730        assert_eq!(values.cache.envelope_buffer_size, 1_000_000);
2731        assert_eq!(values.cache.envelope_expiry, 1800);
2732    }
2733
2734    #[test]
2735    fn test_emit_outcomes() {
2736        for (serialized, deserialized) in &[
2737            ("true", EmitOutcomes::AsOutcomes),
2738            ("false", EmitOutcomes::None),
2739            ("\"as_client_reports\"", EmitOutcomes::AsClientReports),
2740        ] {
2741            let value: EmitOutcomes = serde_json::from_str(serialized).unwrap();
2742            assert_eq!(value, *deserialized);
2743            assert_eq!(serde_json::to_string(&value).unwrap(), *serialized);
2744        }
2745    }
2746
2747    #[test]
2748    fn test_emit_outcomes_invalid() {
2749        assert!(serde_json::from_str::<EmitOutcomes>("asdf").is_err());
2750    }
2751}