Add config query for scylla

nodenet/src/conn.rs

@@ -9,6 +9,8 @@ use netpod::histo::HistoLog2;
 use netpod::query::RawEventsQuery;
 use netpod::{log::*, AggKind};
 use netpod::{EventQueryJsonStringFrame, NodeConfigCached, PerfOpts};
+use scylla::frame::response::cql_to_rust::FromRowError as ScyFromRowError;
+use scylla::transport::errors::{NewSessionError as ScyNewSessionError, QueryError as ScyQueryError};
 use std::net::SocketAddr;
 use std::pin::Pin;
 use tokio::io::AsyncWriteExt;
@@ -16,6 +18,37 @@ use tokio::net::tcp::OwnedWriteHalf;
 use tokio::net::TcpStream;
 use tracing::Instrument;
 
+trait ErrConv<T> {
+    fn err_conv(self) -> Result<T, Error>;
+}
+
+impl<T> ErrConv<T> for Result<T, ScyQueryError> {
+    fn err_conv(self) -> Result<T, Error> {
+        match self {
+            Ok(k) => Ok(k),
+            Err(e) => Err(Error::with_msg_no_trace(format!("{e:?}"))),
+        }
+    }
+}
+
+impl<T> ErrConv<T> for Result<T, ScyNewSessionError> {
+    fn err_conv(self) -> Result<T, Error> {
+        match self {
+            Ok(k) => Ok(k),
+            Err(e) => Err(Error::with_msg_no_trace(format!("{e:?}"))),
+        }
+    }
+}
+
+impl<T> ErrConv<T> for Result<T, ScyFromRowError> {
+    fn err_conv(self) -> Result<T, Error> {
+        match self {
+            Ok(k) => Ok(k),
+            Err(e) => Err(Error::with_msg_no_trace(format!("{e:?}"))),
+        }
+    }
+}
+
 pub async fn events_service(node_config: NodeConfigCached) -> Result<(), Error> {
     let addr = format!("{}:{}", node_config.node.listen, node_config.node.port_raw);
     let lis = tokio::net::TcpListener::bind(addr).await?;
@@ -117,6 +150,36 @@ async fn events_conn_handler_inner_try(
     };
     debug!("---  got query   evq {:?}", evq);
 
+    // TODO make scylla usage configurable in config
+    if evq.channel.backend == "scylla" {
+        // Find the "series" id.
+        let scy = scylla::SessionBuilder::new()
+            .known_node("sf-daqbuf-34:8340")
+            .build()
+            .await
+            .err_conv();
+        let scy = match scy {
+            Ok(k) => k,
+            Err(e) => return Err((e, netout))?,
+        };
+        let cql = "select dtype, series from series_by_channel where facility = ? and channel_name = ?";
+        let res = scy.query(cql, ()).await.err_conv();
+        let res = match res {
+            Ok(k) => k,
+            Err(e) => return Err((e, netout))?,
+        };
+        let rows = res.rows_typed_or_empty::<(i32, i32)>();
+        for r in rows {
+            let r = match r.err_conv() {
+                Ok(k) => k,
+                Err(e) => return Err((e, netout))?,
+            };
+            info!("got row  {r:?}");
+        }
+        error!("TODO scylla fetch continue here");
+        err::todo();
+    }
+
     let mut p1: Pin<Box<dyn Stream<Item = Box<dyn Framable>> + Send>> =
         if let Some(aa) = &node_config.node.channel_archiver {
             match archapp_wrap::archapp::archeng::pipe::make_event_pipe(&evq, node_config.clone(), aa.clone()).await {