Cleanup: src.ctf.lttng-live: coding style

[babeltrace.git] / src / plugins / ctf / lttng-live / data-stream.c

diff --git a/src/plugins/ctf/lttng-live/data-stream.c b/src/plugins/ctf/lttng-live/data-stream.c
index 25a8e4cf1e66bdea7b9c18d4a52f9ada2a381bfd..6021c06580e33ebdedcc8977b033cce4e43426d6 100644 (file)
--- a/src/plugins/ctf/lttng-live/data-stream.c
+++ b/src/plugins/ctf/lttng-live/data-stream.c
@@ -23,8 +23,10 @@
  * SOFTWARE.
  */
 
-#define BT_LOG_TAG "PLUGIN-CTF-LTTNG-LIVE-SRC-DS"
-#include "logging.h"
+#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 <stdio.h>
 #include <stdint.h>
@@ -56,12 +58,18 @@ enum bt_msg_iter_medium_status medop_request_bytes(
        uint64_t len_left;
        uint64_t read_len;
 
+       if (stream->has_stream_hung_up) {
+               status = BT_MSG_ITER_MEDIUM_STATUS_EOF;
+               goto end;
+       }
+
        len_left = stream->base_offset + stream->len - stream->offset;
        if (!len_left) {
                stream->state = LTTNG_LIVE_STREAM_ACTIVE_NO_DATA;
                status = BT_MSG_ITER_MEDIUM_STATUS_AGAIN;
-               return status;
+               goto end;
        }
+
        read_len = MIN(request_sz, stream->buflen);
        read_len = MIN(read_len, len_left);
        status = lttng_live_get_stream_bytes(live_msg_iter,
@@ -70,6 +78,7 @@ enum bt_msg_iter_medium_status medop_request_bytes(
        *buffer_addr = stream->buf;
        *buffer_sz = recv_len;
        stream->offset += recv_len;
+end:
        return status;
 }
 
@@ -78,11 +87,13 @@ bt_stream *medop_borrow_stream(bt_stream_class *stream_class,
                int64_t stream_id, void *data)
 {
        struct lttng_live_stream_iterator *lttng_live_stream = 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_LOGD("Creating stream %s (ID: %" PRIu64 ") out of stream "
+               BT_COMP_LOGI("Creating stream %s (ID: %" PRIu64 ") out of stream "
                        "class %" PRId64, lttng_live_stream->name->str,
                        stream_id, stream_class_id);
 
@@ -103,15 +114,18 @@ bt_stream *medop_borrow_stream(bt_stream_class *stream_class,
                }
 
                if (!lttng_live_stream->stream) {
-                       BT_LOGE("Cannot create stream %s (stream class ID "
+                       BT_COMP_LOGE("Cannot create stream %s (stream class ID "
                                "%" PRId64 ", stream ID %" PRIu64 ")",
                                lttng_live_stream->name->str,
                                stream_class_id, stream_id);
+                       goto end;
                }
+
                bt_stream_set_name(lttng_live_stream->stream,
                        lttng_live_stream->name->str);
        }
 
+end:
        return lttng_live_stream->stream;
 }
 
@@ -128,6 +142,8 @@ enum lttng_live_iterator_status lttng_live_lazy_msg_init(
        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;
@@ -143,19 +159,24 @@ enum lttng_live_iterator_status lttng_live_lazy_msg_init(
                        struct ctf_trace_class *ctf_tc;
                        struct lttng_live_stream_iterator *stream_iter =
                                g_ptr_array_index(trace->stream_iterators,
-                                               stream_iter_idx);
+                                       stream_iter_idx);
 
                        if (stream_iter->msg_iter) {
                                continue;
                        }
                        ctf_tc = ctf_metadata_decoder_borrow_ctf_trace_class(
-                                               trace->metadata->decoder);
+                               trace->metadata->decoder);
                        stream_iter->msg_iter = bt_msg_iter_create(ctf_tc,
-                                       lttng_live->max_query_size, medops,
-                                       stream_iter);
+                               lttng_live->max_query_size, medops, stream_iter,
+                               log_level, self_comp);
                        if (!stream_iter->msg_iter) {
                                goto error;
                        }
+
+                       bt_msg_iter_set_emit_stream_end_message(
+                               stream_iter->msg_iter, true);
+                       bt_msg_iter_set_emit_stream_beginning_message(
+                               stream_iter->msg_iter, true);
                }
        }
 
@@ -176,10 +197,14 @@ struct lttng_live_stream_iterator *lttng_live_stream_iterator_create(
        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;
 
        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;
 
@@ -188,6 +213,8 @@ struct lttng_live_stream_iterator *lttng_live_stream_iterator_create(
                goto error;
        }
 
+       stream_iter->log_level = log_level;
+       stream_iter->self_comp = self_comp;
        trace = lttng_live_borrow_trace(session, ctf_trace_id);
        if (!trace) {
                goto error;
@@ -202,14 +229,19 @@ struct lttng_live_stream_iterator *lttng_live_stream_iterator_create(
        if (trace->trace) {
                struct ctf_trace_class *ctf_tc =
                        ctf_metadata_decoder_borrow_ctf_trace_class(
-                                       trace->metadata->decoder);
+                               trace->metadata->decoder);
                BT_ASSERT(!stream_iter->msg_iter);
                stream_iter->msg_iter = bt_msg_iter_create(ctf_tc,
-                               lttng_live->max_query_size, medops,
-                               stream_iter);
+                       lttng_live->max_query_size, medops, stream_iter,
+                       log_level, self_comp);
                if (!stream_iter->msg_iter) {
                        goto error;
                }
+
+               bt_msg_iter_set_emit_stream_end_message(
+                       stream_iter->msg_iter, true);
+               bt_msg_iter_set_emit_stream_beginning_message(
+                       stream_iter->msg_iter, true);
        }
        stream_iter->buf = g_new0(uint8_t, lttng_live->max_query_size);
        if (!stream_iter->buf) {
@@ -223,7 +255,7 @@ struct lttng_live_stream_iterator *lttng_live_stream_iterator_create(
        }
 
        g_string_printf(stream_iter->name, STREAM_NAME_PREFIX "%" PRIu64,
-                       stream_iter->viewer_stream_id);
+               stream_iter->viewer_stream_id);
        g_ptr_array_add(trace->stream_iterators, stream_iter);
 
        /* Track the number of active stream iterator. */
@@ -252,9 +284,7 @@ void lttng_live_stream_iterator_destroy(
        if (stream_iter->msg_iter) {
                bt_msg_iter_destroy(stream_iter->msg_iter);
        }
-       if (stream_iter->buf) {
-               g_free(stream_iter->buf);
-       }
+       g_free(stream_iter->buf);
        if (stream_iter->name) {
                g_string_free(stream_iter->name, TRUE);
        }