cleanup

Author: alamb

datafusion/sql/src/statement.rs

@@ -16,7 +16,7 @@
 // under the License.
 
 use crate::parser::{
-    CopyToSource, CopyToStatement, CreateExternalTable, DFParser, DescribeTableStmt,
+    CopyToStatement, CreateExternalTable, DFParser, DescribeTableStmt,
     Statement as DFStatement,
 };
 use crate::planner::{
@@ -34,7 +34,7 @@ use datafusion_expr::logical_plan::builder::project;
 use datafusion_expr::logical_plan::DdlStatement;
 use datafusion_expr::utils::expr_to_columns;
 use datafusion_expr::{
-    cast, col, Analyze, CopyTo, CreateCatalog, CreateCatalogSchema,
+    cast, col, Analyze, CreateCatalog, CreateCatalogSchema,
     CreateExternalTable as PlanCreateExternalTable, CreateMemoryTable, CreateView,
     DescribeTable, DmlStatement, DropCatalogSchema, DropTable, DropView, EmptyRelation,
     Explain, ExprSchemable, Filter, LogicalPlan, LogicalPlanBuilder, PlanType, Prepare,
@@ -539,51 +539,11 @@ impl<'a, S: ContextProvider> SqlToRel<'a, S> {
         }))
     }
 
-    fn copy_to_plan(&self, statement: CopyToStatement) -> Result<LogicalPlan> {
-        let CopyToStatement {
-            source,
-            target,
-            options,
-        } = statement;
-
-        let input = match source {
-            CopyToSource::Relation(table_name) => {
-                let table_ref = self.object_name_to_table_reference(table_name)?;
-                let provider =
-                    self.schema_provider.get_table_provider(table_ref.clone())?;
-                let projection = None;
-                LogicalPlanBuilder::scan(table_ref, provider, projection)?.build()
-            }
-            CopyToSource::Query(query) => {
-                let mut planner_context = PlannerContext::new();
-                self.query_to_plan(query, &mut planner_context)
-            }
-        }?;
-
-        // convert options
-        let options = options
-            .into_iter()
-            .map(|(k, v)| {
-                let scalar = match self.parse_value(v, &[])? {
-                    datafusion_expr::Expr::Literal(scalar) => Ok(scalar),
-                    expr => Err(DataFusionError::Plan(format!(
-                        "COPY options expects a literal for {k}, got {expr}"
-                    ))),
-                }?;
-                Ok((k, scalar))
-            })
-            // check for errors
-            .collect::<Result<Vec<_>>>()?
-            // collect into hashMap
-            .into_iter()
-            .collect::<HashMap<_, _>>();
-
-        Ok(LogicalPlan::CopyTo(CopyTo {
-            input: Arc::new(input),
-            target,
-            options,
-            dummy_schema: DFSchemaRef::new(DFSchema::empty()),
-        }))
+    fn copy_to_plan(&self, _statement: CopyToStatement) -> Result<LogicalPlan> {
+        // TODO: implement as part of https://github.com/apache/arrow-datafusion/issues/5654
+        Err(DataFusionError::NotImplemented(
+            "`COPY .. TO ..` statement is yet supported".to_string(),
+        ))
     }
 
     fn build_order_by(