diff --git a/server/src/main.rs b/server/src/main.rs index 8ce6785..ea69c48 100644 --- a/server/src/main.rs +++ b/server/src/main.rs @@ -210,8 +210,14 @@ async fn handle_query(writer: &mut W, state: &SharedDbState, query: String, t } false }, - Response::TableCreated => true, - Response::IndexCreated => true, + Response::TableCreated => { + writer.write_command_complete(CompleteStatus::CreateTable).await?; + true + }, + Response::IndexCreated => { + writer.write_command_complete(CompleteStatus::CreateIndex).await?; + true + }, } }; diff --git a/server/src/proto_wrapper.rs b/server/src/proto_wrapper.rs index bd74ba6..03ce8f5 100644 --- a/server/src/proto_wrapper.rs +++ b/server/src/proto_wrapper.rs @@ -13,6 +13,8 @@ pub enum CompleteStatus { }, Delete(usize), Select(usize), + CreateTable, + CreateIndex, } impl CompleteStatus { @@ -21,6 +23,8 @@ impl CompleteStatus { CompleteStatus::Insert { oid, rows } => format!("INSERT {} {}", oid, rows), CompleteStatus::Delete(rows) => format!("DELETE {}", rows), CompleteStatus::Select(rows) => format!("SELECT {}", rows), + CompleteStatus::CreateTable => "CREATE TABLE".to_string(), + CompleteStatus::CreateIndex => "CREATE INDEX".to_string(), } } }