reifydb-engine 0.9.1

Query execution and processing engine for ReifyDB
Documentation 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80

// SPDX-License-Identifier: Apache-2.0
// Copyright (c) 2026 ReifyDB

use reifydb_core::{common::CommitVersion, interface::catalog::config::ConfigKey};
use reifydb_test_harness::engine::TestEngine;
use reifydb_value::value::Value;

fn insert_wide_rows(t: &TestEngine, count: usize) {
	t.admin("CREATE NAMESPACE test");
	t.admin("CREATE TABLE test::wide { n: int4 }");
	let rows: Vec<String> = (0..count).map(|i| format!("{{n:{i}}}")).collect();
	t.command(&format!("INSERT test::wide [{}]", rows.join(",")));
}

fn set_query_memory_limit(t: &TestEngine, bytes: u64) {
	t.inner()
		.catalog()
		.cache()
		.set_config(ConfigKey::QueryMemoryLimit, CommitVersion(1), Value::Uint8(bytes))
		.expect("failed to set QUERY_MEMORY_LIMIT");
}

#[test]
fn sort_over_query_memory_limit_surfaces_query_006() {
	let t = TestEngine::new();
	insert_wide_rows(&t, 50);
	set_query_memory_limit(&t, 8);

	let err = t.query_err("FROM test::wide SORT {n}");
	assert!(err.contains("QUERY_006"), "expected QUERY_006, got: {}", err);
	assert!(err.contains("exceeded its memory limit"), "expected memory-limit message, got: {}", err);
}

#[test]
fn distinct_over_query_memory_limit_surfaces_query_006() {
	let t = TestEngine::new();
	insert_wide_rows(&t, 50);
	set_query_memory_limit(&t, 8);

	let err = t.query_err("FROM test::wide DISTINCT {n}");
	assert!(err.contains("QUERY_006"), "expected QUERY_006, got: {}", err);
	assert!(err.contains("exceeded its memory limit"), "expected memory-limit message, got: {}", err);
}

#[test]
fn top_k_over_query_memory_limit_surfaces_query_006() {
	let t = TestEngine::new();
	insert_wide_rows(&t, 50);
	set_query_memory_limit(&t, 8);

	let err = t.query_err("FROM test::wide SORT {n} TAKE 5");
	assert!(err.contains("QUERY_006"), "expected QUERY_006, got: {}", err);
	assert!(err.contains("exceeded its memory limit"), "expected memory-limit message, got: {}", err);
}

#[test]
fn sort_under_query_memory_limit_still_succeeds() {
	let t = TestEngine::new();
	insert_wide_rows(&t, 5);
	set_query_memory_limit(&t, 1024 * 1024 * 1024);

	let frames = t.query("FROM test::wide SORT {n}");
	assert_eq!(TestEngine::row_count(&frames), 5);
}

#[test]
fn raising_query_memory_limit_lets_previously_failing_query_succeed() {
	let t = TestEngine::new();
	insert_wide_rows(&t, 50);

	set_query_memory_limit(&t, 8);
	let err = t.query_err("FROM test::wide SORT {n}");
	assert!(err.contains("QUERY_006"), "expected QUERY_006, got: {}", err);

	// query_budget() reads the config key fresh at query start, so raising the limit
	// must let the very same query succeed without a restart.
	set_query_memory_limit(&t, 1024 * 1024 * 1024);
	let frames = t.query("FROM test::wide SORT {n}");
	assert_eq!(TestEngine::row_count(&frames), 50);
}