Ingestor, scaffolding

ingestor/src/collection.rs

@@ -0,0 +1,25 @@
+use rocketman::types::event::Event;
+use anyhow::Result;
+
+enum Ingestor {
+  Jetstream(SpoorJetstream)
+}
+
+struct SpoorJetstream;
+
+#[async_trait]
+impl LexiconIngestor for SpoorJetstream {
+  async fn ingest(&self, message: Event<Value>) -> Result<()> {
+    info!("{:?}", message);
+    // if let Some(commit) = &message.commit {
+    //   match commit.operation {
+    //     Operation::Create | Operation::Update => {}
+    //     Operation::Delete => {}
+    //   }
+    // } else {
+    //   return Err("Message has no commit");
+    // }
+    Ok(())
+  }
+}
+

ingestor/src/ingestor.rs

@@ -0,0 +1,54 @@
+use atproto::{
+  Collection,
+  lexicons::my::spoor::log::{Activity, Session},
+};
+use async_trait::async_trait;
+use rocketman::{
+  options::JetstreamOptions,
+  ingestion::LexiconIngestor,
+  // types::event::{Event, Operation},
+  types::event::Event,
+  connection::JetstreamConnection,
+  handler,
+};
+use serde_json::value::Value;
+use std::{
+  collections::HashMap,
+  sync::{Arc, Mutex},
+};
+use tracing::{error, info};
+
+pub async fn start_ingestor() {
+  info!("Starting ingestor");
+  let nsids = vec![
+    Activity::NSID.to_string(),
+    Session::NSID.to_string(),
+  ];
+  let opts = JetstreamOptions::builder().wanted_collections(nsids.clone()).build();
+  let jetstream = JetstreamConnection::new(opts);
+
+  let mut ingesters: HashMap<String, Box<dyn LexiconIngestor + Send + Sync>> = HashMap::new();
+  for nsid in nsids {
+    ingesters.insert(nsid, Box::new(SpoorJetstream));
+  }
+
+  let cursor: Arc<Mutex<Option<u64>>> = Arc::new(Mutex::new(None));
+
+  let msg_rx = jetstream.get_msg_rx();
+  let reconnect_tx = jetstream.get_reconnect_tx();
+
+  let cursor_clone = cursor.clone();
+  tokio::spawn(async move {
+    while let Ok(message) = msg_rx.recv_async().await {
+      if let Err(e) = handler::handle_message(message, &ingesters, 
+          reconnect_tx.clone(), cursor_clone.clone()).await {
+        error!("Error processing message: {}", e);
+      }
+    }
+  });
+
+  if let Err(e) = jetstream.connect(cursor.clone()).await {
+    error!("Failed to connect to Jetstream: {}", e);
+    std::process::exit(1);
+  }
+}

ingestor/src/main.rs

@@ -0,0 +1,8 @@
+
+#[tokio::main]
+async fn main() {
+  let subscriber = tracing_subscriber::FmtSubscriber::new();
+  let _ = tracing::subscriber::set_global_default(subscriber);
+
+  
+}