rename AdminMessage to SignalAlert message which is less confusing

This commit is contained in:
Chris Beck
2025-12-05 22:05:58 -07:00
parent f3b66541ed
commit 010a5a9336
2 changed files with 18 additions and 18 deletions
+5 -5
View File
@@ -1,6 +1,6 @@
use super::log_buffer::LogBuffer;
use super::route::{Destination, Limit, Route};
use super::{AdminMessage, LimitResult, Limiter, LimiterSet};
use super::{SignalAlertMessage, LimitResult, Limiter, LimiterSet};
use crate::{
concurrent_map::ConcurrentMap,
log_format::LogFormatConfig,
@@ -74,7 +74,7 @@ pub struct LogHandlerConfig {
#[derive(Debug)]
pub struct LogHandler {
config: LogHandlerConfig,
admin_mq_tx: UnboundedSender<AdminMessage>,
signal_alert_mq_tx: UnboundedSender<SignalAlertMessage>,
/// Log buffers keyed by origin (app + host). Lazily created.
log_buffers: ConcurrentMap<Origin, LogBuffer>,
/// Routes with their associated limiter sets.
@@ -85,7 +85,7 @@ pub struct LogHandler {
impl LogHandler {
/// Initialize a new log handler
pub fn new(config: LogHandlerConfig, admin_mq_tx: UnboundedSender<AdminMessage>) -> Self {
pub fn new(config: LogHandlerConfig, signal_alert_mq_tx: UnboundedSender<SignalAlertMessage>) -> Self {
let routes = config
.routes
.iter()
@@ -100,7 +100,7 @@ impl LogHandler {
Self {
config,
admin_mq_tx,
signal_alert_mq_tx,
log_buffers: ConcurrentMap::new(),
routes,
overall_limits,
@@ -192,7 +192,7 @@ impl LogHandler {
// Send alert if we have formatted text
if let Some(text) = formatted_text {
let destination_override = rate_limit_result.ok().flatten();
if let Err(_err) = self.admin_mq_tx.send(AdminMessage {
if let Err(_err) = self.signal_alert_mq_tx.send(SignalAlertMessage {
origin: Some(origin),
text,
attachment_paths: Default::default(),
+13 -13
View File
@@ -162,7 +162,7 @@ fn parse_gateway_command(s: &str) -> Result<GatewayCommand, String> {
/// A message queued to be sent to all admins.
/// This is generally an alert message, which may have attached images.
#[derive(Clone, Debug, Default)]
struct AdminMessage {
struct SignalAlertMessage {
/// The origin of the message (app + host), if from syslog
origin: Option<Origin>,
text: String,
@@ -188,8 +188,8 @@ struct AdminMessage {
/// and then the buffer is purged. This serves as a minimal log-aggregation and alerting system.
pub struct Gateway {
config: GatewayConfig,
admin_mq_tx: UnboundedSender<AdminMessage>,
admin_mq_rx: Mutex<UnboundedReceiver<AdminMessage>>,
signal_alert_mq_tx: UnboundedSender<SignalAlertMessage>,
signal_alert_mq_rx: Mutex<UnboundedReceiver<SignalAlertMessage>>,
token: CancellationToken,
prometheus: Option<Prometheus>,
/// Log handler for processing log messages from all origins.
@@ -205,7 +205,7 @@ impl Gateway {
token: CancellationToken,
message_handler: Option<Box<dyn MessageHandler>>,
) -> Self {
let (admin_mq_tx, admin_mq_rx) = unbounded_channel();
let (signal_alert_mq_tx, signal_alert_mq_rx) = unbounded_channel();
let prometheus = config
.prometheus
@@ -214,12 +214,12 @@ impl Gateway {
.transpose()
.expect("Invalid prometheus config");
let log_handler = LogHandler::new(config.log_handler.clone(), admin_mq_tx.clone());
let log_handler = LogHandler::new(config.log_handler.clone(), signal_alert_mq_tx.clone());
Self {
config,
admin_mq_tx,
admin_mq_rx: Mutex::new(admin_mq_rx),
signal_alert_mq_tx,
signal_alert_mq_rx: Mutex::new(signal_alert_mq_rx),
token,
prometheus,
log_handler,
@@ -364,8 +364,8 @@ impl Gateway {
async fn do_run(&self, signal_cli: &impl RpcClient) -> Result<(), RpcClientError> {
self.update_trust(signal_cli).await;
let mut admin_mq_rx = self
.admin_mq_rx
let mut signal_alert_mq_rx = self
.signal_alert_mq_rx
.try_lock()
.expect("Mutex should not be contended");
let mut signal_rx = signal_cli
@@ -378,7 +378,7 @@ impl Gateway {
info!("Stop requested");
return Ok(());
},
outbound_admin_msg = admin_mq_rx.recv() => {
outbound_admin_msg = signal_alert_mq_rx.recv() => {
if let Some(msg) = outbound_admin_msg {
// Log summary, or first 500 bytes of text if no summary provided
let summary = msg.summary.as_deref().unwrap_or_else(|| {
@@ -413,7 +413,7 @@ impl Gateway {
attachments,
}.send(signal_cli).await?;
} else {
warn!("admin_mq_rx is closed, halting service");
warn!("signal_alert_mq_rx is closed, halting service");
self.token.cancel();
return Ok(());
}
@@ -762,8 +762,8 @@ impl Gateway {
.collect::<Vec<_>>()
.join(" ");
self.admin_mq_tx
.send(AdminMessage {
self.signal_alert_mq_tx
.send(SignalAlertMessage {
origin: None, // Prometheus alerts don't have a syslog origin
text,
attachment_paths,