[DQ] Reuse dq validation states (#10222)

Author: rvu1024

ydb/library/yql/providers/common/transform/yql_lazy_init.h

@@ -26,6 +26,10 @@ class TLazyInitHolder
         return Value.Get();
     }
 
+    inline explicit operator bool() const noexcept {
+        return !!Value.Get();
+    }
+
 private:
     TFactory Factory;
     mutable THolder<T> Value;

ydb/library/yql/providers/dq/provider/ya.make

@@ -35,6 +35,7 @@ PEERDIR(
     library/cpp/yson
     ydb/public/lib/yson_value
     ydb/public/sdk/cpp/client/ydb_driver
+    ydb/library/yql/ast
     ydb/library/yql/core
     ydb/library/yql/core/issue
     ydb/library/yql/utils/backtrace
@@ -60,6 +61,7 @@ PEERDIR(
     ydb/library/yql/providers/dq/actors
     ydb/library/yql/providers/result/expr_nodes
     ydb/library/yql/minikql
+    ydb/library/yql/public/issue
 )
 
 YQL_LAST_ABI_VERSION()

ydb/library/yql/providers/dq/provider/yql_dq_datasource.cpp

@@ -51,6 +51,7 @@ class TDqDataProviderSource: public TDataProviderBase {
         , TypeAnnotationTransformer_([] () { return CreateDqsDataSourceTypeAnnotationTransformer(); })
         , ConstraintsTransformer_([] () { return CreateDqDataSourceConstraintTransformer(); })
         , StatisticsTransformer_([this]() { return CreateDqsStatisticsTransformer(State_, TBaseProviderContext::Instance()); })
+        , ExecutionValidator_([this]() { return MakeHolder<TDqExecutionValidator>(State_); })
     { }
 
     TStringBuf GetName() const override {
@@ -262,7 +263,7 @@ class TDqDataProviderSource: public TDataProviderBase {
     }
 
     bool ValidateExecution(const TExprNode& node, TExprContext& ctx) override {
-        return ValidateDqExecution(node, *State_->TypeCtx, ctx, State_);
+        return ExecutionValidator_->ValidateDqExecution(node, ctx);
     }
 
     bool CanParse(const TExprNode& node) override {
@@ -276,9 +277,16 @@ class TDqDataProviderSource: public TDataProviderBase {
     void Reset() final {
         if (ExecTransformer_) {
             ExecTransformer_->Rewind();
+        }
+        if (TypeAnnotationTransformer_) {
             TypeAnnotationTransformer_->Rewind();
+        }
+        if (ConstraintsTransformer_) {
             ConstraintsTransformer_->Rewind();
         }
+        if (ExecutionValidator_) {
+            ExecutionValidator_->Rewind();
+        }
     }
 
 private:
@@ -288,6 +296,7 @@ class TDqDataProviderSource: public TDataProviderBase {
     TLazyInitHolder<TVisitorTransformerBase> TypeAnnotationTransformer_;
     TLazyInitHolder<IGraphTransformer> ConstraintsTransformer_;
     TLazyInitHolder<IGraphTransformer> StatisticsTransformer_;
+    TLazyInitHolder<TDqExecutionValidator> ExecutionValidator_;
 };
 
 }