[babeltrace.git] / src / plugins / utils / counter / counter.c

/*
 * Copyright 2017 Philippe Proulx <pproulx@efficios.com>
 *
 * Permission is hereby granted, free of charge, to any person obtaining a copy
 * of this software and associated documentation files (the "Software"), to deal
 * in the Software without restriction, including without limitation the rights
 * to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
 * copies of the Software, and to permit persons to whom the Software is
 * furnished to do so, subject to the following conditions:
 *
 * The above copyright notice and this permission notice shall be included in
 * all copies or substantial portions of the Software.
 *
 * THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
 * IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
 * FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
 * AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
 * LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
 * OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
 * SOFTWARE.
 */

#define BT_COMP_LOG_SELF_COMP (counter->self_comp)
#define BT_LOG_OUTPUT_LEVEL (counter->log_level)
#define BT_LOG_TAG "PLUGIN/FLT.UTILS.COUNTER"
#include "plugins/comp-logging.h"

#include <babeltrace2/babeltrace.h>
#include "common/macros.h"
#include "common/common.h"
#include "common/assert.h"
#include <inttypes.h>
#include <stdint.h>

#include "counter.h"

#define PRINTF_COUNT(_what, _var, args...)				\
	do {								\
		if (counter->count._var != 0 || !counter->hide_zero) {	\
			printf("%15" PRIu64 " %s message%s\n",		\
				counter->count._var,			\
				(_what),				\
				counter->count._var == 1 ? "" : "s");	\
		}							\
	} while (0)

static
const char * const in_port_name = "in";

static
uint64_t get_total_count(struct counter *counter)
{
	return counter->count.event +
		counter->count.stream_begin +
		counter->count.stream_end +
		counter->count.stream_activity_begin +
		counter->count.stream_activity_end +
		counter->count.packet_begin +
		counter->count.packet_end +
		counter->count.disc_events +
		counter->count.disc_packets +
		counter->count.msg_iter_inactivity +
		counter->count.other;
}

static
void print_count(struct counter *counter)
{
	uint64_t total = get_total_count(counter);

	PRINTF_COUNT("Event", event);
	PRINTF_COUNT("Stream beginning", stream_begin);
	PRINTF_COUNT("Stream end", stream_end);
	PRINTF_COUNT("Stream activity beginning", stream_activity_begin);
	PRINTF_COUNT("Stream activity end", stream_activity_end);
	PRINTF_COUNT("Packet beginning", packet_begin);
	PRINTF_COUNT("Packet end", packet_end);
	PRINTF_COUNT("Discarded event", disc_events);
	PRINTF_COUNT("Discarded packet", disc_packets);
	PRINTF_COUNT("Message iterator inactivity", msg_iter_inactivity);

	if (counter->count.other > 0) {
		PRINTF_COUNT("Other (unknown)", other);
	}

	printf("%s%15" PRIu64 " message%s (TOTAL)%s\n",
		bt_common_color_bold(), total, total == 1 ? "" : "s",
		bt_common_color_reset());
	counter->last_printed_total = total;
}

static
void try_print_count(struct counter *counter, uint64_t msg_count)
{
	if (counter->step == 0) {
		/* No update */
		return;
	}

	counter->at += msg_count;

	if (counter->at >= counter->step) {
		counter->at = 0;
		print_count(counter);
		putchar('\n');
	}
}

static
void try_print_last(struct counter *counter)
{
	const uint64_t total = get_total_count(counter);

	if (total != counter->last_printed_total) {
		print_count(counter);
	}
}

void destroy_private_counter_data(struct counter *counter)
{
	bt_self_component_port_input_message_iterator_put_ref(counter->msg_iter);
	g_free(counter);
}

BT_HIDDEN
void counter_finalize(bt_self_component_sink *comp)
{
	struct counter *counter;

	BT_ASSERT(comp);
	counter = bt_self_component_get_data(
			bt_self_component_sink_as_self_component(comp));
	BT_ASSERT(counter);
	try_print_last(counter);
	bt_self_component_port_input_message_iterator_put_ref(counter->msg_iter);
	g_free(counter);
}

