[SchemeShard] Add required path creation (#11445)

Author: Enjection

ydb/core/tx/schemeshard/generated/codegen/main.cpp

@@ -0,0 +1,77 @@
+#include <util/stream/file.h>
+#include <util/stream/output.h>
+#include <util/string/builder.h>
+#include <util/system/src_location.h>
+#include <ydb/core/protos/flat_scheme_op.pb.h>
+#include <google/protobuf/descriptor.pb.h>
+
+#include <jinja2cpp/template_env.h>
+#include <jinja2cpp/template.h>
+#include <jinja2cpp/value.h>
+#include <jinja2cpp/reflected_value.h>
+#include <cstdint>
+#include <string>
+#include <vector>
+#include <iostream>
+
+std::string replace(
+    const std::string& str,
+    const std::string& from,
+    const std::string& to)
+{
+    std::string result;
+
+    size_t pos = str.find(from);
+    result.append(str, 0, pos);
+    result.append(to);
+    result.append(str, pos + from.size(), std::string::npos);
+
+    return result;
+}
+
+int main(int argc, char** argv) {
+    if (argc < 3) {
+        Cerr << "Usage: " << argv[0] << " INPUT OUTPUT ..." << Endl;
+        return 1;
+    }
+
+    std::vector<std::string> opTypes;
+    const auto* d = NKikimrSchemeOp::EOperationType_descriptor();
+    for (int i = 0; i < d->value_count(); ++i) {
+        const auto* v = d->value(i);
+        if (v) {
+            auto name = v->full_name();
+            name = replace(name, ".", "::");
+            opTypes.emplace_back(name);
+        }
+    }
+
+    jinja2::TemplateEnv env;
+    env.AddGlobal("generator", jinja2::Reflect(std::string(__SOURCE_FILE__)));
+    env.AddGlobal("opTypes", jinja2::Reflect(opTypes));
+
+    for (int i = 1; i < argc; i += 2) {
+        if (!(i + 1 < argc)) {
+            Cerr << "ERROR: missing output for " << argv[i] << Endl;
+            return 1;
+        }
+
+        jinja2::Template t(&env);
+        auto loaded = t.Load(TFileInput(argv[i]).ReadAll(), argv[i]);
+        if (!loaded) {
+            Cerr << "ERROR: " << loaded.error().ToString() << Endl;
+            return 1;
+        }
+
+        auto rendered = t.RenderAsString({});
+        if (!rendered) {
+            Cerr << "ERROR: " << rendered.error().ToString() << Endl;
+            return 1;
+        }
+
+        TFileOutput(argv[i + 1]).Write(rendered.value());
+        Cout << "Generated " << argv[i + 1] << " from " << argv[i] << Endl;
+    }
+
+    return 0;
+}

ydb/core/tx/schemeshard/generated/codegen/ya.make

@@ -0,0 +1,10 @@
+PROGRAM()
+
+SRCS(main.cpp)
+
+PEERDIR(
+    contrib/libs/jinja2cpp
+    ydb/core/protos
+)
+
+END()

ydb/core/tx/schemeshard/generated/dispatch_op.h.in

@@ -0,0 +1,20 @@
+// Auto-generated by {{ generator }}, do not modify.
+#pragma once
+
+#include <ydb/core/tx/schemeshard/generated/traits.h>
+
+namespace NKikimr::NSchemeShard::NGenerated {
+
+template <class TFn>
+constexpr auto DispatchOp(const TTxTransaction& tx, TFn fn) {
+    switch (tx.GetOperationType()) {
+{% for opType in opTypes %}
+        case {{ opType }}:
+            return fn(TSchemeTxTraits<{{ opType }}>{});
+{% endfor %}
+        default:
+            return fn(TSchemeTxTraitsFallback{});
+    }
+}
+
+} // namespace NKikimr::NSchemeShard::NGenerated