src.ctf.lttng-live: make lttng_live_trace::trace a bt2::Trace::Shared
[babeltrace.git] / src / plugins / ctf / lttng-live / data-stream.cpp
index 9ad1e9ef8412186cb8ef465d75cbe5d873857492..cf49b7e39a84839646ebadcde026d2bd2e585d68 100644 (file)
 #include <glib.h>
 #include <inttypes.h>
 #include <stdio.h>
+#include <stdlib.h>
 
 #include <babeltrace2/babeltrace.h>
 
-#include "plugins/ctf/common/metadata/decoder.hpp"
-
-#define BT_COMP_LOG_SELF_COMP self_comp
-#define BT_LOG_OUTPUT_LEVEL   log_level
-#define BT_LOG_TAG            "PLUGIN/SRC.CTF.LTTNG-LIVE/DS"
-#include "logging/comp-logging.h"
-
 #include "common/assert.h"
-#include "compat/mman.h"
+#include "compat/mman.h" /* IWYU pragma: keep  */
+#include "cpp-common/vendor/fmt/format.h"
 
-#include "../common/msg-iter/msg-iter.hpp"
+#include "../common/src/msg-iter/msg-iter.hpp"
 #include "data-stream.hpp"
 
 #define STREAM_NAME_PREFIX "stream-"