BT_HIDDEN
bt_component_class_init_method_status counter_init(
		bt_self_component_sink *component,
		const bt_value *params,
		__attribute__((unused)) void *init_method_data)
{
	bt_component_class_init_method_status status =
		BT_COMPONENT_CLASS_INIT_METHOD_STATUS_OK;
	bt_self_component_add_port_status add_port_status;
	struct counter *counter = g_new0(struct counter, 1);
	const bt_value *step = NULL;
	const bt_value *hide_zero = NULL;

	if (!counter) {
		status = BT_COMPONENT_CLASS_INIT_METHOD_STATUS_MEMORY_ERROR;
		goto error;
	}

	counter->self_comp =
		bt_self_component_sink_as_self_component(component);
	counter->log_level = bt_component_get_logging_level(
		bt_self_component_as_component(counter->self_comp));
	add_port_status = bt_self_component_sink_add_input_port(component,
		"in", NULL, NULL);
	switch (add_port_status) {
	case BT_SELF_COMPONENT_ADD_PORT_STATUS_ERROR:
		status = BT_COMPONENT_CLASS_INIT_METHOD_STATUS_ERROR;
		goto error;
	case BT_SELF_COMPONENT_ADD_PORT_STATUS_MEMORY_ERROR:
		status = BT_COMPONENT_CLASS_INIT_METHOD_STATUS_MEMORY_ERROR;
		goto error;
	default:
		break;
	}

	counter->last_printed_total = -1ULL;
	counter->step = 10000;
	step = bt_value_map_borrow_entry_value_const(params, "step");
	if (step) {
		if (!bt_value_is_unsigned_integer(step)) {
			BT_COMP_LOGE("`step` parameter: expecting an unsigned integer value: "
				"type=%s", bt_common_value_type_string(
					bt_value_get_type(step)));
			goto error;
		}

		counter->step = bt_value_unsigned_integer_get(step);
	}

	hide_zero = bt_value_map_borrow_entry_value_const(params, "hide-zero");
	if (hide_zero) {
		if (!bt_value_is_bool(hide_zero)) {
			BT_COMP_LOGE("`hide-zero` parameter: expecting a boolean value: "
				"type=%s", bt_common_value_type_string(
					bt_value_get_type(hide_zero)));
			goto error;
		}

		counter->hide_zero = (bool) bt_value_bool_get(hide_zero);
	}

	bt_self_component_set_data(
		bt_self_component_sink_as_self_component(component),
		counter);
	goto end;

error:
	destroy_private_counter_data(counter);

	if (status == BT_COMPONENT_CLASS_INIT_METHOD_STATUS_OK) {
		status = BT_COMPONENT_CLASS_INIT_METHOD_STATUS_ERROR;
	}

end:
	return status;
}

BT_HIDDEN
bt_component_class_sink_graph_is_configured_method_status
counter_graph_is_configured(
		bt_self_component_sink *comp)
{
	bt_component_class_sink_graph_is_configured_method_status status =
		BT_COMPONENT_CLASS_SINK_GRAPH_IS_CONFIGURED_METHOD_STATUS_OK;
	struct counter *counter;
	bt_self_component_port_input_message_iterator *iterator;

	counter = bt_self_component_get_data(
		bt_self_component_sink_as_self_component(comp));
	BT_ASSERT(counter);
	iterator = bt_self_component_port_input_message_iterator_create(
		bt_self_component_sink_borrow_input_port_by_name(comp,
			in_port_name));
	if (!iterator) {
		status = BT_COMPONENT_CLASS_SINK_GRAPH_IS_CONFIGURED_METHOD_STATUS_MEMORY_ERROR;
		goto end;
	}

	BT_SELF_COMPONENT_PORT_INPUT_MESSAGE_ITERATOR_MOVE_REF(
		counter->msg_iter, iterator);

end:
	return status;
}

