wip

libs/nats-libs/src/lib.rs

@@ -1 +1,2 @@
-pub mod functions;
+pub mod job;
+pub mod stream;

libs/nats-libs/src/stream.rs

@@ -0,0 +1,46 @@
+use std::num::NonZeroU64;
+
+use async_nats::jetstream::{Context, stream::Stream};
+use async_trait::async_trait;
+use types::error::{CorroError, CorroErrorType};
+
+#[async_trait]
+pub trait JetstreamClient {
+    async fn lock(&self, id: NonZeroU64) -> Result<u64, CorroError>;
+}
+
+#[async_trait]
+impl JetstreamClient for Context {
+    async fn lock(&self, id: NonZeroU64) -> Result<u64, CorroError> {
+        match self
+            .publish(format!("locks.{id}"), "".into())
+            .await
+            .unwrap()
+            .await
+        {
+            Ok(ack) => Ok(ack.sequence),
+            Err(why) => Err(CorroError {
+                error_type: CorroErrorType::NatsError,
+                message: why.to_string(),
+            }),
+        }
+    }
+}
+
+#[async_trait]
+pub trait StreamClient {
+    async fn unlock(&self, sequence: u64) -> Result<bool, CorroError>;
+}
+
+#[async_trait]
+impl StreamClient for Stream {
+    async fn unlock(&self, sequence: u64) -> Result<bool, CorroError> {
+        match self.delete_message(sequence).await {
+            Ok(result) => Ok(result),
+            Err(why) => Err(CorroError {
+                error_type: CorroErrorType::NatsError,
+                message: why.to_string(),
+            }),
+        }
+    }
+}