Fix sql flags for module resolver in query replay (#10405)

Author: nepal

ydb/library/yql/core/facade/yql_facade.cpp

@@ -607,6 +607,30 @@ void TProgram::HandleSourceCode(TString& sourceCode) {
     }
 }
 
+namespace {
+
+THashSet<TString> ExtractSqlFlags(const NYT::TNode& dataNode) {
+    THashSet<TString> result;
+    for (const auto& f : dataNode["SqlFlags"].AsList()) {
+        result.insert(f.AsString());
+    }
+    return result;
+}
+
+} // namespace
+
+void UpdateSqlFlagsFromQContext(const TQContext& qContext, THashSet<TString>& flags) {
+    if (qContext.CanRead()) {
+        auto loaded = qContext.GetReader()->Get({FacadeComponent, TranslationLabel}).GetValueSync();
+        if (!loaded) {
+            return;
+        }
+
+        auto dataNode = NYT::NodeFromYsonString(loaded->Value);
+        flags = ExtractSqlFlags(dataNode);
+    }
+}
+
 void TProgram::HandleTranslationSettings(NSQLTranslation::TTranslationSettings& loadedSettings,
     const NSQLTranslation::TTranslationSettings*& currentSettings)
 {
@@ -643,11 +667,7 @@ void TProgram::HandleTranslationSettings(NSQLTranslation::TTranslationSettings&
             loadedSettings.ClusterMapping[c.first] = c.second.AsString();
         }
 
-        loadedSettings.Flags.clear();
-        for (const auto& f : dataNode["SqlFlags"].AsList()) {
-            loadedSettings.Flags.insert(f.AsString());
-        }
-    
+        loadedSettings.Flags = ExtractSqlFlags(dataNode);
         loadedSettings.V0Behavior = (NSQLTranslation::EV0Behavior)dataNode["V0Behavior"].AsUint64();
         loadedSettings.V0WarnAsError = NSQLTranslation::ISqlFeaturePolicy::Make(dataNode["V0WarnAsError"].AsBool());
         loadedSettings.DqDefaultAuto = NSQLTranslation::ISqlFeaturePolicy::Make(dataNode["DqDefaultAuto"].AsBool());

ydb/library/yql/core/facade/yql_facade.h

@@ -446,4 +446,6 @@ class TProgram: public TThrRefBase, private TNonCopyable
     TIssues FinalIssues_;
 };
 
+void UpdateSqlFlagsFromQContext(const TQContext& qContext, THashSet<TString>& flags);
+
 } // namspace NYql

ydb/library/yql/tools/dqrun/dqrun.cpp

@@ -911,6 +911,7 @@ int RunMain(int argc, const char* argv[])
 
     TGatewaysConfig gatewaysConfig;
     ReadGatewaysConfig(gatewaysCfgFile, &gatewaysConfig, sqlFlags);
+    UpdateSqlFlagsFromQContext(qContext, sqlFlags);
     PatchGatewaysConfig(&gatewaysConfig, mrJobBin, mrJobUdfsDir, numYtThreads, res.Has("keep-temp"));
     if (runOptions.AnalyzeQuery) {
         auto* setting = gatewaysConfig.MutableDq()->AddDefaultSettings();
@@ -1136,7 +1137,7 @@ int RunMain(int argc, const char* argv[])
 
         moduleResolver = std::make_shared<TModuleResolver>(std::move(modules), ctx.NextUniqueId, clusterMapping, sqlFlags, hasValidate);
     } else {
-        if (!GetYqlDefaultModuleResolver(ctx, moduleResolver, clusters)) {
+        if (GetYqlModuleResolver(ctx, moduleResolver, {}, clusters, sqlFlags).empty()) {
             *runOptions.ErrStream << "Errors loading default YQL libraries:" << Endl;
             ctx.IssueManager.GetIssues().PrintTo(*runOptions.ErrStream);
             return 1;