模式创建

您可以在 schema 函数中创建（重新）创建基准测试所需的自己的键空间和表。如果存在，schema 函数还应删除旧模式。通过运行 latte schema 命令来执行 schema 函数。

pub async fn schema(ctx) {
  ctx.execute("CREATE KEYSPACE IF NOT EXISTS test \
                 WITH REPLICATION = { 'class' : 'SimpleStrategy', 'replication_factor' : 1 }").await?;
  ctx.execute("DROP TABLE IF NOT EXISTS test.test").await?;
  ctx.execute("CREATE TABLE test.test(id bigint, data varchar)").await?;
}

预处理语句

调用 ctx.execute 不是最佳选择，因为它不使用预处理语句。您可以在 prepare 函数中准备语句并将它们注册到上下文对象上。

const INSERT = "my_insert";
const SELECT = "my_select";

pub async fn prepare(ctx) {
  ctx.prepare(INSERT, "INSERT INTO test.test(id, data) VALUES (?, ?)").await?;
  ctx.prepare(SELECT, "SELECT * FROM test.test WHERE id = ?").await?;
}

pub async fn run(ctx, i) {
  ctx.execute_prepared(SELECT, [i]).await
}

查询参数也可以通过名称绑定和传递

const INSERT = "my_insert";

pub async fn prepare(ctx) {
  ctx.prepare(INSERT, "INSERT INTO test.test(id, data) VALUES (:id, :data)").await?;
}

pub async fn run(ctx, i) {
  ctx.execute_prepared(INSERT, #{id: 5, data: "foo"}).await
}

填充数据库

当读取查询返回非空结果集时，它们更有趣。

要能够使用 latte load 将数据加载到表中，您需要在上下文对象上设置加载周期数并定义 load 函数。

pub async fn prepare(ctx) {
  ctx.load_cycle_count = 1000000;
}

pub async fn load(ctx, i) {
  ctx.execute_prepared(INSERT, [i, "Lorem ipsum dolor sit amet"]).await
}

我们还建议定义 erase 函数以在加载前清除数据，这样无论数据库中之前存在什么数据，您都可以始终获得相同的数据集。

pub async fn erase(ctx) {
  ctx.execute("TRUNCATE TABLE test.test").await
}

生成数据

Latte 附带一组数据生成函数库。这些函数在 latte 包中可用。通常，这些函数接受一个整数 i 周期号，因此您可以生成一致的数字。数据生成函数是纯函数，即多次使用相同的参数调用它们会始终产生相同的结果。

• latte::uuid(i) – 生成一个随机（类型 4）UUID
• latte::hash(i) – 生成一个非负整数哈希值
• latte::hash2(a, b) – 生成两个整数的非负整数哈希值
• latte::hash_range(i, max) – 在范围 0..max 内生成一个整数值
• latte::hash_select(i, vector) – 根据哈希选择向量中的一个元素
• latte::blob(i, len) – 生成一个长度为 len 的随机二进制 blob
• latte::normal(i, mean, std_dev) – 从正态分布生成一个浮点数
• latte::uniform(i, min, max) – 从均匀分布生成一个浮点数

类型转换

Rune使用64位表示法来处理整数和浮点数。从版本0.28开始，Rune数字会自动转换为正确的目标查询参数类型，因此您无需进行显式转换。例如，您可以将整数作为Cassandra类型的参数传递smallint。如果数字太大，无法适应目标类型的允许范围，则会引发运行时错误。

以下方法可用

• x.to_integer() – 将浮点数转换为整数
• x.to_float() – 将整数转换为浮点数
• x.to_string() – 将浮点数或整数转换为字符串
• x.clamp(min, max) – 将整数或浮点数值的范围限制在给定的范围内

您也可以通过调用实例函数to_integer或to_float来在浮点数和整数之间进行转换。

文本资源

可以使用fs模块中的函数从文件或资源中加载文本数据

• fs::read_to_string(file_path) – 返回文件内容作为字符串
• fs::read_lines(file_path) – 将文件行读入字符串向量
• fs::read_resource_to_string(resource_name) – 返回内置资源内容作为字符串
• fs::read_resource_lines(resource_name) – 返回内置资源行作为字符串向量

资源嵌入在程序二进制文件中。您可以在源树中的resources文件夹中找到它们。

为了减少内存分配的成本，最好在prepare函数中只加载一次资源，并将它们存储在上下文的data字段中，以便在load和run中未来使用。

pub async fn prepare(ctx) {
  ctx.data.last_names = fs::read_lines("lastnames.txt")?;
  // ... prepare queries
}

pub async fn run(ctx, i) {
  let random_last_name = latte::hash_select(i, ctx.data.last_names);
  // ... use random_last_name in queries
}

参数化工作负载

可以通过从命令行调用提供的参数来参数化工作负载。使用宏latte::param!(param_name, default_value)来初始化从命令行参数来的脚本常量

const ROW_COUNT = latte::param!("row_count", 1000000);

pub async fn prepare(ctx) {
  ctx.load_cycle_count = ROW_COUNT;
} 

然后您可以使用-P来设置参数

latte run <workload> -P row_count=200

错误处理

在执行工作负载脚本期间出现的错误分为三类

• 编译错误 – 在脚本加载时检测到的错误；例如，语法错误或引用未定义的变量。这些错误会立即发出信号，并在连接到数据库之前终止基准测试。
• 运行时错误/恐慌 – 例如，除以零或数组越界访问。它们会立即终止基准测试。
• 错误返回值 - 例如当查询执行返回错误结果时。这些值只有在实际从函数返回时才有效（使用?将它们传播到调用链）。除Cassandra超载错误外的所有错误都会立即终止基准测试。
  在主运行阶段发生的超载错误（例如超时）将被计数并在基准测试报告中报告。

其他函数

• ctx.elapsed_secs() - 返回自开始工作负载以来经过的秒数，作为浮点数