intermediate changes

ref:cde9a383711a11544ce7e107a78147fb96cc4029

1 files changed, 21 insertions, 0 deletions

diff --git a/library/cpp/lwtrace/custom_action.cpp b/library/cpp/lwtrace/custom_action.cpp
new file mode 100644
index 00000000000..a379b34ec05
--- /dev/null
+++ b/library/cpp/lwtrace/custom_action.cpp
@@ -0,0 +1,21 @@
+#include "custom_action.h"
+
+#include "control.h"
+
+using namespace NLWTrace;
+
+TCustomActionExecutor* TCustomActionFactory::Create(TProbe* probe, const TCustomAction& action, TSession* trace) const {
+    auto iter = Callbacks.find(action.GetName());
+    if (iter != Callbacks.end()) {
+        return iter->second(probe, action, trace);
+    } else {
+        return nullptr;
+    }
+}
+
+void TCustomActionFactory::Register(const TString& name, const TCustomActionFactory::TCallback& callback) {
+    if (Callbacks.contains(name)) {
+        ythrow yexception() << "duplicate custom action '" << name << "'";
+    }
+    Callbacks[name] = callback;
+}