fix

Author: Sean Loiselle

src/adapter/src/client.rs

@@ -537,6 +537,36 @@ impl SessionClient {
         let mut results = vec![];
 
         for stmt in stmts {
+            let execute_responses = Plan::generated_from((&stmt).into())
+                .into_iter()
+                .map(ExecuteResponse::generated_from)
+                .flatten()
+                .collect::<Vec<_>>();
+
+            if execute_responses.iter().any(|execute_response| {
+                matches!(
+                    execute_response,
+                    ExecuteResponseKind::Fetch
+                        | ExecuteResponseKind::SetVariable
+                        | ExecuteResponseKind::Tailing
+                        | ExecuteResponseKind::CopyTo
+                        | ExecuteResponseKind::CopyFrom
+                        | ExecuteResponseKind::Raise
+                        | ExecuteResponseKind::DeclaredCursor
+                        | ExecuteResponseKind::ClosedCursor
+                )
+            }) && !matches!(
+                stmt,
+                // Both `SelectStatement` and `CopyStatement` generate
+                // `PeekPlan`, but `SELECT` should be permitted and `COPY` not.
+                Statement::Select(SelectStatement { query: _, as_of: _ })
+            ) {
+                results.push(SimpleResult::err(format!(
+                    "unsupported via this API: {}",
+                    stmt.to_ast_string()
+                )));
+            }
+
             if matches!(self.session().transaction(), TransactionStatus::Failed(_)) {
                 break;
             }

src/sql/src/plan.rs

@@ -147,7 +147,7 @@ impl Plan {
     /// Expresses which [`StatementKind`] can generate which set of
     /// [`PlanKind`].
     pub fn generated_from(stmt: StatementKind) -> Vec<PlanKind> {
-        match stmt {
+        match dbg!(stmt) {
             StatementKind::AlterConnection => vec![PlanKind::AlterNoop, PlanKind::RotateKeys],
             StatementKind::AlterIndex => vec![
                 PlanKind::AlterIndexResetOptions,
@@ -168,7 +168,12 @@ impl Plan {
             StatementKind::AlterSystemSet => vec![PlanKind::AlterNoop, PlanKind::AlterSystemSet],
             StatementKind::Close => vec![PlanKind::Close],
             StatementKind::Commit => vec![PlanKind::CommitTransaction],
-            StatementKind::Copy => vec![PlanKind::CopyFrom, PlanKind::Peek, PlanKind::SendDiffs],
+            StatementKind::Copy => vec![
+                PlanKind::CopyFrom,
+                PlanKind::Peek,
+                PlanKind::SendDiffs,
+                PlanKind::Tail,
+            ],
             StatementKind::CreateCluster => vec![PlanKind::CreateComputeInstance],
             StatementKind::CreateClusterReplica => vec![PlanKind::CreateComputeInstanceReplica],
             StatementKind::CreateConnection => vec![PlanKind::CreateConnection],
@@ -207,6 +212,7 @@ impl Plan {
             StatementKind::SetVariable => vec![PlanKind::SetVariable],
             StatementKind::Show => vec![
                 PlanKind::Peek,
+                PlanKind::SendRows,
                 PlanKind::ShowVariable,
                 PlanKind::ShowAllVariables,
             ],