X-Git-Url: http://git.efficios.com/?a=blobdiff_plain;f=plugins%2Futils%2Fcounter%2Fcounter.c;h=7b82ae4a84e084229e9bd2eb081d6fdc37f0793b;hb=05e2128659970c32648a01255ed870449f05d518;hp=9ef22372d617216e984f72b81a49f907f98cedf3;hpb=9d408fcae74602e3591f66623ceb85f482d948ed;p=babeltrace.git diff --git a/plugins/utils/counter/counter.c b/plugins/utils/counter/counter.c index 9ef22372..7b82ae4a 100644 --- a/plugins/utils/counter/counter.c +++ b/plugins/utils/counter/counter.c @@ -24,7 +24,7 @@ #include #include #include -#include +#include #include #include @@ -48,28 +48,20 @@ uint64_t get_total_count(struct counter *counter) counter->count.packet_begin + counter->count.packet_end + counter->count.inactivity + - counter->count.discarded_events + - counter->count.discarded_packets + counter->count.other; } static -void print_count(struct counter *counter, uint64_t total) +void print_count(struct counter *counter) { + uint64_t total = get_total_count(counter); + PRINTF_COUNT("event", "events", event); PRINTF_COUNT("stream beginning", "stream beginnings", stream_begin); PRINTF_COUNT("stream end", "stream ends", stream_end); PRINTF_COUNT("packet beginning", "packet beginnings", packet_begin); PRINTF_COUNT("packet end", "packet ends", packet_end); PRINTF_COUNT("inactivity", "inactivities", inactivity); - PRINTF_COUNT("discarded events notification", - "discarded events notifications", discarded_events_notifs); - PRINTF_COUNT(" known discarded event", " known discarded events", - discarded_events); - PRINTF_COUNT("discarded packets notification", - "discarded packets notifications", discarded_packets_notifs); - PRINTF_COUNT(" known discarded packet", " known discarded packets", - discarded_packets); if (counter->count.other > 0) { PRINTF_COUNT(" other (unknown) notification", @@ -83,19 +75,18 @@ void print_count(struct counter *counter, uint64_t total) } static -void try_print_count(struct counter *counter) +void try_print_count(struct counter *counter, uint64_t notif_count) { - uint64_t total; - if (counter->step == 0) { /* No update */ return; } - total = get_total_count(counter); + counter->at += notif_count; - if (total % counter->step == 0) { - print_count(counter, total); + if (counter->at >= counter->step) { + counter->at = 0; + print_count(counter); putchar('\n'); } } @@ -106,204 +97,192 @@ void try_print_last(struct counter *counter) const uint64_t total = get_total_count(counter); if (total != counter->last_printed_total) { - print_count(counter, total); + print_count(counter); } } void destroy_private_counter_data(struct counter *counter) { - bt_put(counter->notif_iter); + bt_object_put_ref(counter->notif_iter); g_free(counter); } -void counter_finalize(struct bt_private_component *component) +BT_HIDDEN +void counter_finalize(struct bt_self_component_sink *comp) { struct counter *counter; - assert(component); - counter = bt_private_component_get_user_data(component); - assert(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_put(counter->notif_iter); + bt_object_put_ref(counter->notif_iter); g_free(counter); } -enum bt_component_status counter_init(struct bt_private_component *component, - struct bt_value *params, UNUSED_VAR void *init_method_data) +BT_HIDDEN +enum bt_self_component_status counter_init( + struct bt_self_component_sink *component, + const struct bt_value *params, + UNUSED_VAR void *init_method_data) { - enum bt_component_status ret; + enum bt_self_component_status ret; struct counter *counter = g_new0(struct counter, 1); - struct bt_value *step = NULL; - struct bt_value *hide_zero = NULL; + const struct bt_value *step = NULL; + const struct bt_value *hide_zero = NULL; if (!counter) { - ret = BT_COMPONENT_STATUS_NOMEM; - goto end; + ret = BT_SELF_COMPONENT_STATUS_NOMEM; + goto error; } - ret = bt_private_component_sink_add_input_private_port(component, + ret = bt_self_component_sink_add_input_port(component, "in", NULL, NULL); - if (ret != BT_COMPONENT_STATUS_OK) { - goto end; + if (ret != BT_SELF_COMPONENT_STATUS_OK) { + goto error; } counter->last_printed_total = -1ULL; counter->step = 1000; - step = bt_value_map_get(params, "step"); - if (bt_value_is_integer(step)) { + step = bt_value_map_borrow_entry_value_const(params, "step"); + if (step && bt_value_is_integer(step)) { int64_t val; - int vret = bt_value_integer_get(step, &val); - - assert(vret == 0); + val = bt_value_integer_get(step); if (val >= 0) { counter->step = (uint64_t) val; } } - hide_zero = bt_value_map_get(params, "hide-zero"); - if (bt_value_is_bool(hide_zero)) { + hide_zero = bt_value_map_borrow_entry_value_const(params, "hide-zero"); + if (hide_zero && bt_value_is_bool(hide_zero)) { bt_bool val; - int vret = bt_value_bool_get(hide_zero, &val); - assert(vret == 0); + val = bt_value_bool_get(hide_zero); counter->hide_zero = (bool) val; } - ret = bt_private_component_set_user_data(component, counter); - if (ret != BT_COMPONENT_STATUS_OK) { - goto error; - } - + bt_self_component_set_data( + bt_self_component_sink_as_self_component(component), + counter); goto end; error: destroy_private_counter_data(counter); end: - bt_put(step); - bt_put(hide_zero); return ret; } -void counter_port_connected( - struct bt_private_component *component, - struct bt_private_port *self_port, - struct bt_port *other_port) +BT_HIDDEN +enum bt_self_component_status counter_port_connected( + struct bt_self_component_sink *comp, + struct bt_self_component_port_input *self_port, + struct bt_port_output *other_port) { + enum bt_self_component_status status = BT_SELF_COMPONENT_STATUS_OK; struct counter *counter; - struct bt_notification_iterator *iterator; - struct bt_private_connection *connection; - enum bt_connection_status conn_status; - - counter = bt_private_component_get_user_data(component); - assert(counter); - connection = bt_private_port_get_private_connection(self_port); - assert(connection); - conn_status = bt_private_connection_create_notification_iterator( - connection, NULL, &iterator); - if (conn_status != BT_CONNECTION_STATUS_OK) { - counter->error = true; + struct bt_self_component_port_input_notification_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_notification_iterator_create( + self_port); + if (!iterator) { + status = BT_SELF_COMPONENT_STATUS_NOMEM; goto end; } - BT_MOVE(counter->notif_iter, iterator); + BT_OBJECT_MOVE_REF(counter->notif_iter, iterator); end: - bt_put(connection); + return status; } -enum bt_component_status counter_consume(struct bt_private_component *component) +BT_HIDDEN +enum bt_self_component_status counter_consume( + struct bt_self_component_sink *comp) { - enum bt_component_status ret = BT_COMPONENT_STATUS_OK; - struct bt_notification *notif = NULL; + enum bt_self_component_status ret = BT_SELF_COMPONENT_STATUS_OK; struct counter *counter; enum bt_notification_iterator_status it_ret; - int64_t count; + uint64_t notif_count; + bt_notification_array notifs; - counter = bt_private_component_get_user_data(component); - assert(counter); - - if (unlikely(counter->error)) { - ret = BT_COMPONENT_STATUS_ERROR; - goto end; - } + counter = bt_self_component_get_data( + bt_self_component_sink_as_self_component(comp)); + BT_ASSERT(counter); if (unlikely(!counter->notif_iter)) { try_print_last(counter); - ret = BT_COMPONENT_STATUS_END; + ret = BT_SELF_COMPONENT_STATUS_END; goto end; } - /* Consume one notification */ - it_ret = bt_notification_iterator_next(counter->notif_iter); + /* Consume notifications */ + it_ret = bt_self_component_port_input_notification_iterator_next( + counter->notif_iter, ¬ifs, ¬if_count); if (it_ret < 0) { - ret = BT_COMPONENT_STATUS_ERROR; + ret = BT_SELF_COMPONENT_STATUS_ERROR; goto end; } switch (it_ret) { - case BT_NOTIFICATION_ITERATOR_STATUS_AGAIN: - ret = BT_COMPONENT_STATUS_AGAIN; - goto end; - case BT_NOTIFICATION_ITERATOR_STATUS_END: - try_print_last(counter); - ret = BT_COMPONENT_STATUS_END; - goto end; case BT_NOTIFICATION_ITERATOR_STATUS_OK: { - struct bt_notification *notif = - bt_notification_iterator_get_notification(counter->notif_iter); - - assert(notif); - switch (bt_notification_get_type(notif)) { - case BT_NOTIFICATION_TYPE_EVENT: - counter->count.event++; - break; - case BT_NOTIFICATION_TYPE_INACTIVITY: - counter->count.inactivity++; - break; - case BT_NOTIFICATION_TYPE_STREAM_BEGIN: - counter->count.stream_begin++; - break; - case BT_NOTIFICATION_TYPE_STREAM_END: - counter->count.stream_end++; - break; - case BT_NOTIFICATION_TYPE_PACKET_BEGIN: - counter->count.packet_begin++; - break; - case BT_NOTIFICATION_TYPE_PACKET_END: - counter->count.packet_end++; - break; - case BT_NOTIFICATION_TYPE_DISCARDED_EVENTS: - counter->count.discarded_events_notifs++; - count = bt_notification_discarded_events_get_count( - notif); - if (count >= 0) { - counter->count.discarded_events += count; + uint64_t i; + + for (i = 0; i < notif_count; i++) { + struct bt_notification *notif = notifs[i]; + + BT_ASSERT(notif); + switch (bt_notification_get_type(notif)) { + case BT_NOTIFICATION_TYPE_EVENT: + counter->count.event++; + break; + case BT_NOTIFICATION_TYPE_INACTIVITY: + counter->count.inactivity++; + break; + case BT_NOTIFICATION_TYPE_STREAM_BEGIN: + counter->count.stream_begin++; + break; + case BT_NOTIFICATION_TYPE_STREAM_END: + counter->count.stream_end++; + break; + case BT_NOTIFICATION_TYPE_PACKET_BEGIN: + counter->count.packet_begin++; + break; + case BT_NOTIFICATION_TYPE_PACKET_END: + counter->count.packet_end++; + break; + default: + counter->count.other++; } - break; - case BT_NOTIFICATION_TYPE_DISCARDED_PACKETS: - counter->count.discarded_packets_notifs++; - count = bt_notification_discarded_packets_get_count( - notif); - if (count >= 0) { - counter->count.discarded_packets += count; - } - break; - default: - counter->count.other++; + + bt_object_put_ref(notif); } - bt_put(notif); + ret = BT_SELF_COMPONENT_STATUS_OK; + break; } + case BT_NOTIFICATION_ITERATOR_STATUS_AGAIN: + ret = BT_SELF_COMPONENT_STATUS_AGAIN; + goto end; + case BT_NOTIFICATION_ITERATOR_STATUS_END: + try_print_last(counter); + ret = BT_SELF_COMPONENT_STATUS_END; + goto end; + case BT_NOTIFICATION_ITERATOR_STATUS_NOMEM: + ret = BT_SELF_COMPONENT_STATUS_NOMEM; + goto end; default: break; } - try_print_count(counter); + try_print_count(counter, notif_count); end: - bt_put(notif); return ret; }