BT_HIDDEN
bt_component_class_sink_consume_method_status counter_consume(
		bt_self_component_sink *comp)
{
	bt_component_class_sink_consume_method_status status =
		BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_OK;
	struct counter *counter;
	bt_message_iterator_next_status next_status;
	uint64_t msg_count;
	bt_message_array_const msgs;

	counter = bt_self_component_get_data(
			bt_self_component_sink_as_self_component(comp));
	BT_ASSERT(counter);

	if (G_UNLIKELY(!counter->msg_iter)) {
		try_print_last(counter);
		status = BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_END;
		goto end;
	}

	/* Consume messages */
	next_status = bt_self_component_port_input_message_iterator_next(
		counter->msg_iter, &msgs, &msg_count);
	if (next_status < 0) {
		status = (int) next_status;
		goto end;
	}

	switch (next_status) {
	case BT_MESSAGE_ITERATOR_NEXT_STATUS_OK:
	{
		uint64_t i;

		for (i = 0; i < msg_count; i++) {
			const bt_message *msg = msgs[i];

			BT_ASSERT(msg);
			switch (bt_message_get_type(msg)) {
			case BT_MESSAGE_TYPE_EVENT:
				counter->count.event++;
				break;
			case BT_MESSAGE_TYPE_PACKET_BEGINNING:
				counter->count.packet_begin++;
				break;
			case BT_MESSAGE_TYPE_PACKET_END:
				counter->count.packet_end++;
				break;
			case BT_MESSAGE_TYPE_MESSAGE_ITERATOR_INACTIVITY:
				counter->count.msg_iter_inactivity++;
				break;
			case BT_MESSAGE_TYPE_STREAM_BEGINNING:
				counter->count.stream_begin++;
				break;
			case BT_MESSAGE_TYPE_STREAM_END:
				counter->count.stream_end++;
				break;
			case BT_MESSAGE_TYPE_STREAM_ACTIVITY_BEGINNING:
				counter->count.stream_activity_begin++;
				break;
			case BT_MESSAGE_TYPE_STREAM_ACTIVITY_END:
				counter->count.stream_activity_end++;
				break;
			case BT_MESSAGE_TYPE_DISCARDED_EVENTS:
				counter->count.disc_events++;
				break;
			case BT_MESSAGE_TYPE_DISCARDED_PACKETS:
				counter->count.disc_packets++;
				break;
			default:
				counter->count.other++;
			}

			bt_message_put_ref(msg);
		}

		status = BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_OK;
		break;
	}
	case BT_MESSAGE_ITERATOR_NEXT_STATUS_AGAIN:
		status = BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_AGAIN;
		goto end;
	case BT_MESSAGE_ITERATOR_NEXT_STATUS_END:
		try_print_last(counter);
		status = BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_END;
		goto end;
	case BT_MESSAGE_ITERATOR_NEXT_STATUS_MEMORY_ERROR:
		status = BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_MEMORY_ERROR;
		goto end;
	default:
		break;
	}

	try_print_count(counter, msg_count);

end:
	return status;
}
Commit	Line	Data
14b7ef9e PP	1	/*
	2	* Copyright 2017 Philippe Proulx <pproulx@efficios.com>
	3	*
	4	* Permission is hereby granted, free of charge, to any person obtaining a copy
	5	* of this software and associated documentation files (the "Software"), to deal
	6	* in the Software without restriction, including without limitation the rights
	7	* to use, copy, modify, merge, publish, distribute, sublicense, and/or sell
	8	* copies of the Software, and to permit persons to whom the Software is
	9	* furnished to do so, subject to the following conditions:
	10	*
	11	* The above copyright notice and this permission notice shall be included in
	12	* all copies or substantial portions of the Software.
	13	*
	14	* THE SOFTWARE IS PROVIDED "AS IS", WITHOUT WARRANTY OF ANY KIND, EXPRESS OR
	15	* IMPLIED, INCLUDING BUT NOT LIMITED TO THE WARRANTIES OF MERCHANTABILITY,
	16	* FITNESS FOR A PARTICULAR PURPOSE AND NONINFRINGEMENT. IN NO EVENT SHALL THE
	17	* AUTHORS OR COPYRIGHT HOLDERS BE LIABLE FOR ANY CLAIM, DAMAGES OR OTHER
	18	* LIABILITY, WHETHER IN AN ACTION OF CONTRACT, TORT OR OTHERWISE, ARISING FROM,
	19	* OUT OF OR IN CONNECTION WITH THE SOFTWARE OR THE USE OR OTHER DEALINGS IN THE
	20	* SOFTWARE.
	21	*/
	22
