Plan checker sessionPass ConnectorSession to PlanChecker

presto-hive/src/test/java/com/facebook/presto/hive/AbstractTestHiveClient.java

@@ -1235,6 +1235,12 @@ public RuntimeStats getRuntimeStats()
             {
                 return session.getRuntimeStats();
             }
+
+            @Override
+            public ConnectorSession forConnectorId(ConnectorId connectorId)
+            {
+                return this;
+            }
         };
     }
 

presto-main/src/main/java/com/facebook/presto/FullConnectorSession.java

@@ -199,4 +199,10 @@ public RuntimeStats getRuntimeStats()
     {
         return session.getRuntimeStats();
     }
+
+    @Override
+    public ConnectorSession forConnectorId(ConnectorId connectorId)
+    {
+        return new FullConnectorSession(session, identity);
+    }
 }

presto-main/src/main/java/com/facebook/presto/sql/planner/sanity/PlanChecker.java

@@ -87,7 +87,7 @@ public void validateFinalPlan(PlanNode planNode, Session session, Metadata metad
         checkers.get(Stage.FINAL).forEach(checker -> checker.validate(planNode, session, metadata, warningCollector));
         for (PlanCheckerProvider provider : planCheckerProviderManager.getPlanCheckerProviders()) {
             for (com.facebook.presto.spi.plan.PlanChecker checker : provider.getFinalPlanCheckers()) {
-                checker.validate(planNode, warningCollector);
+                checker.validate(planNode, warningCollector, session.toConnectorSession());
             }
         }
     }
@@ -97,7 +97,7 @@ public void validateIntermediatePlan(PlanNode planNode, Session session, Metadat
         checkers.get(Stage.INTERMEDIATE).forEach(checker -> checker.validate(planNode, session, metadata, warningCollector));
         for (PlanCheckerProvider provider : planCheckerProviderManager.getPlanCheckerProviders()) {
             for (com.facebook.presto.spi.plan.PlanChecker checker : provider.getIntermediatePlanCheckers()) {
-                checker.validate(planNode, warningCollector);
+                checker.validate(planNode, warningCollector, session.toConnectorSession());
             }
         }
     }
@@ -107,15 +107,18 @@ public void validatePlanFragment(PlanFragment planFragment, Session session, Met
         checkers.get(Stage.FRAGMENT).forEach(checker -> checker.validateFragment(planFragment, session, metadata, warningCollector));
         for (PlanCheckerProvider provider : planCheckerProviderManager.getPlanCheckerProviders()) {
             for (com.facebook.presto.spi.plan.PlanChecker checker : provider.getFragmentPlanCheckers()) {
-                checker.validateFragment(new SimplePlanFragment(
-                        planFragment.getId(),
-                        planFragment.getRoot(),
-                        planFragment.getVariables(),
-                        planFragment.getPartitioning(),
-                        planFragment.getTableScanSchedulingOrder(),
-                        planFragment.getPartitioningScheme(),
-                        planFragment.getStageExecutionDescriptor(),
-                        planFragment.isOutputTableWriterFragment()), warningCollector);
+                checker.validateFragment(
+                        new SimplePlanFragment(
+                                planFragment.getId(),
+                                planFragment.getRoot(),
+                                planFragment.getVariables(),
+                                planFragment.getPartitioning(),
+                                planFragment.getTableScanSchedulingOrder(),
+                                planFragment.getPartitioningScheme(),
+                                planFragment.getStageExecutionDescriptor(),
+                                planFragment.isOutputTableWriterFragment()),
+                        warningCollector,
+                        session.toConnectorSession());
             }
         }
     }

presto-main/src/main/java/com/facebook/presto/testing/TestingConnectorSession.java

@@ -17,6 +17,7 @@
 import com.facebook.presto.common.function.SqlFunctionProperties;
 import com.facebook.presto.common.type.TimeZoneKey;
 import com.facebook.presto.execution.QueryIdGenerator;
+import com.facebook.presto.spi.ConnectorId;
 import com.facebook.presto.spi.ConnectorSession;
 import com.facebook.presto.spi.PrestoException;
 import com.facebook.presto.spi.WarningCollector;
@@ -224,6 +225,26 @@ public RuntimeStats getRuntimeStats()
         return new RuntimeStats();
     }
 
