Merge be85339 into f5dc352

ydb-platform · Feb 25, 2024 · 8441805 · 8441805
2 parents f5dc352 + be85339
commit 8441805
Showing 1 changed file with 6 additions and 2 deletions.
diff --git a/ydb/library/yql/sql/pg/pg_sql.cpp b/ydb/library/yql/sql/pg/pg_sql.cpp
@@ -2169,7 +2169,7 @@ class TConverter : public IPGParseEvents {
                 AddError(TStringBuilder() << "VariableSetStmt, expected string literal for " << value->name << " option");
                 return nullptr;
             }
-        } else if (name.StartsWith("dq.") || name.StartsWith("yt.") || name.StartsWith("s3.")) {
+        } else if (name.StartsWith("dq.") || name.StartsWith("yt.") || name.StartsWith("s3.") || name.StartsWith("ydb.")) {
             if (ListLength(value->args) != 1) {
                 AddError(TStringBuilder() << "VariableSetStmt, expected 1 arg, but got: " << ListLength(value->args));
                 return nullptr;
@@ -2184,8 +2184,12 @@ class TConverter : public IPGParseEvents {
                     providerName = NYql::DqProviderName;
                 } else if (name.StartsWith("yt.")) {
                     providerName = NYql::YtProviderName;
-                } else {
+                } else if (name.StartsWith("s3.")) {
                     providerName = NYql::S3ProviderName;
+                } else if (name.StartsWith("ydb.")) {
+                    providerName = NYql::YdbProviderName;
+                } else {
+                    Y_ASSERT(0);
                 }
 
                 auto providerSource = L(A("DataSource"), QA(providerName), QA("$all"));