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