+    @Override
+    public ConnectorSession forConnectorId(ConnectorId connectorId)
+    {
+        return new TestingConnectorSession(
+                sqlFunctionProperties.getSessionUser(),
+                identity,
+                source,
+                traceToken,
+                timeZoneKey,
+                locale,
+                startTime,
+                ImmutableList.copyOf(properties.values()),
+                propertyValues,
+                sqlFunctionProperties.isLegacyRowFieldOrdinalAccessEnabled(),
+                clientInfo,
+                clientTags,
+                schema,
+                sessionFunctions);
+    }
+
     @Override
     public String toString()
     {

presto-native-sidecar-plugin/src/main/java/com/facebook/presto/sidecar/nativechecker/NativePlanChecker.java

@@ -16,6 +16,7 @@
 import com.facebook.airlift.json.JsonCodec;
 import com.facebook.airlift.log.Logger;
 import com.facebook.presto.spi.ConnectorId;
+import com.facebook.presto.spi.ConnectorSession;
 import com.facebook.presto.spi.NodeManager;
 import com.facebook.presto.spi.PrestoException;
 import com.facebook.presto.spi.TableHandle;
@@ -61,13 +62,13 @@ public NativePlanChecker(NodeManager nodeManager, JsonCodec<SimplePlanFragment>
     }
 
     @Override
-    public void validate(PlanNode planNode, WarningCollector warningCollector)
+    public void validate(PlanNode planNode, WarningCollector warningCollector, ConnectorSession session)
     {
         // NO-OP, only validating fragments
     }
 
     @Override
-    public void validateFragment(SimplePlanFragment planFragment, WarningCollector warningCollector)
+    public void validateFragment(SimplePlanFragment planFragment, WarningCollector warningCollector, ConnectorSession session)
     {
         if (!planFragment.getPartitioning().isCoordinatorOnly() && !isInternalSystemConnector(planFragment.getRoot())) {
             runValidation(planFragment);

presto-native-sidecar-plugin/src/test/java/com/facebook/presto/sidecar/TestPlanCheckerProvider.java

@@ -93,15 +93,15 @@ public void testNativePlanMockValidate()
             PlanConversionResponse responseOk = new PlanConversionResponse(ImmutableList.of());
             String responseOkString = PLAN_CONVERSION_RESPONSE_JSON_CODEC.toJson(responseOk);
             server.enqueue(new MockResponse().setBody(responseOkString));
-            checker.validateFragment(fragment, null);
+            checker.validateFragment(fragment, null, null);
 
             String errorMessage = "native conversion error";
             ErrorCode errorCode = StandardErrorCode.NOT_SUPPORTED.toErrorCode();
             PlanConversionResponse responseError = new PlanConversionResponse(ImmutableList.of(new PlanConversionFailureInfo("MockError", errorMessage, null, ImmutableList.of(), ImmutableList.of(), errorCode)));
             String responseErrorString = PLAN_CONVERSION_RESPONSE_JSON_CODEC.toJson(responseError);
             server.enqueue(new MockResponse().setResponseCode(500).setBody(responseErrorString));
             PrestoException error = expectThrows(PrestoException.class,
-                    () -> checker.validateFragment(fragment, null));
+                    () -> checker.validateFragment(fragment, null, null));
             assertEquals(error.getErrorCode(), errorCode);
             assertTrue(error.getMessage().contains(errorMessage));
         }

presto-spi/src/main/java/com/facebook/presto/spi/ConnectorSession.java

@@ -66,4 +66,10 @@ default boolean isReadConstraints()
     WarningCollector getWarningCollector();
 
     RuntimeStats getRuntimeStats();
+
+    /**
+     * returns a ConnectorSession for a specific ConnectorId
+     * @return
+     */
+    ConnectorSession forConnectorId(ConnectorId connectorId);
 }

presto-spi/src/main/java/com/facebook/presto/spi/plan/PlanChecker.java

@@ -14,14 +14,15 @@
 
 package com.facebook.presto.spi.plan;
 
+import com.facebook.presto.spi.ConnectorSession;
 import com.facebook.presto.spi.WarningCollector;
 
 public interface PlanChecker
 {
-    void validate(PlanNode planNode, WarningCollector warningCollector);
+    void validate(PlanNode planNode, WarningCollector warningCollector, ConnectorSession session);
 
-    default void validateFragment(SimplePlanFragment planFragment, WarningCollector warningCollector)
+    default void validateFragment(SimplePlanFragment planFragment, WarningCollector warningCollector, ConnectorSession session)
     {
-        validate(planFragment.getRoot(), warningCollector);
+        validate(planFragment.getRoot(), warningCollector, session);
     }
 }

presto-spi/src/test/java/com/facebook/presto/spi/TestingSession.java

@@ -130,6 +130,12 @@ public RuntimeStats getRuntimeStats()
         {
             return new RuntimeStats();
         }
+
+        @Override
+        public ConnectorSession forConnectorId(ConnectorId connectorId)
+        {
+            return this;
+        }
     };
 
     private TestingSession() {}

presto-tests/src/test/java/com/facebook/presto/execution/TriggerFailurePlanChecker.java

@@ -13,6 +13,7 @@
  */
 package com.facebook.presto.execution;
 
+import com.facebook.presto.spi.ConnectorSession;
 import com.facebook.presto.spi.PrestoException;
 import com.facebook.presto.spi.StandardErrorCode;
 import com.facebook.presto.spi.WarningCollector;
@@ -36,7 +37,7 @@ public TriggerFailurePlanChecker(AtomicBoolean triggerValidationFailure)
     }
 
     @Override
-    public void validate(PlanNode planNode, WarningCollector warningCollector)
+    public void validate(PlanNode planNode, WarningCollector warningCollector, ConnectorSession session)
     {
         if (triggerValidationFailure.get()) {
             throw new PrestoException(FAILURE_ERROR_CODE, FAILURE_MESSAGE);