+++ /dev/null
-/*
- * 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_LOG_TAG "PLUGIN-UTILS-COUNTER-FLT"
-#include "logging.h"
-
-#include <babeltrace2/babeltrace.h>
-#include <babeltrace2/babeltrace-internal.h>
-#include <babeltrace2/common-internal.h>
-#include <plugins-common.h>
-#include <babeltrace2/assert-internal.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_self_component_status counter_init(
- bt_self_component_sink *component,
- const bt_value *params,
- UNUSED_VAR void *init_method_data)
-{
- bt_self_component_status ret;
- struct counter *counter = g_new0(struct counter, 1);
- const bt_value *step = NULL;
- const bt_value *hide_zero = NULL;
-
- if (!counter) {
- ret = BT_SELF_COMPONENT_STATUS_NOMEM;
- goto error;
- }
-
- ret = bt_self_component_sink_add_input_port(component,
- "in", NULL, NULL);
- if (ret != BT_SELF_COMPONENT_STATUS_OK) {
- goto error;
- }
-
- 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_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_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);
- ret = BT_SELF_COMPONENT_STATUS_ERROR;
-
-end:
- return ret;
-}
-
-BT_HIDDEN
-bt_self_component_status counter_graph_is_configured(
- bt_self_component_sink *comp)
-{
- bt_self_component_status status = BT_SELF_COMPONENT_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_SELF_COMPONENT_STATUS_NOMEM;
- goto end;
- }
-
- BT_SELF_COMPONENT_PORT_INPUT_MESSAGE_ITERATOR_MOVE_REF(
- counter->msg_iter, iterator);
-
-end:
- return status;
-}
-
-BT_HIDDEN
-bt_self_component_status counter_consume(
- bt_self_component_sink *comp)
-{
- bt_self_component_status ret = BT_SELF_COMPONENT_STATUS_OK;
- struct counter *counter;
- bt_message_iterator_status it_ret;
- 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 (unlikely(!counter->msg_iter)) {
- try_print_last(counter);
- ret = BT_SELF_COMPONENT_STATUS_END;
- goto end;
- }
-
- /* Consume messages */
- it_ret = bt_self_component_port_input_message_iterator_next(
- counter->msg_iter, &msgs, &msg_count);
- if (it_ret < 0) {
- ret = BT_SELF_COMPONENT_STATUS_ERROR;
- goto end;
- }
-
- switch (it_ret) {
- case BT_MESSAGE_ITERATOR_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);
- }
-
- ret = BT_SELF_COMPONENT_STATUS_OK;
- break;
- }
- case BT_MESSAGE_ITERATOR_STATUS_AGAIN:
- ret = BT_SELF_COMPONENT_STATUS_AGAIN;
- goto end;
- case BT_MESSAGE_ITERATOR_STATUS_END:
- try_print_last(counter);
- ret = BT_SELF_COMPONENT_STATUS_END;
- goto end;
- case BT_MESSAGE_ITERATOR_STATUS_NOMEM:
- ret = BT_SELF_COMPONENT_STATUS_NOMEM;
- goto end;
- default:
- break;
- }
-
- try_print_count(counter, msg_count);
-
-end:
- return ret;
-}