Merge pull request #15 from worldcoin/0xkitsune/queue-utils

feat(utils): add sqs query to utils
worldcoin · Feb 1, 2024 · 0dbd1cf · 0dbd1cf
2 parents 0cda30e + 8273107
commit 0dbd1cf
Showing 1 changed file with 23 additions and 2 deletions.
diff --git a/bin/utils.rs b/bin/utils.rs
@@ -4,7 +4,8 @@ use rand::{thread_rng, Rng};
 
 #[derive(Debug, Clone, Parser)]
 enum Opt {
-    RandomQuery(RandomQuery),
+    HttpQuery(RandomQuery),
+    SQSQuery(RandomQuery),
 }
 
 #[derive(Debug, Clone, Args)]
@@ -20,7 +21,7 @@ async fn main() -> eyre::Result<()> {
     let args = Opt::parse();
 
     match args {
-        Opt::RandomQuery(args) => {
+        Opt::HttpQuery(args) => {
             let mut rng = thread_rng();
 
             let template: Template = rng.gen();
@@ -32,6 +33,26 @@ async fn main() -> eyre::Result<()> {
                 .await?
                 .error_for_status()?;
         }
+
+        Opt::SQSQuery(args) => {
+            let aws_config = aws_config::load_defaults(
+                aws_config::BehaviorVersion::latest(),
+            )
+            .await;
+
+            let aws_client = aws_sdk_sqs::Client::new(&aws_config);
+
+            let mut rng = thread_rng();
+
+            let template: Template = rng.gen();
+
+            aws_client
+                .send_message()
+                .queue_url(args.url)
+                .message_body(serde_json::to_string(&template)?)
+                .send()
+                .await?;
+        }
     }
 
     Ok(())