diff --git a/core/trino-main/src/main/java/io/trino/operator/ExchangeOperator.java b/core/trino-main/src/main/java/io/trino/operator/ExchangeOperator.java index 2c1c42b8e0df..ad44dc66b8f5 100644 --- a/core/trino-main/src/main/java/io/trino/operator/ExchangeOperator.java +++ b/core/trino-main/src/main/java/io/trino/operator/ExchangeOperator.java @@ -70,9 +70,9 @@ public ExchangeOperatorFactory( ExchangeManagerRegistry exchangeManagerRegistry) { this.operatorId = operatorId; - this.sourceId = sourceId; - this.directExchangeClientSupplier = directExchangeClientSupplier; - this.serdeFactory = serdeFactory; + this.sourceId = requireNonNull(sourceId, "sourceId is null"); + this.directExchangeClientSupplier = requireNonNull(directExchangeClientSupplier, "directExchangeClientSupplier is null"); + this.serdeFactory = requireNonNull(serdeFactory, "serdeFactory is null"); this.retryPolicy = requireNonNull(retryPolicy, "retryPolicy is null"); this.exchangeManagerRegistry = requireNonNull(exchangeManagerRegistry, "exchangeManagerRegistry is null"); } diff --git a/core/trino-main/src/main/java/io/trino/sql/planner/LocalExecutionPlanner.java b/core/trino-main/src/main/java/io/trino/sql/planner/LocalExecutionPlanner.java index 2ec18558dd32..cbdf2c3203ac 100644 --- a/core/trino-main/src/main/java/io/trino/sql/planner/LocalExecutionPlanner.java +++ b/core/trino-main/src/main/java/io/trino/sql/planner/LocalExecutionPlanner.java @@ -508,7 +508,7 @@ public LocalExecutionPlanner( this.pageSourceManager = requireNonNull(pageSourceManager, "pageSourceManager is null"); this.indexManager = requireNonNull(indexManager, "indexManager is null"); this.nodePartitioningManager = requireNonNull(nodePartitioningManager, "nodePartitioningManager is null"); - this.directExchangeClientSupplier = directExchangeClientSupplier; + this.directExchangeClientSupplier = requireNonNull(directExchangeClientSupplier, "directExchangeClientSupplier is null"); this.pageSinkManager = requireNonNull(pageSinkManager, "pageSinkManager is null"); this.expressionCompiler = requireNonNull(expressionCompiler, "expressionCompiler is null"); this.pageFunctionCompiler = requireNonNull(pageFunctionCompiler, "pageFunctionCompiler is null"); diff --git a/core/trino-main/src/main/java/io/trino/testing/PlanTester.java b/core/trino-main/src/main/java/io/trino/testing/PlanTester.java index 1b1a226da4e9..28a3992337d0 100644 --- a/core/trino-main/src/main/java/io/trino/testing/PlanTester.java +++ b/core/trino-main/src/main/java/io/trino/testing/PlanTester.java @@ -745,7 +745,9 @@ private List createDrivers(Session session, @Language("SQL") String sql) indexManager, nodePartitioningManager, pageSinkManager, - null, + (_, _, _, _, _, _) -> { + throw new UnsupportedOperationException(); + }, expressionCompiler, pageFunctionCompiler, joinFilterFunctionCompiler,