Generating lists

Cargo.toml

@@ -1,7 +1,7 @@
 [package]
 name = "latte-cli"
 description = "A database benchmarking tool for Apache Cassandra"
-version = "0.26.0"
+version = "0.27.0"
 authors = ["Piotr Kołaczkowski <[email protected]>"]
 edition = "2021"
 readme = "README.md"
@@ -38,7 +38,7 @@ rand = "0.8"
 regex = "1.5"
 rune = "0.12"
 rust-embed = "8"
-scylla = { version = "0.13", features = ["ssl"] }
+scylla = { path = "../scylla-rust-driver/scylla", features = ["ssl"] }
 search_path = "0.1"
 serde = { version = "1.0.116", features = ["derive"] }
 serde_json = "1.0.57"

src/context.rs

@@ -20,7 +20,7 @@ use rune::ast;
 use rune::ast::Kind;
 use rune::macros::{quote, MacroContext, TokenStream};
 use rune::parse::Parser;
-use rune::runtime::{Object, Shared, TypeInfo, VmError};
+use rune::runtime::{Function, Object, Shared, TypeInfo, VmError};
 use rune::{Any, Value};
 use rust_embed::RustEmbed;
 use scylla::_macro_internal::ColumnType;
@@ -616,6 +616,15 @@ mod bind {
                     .try_collect()?;
                 Ok(CqlValue::Set(elements))
             }
+            (Value::Vec(v), ColumnType::Vector(elt, dim)) => {
+                let v = v.borrow_ref().unwrap();
+                let elements = v
+                    .as_ref()
+                    .iter()
+                    .map(|v| to_scylla_value(v, elt))
+                    .try_collect()?;
+                Ok(CqlValue::Vector(elements))
+            }
             (
                 Value::Object(v),
                 ColumnType::UserDefinedType {
@@ -858,6 +867,15 @@ pub fn clamp_int(value: i64, min: i64, max: i64) -> i64 {
     value.clamp(min, max)
 }
 
+pub fn list(seed: i64, len: usize, generator: Function) -> Result<Vec<Value>, VmError> {
+    let mut result = Vec::with_capacity(len);
+    for i in 0..len {
+        let value = generator.call((hash2(seed, i as i64),))?;
+        result.push(value);
+    }
+    Ok(result)
+}
+
 /// Generates random blob of data of given length.
 /// Parameter `seed` is used to seed the RNG.
 pub fn blob(seed: i64, len: usize) -> rune::runtime::Bytes {

src/workload.rs

@@ -129,6 +129,8 @@ impl Program {
         let mut latte_module = Module::with_crate("latte");
         latte_module.function(&["blob"], context::blob).unwrap();
         latte_module.function(&["text"], context::text).unwrap();
+        latte_module.function(&["list"], context::list).unwrap();
+
         latte_module
             .function(&["now_timestamp"], context::now_timestamp)
             .unwrap();