use arrow_array::RecordBatch;
use reifydb_catalog::{
catalog::view::ViewToCreate,
store::{row_settings::create::create_row_settings, view::create::ViewStorage},
};
use reifydb_core::{
error::diagnostic::catalog::view_already_exists,
interface::catalog::{change::CatalogTrackViewChangeOperations, storage::StorageId},
row::RowSettings,
value::batch::single_row,
};
use reifydb_evaluate::stack::SymbolTable;
use reifydb_rql::nodes::{CompiledViewStorageKind, CreateDeferredViewNode};
use reifydb_transaction::transaction::{Transaction, admin::AdminTransaction};
use reifydb_value::{return_error, value::Value};
use super::{create_deferred_view_flow, extract_view_sort};
use crate::{Result, vm::services::Services};
pub(crate) fn create_deferred_view(
services: &Services,
txn: &mut AdminTransaction,
symbols: &SymbolTable,
plan: CreateDeferredViewNode,
) -> Result<RecordBatch> {
if let Some(view) = services.catalog.find_view_by_name(
&mut Transaction::Admin(txn),
plan.namespace.id(),
plan.view.text(),
)? {
if plan.if_not_exists {
return single_row([
("id", Value::Uint8(view.id().0)),
("namespace", Value::Utf8(plan.namespace.name().to_string())),
("view", Value::Utf8(plan.view.text().to_string())),
("created", Value::Boolean(false)),
]);
}
return_error!(view_already_exists(plan.view.clone(), plan.namespace.name(), view.name(),));
}
let storage = match &plan.storage_kind {
CompiledViewStorageKind::Table {
partition_by,
} => ViewStorage::Table {
partition_by: partition_by.clone(),
},
CompiledViewStorageKind::RingBuffer {
capacity,
partition_by,
} => ViewStorage::RingBuffer {
capacity: *capacity,
partition_by: partition_by.clone(),
},
CompiledViewStorageKind::Series {
key,
partition_by,
} => ViewStorage::Series {
key: key.clone(),
tag: None,
partition_by: partition_by.clone(),
},
};
let sort = extract_view_sort(&plan.as_clause, &plan.columns);
let result = services.catalog.create_deferred_view(
txn,
ViewToCreate {
name: plan.view.clone(),
namespace: plan.namespace.id(),
columns: plan.columns,
storage,
sort,
},
)?;
txn.track_view_created(result.clone())?;
if let Some(ttl) = &plan.ttl {
create_row_settings(
txn,
StorageId::View(result.id()),
&RowSettings {
ttl: Some(ttl.clone()),
persistent: plan.persistent,
},
)?;
}
create_deferred_view_flow(
&services.catalog,
&services.routines,
&services.operators,
txn,
symbols,
&result,
*plan.as_clause,
&services.ioc,
)?;
single_row([
("id", Value::Uint8(result.id().0)),
("namespace", Value::Utf8(plan.namespace.name().to_string())),
("view", Value::Utf8(plan.view.text().to_string())),
("created", Value::Boolean(true)),
])
}
#[cfg(test)]
pub mod tests {
use reifydb_test_harness::engine::create_test_admin_transaction_with_internal_shape;
use reifydb_value::{
params::Params,
value::{Value, column_view::ColumnView, frame::frame::Frame},
};
use crate::vm::{Admin, executor::Executor};
#[test]
fn test_create_view() {
let instance = Executor::testing();
let mut txn = create_test_admin_transaction_with_internal_shape();
let r = instance.admin(
&mut txn,
Admin {
rql: "CREATE NAMESPACE test_namespace",
params: Params::default(),
},
);
if let Some(e) = r.error {
panic!("{e:?}");
}
let r = instance.admin(
&mut txn,
Admin {
rql: "CREATE TABLE test_namespace::src { id: Int4 }",
params: Params::default(),
},
);
if let Some(e) = r.error {
panic!("{e:?}");
}
let r = instance.admin(
&mut txn,
Admin {
rql: "CREATE DEFERRED VIEW test_namespace::test_view { id: Int4 } AS { FROM test_namespace::src }",
params: Params::default(),
},
);
if let Some(e) = r.error {
panic!("{e:?}");
}
let frame = &r[0];
assert_eq!(value_at(frame, 0), Value::Uint8(16387));
assert_eq!(value_at(frame, 1), Value::Utf8("test_namespace".to_string()));
assert_eq!(value_at(frame, 2), Value::Utf8("test_view".to_string()));
assert_eq!(value_at(frame, 3), Value::Boolean(true));
let r = instance.admin(
&mut txn,
Admin {
rql: "CREATE DEFERRED VIEW test_namespace::test_view { id: Int4 } AS { FROM test_namespace::src }",
params: Params::default(),
},
);
assert!(r.is_err());
assert_eq!(r.error.unwrap().diagnostic().code, "CA_003");
}
#[test]
fn test_create_same_view_in_different_shape() {
let instance = Executor::testing();
let mut txn = create_test_admin_transaction_with_internal_shape();
let r = instance.admin(
&mut txn,
Admin {
rql: "CREATE NAMESPACE test_namespace",
params: Params::default(),
},
);
if let Some(e) = r.error {
panic!("{e:?}");
}
let r = instance.admin(
&mut txn,
Admin {
rql: "CREATE NAMESPACE another_shape",
params: Params::default(),
},
);
if let Some(e) = r.error {
panic!("{e:?}");
}
let r = instance.admin(
&mut txn,
Admin {
rql: "CREATE TABLE test_namespace::src { id: Int4 }",
params: Params::default(),
},
);
if let Some(e) = r.error {
panic!("{e:?}");
}
let r = instance.admin(
&mut txn,
Admin {
rql: "CREATE TABLE another_shape::src { id: Int4 }",
params: Params::default(),
},
);
if let Some(e) = r.error {
panic!("{e:?}");
}
let r = instance.admin(
&mut txn,
Admin {
rql: "CREATE DEFERRED VIEW test_namespace::test_view { id: Int4 } AS { FROM test_namespace::src }",
params: Params::default(),
},
);
if let Some(e) = r.error {
panic!("{e:?}");
}
let frame = &r[0];
assert_eq!(value_at(frame, 0), Value::Uint8(16388));
assert_eq!(value_at(frame, 1), Value::Utf8("test_namespace".to_string()));
assert_eq!(value_at(frame, 2), Value::Utf8("test_view".to_string()));
assert_eq!(value_at(frame, 3), Value::Boolean(true));
let r = instance.admin(
&mut txn,
Admin {
rql: "CREATE DEFERRED VIEW another_shape::test_view { id: Int4 } AS { FROM another_shape::src }",
params: Params::default(),
},
);
if let Some(e) = r.error {
panic!("{e:?}");
}
let frame = &r[0];
assert_eq!(value_at(frame, 0), Value::Uint8(16389));
assert_eq!(value_at(frame, 1), Value::Utf8("another_shape".to_string()));
assert_eq!(value_at(frame, 2), Value::Utf8("test_view".to_string()));
assert_eq!(value_at(frame, 3), Value::Boolean(true));
}
fn value_at(frame: &Frame, column: usize) -> Value {
ColumnView::try_from((frame.batch.column(column), frame.batch.schema_ref().field(column)))
.unwrap()
.get_value(0)
}
}