intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 69 insertions, 0 deletions

diff --git a/library/cpp/lwtrace/start.cpp b/library/cpp/lwtrace/start.cpp
new file mode 100644
index 0000000000..121d5472b6
--- /dev/null
+++ b/library/cpp/lwtrace/start.cpp
@@ -0,0 +1,69 @@
+#include "start.h"
+
+#include "all.h"
+
+#include <google/protobuf/text_format.h>
+
+#include <util/generic/singleton.h>
+#include <util/stream/file.h>
+#include <util/stream/output.h>
+#include <util/system/env.h>
+
+#include <stdlib.h>
+
+using namespace NLWTrace;
+
+namespace {
+    struct TTraceManagerHolder {
+        TManager TraceManager;
+        TTraceManagerHolder()
+            : TraceManager(*Singleton<TProbeRegistry>(), true)
+        {
+        }
+    };
+
+    void TraceFromEnv(TString path) {
+        TString script = TUnbufferedFileInput(path).ReadAll();
+        TQuery query;
+        bool ok = google::protobuf::TextFormat::ParseFromString(script, &query);
+        Y_VERIFY(ok, "failed to parse protobuf");
+        Singleton<TTraceManagerHolder>()->TraceManager.New("env", query);
+    }
+
+} // anonymous namespace
+
+void NLWTrace::StartLwtraceFromEnv() {
+    static bool started = false;
+    if (started) {
+        return;
+    } else {
+        started = true;
+    }
+
+    TString path = GetEnv("LWTRACE");
+    if (!path) {
+        return;
+    }
+
+    try {
+        TraceFromEnv(path);
+    } catch (...) {
+        Cerr << "failed to load lwtrace script: " << CurrentExceptionMessage() << "\n";
+        abort();
+    }
+}
+
+void NLWTrace::StartLwtraceFromEnv(std::function<void(TManager&)> prepare) {
+    TString path = GetEnv("LWTRACE");
+    if (Y_LIKELY(!path)) {
+        return;
+    }
+
+    try {
+        prepare(Singleton<TTraceManagerHolder>()->TraceManager);
+        TraceFromEnv(path);
+    } catch (...) {
+        Cerr << "failed to load lwtrace script: " << CurrentExceptionMessage() << "\n";
+        abort();
+    }
+}