feat: initial work on elasticsearch integration

Signed-off-by: Zomatree <me@zomatree.live>
2026-03-29 05:43:58 +01:00
parent d9643ebd8d
commit 8009b3bf53
49 changed files with 1526 additions and 128 deletions
@@ -1,7 +1,7 @@
 use std::collections::HashSet;

 use crate::events::rabbit::*;
-use crate::User;
+use crate::{Message, User};
 use amqprs::channel::{BasicPublishArguments, ExchangeDeclareArguments};
 use amqprs::connection::OpenConnectionArguments;
 use amqprs::{channel::Channel, connection::Connection, error::Error as AMQPError};
@@ -52,6 +52,15 @@ impl AMQP {
            .await
            .expect("Failed to declare exchange");

+        channel
+            .exchange_declare(
+                ExchangeDeclareArguments::new(&config.elasticsearch.exchange, "direct")
+                    .durable(true)
+                    .finish(),
+            )
+            .await
+            .expect("Failed to declare exchange");
+
        AMQP::new(connection, channel)
    }

@@ -316,4 +325,104 @@ impl AMQP {
            )
            .await
    }
+
+    pub async fn new_message_search(&self, message: Message) -> Result<(), AMQPError> {
+        let config = revolt_config::config().await;
+
+        let payload = to_string(&message).unwrap();
+
+        debug!(
+            "Sending new message search payload on channel {}: {}",
+            config.elasticsearch.message_queue, payload
+        );
+
+        self.channel
+            .basic_publish(
+                BasicProperties::default()
+                    .with_content_type("application/json")
+                    .with_persistence(true)
+                    .finish(),
+                payload.into(),
+                BasicPublishArguments::new(
+                    &config.elasticsearch.exchange,
+                    &config.elasticsearch.message_queue,
+                ),
+            )
+            .await
+    }
+
+    pub async fn edit_message_search(&self, message: Message) -> Result<(), AMQPError> {
+        let config = revolt_config::config().await;
+
+        let payload = to_string(&message).unwrap();
+
+        debug!(
+            "Sending edit message search payload on channel {}: {}",
+            config.elasticsearch.message_edit_queue, payload
+        );
+
+        self.channel
+            .basic_publish(
+                BasicProperties::default()
+                    .with_content_type("application/json")
+                    .with_persistence(true)
+                    .finish(),
+                payload.into(),
+                BasicPublishArguments::new(
+                    &config.elasticsearch.exchange,
+                    &config.elasticsearch.message_edit_queue,
+                ),
+            )
+            .await
+    }
+
+    pub async fn delete_message_search(&self, message_id: String) -> Result<(), AMQPError> {
+        let config = revolt_config::config().await;
+
+        let payload = to_string(&MessageDeletePayload { message_id }).unwrap();
+
+        debug!(
+            "Sending delete message search payload on channel {}: {}",
+            config.elasticsearch.message_delete_queue, payload
+        );
+
+        self.channel
+            .basic_publish(
+                BasicProperties::default()
+                    .with_content_type("application/json")
+                    .with_persistence(true)
+                    .finish(),
+                payload.into(),
+                BasicPublishArguments::new(
+                    &config.elasticsearch.exchange,
+                    &config.elasticsearch.message_delete_queue,
+                ),
+            )
+            .await
+    }
+
+    pub async fn delete_channel_search(&self, channel_id: String) -> Result<(), AMQPError> {
+        let config = revolt_config::config().await;
+
+        let payload = to_string(&ChannelDeletePayload { channel_id }).unwrap();
+
+        debug!(
+            "Sending delete channel search payload on channel {}: {}",
+            config.elasticsearch.channel_delete_queue, payload
+        );
+
+        self.channel
+            .basic_publish(
+                BasicProperties::default()
+                    .with_content_type("application/json")
+                    .with_persistence(true)
+                    .finish(),
+                payload.into(),
+                BasicPublishArguments::new(
+                    &config.elasticsearch.exchange,
+                    &config.elasticsearch.channel_delete_queue,
+                ),
+            )
+            .await
+    }
 }