src.ctf.lttng-live: introduce lttng_live_metadata::UP and use it

[babeltrace.git] / src / plugins / ctf / lttng-live / lttng-live.hpp

diff --git a/src/plugins/ctf/lttng-live/lttng-live.hpp b/src/plugins/ctf/lttng-live/lttng-live.hpp
index 4f84ca7939354680a505bc9a61558c1b079e9af2..a8ec55fc14055625750a9130d1ff43f527dd60ad 100644 (file)
--- a/src/plugins/ctf/lttng-live/lttng-live.hpp
+++ b/src/plugins/ctf/lttng-live/lttng-live.hpp
@@ -16,6 +16,8 @@
 
 #include <babeltrace2/babeltrace.h>
 
+#include "cpp-common/vendor/fmt/format.h" /* IWYU pragma: keep */
+
 #include "../common/src/metadata/tsdl/decoder.hpp"
 #include "../common/src/msg-iter/msg-iter.hpp"
 #include "viewer-connection.hpp"
@@ -40,11 +42,37 @@ enum lttng_live_stream_state
     LTTNG_LIVE_STREAM_EOF,
 };
 
+inline const char *format_as(const lttng_live_stream_state state) noexcept
+{
+    switch (state) {
+    case LTTNG_LIVE_STREAM_ACTIVE_NO_DATA:
+        return "ACTIVE_NO_DATA";
+
+    case LTTNG_LIVE_STREAM_QUIESCENT_NO_DATA:
+        return "QUIESCENT_NO_DATA";
+
+    case LTTNG_LIVE_STREAM_QUIESCENT:
+        return "QUIESCENT";
+
+    case LTTNG_LIVE_STREAM_ACTIVE_DATA:
+        return "ACTIVE_DATA";
+
+    case LTTNG_LIVE_STREAM_EOF:
+        return "EOF";
+    }
+
+    bt_common_abort();
+}
+
 /* Iterator over a live stream. */
 struct lttng_live_stream_iterator
 {
-    bt_logging_level log_level = (bt_logging_level) 0;
-    bt_self_component *self_comp = nullptr;
+    explicit lttng_live_stream_iterator(const bt2c::Logger& parentLogger) :
+        logger {parentLogger, "PLUGIN/SRC.CTF.LTTNG-LIVE/STREAM-ITER"}
+    {
+    }
+
+    bt2c::Logger logger;
 
     /* Owned by this. */
     bt_stream *stream = nullptr;
@@ -112,12 +140,19 @@ struct lttng_live_stream_iterator
 
 struct lttng_live_metadata
 {
-    bt_logging_level log_level = (bt_logging_level) 0;
-    bt_self_component *self_comp = nullptr;
+    using UP = std::unique_ptr<lttng_live_metadata>;
+
+    explicit lttng_live_metadata(const bt2c::Logger& parentLogger) :
+        logger {parentLogger, "PLUGIN/SRC.CTF.LTTNG-LIVE/METADATA"}
+    {
+    }
+
+    bt2c::Logger logger;
 
     uint64_t stream_id = 0;
+
     /* Weak reference. */
-    struct ctf_metadata_decoder *decoder = nullptr;
+    ctf_metadata_decoder_up decoder;
 };
 
 enum lttng_live_metadata_stream_state
@@ -144,8 +179,12 @@ enum lttng_live_metadata_stream_state
 
 struct lttng_live_trace
 {
-    bt_logging_level log_level = (bt_logging_level) 0;
-    bt_self_component *self_comp = nullptr;
+    explicit lttng_live_trace(const bt2c::Logger& parentLogger) :
+        logger {parentLogger, "PLUGIN/SRC.CTF.LTTNG-LIVE/TRACE"}
+    {
+    }
+
+    bt2c::Logger logger;
 
     /* Back reference to session. */
     struct lttng_live_session *session = nullptr;
@@ -153,13 +192,11 @@ struct lttng_live_trace
     /* ctf trace ID within the session. */
     uint64_t id = 0;
 
-    /* Owned by this. */
-    bt_trace *trace = nullptr;
+    bt2::Trace::Shared trace;
 
-    /* Weak reference. */
-    bt_trace_class *trace_class = nullptr;
+    bt2::TraceClass::Shared trace_class;
 
-    struct lttng_live_metadata *metadata = nullptr;
+    lttng_live_metadata::UP metadata;
 
     const bt_clock_class *clock_class = nullptr;
 
@@ -173,7 +210,13 @@ struct lttng_live_trace
 
 struct lttng_live_session
 {
-    bt_logging_level log_level = (bt_logging_level) 0;
+    explicit lttng_live_session(const bt2c::Logger& parentLogger) :
+        logger {parentLogger, "PLUGIN/SRC.CTF.LTTNG-LIVE/SESSION"}
+    {
+    }
+
+    bt2c::Logger logger;
+
     bt_self_component *self_comp = nullptr;
 
     /* Weak reference. */
@@ -208,14 +251,21 @@ enum session_not_found_action
  */
 struct lttng_live_component
 {
-    bt_logging_level log_level = (bt_logging_level) 0;
+    using UP = std::unique_ptr<lttng_live_component>;
+
+    explicit lttng_live_component(bt2c::Logger loggerParam) noexcept :
+        logger {std::move(loggerParam)}
+    {
+    }
+
+    bt2c::Logger logger;
 
     /* Weak reference. */
     bt_self_component *self_comp = nullptr;
 
     struct
     {
-        GString *url = nullptr;
+        std::string url;
         enum session_not_found_action sess_not_found_act = SESSION_NOT_FOUND_ACTION_CONTINUE;
     } params;
 
@@ -230,7 +280,13 @@ struct lttng_live_component
 
 struct lttng_live_msg_iter
 {
-    bt_logging_level log_level = (bt_logging_level) 0;
+    explicit lttng_live_msg_iter(const bt2c::Logger& parentLogger) :
+        logger {parentLogger, "PLUGIN/SRC.CTF.LTTNG-LIVE/MSG-ITER"}
+    {
+    }
+
+    bt2c::Logger logger;
+
     bt_self_component *self_comp = nullptr;
 
     /* Weak reference. */
@@ -275,6 +331,37 @@ enum lttng_live_iterator_status
     LTTNG_LIVE_ITERATOR_STATUS_UNSUPPORTED = -4,
 };
 
+inline const char *format_as(const lttng_live_iterator_status status) noexcept
+{
+    switch (status) {
+    case LTTNG_LIVE_ITERATOR_STATUS_CONTINUE:
+        return "LTTNG_LIVE_ITERATOR_STATUS_CONTINUE";
+
+    case LTTNG_LIVE_ITERATOR_STATUS_AGAIN:
+        return "LTTNG_LIVE_ITERATOR_STATUS_AGAIN";
+
+    case LTTNG_LIVE_ITERATOR_STATUS_END:
+        return "LTTNG_LIVE_ITERATOR_STATUS_END";
+
+    case LTTNG_LIVE_ITERATOR_STATUS_OK:
+        return "LTTNG_LIVE_ITERATOR_STATUS_OK";
+
+    case LTTNG_LIVE_ITERATOR_STATUS_INVAL:
+        return "LTTNG_LIVE_ITERATOR_STATUS_INVAL";
+
+    case LTTNG_LIVE_ITERATOR_STATUS_ERROR:
+        return "LTTNG_LIVE_ITERATOR_STATUS_ERROR";
+
+    case LTTNG_LIVE_ITERATOR_STATUS_NOMEM:
+        return "LTTNG_LIVE_ITERATOR_STATUS_NOMEM";
+
+    case LTTNG_LIVE_ITERATOR_STATUS_UNSUPPORTED:
+        return "LTTNG_LIVE_ITERATOR_STATUS_UNSUPPORTED";
+    }
+
+    bt_common_abort();
+}
+
 bt_component_class_initialize_method_status
 lttng_live_component_init(bt_self_component_source *self_comp,
                           bt_self_component_source_configuration *config, const bt_value *params,
@@ -322,7 +409,7 @@ int lttng_live_add_session(struct lttng_live_msg_iter *lttng_live_msg_iter, uint
  * written to the file.
  */
 enum lttng_live_get_one_metadata_status
-lttng_live_get_one_metadata_packet(struct lttng_live_trace *trace, FILE *fp, size_t *reply_len);
+lttng_live_get_one_metadata_packet(struct lttng_live_trace *trace, std::vector<char>& buf);
 
 enum lttng_live_iterator_status
 lttng_live_get_next_index(struct lttng_live_msg_iter *lttng_live_msg_iter,