YQL-19747 statements (hints etc)

commit_hash:1288e94c1f35aed35f40ac5e9b59e708b7cfafad

Author: vitstn

yql/essentials/data/language/statements_opensource.json

@@ -0,0 +1 @@
+{"read":{"yt":{"hints":[{"name":"infer_scheme"},{"name":"force_infer_schema"},{"name":"inline"},{"name":"xlock"},{"name":"unordered"},{"name":"non_unique"},{"name":"ignore_type_v3"}]}},"insert":{"yt":{"hints":[{"name":"truncate"},{"name":"compression_codec"},{"name":"erasure_codec"},{"name":"expiration"},{"name":"replication_factor"},{"name":"user_attrs"},{"name":"media"},{"name":"primary_medium"},{"name":"keep_meta"},{"name":"monotonic_keys"},{"name":"column_groups"},{"name":"security_tags"}]}},"replace":{},"upsert":{},"update":{},"delete":{},"create_table":{},"create_view":{}}

yql/essentials/sql/v1/sql_query.cpp

@@ -4002,4 +4002,15 @@ void EnumeratePragmas(std::function<void(std::string_view)> callback) {
     callback("yson.DisableCastToString");
 }
 
+void EnumerateStmtContexts(std::function<void(std::string_view)> callback) {
+    callback("read");
+    callback("insert");
+    callback("replace");
+    callback("upsert");
+    callback("update");
+    callback("delete");
+    callback("create_table");
+    callback("create_view");
+}
+
 } // namespace NSQLTranslationV1

yql/essentials/sql/v1/sql_query.h

@@ -85,5 +85,6 @@ class TSqlQuery: public TSqlTranslation {
 };
 
 void EnumeratePragmas(std::function<void(std::string_view)> callback);
+void EnumerateStmtContexts(std::function<void(std::string_view)> callback);
 
 } // namespace NSQLTranslationV1

yt/yql/providers/yt/provider/ya.make

@@ -73,6 +73,7 @@ SRCS(
 
 PEERDIR(
     library/cpp/yson/node
+    library/cpp/json/writer
     library/cpp/disjoint_sets
     yt/cpp/mapreduce/common
     yt/cpp/mapreduce/interface

yt/yql/providers/yt/provider/yql_yt_op_settings.cpp

@@ -4,6 +4,7 @@
 #include <yt/yql/providers/yt/common/yql_yt_settings.h>
 #include <yt/yql/providers/yt/provider/yql_yt_block_io_utils.h>
 #include <yql/essentials/providers/common/provider/yql_provider.h>
+#include <yql/essentials/providers/common/provider/yql_provider_names.h>
 #include <yql/essentials/core/yql_expr_type_annotation.h>
 #include <yql/essentials/core/yql_opt_utils.h>
 
@@ -1322,4 +1323,62 @@ EYtSettingTypes operator|(EYtSettingType left, EYtSettingType right) {
     return EYtSettingTypes(left) | EYtSettingTypes(right);
 }
 
+void YtWriteHint(std::string_view name, NJsonWriter::TBuf& json) {
+    json.BeginObject();
+    json.WriteKey("name");
+    json.WriteString(name);
+    json.EndObject();
+}
+
+void YtWriteHints(EYtSettingTypes flags, NJsonWriter::TBuf& json) {
+    for (ui32 i = 0; i < (ui32)EYtSettingType::LAST; ++i) {
+        if (flags.HasFlags((EYtSettingType)i)) {
+            YtWriteHint(ToString((EYtSettingType)i), json);
+        }
+    }
+}
+
+void YtWriteStmtContext(std::string_view ctxName, NJsonWriter::TBuf& json) {
+    if (ctxName == "read") {
+        json.WriteKey(YtProviderName);
+        json.BeginObject();
+        json.WriteKey("hints");
+        json.BeginList();
+        YtWriteHints(
+            EYtSettingType::InferScheme |
+            EYtSettingType::ForceInferScheme |
+            EYtSettingType::Inline |
+            EYtSettingType::XLock |
+            EYtSettingType::Unordered |
+            EYtSettingType::NonUnique |
+            EYtSettingType::IgnoreTypeV3,
+            json
+        );
+        json.EndList();
+        json.EndObject();
+    } else if (ctxName == "insert") {
+        json.WriteKey(YtProviderName);
+        json.BeginObject();
+        json.WriteKey("hints");
+        json.BeginList();
+        YtWriteHint("truncate", json);
+        YtWriteHints(
+            EYtSettingType::CompressionCodec |
+            EYtSettingType::ErasureCodec |
+            EYtSettingType::Expiration |
+            EYtSettingType::ReplicationFactor |
+            EYtSettingType::UserAttrs |
+            EYtSettingType::Media |
+            EYtSettingType::PrimaryMedium |
+            EYtSettingType::KeepMeta |
+            EYtSettingType::MonotonicKeys |
+            EYtSettingType::ColumnGroups |
+            EYtSettingType::SecurityTags,
+            json
+        );
+        json.EndList();
+        json.EndObject();
+    }
+}
+
 } // NYql

yt/yql/providers/yt/provider/yql_yt_op_settings.h

@@ -1,6 +1,7 @@
 #pragma once
 
 #include <yql/essentials/ast/yql_expr.h>
+#include <library/cpp/json/writer/json.h>
 
 #include <util/generic/flags.h>
 #include <util/generic/strbuf.h>
@@ -64,12 +65,12 @@ enum class EYtSettingType: ui64 {
     WarnNonExisting          /* "warn_non_existing" "warnnonexisting" */,
     XLock                    /* "xlock" */,
     Unordered                /* "unordered" */,
-    NonUnique                /* "nonUnique" */,
+    NonUnique                /* "non_unique" "nonUnique" */,
     UserSchema               /* "userschema" */,
     UserColumns              /* "usercolumns" */,
     StatColumns              /* "statcolumns" */,
     SysColumns               /* "syscolumns" */,
-    IgnoreTypeV3             /* "ignoretypev3" "ignore_type_v3" */,
+    IgnoreTypeV3             /* "ignore_type_v3" "ignoretypev3" */,
     // Table content
     MemUsage                 /* "memUsage" */,
     ItemsCount               /* "itemsCount" */,
@@ -228,6 +229,7 @@ TMaybe<ui64> GetMaxJobSizeForFirstAsPrimary(const TExprNode& settings);
 bool UseJoinReduceForSecondAsPrimary(const TExprNode& settings);
 
 ui32 GetMinChildrenForIndexedKeyFilter(EYtSettingType type);
+void YtWriteStmtContext(std::string_view ctxName, NJsonWriter::TBuf& json);
 
 } // NYql