use {
super::SharedMemoryStorage,
async_trait::async_trait,
gluesql_core::{ast::ColumnDef, result::MutResult, store::AlterTable},
memory_storage::MemoryStorage,
std::sync::Arc,
};
#[async_trait(?Send)]
impl AlterTable for SharedMemoryStorage {
async fn rename_schema(self, table_name: &str, new_table_name: &str) -> MutResult<Self, ()> {
let database = Arc::clone(&self.database);
let mut database = database.write().await;
if let Err(error) = MemoryStorage::rename_schema(&mut database, table_name, new_table_name)
{
return Err((self, error));
}
Ok((self, ()))
}
async fn rename_column(
self,
table_name: &str,
old_column_name: &str,
new_column_name: &str,
) -> MutResult<Self, ()> {
let database = Arc::clone(&self.database);
let mut database = database.write().await;
if let Err(error) = MemoryStorage::rename_column(
&mut database,
table_name,
old_column_name,
new_column_name,
) {
return Err((self, error));
}
Ok((self, ()))
}
async fn add_column(self, table_name: &str, column_def: &ColumnDef) -> MutResult<Self, ()> {
let database = Arc::clone(&self.database);
let mut database = database.write().await;
if let Err(error) = MemoryStorage::add_column(&mut database, table_name, column_def) {
return Err((self, error));
}
Ok((self, ()))
}
async fn drop_column(
self,
table_name: &str,
column_name: &str,
if_exists: bool,
) -> MutResult<Self, ()> {
let database = Arc::clone(&self.database);
let mut database = database.write().await;
if let Err(error) =
MemoryStorage::drop_column(&mut database, table_name, column_name, if_exists)
{
return Err((self, error));
}
Ok((self, ()))
}
}