aljaz/turso
1
0
Fork 0
mirror of https://github.com/aljazceru/turso.git synced 2026-02-18 14:35:14 +01:00
Code Issues Packages Projects Releases Wiki Activity

refactor: remove PseudoTable from Table

Browse Source 
the only reason for `PseudoTable` to exist, is to provide column
information for `PseudoCursor` creation. this should not be part of the
schema.
This commit is contained in:
Levy A.
2025-06-30 13:50:26 -03:00
parent 3907b387b3
commit ffd6844b5b
11 changed files with 19 additions and 37 deletions
Download Patch File Download Diff File Expand all files Collapse all files

17
core/schema.rs
View File

@@ -136,7 +136,6 @@ impl Schema {
#[derive(Clone, Debug)]
pub enum Table {
BTree(Rc<BTreeTable>),
Pseudo(Rc<PseudoTable>),
Virtual(Rc<VirtualTable>),
FromClauseSubquery(FromClauseSubquery),
}
@@ -145,7 +144,6 @@ impl Table {
pub fn get_root_page(&self) -> usize {
match self {
Table::BTree(table) => table.root_page,
Table::Pseudo(_) => unimplemented!(),
Table::Virtual(_) => unimplemented!(),
Table::FromClauseSubquery(_) => unimplemented!(),
}
@@ -154,7 +152,6 @@ impl Table {
pub fn get_name(&self) -> &str {
match self {
Self::BTree(table) => &table.name,
Self::Pseudo(_) => "",
Self::Virtual(table) => &table.name,
Self::FromClauseSubquery(from_clause_subquery) => &from_clause_subquery.name,
}
@@ -163,7 +160,6 @@ impl Table {
pub fn get_column_at(&self, index: usize) -> Option<&Column> {
match self {
Self::BTree(table) => table.columns.get(index),
Self::Pseudo(table) => unimplemented!(),
Self::Virtual(table) => table.columns.get(index),
Self::FromClauseSubquery(from_clause_subquery) => {
from_clause_subquery.columns.get(index)
@@ -174,7 +170,6 @@ impl Table {
pub fn columns(&self) -> &Vec<Column> {
match self {
Self::BTree(table) => &table.columns,
Self::Pseudo(table) => unimplemented!(),
Self::Virtual(table) => &table.columns,
Self::FromClauseSubquery(from_clause_subquery) => &from_clause_subquery.columns,
}
@@ -183,7 +178,6 @@ impl Table {
pub fn btree(&self) -> Option<Rc<BTreeTable>> {
match self {
Self::BTree(table) => Some(table.clone()),
Self::Pseudo(_) => None,
Self::Virtual(_) => None,
Self::FromClauseSubquery(_) => None,
}
@@ -201,7 +195,6 @@ impl PartialEq for Table {
fn eq(&self, other: &Self) -> bool {
match (self, other) {
(Self::BTree(a), Self::BTree(b)) => Rc::ptr_eq(a, b),
(Self::Pseudo(a), Self::Pseudo(b)) => Rc::ptr_eq(a, b),
(Self::Virtual(a), Self::Virtual(b)) => Rc::ptr_eq(a, b),
_ => false,
}
@@ -291,12 +284,12 @@ impl BTreeTable {
}
}
#[derive(Debug, Default)]
pub struct PseudoTable {
#[derive(Debug, Default, Clone, Copy)]
pub struct PseudoCursorType {
pub column_count: usize,
}
impl PseudoTable {
impl PseudoCursorType {
pub fn new() -> Self {
Self { column_count: 0 }
}
@@ -557,8 +550,8 @@ fn create_table(
})
}
pub fn _build_pseudo_table(columns: &[ResultColumn]) -> PseudoTable {
let table = PseudoTable::new();
pub fn _build_pseudo_table(columns: &[ResultColumn]) -> PseudoCursorType {
let table = PseudoCursorType::new();
for column in columns {
match column {
ResultColumn::Expr(expr, _as_name) => {

2
core/translate/display.rs
View File

@@ -321,7 +321,7 @@ impl ToSqlString for JoinedTable {
) -> String {
let table_or_subquery =
match &self.table {
Table::BTree(..) | Table::Pseudo(..) | Table::Virtual(..) => {
Table::BTree(..) | Table::Virtual(..) => {
self.table.get_name().to_string()
}
Table::FromClauseSubquery(from_clause_subquery) => {

1
core/translate/expr.rs
View File

@@ -1930,7 +1930,6 @@ pub fn translate_expr(
});
Ok(target_register)
}
Table::Pseudo(_) => panic!("Column access on pseudo table"),
}
}
ast::Expr::RowId {

10
core/translate/group_by.rs
View File

@@ -1,12 +1,10 @@
use std::rc::Rc;
use turso_sqlite3_parser::ast;
use crate::translate::expr::{walk_expr, WalkControl};
use crate::translate::plan::ResultSetColumn;
use crate::{
function::AggFunc,
schema::PseudoTable,
schema::PseudoCursorType,
translate::collate::CollationSeq,
util::exprs_are_equivalent,
vdbe::{
@@ -306,11 +304,9 @@ pub fn group_by_create_pseudo_table(
) -> usize {
// Create a pseudo-table to read one row at a time from the sorter
// This allows us to use standard table access operations on the sorted data
let pseudo_table = Rc::new(PseudoTable {
program.alloc_cursor_id(CursorType::Pseudo(PseudoCursorType {
column_count: sorter_column_count,
});
program.alloc_cursor_id(CursorType::Pseudo(pseudo_table.clone()))
}))
}
/// In case sorting is needed for GROUP BY, sorts the rows in the GROUP BY sorter

7
core/translate/index.rs
View File

@@ -2,7 +2,7 @@ use std::sync::Arc;
use crate::vdbe::insn::CmpInsFlags;
use crate::{
schema::{BTreeTable, Column, Index, IndexColumn, PseudoTable, Schema},
schema::{BTreeTable, Column, Index, IndexColumn, PseudoCursorType, Schema},
storage::pager::CreateBTreeFlags,
util::normalize_ident,
vdbe::{
@@ -83,8 +83,9 @@ pub fn translate_create_index(
let btree_cursor_id = program.alloc_cursor_id(CursorType::BTreeIndex(idx.clone()));
let table_cursor_id = program.alloc_cursor_id(CursorType::BTreeTable(tbl.clone()));
let sorter_cursor_id = program.alloc_cursor_id(CursorType::Sorter);
let pseudo_table = PseudoTable::new_with_columns(tbl.columns.clone());
let pseudo_cursor_id = program.alloc_cursor_id(CursorType::Pseudo(pseudo_table.into()));
let pseudo_cursor_id = program.alloc_cursor_id(CursorType::Pseudo(PseudoCursorType {
column_count: tbl.columns.len(),
}));
// Create a new B-Tree and store the root page index in a register
let root_page_reg = program.alloc_register();

2
core/translate/main_loop.rs
View File

@@ -563,7 +563,6 @@ pub fn open_loop(
end_offset: loop_end,
});
}
Table::Pseudo(_) => panic!("Pseudo tables should not loop"),
}
if let Some(table_cursor_id) = table_cursor_id {
@@ -1042,7 +1041,6 @@ pub fn close_loop(
target_pc: loop_labels.loop_start,
});
}
other => unreachable!("Unsupported table reference type: {:?}", other),
}
program.preassign_label_to_next_insn(loop_labels.loop_end);
}

9
core/translate/order_by.rs
View File

@@ -1,9 +1,7 @@
use std::rc::Rc;
use turso_sqlite3_parser::ast::{self, SortOrder};
use crate::{
schema::PseudoTable,
schema::PseudoCursorType,
translate::collate::CollationSeq,
util::exprs_are_equivalent,
vdbe::{
@@ -95,10 +93,9 @@ pub fn emit_order_by(
.map(|v| v.len())
.unwrap_or(0);
let pseudo_table = Rc::new(PseudoTable {
let pseudo_cursor = program.alloc_cursor_id(CursorType::Pseudo(PseudoCursorType {
column_count: sorter_column_count,
});
let pseudo_cursor = program.alloc_cursor_id(CursorType::Pseudo(pseudo_table.clone()));
}));
let SortMetadata {
sort_cursor,
reg_sorter_data,

1
core/translate/plan.rs
View File

@@ -989,7 +989,6 @@ impl JoinedTable {
let index_cursor_id = None;
Ok((table_cursor_id, index_cursor_id))
}
Table::Pseudo(_) => Ok((None, None)),
Table::FromClauseSubquery(..) => Ok((None, None)),
}
}

1
core/translate/schema.rs
View File

@@ -749,7 +749,6 @@ pub fn translate_drop_table(
db: 0, // TODO change this for multiple databases
});
}
Table::Pseudo(..) => unimplemented!(),
Table::FromClauseSubquery(..) => panic!("FromClauseSubquery can't be dropped"),
};

4
core/vdbe/builder.rs
View File

@@ -6,7 +6,7 @@ use turso_sqlite3_parser::ast::{self, TableInternalId};
use crate::{
numeric::Numeric,
parameters::Parameters,
schema::{BTreeTable, Index, PseudoTable, Table},
schema::{BTreeTable, Index, PseudoCursorType, Table},
translate::{
collate::CollationSeq,
emitter::TransactionMode,
@@ -116,7 +116,7 @@ pub struct ProgramBuilder {
pub enum CursorType {
BTreeTable(Rc<BTreeTable>),
BTreeIndex(Arc<Index>),
Pseudo(Rc<PseudoTable>),
Pseudo(PseudoCursorType),
Sorter,
VirtualTable(Rc<VirtualTable>),
}

2
core/vdbe/explain.rs
View File

@@ -539,7 +539,7 @@ pub fn insn_to_str(
let name = &index.columns.get(*column).unwrap().name;
Some(name)
}
CursorType::Pseudo(pseudo_table) => None,
CursorType::Pseudo(_) => None,
CursorType::Sorter => None,
CursorType::VirtualTable(v) => v.columns.get(*column).unwrap().name.as_ref(),
};