627c70e6	23	#define BT_COMP_LOG_SELF_COMP (counter->self_comp)
c58e2582	24	#define BT_LOG_OUTPUT_LEVEL (counter->log_level)
b03487ab	25	#define BT_LOG_TAG "PLUGIN/FLT.UTILS.COUNTER"
627c70e6	26	#include "plugins/comp-logging.h"
68d9d039	27
71c5da58	28	#include <babeltrace2/babeltrace.h>
85e7137b	29	#include "common/macros.h"
57952005	30	#include "common/common.h"
57952005	31	#include "common/assert.h"
14b7ef9e PP	32	#include <inttypes.h>
	33	#include <stdint.h>
	34
	35	#include "counter.h"
	36
168baad4	37	#define PRINTF_COUNT(_what, _var, args...) \
14b7ef9e PP	38	do { \
14b7ef9e PP	39	if (counter->count._var != 0 \|\| !counter->hide_zero) { \
168baad4	40	printf("%15" PRIu64 " %s message%s\n", \
14b7ef9e	41	counter->count._var, \
168baad4 PP	42	(_what), \
168baad4 PP	43	counter->count._var == 1 ? "" : "s"); \
14b7ef9e PP	44	} \
	45	} while (0)
	46
1043fdea PP	47	static
	48	const char * const in_port_name = "in";
	49
14b7ef9e PP	50	static
	51	uint64_t get_total_count(struct counter *counter)
	52	{
	53	return counter->count.event +
	54	counter->count.stream_begin +
	55	counter->count.stream_end +
168baad4 PP	56	counter->count.stream_activity_begin +
168baad4 PP	57	counter->count.stream_activity_end +
14b7ef9e PP	58	counter->count.packet_begin +
14b7ef9e PP	59	counter->count.packet_end +
168baad4 PP	60	counter->count.disc_events +
168baad4 PP	61	counter->count.disc_packets +
40bf6fd0	62	counter->count.msg_iter_inactivity +
14b7ef9e PP	63	counter->count.other;
	64	}
	65
	66	static
3fd7b79d	67	void print_count(struct counter *counter)
14b7ef9e	68	{
3fd7b79d PP	69	uint64_t total = get_total_count(counter);
3fd7b79d PP	70
168baad4 PP	71	PRINTF_COUNT("Event", event);
	72	PRINTF_COUNT("Stream beginning", stream_begin);
	73	PRINTF_COUNT("Stream end", stream_end);
	74	PRINTF_COUNT("Stream activity beginning", stream_activity_begin);
	75	PRINTF_COUNT("Stream activity end", stream_activity_end);
	76	PRINTF_COUNT("Packet beginning", packet_begin);
	77	PRINTF_COUNT("Packet end", packet_end);
	78	PRINTF_COUNT("Discarded event", disc_events);
	79	PRINTF_COUNT("Discarded packet", disc_packets);
	80	PRINTF_COUNT("Message iterator inactivity", msg_iter_inactivity);
14b7ef9e PP	81
14b7ef9e PP	82	if (counter->count.other > 0) {
168baad4	83	PRINTF_COUNT("Other (unknown)", other);
14b7ef9e PP	84	}
14b7ef9e PP	85
b09a5592	86	printf("%s%15" PRIu64 " message%s (TOTAL)%s\n",
14b7ef9e PP	87	bt_common_color_bold(), total, total == 1 ? "" : "s",
	88	bt_common_color_reset());
	89	counter->last_printed_total = total;
	90	}
	91
	92	static
