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 <pkolaczk@gmail.com>"]
 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 {
@@ -907,6 +916,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();