reduce code copy-pastes

Author: badmonster0

src/ops/shared/postgres.rs

@@ -1,7 +1,10 @@
 use crate::prelude::*;
 
+use crate::ops::sdk::*;
 use crate::settings::DatabaseConnectionSpec;
 use sqlx::PgPool;
+use sqlx::postgres::types::PgRange;
+use std::ops::Bound;
 
 pub async fn get_db_pool(
     db_ref: Option<&spec::AuthEntryReference<DatabaseConnectionSpec>>,
@@ -18,3 +21,54 @@ pub async fn get_db_pool(
     };
     Ok(db_pool)
 }
+
+pub fn key_value_fields_iter<'a>(
+    key_fields_schema: &[FieldSchema],
+    key_value: &'a KeyValue,
+) -> Result<&'a [KeyValue]> {
+    let slice = if key_fields_schema.len() == 1 {
+        std::slice::from_ref(key_value)
+    } else {
+        match key_value {
+            KeyValue::Struct(fields) => fields,
+            _ => bail!("expect struct key value"),
+        }
+    };
+    Ok(slice)
+}
+
+pub fn bind_key_field<'arg>(
+    builder: &mut sqlx::QueryBuilder<'arg, sqlx::Postgres>,
+    key_value: &'arg KeyValue,
+) -> Result<()> {
+    match key_value {
+        KeyValue::Bytes(v) => {
+            builder.push_bind(&**v);
+        }
+        KeyValue::Str(v) => {
+            builder.push_bind(&**v);
+        }
+        KeyValue::Bool(v) => {
+            builder.push_bind(v);
+        }
+        KeyValue::Int64(v) => {
+            builder.push_bind(v);
+        }
+        KeyValue::Range(v) => {
+            builder.push_bind(PgRange {
+                start: Bound::Included(v.start as i64),
+                end: Bound::Excluded(v.end as i64),
+            });
+        }
+        KeyValue::Uuid(v) => {
+            builder.push_bind(v);
+        }
+        KeyValue::Date(v) => {
+            builder.push_bind(v);
+        }
+        KeyValue::Struct(fields) => {
+            builder.push_bind(sqlx::types::Json(fields));
+        }
+    }
+    Ok(())
+}