b09a5592	93	void try_print_count(struct counter *counter, uint64_t msg_count)
14b7ef9e	94	{
14b7ef9e PP	95	if (counter->step == 0) {
	96	/* No update */
	97	return;
	98	}
	99
b09a5592	100	counter->at += msg_count;
14b7ef9e	101
3fd7b79d PP	102	if (counter->at >= counter->step) {
	103	counter->at = 0;
	104	print_count(counter);
14b7ef9e PP	105	putchar('\n');
	106	}
	107	}
	108
	109	static
	110	void try_print_last(struct counter *counter)
	111	{
	112	const uint64_t total = get_total_count(counter);
	113
	114	if (total != counter->last_printed_total) {
3fd7b79d	115	print_count(counter);
14b7ef9e PP	116	}
	117	}
	118
	119	void destroy_private_counter_data(struct counter *counter)
	120	{
b09a5592	121	bt_self_component_port_input_message_iterator_put_ref(counter->msg_iter);
14b7ef9e PP	122	g_free(counter);
	123	}
	124
834e9996	125	BT_HIDDEN
8eee8ea2	126	void counter_finalize(bt_self_component_sink *comp)
14b7ef9e PP	127	{
	128	struct counter *counter;
	129
834e9996 PP	130	BT_ASSERT(comp);
834e9996 PP	131	counter = bt_self_component_get_data(
bb61965b	132	bt_self_component_sink_as_self_component(comp));
8b45963b	133	BT_ASSERT(counter);
14b7ef9e	134	try_print_last(counter);
b09a5592	135	bt_self_component_port_input_message_iterator_put_ref(counter->msg_iter);
14b7ef9e PP	136	g_free(counter);
	137	}
	138