@@ -68,15 +63,13 @@ end:
 static bt_stream *medop_borrow_stream(bt_stream_class *stream_class, int64_t stream_id, void *data)
 {
     lttng_live_stream_iterator *lttng_live_stream = (lttng_live_stream_iterator *) data;
-    bt_logging_level log_level = lttng_live_stream->log_level;
-    bt_self_component *self_comp = lttng_live_stream->self_comp;
 
     if (!lttng_live_stream->stream) {
         uint64_t stream_class_id = bt_stream_class_get_id(stream_class);
 
-        BT_COMP_LOGI("Creating stream %s (ID: %" PRIu64 ") out of stream "
-                     "class %" PRId64,
-                     lttng_live_stream->name->str, stream_id, stream_class_id);
+        BT_CPPLOGI_SPEC(lttng_live_stream->logger,
+                        "Creating stream {} (ID: {}) out of stream class {}",
+                        lttng_live_stream->name->str, stream_id, stream_class_id);
 
         if (stream_id < 0) {
             /*
@@ -85,18 +78,19 @@ static bt_stream *medop_borrow_stream(bt_stream_class *stream_class, int64_t str
              * LTTng. In these cases, use the viewer_stream_id that
              * is unique for a live viewer session.
              */
-            lttng_live_stream->stream = bt_stream_create_with_id(
-                stream_class, lttng_live_stream->trace->trace, lttng_live_stream->viewer_stream_id);
+            lttng_live_stream->stream =
+                bt_stream_create_with_id(stream_class, lttng_live_stream->trace->trace->libObjPtr(),
+                                         lttng_live_stream->viewer_stream_id);
         } else {
             lttng_live_stream->stream = bt_stream_create_with_id(
-                stream_class, lttng_live_stream->trace->trace, (uint64_t) stream_id);
+                stream_class, lttng_live_stream->trace->trace->libObjPtr(), (uint64_t) stream_id);
         }
 
         if (!lttng_live_stream->stream) {
-            BT_COMP_LOGE_APPEND_CAUSE(self_comp,
-                                      "Cannot create stream %s (stream class ID "
-                                      "%" PRId64 ", stream ID %" PRIu64 ")",
-                                      lttng_live_stream->name->str, stream_class_id, stream_id);
+            BT_CPPLOGE_APPEND_CAUSE_SPEC(
+                lttng_live_stream->logger,
+                "Cannot create stream {} (stream class ID {}, stream ID {})",
+                lttng_live_stream->name->str, stream_class_id, stream_id);
             goto end;
         }
 
@@ -119,16 +113,15 @@ enum lttng_live_iterator_status lttng_live_lazy_msg_init(struct lttng_live_sessi
 {
     struct lttng_live_component *lttng_live = session->lttng_live_msg_iter->lttng_live_comp;
     uint64_t trace_idx, stream_iter_idx;
-    bt_logging_level log_level = session->log_level;
-    bt_self_component *self_comp = session->self_comp;
 
     if (!session->lazy_stream_msg_init) {
         return LTTNG_LIVE_ITERATOR_STATUS_OK;
     }
 
-    BT_COMP_LOGD("Lazily initializing self message iterator for live session: "
-                 "session-id=%" PRIu64 ", self-msg-iter-addr=%p",
-                 session->id, self_msg_iter);
+    BT_CPPLOGD_SPEC(session->logger,
+                    "Lazily initializing self message iterator for live session: "
+                    "session-id={}, self-msg-iter-addr={}",
+                    session->id, fmt::ptr(self_msg_iter));
 
     for (trace_idx = 0; trace_idx < session->traces->len; trace_idx++) {
         struct lttng_live_trace *trace =
@@ -144,16 +137,19 @@ enum lttng_live_iterator_status lttng_live_lazy_msg_init(struct lttng_live_sessi
             if (stream_iter->msg_iter) {
                 continue;
             }
-            ctf_tc = ctf_metadata_decoder_borrow_ctf_trace_class(trace->metadata->decoder);
-            BT_COMP_LOGD("Creating CTF message iterator: "
-                         "session-id=%" PRIu64 ", ctf-tc-addr=%p, "
-                         "stream-iter-name=%s, self-msg-iter-addr=%p",
-                         session->id, ctf_tc, stream_iter->name->str, self_msg_iter);
+            ctf_tc = ctf_metadata_decoder_borrow_ctf_trace_class(trace->metadata->decoder.get());
+            BT_CPPLOGD_SPEC(stream_iter->logger,
+                            "Creating CTF message iterator: session-id={}, ctf-tc-addr={}, "
+                            "stream-iter-name={}, self-msg-iter-addr={}",
+                            session->id, fmt::ptr(ctf_tc), stream_iter->name->str,
+                            fmt::ptr(self_msg_iter));
             stream_iter->msg_iter =
                 ctf_msg_iter_create(ctf_tc, lttng_live->max_query_size, medops, stream_iter,
-                                    log_level, self_comp, self_msg_iter);
+                                    self_msg_iter, stream_iter->logger)
+                    .release();
             if (!stream_iter->msg_iter) {
-                BT_COMP_LOGE_APPEND_CAUSE(self_comp, "Failed to create CTF message iterator");
+                BT_CPPLOGE_APPEND_CAUSE_SPEC(stream_iter->logger,
+                                             "Failed to create CTF message iterator");
                 goto error;
             }
         }
@@ -171,35 +167,21 @@ struct lttng_live_stream_iterator *
 lttng_live_stream_iterator_create(struct lttng_live_session *session, uint64_t ctf_trace_id,
                                   uint64_t stream_id, bt_self_message_iterator *self_msg_iter)
 {
-    struct lttng_live_stream_iterator *stream_iter;
-    struct lttng_live_component *lttng_live;
-    struct lttng_live_trace *trace;
-    bt_logging_level log_level;
-    bt_self_component *self_comp;
+    lttng_live_stream_iterator *stream_iter = nullptr;
 
     BT_ASSERT(session);
     BT_ASSERT(session->lttng_live_msg_iter);
     BT_ASSERT(session->lttng_live_msg_iter->lttng_live_comp);
-    log_level = session->log_level;
-    self_comp = session->self_comp;
-
-    lttng_live = session->lttng_live_msg_iter->lttng_live_comp;
-
-    stream_iter = g_new0(struct lttng_live_stream_iterator, 1);
-    if (!stream_iter) {
-        BT_COMP_LOGE_APPEND_CAUSE(self_comp,
-                                  "Failed to allocate struct lttng_live_stream_iterator");
-        goto error;
-    }
 
-    stream_iter->log_level = log_level;
-    stream_iter->self_comp = self_comp;
-    trace = lttng_live_session_borrow_or_create_trace_by_id(session, ctf_trace_id);
+    lttng_live_component *lttng_live = session->lttng_live_msg_iter->lttng_live_comp;
+    lttng_live_trace *trace =
+        lttng_live_session_borrow_or_create_trace_by_id(session, ctf_trace_id);
     if (!trace) {
-        BT_COMP_LOGE_APPEND_CAUSE(self_comp, "Failed to borrow CTF trace.");
+        BT_CPPLOGE_APPEND_CAUSE_SPEC(session->logger, "Failed to borrow CTF trace.");
         goto error;
     }
 
+    stream_iter = new lttng_live_stream_iterator {session->logger};
     stream_iter->trace = trace;
     stream_iter->state = LTTNG_LIVE_STREAM_ACTIVE_NO_DATA;
     stream_iter->viewer_stream_id = stream_id;
@@ -212,26 +194,29 @@ lttng_live_stream_iterator_create(struct lttng_live_session *session, uint64_t c
 
     if (trace->trace) {
         struct ctf_trace_class *ctf_tc =
-            ctf_metadata_decoder_borrow_ctf_trace_class(trace->metadata->decoder);
+            ctf_metadata_decoder_borrow_ctf_trace_class(trace->metadata->decoder.get());
         BT_ASSERT(!stream_iter->msg_iter);
-        stream_iter->msg_iter =
-            ctf_msg_iter_create(ctf_tc, lttng_live->max_query_size, medops, stream_iter, log_level,
-                                self_comp, self_msg_iter);
+        stream_iter->msg_iter = ctf_msg_iter_create(ctf_tc, lttng_live->max_query_size, medops,
+                                                    stream_iter, self_msg_iter, stream_iter->logger)
+                                    .release();
         if (!stream_iter->msg_iter) {
-            BT_COMP_LOGE_APPEND_CAUSE(self_comp, "Failed to create CTF message iterator");
+            BT_CPPLOGE_APPEND_CAUSE_SPEC(stream_iter->logger,
+                                         "Failed to create CTF message iterator");
             goto error;
         }
     }
     stream_iter->buf = g_new0(uint8_t, lttng_live->max_query_size);
     if (!stream_iter->buf) {
-        BT_COMP_LOGE_APPEND_CAUSE(self_comp, "Failed to allocate live stream iterator buffer");
+        BT_CPPLOGE_APPEND_CAUSE_SPEC(stream_iter->logger,
+                                     "Failed to allocate live stream iterator buffer");
         goto error;
     }
 
     stream_iter->buflen = lttng_live->max_query_size;
     stream_iter->name = g_string_new(NULL);
     if (!stream_iter->name) {
-        BT_COMP_LOGE_APPEND_CAUSE(self_comp, "Failed to allocate live stream iterator name buffer");
+        BT_CPPLOGE_APPEND_CAUSE_SPEC(stream_iter->logger,
+                                     "Failed to allocate live stream iterator name buffer");
         goto error;
     }
 
@@ -273,5 +258,5 @@ void lttng_live_stream_iterator_destroy(struct lttng_live_stream_iterator *strea
     /* Track the number of active stream iterator. */
     stream_iter->trace->session->lttng_live_msg_iter->active_stream_iter--;
 
-    g_free(stream_iter);
+    delete stream_iter;
 }
This page took 0.026124 seconds and 4 git commands to generate.