diff --git a/axiom/connectors/hive/tests/LocalHiveConnectorMetadataTest.cpp b/axiom/connectors/hive/tests/LocalHiveConnectorMetadataTest.cpp index 245df04c8..1fb2139b0 100644 --- a/axiom/connectors/hive/tests/LocalHiveConnectorMetadataTest.cpp +++ b/axiom/connectors/hive/tests/LocalHiveConnectorMetadataTest.cpp @@ -142,12 +142,10 @@ class LocalHiveConnectorMetadataTest /*explain=*/false); auto builder = exec::test::PlanBuilder().values({values}); - auto insertHandle = std::make_shared( - velox::exec::test::kHiveConnectorId, handle->veloxHandle()); auto plan = builder.startTableWriter() .outputDirectoryPath(outputPath) .targetColumns(table->type()) - .insertHandle(insertHandle) + .insertHandle(handle->veloxHandle()) .fileFormat(format) .endTableWriter() .planNode(); diff --git a/axiom/optimizer/ToVelox.cpp b/axiom/optimizer/ToVelox.cpp index dc1713e3b..ef2264285 100644 --- a/axiom/optimizer/ToVelox.cpp +++ b/axiom/optimizer/ToVelox.cpp @@ -2319,7 +2319,9 @@ velox::core::PlanNodePtr ToVelox::makeWrite( VELOX_CHECK(!finishWrite_, "Only single TableWrite per query supported"); auto insertTableHandle = std::make_shared( - connectorId, handle->veloxHandle()); + connectorId, + handle->veloxHandle(), + /*notNullColumns=*/folly::F14FastSet{}); finishWrite_ = { metadata, connectorId, diff --git a/axiom/optimizer/tests/TestConnectorQueryTest.cpp b/axiom/optimizer/tests/TestConnectorQueryTest.cpp index 5b9e4800b..a1d3481a9 100644 --- a/axiom/optimizer/tests/TestConnectorQueryTest.cpp +++ b/axiom/optimizer/tests/TestConnectorQueryTest.cpp @@ -47,7 +47,8 @@ class TestConnectorQueryTest : public QueryTestBase, auto handle = std::make_shared( kTestConnectorId, std::make_shared(SchemaTableName{ - std::string(connector::TestConnector::kDefaultSchema), tableName})); + std::string(connector::TestConnector::kDefaultSchema), tableName}), + /*notNullColumns=*/folly::F14FastSet{}); auto write = std::make_shared( "writenodeid", source->outputType(), diff --git a/axiom/optimizer/v2/EmitPass.cpp b/axiom/optimizer/v2/EmitPass.cpp index 81f74fed0..e816b3ba0 100644 --- a/axiom/optimizer/v2/EmitPass.cpp +++ b/axiom/optimizer/v2/EmitPass.cpp @@ -2092,7 +2092,9 @@ velox::core::PlanNodePtr Emitter::emitTableWrite(const TableWrite& tableWrite) { VELOX_CHECK(!finishWrite_, "Only one TableWrite per query is supported"); auto insertTableHandle = std::make_shared( - connectorId, handle->veloxHandle()); + connectorId, + handle->veloxHandle(), + /*notNullColumns=*/folly::F14FastSet{}); finishWrite_ = FinishWrite{ metadata, connectorId,