834e9996	139	BT_HIDDEN
fb25b9e3	140	bt_component_class_init_method_status counter_init(
8eee8ea2 PP	141	bt_self_component_sink *component,
8eee8ea2 PP	142	const bt_value *params,
f807570c	143	__attribute__((unused)) void *init_method_data)
14b7ef9e	144	{
fb25b9e3 PP	145	bt_component_class_init_method_status status =
	146	BT_COMPONENT_CLASS_INIT_METHOD_STATUS_OK;
	147	bt_self_component_add_port_status add_port_status;
14b7ef9e	148	struct counter *counter = g_new0(struct counter, 1);
8eee8ea2 PP	149	const bt_value *step = NULL;
8eee8ea2 PP	150	const bt_value *hide_zero = NULL;
14b7ef9e PP	151
14b7ef9e PP	152	if (!counter) {
fb25b9e3	153	status = BT_COMPONENT_CLASS_INIT_METHOD_STATUS_MEMORY_ERROR;
834e9996	154	goto error;
14b7ef9e PP	155	}
14b7ef9e PP	156
627c70e6 PP	157	counter->self_comp =
627c70e6 PP	158	bt_self_component_sink_as_self_component(component);
c58e2582	159	counter->log_level = bt_component_get_logging_level(
627c70e6	160	bt_self_component_as_component(counter->self_comp));
fb25b9e3	161	add_port_status = bt_self_component_sink_add_input_port(component,
14b7ef9e	162	"in", NULL, NULL);
fb25b9e3 PP	163	switch (add_port_status) {
	164	case BT_SELF_COMPONENT_ADD_PORT_STATUS_ERROR:
	165	status = BT_COMPONENT_CLASS_INIT_METHOD_STATUS_ERROR;
834e9996	166	goto error;
fb25b9e3 PP	167	case BT_SELF_COMPONENT_ADD_PORT_STATUS_MEMORY_ERROR:
	168	status = BT_COMPONENT_CLASS_INIT_METHOD_STATUS_MEMORY_ERROR;
	169	goto error;
	170	default:
	171	break;
14b7ef9e PP	172	}
	173
	174	counter->last_printed_total = -1ULL;
a79f976c	175	counter->step = 10000;
ce141536	176	step = bt_value_map_borrow_entry_value_const(params, "step");
68d9d039 PP	177	if (step) {
68d9d039 PP	178	if (!bt_value_is_unsigned_integer(step)) {
627c70e6	179	BT_COMP_LOGE("`step` parameter: expecting an unsigned integer value: "
68d9d039 PP	180	"type=%s", bt_common_value_type_string(
	181	bt_value_get_type(step)));
	182	goto error;
14b7ef9e	183	}
68d9d039 PP	184
68d9d039 PP	185	counter->step = bt_value_unsigned_integer_get(step);
14b7ef9e PP	186	}
14b7ef9e PP	187
ce141536	188	hide_zero = bt_value_map_borrow_entry_value_const(params, "hide-zero");
68d9d039 PP	189	if (hide_zero) {
68d9d039 PP	190	if (!bt_value_is_bool(hide_zero)) {
627c70e6	191	BT_COMP_LOGE("`hide-zero` parameter: expecting a boolean value: "
68d9d039 PP	192	"type=%s", bt_common_value_type_string(
	193	bt_value_get_type(hide_zero)));
	194	goto error;
	195	}
14b7ef9e	196
68d9d039	197	counter->hide_zero = (bool) bt_value_bool_get(hide_zero);
14b7ef9e PP	198	}
14b7ef9e PP	199
834e9996	200	bt_self_component_set_data(
bb61965b	201	bt_self_component_sink_as_self_component(component),
834e9996	202	counter);
14b7ef9e PP	203	goto end;
	204
	205	error:
	206	destroy_private_counter_data(counter);
fb25b9e3 PP	207
	208	if (status == BT_COMPONENT_CLASS_INIT_METHOD_STATUS_OK) {
	209	status = BT_COMPONENT_CLASS_INIT_METHOD_STATUS_ERROR;
	210	}
14b7ef9e PP	211
14b7ef9e PP	212	end:
fb25b9e3	213	return status;
14b7ef9e PP	214	}
14b7ef9e PP	215
834e9996	216	BT_HIDDEN
fb25b9e3 PP	217	bt_component_class_sink_graph_is_configured_method_status
fb25b9e3 PP	218	counter_graph_is_configured(
1043fdea	219	bt_self_component_sink *comp)
14b7ef9e	220	{
fb25b9e3 PP	221	bt_component_class_sink_graph_is_configured_method_status status =
fb25b9e3 PP	222	BT_COMPONENT_CLASS_SINK_GRAPH_IS_CONFIGURED_METHOD_STATUS_OK;
14b7ef9e	223	struct counter *counter;
b09a5592	224	bt_self_component_port_input_message_iterator *iterator;
14b7ef9e	225
834e9996	226	counter = bt_self_component_get_data(
bb61965b	227	bt_self_component_sink_as_self_component(comp));
8b45963b	228	BT_ASSERT(counter);
b09a5592	229	iterator = bt_self_component_port_input_message_iterator_create(
1043fdea PP	230	bt_self_component_sink_borrow_input_port_by_name(comp,
1043fdea PP	231	in_port_name));
834e9996	232	if (!iterator) {
fb25b9e3	233	status = BT_COMPONENT_CLASS_SINK_GRAPH_IS_CONFIGURED_METHOD_STATUS_MEMORY_ERROR;
14b7ef9e PP	234	goto end;
	235	}
	236
b09a5592 PP	237	BT_SELF_COMPONENT_PORT_INPUT_MESSAGE_ITERATOR_MOVE_REF(
b09a5592 PP	238	counter->msg_iter, iterator);
14b7ef9e PP	239
14b7ef9e PP	240	end:
634f394c	241	return status;
14b7ef9e PP	242	}
14b7ef9e PP	243
834e9996	244	BT_HIDDEN
fb25b9e3	245	bt_component_class_sink_consume_method_status counter_consume(
8eee8ea2	246	bt_self_component_sink *comp)
14b7ef9e	247	{
fb25b9e3 PP	248	bt_component_class_sink_consume_method_status status =
fb25b9e3 PP	249	BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_OK;
14b7ef9e	250	struct counter *counter;
fb25b9e3	251	bt_message_iterator_next_status next_status;
b09a5592 PP	252	uint64_t msg_count;
b09a5592 PP	253	bt_message_array_const msgs;
14b7ef9e	254
834e9996	255	counter = bt_self_component_get_data(
bb61965b	256	bt_self_component_sink_as_self_component(comp));
8b45963b	257	BT_ASSERT(counter);
14b7ef9e	258
85e7137b	259	if (G_UNLIKELY(!counter->msg_iter)) {
14b7ef9e	260	try_print_last(counter);
fb25b9e3	261	status = BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_END;
14b7ef9e PP	262	goto end;
	263	}
	264
b09a5592	265	/* Consume messages */
fb25b9e3	266	next_status = bt_self_component_port_input_message_iterator_next(
b09a5592	267	counter->msg_iter, &msgs, &msg_count);
fb25b9e3 PP	268	if (next_status < 0) {
fb25b9e3 PP	269	status = (int) next_status;
14b7ef9e PP	270	goto end;
	271	}
	272
fb25b9e3 PP	273	switch (next_status) {
fb25b9e3 PP	274	case BT_MESSAGE_ITERATOR_NEXT_STATUS_OK:
14b7ef9e	275	{
3fd7b79d PP	276	uint64_t i;
3fd7b79d PP	277
b09a5592 PP	278	for (i = 0; i < msg_count; i++) {
b09a5592 PP	279	const bt_message *msg = msgs[i];
3fd7b79d	280
b09a5592 PP	281	BT_ASSERT(msg);
	282	switch (bt_message_get_type(msg)) {
	283	case BT_MESSAGE_TYPE_EVENT:
3fd7b79d PP	284	counter->count.event++;
3fd7b79d PP	285	break;
168baad4 PP	286	case BT_MESSAGE_TYPE_PACKET_BEGINNING:
	287	counter->count.packet_begin++;
	288	break;
	289	case BT_MESSAGE_TYPE_PACKET_END:
	290	counter->count.packet_end++;
	291	break;
40bf6fd0 FD	292	case BT_MESSAGE_TYPE_MESSAGE_ITERATOR_INACTIVITY:
40bf6fd0 FD	293	counter->count.msg_iter_inactivity++;
3fd7b79d	294	break;
b09a5592	295	case BT_MESSAGE_TYPE_STREAM_BEGINNING:
3fd7b79d PP	296	counter->count.stream_begin++;
3fd7b79d PP	297	break;
b09a5592	298	case BT_MESSAGE_TYPE_STREAM_END:
3fd7b79d PP	299	counter->count.stream_end++;
3fd7b79d PP	300	break;
168baad4 PP	301	case BT_MESSAGE_TYPE_STREAM_ACTIVITY_BEGINNING:
168baad4 PP	302	counter->count.stream_activity_begin++;
3fd7b79d	303	break;
168baad4 PP	304	case BT_MESSAGE_TYPE_STREAM_ACTIVITY_END:
	305	counter->count.stream_activity_end++;
	306	break;
	307	case BT_MESSAGE_TYPE_DISCARDED_EVENTS:
	308	counter->count.disc_events++;
	309	break;
	310	case BT_MESSAGE_TYPE_DISCARDED_PACKETS:
	311	counter->count.disc_packets++;
3fd7b79d	312	break;
3fd7b79d PP	313	default:
3fd7b79d PP	314	counter->count.other++;
14b7ef9e	315	}
3fd7b79d	316
b09a5592	317	bt_message_put_ref(msg);
14b7ef9e	318	}
834e9996	319
fb25b9e3	320	status = BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_OK;
834e9996	321	break;
14b7ef9e	322	}
fb25b9e3 PP	323	case BT_MESSAGE_ITERATOR_NEXT_STATUS_AGAIN:
fb25b9e3 PP	324	status = BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_AGAIN;
834e9996	325	goto end;
fb25b9e3	326	case BT_MESSAGE_ITERATOR_NEXT_STATUS_END:
834e9996	327	try_print_last(counter);
fb25b9e3	328	status = BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_END;
834e9996	329	goto end;
fb25b9e3 PP	330	case BT_MESSAGE_ITERATOR_NEXT_STATUS_MEMORY_ERROR:
fb25b9e3 PP	331	status = BT_COMPONENT_CLASS_SINK_CONSUME_METHOD_STATUS_MEMORY_ERROR;
834e9996	332	goto end;
14b7ef9e PP	333	default:
	334	break;
	335	}
	336
b09a5592	337	try_print_count(counter, msg_count);
14b7ef9e PP	338
14b7ef9e PP	339	end:
fb25b9e3	340	return status;
14b7ef9e	341	}