Remove RawEventsQuery

dbconn/src/dbconn.rs

@@ -5,14 +5,17 @@ pub mod search;
 pub mod pg {
     pub use tokio_postgres::{Client, Error};
 }
+pub mod channelconfig;
+
 use err::Error;
-use netpod::log::*;
+use netpod::{log::*, ScalarType, Shape};
 use netpod::{Channel, Database, NodeConfigCached, ScyllaConfig};
 use scylla::frame::response::cql_to_rust::FromRowError as ScyFromRowError;
 use scylla::transport::errors::{NewSessionError as ScyNewSessionError, QueryError as ScyQueryError};
 use scylla::Session as ScySession;
+use std::sync::Arc;
 use std::time::Duration;
-use tokio_postgres::{Client, NoTls};
+use tokio_postgres::{Client, Client as PgClient, NoTls};
 
 trait ErrConv<T> {
     fn err_conv(self) -> Result<T, Error>;
@@ -188,3 +191,40 @@ pub async fn insert_channel(name: String, facility: i64, dbc: &Client) -> Result
     }
     Ok(())
 }
+
+pub async fn find_series(channel: &Channel, pgclient: Arc<PgClient>) -> Result<(u64, ScalarType, Shape), Error> {
+    info!("find_series  channel {:?}", channel);
+    let rows = if let Some(series) = channel.series() {
+        let q = "select series, facility, channel, scalar_type, shape_dims from series_by_channel where series = $1";
+        pgclient.query(q, &[&(series as i64)]).await.err_conv()?
+    } else {
+        let q = "select series, facility, channel, scalar_type, shape_dims from series_by_channel where facility = $1 and channel = $2";
+        pgclient
+            .query(q, &[&channel.backend(), &channel.name()])
+            .await
+            .err_conv()?
+    };
+    if rows.len() < 1 {
+        return Err(Error::with_public_msg_no_trace(format!(
+            "No series found for {channel:?}"
+        )));
+    }
+    if rows.len() > 1 {
+        error!("Multiple series found for {channel:?}");
+        return Err(Error::with_public_msg_no_trace(
+            "Multiple series found for channel, can not return data for ambiguous series",
+        ));
+    }
+    let row = rows
+        .into_iter()
+        .next()
+        .ok_or_else(|| Error::with_public_msg_no_trace(format!("can not find series for channel")))?;
+    let series = row.get::<_, i64>(0) as u64;
+    let _facility: String = row.get(1);
+    let _channel: String = row.get(2);
+    let a: i32 = row.get(3);
+    let scalar_type = ScalarType::from_scylla_i32(a)?;
+    let a: Vec<i32> = row.get(4);
+    let shape = Shape::from_scylla_shape_dims(&a)?;
+    Ok((series, scalar_type, shape))
+}