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 04496b81530ffcec61c86e773933ad78d6d5b13c..6021c06580e33ebdedcc8977b033cce4e43426d6 100644 (file)
--- a/src/plugins/ctf/lttng-live/data-stream.c
+++ b/src/plugins/ctf/lttng-live/data-stream.c
@@ -159,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, log_level, self_comp);
+                               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);
                }
        }
 
@@ -224,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, log_level, self_comp);
+                       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) {
@@ -245,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. */