aljaz/turso
1
0
Fork 0
mirror of https://github.com/aljazceru/turso.git synced 2025-12-25 12:04:21 +01:00
Code Issues Packages Projects Releases Wiki Activity

support limit

Browse Source
This commit is contained in:
meteorgan
2025-06-16 23:03:04 +08:00
parent d4789d0a05
commit 1fcc2ddd90
2 changed files with 29 additions and 13 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
core/translate/compound_select.rs
View File

@@ -246,6 +246,7 @@ fn emit_compound_select(
&left_index,
right_cursor_id,
target_cursor_id,
limit_ctx,
);
}
_ => {
@@ -362,6 +363,7 @@ fn read_intersect_rows(
index: &Index,
right_cursor_id: usize,
target_cursor: Option<usize>,
limit_ctx: Option<LimitCtx>,
) {
let label_close = program.allocate_label();
let label_loop_start = program.allocate_label();
@@ -413,6 +415,12 @@ fn read_intersect_rows(
count: column_count,
});
}
if let Some(limit_ctx) = limit_ctx {
program.emit_insn(Insn::DecrJumpZero {
reg: limit_ctx.reg_limit,
target_pc: label_close,
});
}
program.preassign_label_to_next_insn(label_next);
program.emit_insn(Insn::Next {
cursor_id: left_cursor_id,

34
testing/select.test
View File

@@ -361,12 +361,6 @@ if {[info exists ::env(SQLITE_EXEC)] && ($::env(SQLITE_EXEC) eq "scripts/limbo-s
y|y}
}
select * from t UNION select * from u UNION select * from v UNION ALL select * from t;
} {x|x
y|y
x|x
y|y}
do_execsql_test_on_specific_db {:memory:} select-intersect-1 {
CREATE TABLE t(x TEXT, y TEXT);
CREATE TABLE u(x TEXT, y TEXT);
@@ -410,11 +404,25 @@ do_execsql_test_on_specific_db {:memory:} select-union-intersect {
select * from t UNION select * from u INTERSECT select * from v;
} {x|x}
#do_execsql_test_on_specific_db {:memory:} select-intersect-with-limit {
# CREATE TABLE t(x TEXT, y TEXT);
# CREATE TABLE u(x TEXT, y TEXT);
# INSERT INTO t VALUES('x','x'),('y','y'), ('z','z');
# INSERT INTO u VALUES('x','x'),('y','y'), ('z','z');
do_execsql_test_on_specific_db {:memory:} select-intersect-with-limit {
CREATE TABLE t(x TEXT, y TEXT);
CREATE TABLE u(x TEXT, y TEXT);
INSERT INTO t VALUES('x','x'),('y','y'), ('z','z');
INSERT INTO u VALUES('x','x'),('y','y'), ('z','z');
# select * from t INTERSECT select * from u limit 2;
#} {x|x}
select * from t INTERSECT select * from u limit 2;
} {x|x
y|y}
do_execsql_test_on_specific_db {:memory:} select-intersect-union-with-limit {
CREATE TABLE t(x TEXT, y TEXT);
CREATE TABLE u(x TEXT, y TEXT);
CREATE TABLE v(x TEXT, y TEXT);
INSERT INTO t VALUES('x','x'),('y','y'), ('z','z');
INSERT INTO u VALUES('d','d'),('e','e'), ('z','z');
INSERT INTO v VALUES('a','a'),('b','b');
select * from t INTERSECT select * from u UNION select * from v limit 3;
} {a|a
b|b
z|z}