From 1743f6bd2431a162e91f80cdfdbef9577ba8c94e Mon Sep 17 00:00:00 2001
From: Jacob Domagala <domagala.jacob@gmail.com>
Date: Sat, 30 Sep 2023 21:05:06 +0200
Subject: [PATCH] #2183: Add LDMS stream_publish to node_lb_data file

---
 src/vt/vrt/collection/balance/node_lb_data.cc | 14 ++++++++++++++
 1 file changed, 14 insertions(+)

diff --git a/src/vt/vrt/collection/balance/node_lb_data.cc b/src/vt/vrt/collection/balance/node_lb_data.cc
index 18828426cb..8ebe1144df 100644
--- a/src/vt/vrt/collection/balance/node_lb_data.cc
+++ b/src/vt/vrt/collection/balance/node_lb_data.cc
@@ -58,6 +58,12 @@
 #include <sys/stat.h>
 #include <memory>
 
+#if vt_check_enabled(ldms)
+#include <ldms/ldms.h>
+#include <ldms/ldmsd_stream.h>
+#include <ovis_util/util.h>
+#endif
+
 #include <fmt-vt/core.h>
 
 namespace vt { namespace vrt { namespace collection { namespace balance {
@@ -286,6 +292,14 @@ void NodeLBData::outputLBDataForPhase(PhaseType phase) {
   auto j = lb_data_->toJson(phase);
   auto writer = static_cast<JSONAppender*>(lb_data_writer_.get());
   writer->addElm(*j);
+
+#if vt_check_enabled(ldms)
+  ldms_t ldms = ldms_xprt_new("sock");
+  const auto jsonStr = j->dump();
+  ldmsd_stream_publish(
+    ldms, "LB_data", LDMSD_STREAM_JSON, jsonStr.c_str(), jsonStr.length()
+  );
+#endif
 }
 
 void NodeLBData::registerCollectionInfo(