Make bt_private_connection_create_notification_iterator() return a status code

[babeltrace.git] / plugins / utils / dummy / dummy.c

diff --git a/plugins/utils/dummy/dummy.c b/plugins/utils/dummy/dummy.c
index 66b29968a6f5ecbf0418e14975a4b425e93a4a9e..c3434a59929b8f1c6b2d71d4f8cd7405f9d8de77 100644 (file)
--- a/plugins/utils/dummy/dummy.c
+++ b/plugins/utils/dummy/dummy.c
@@ -21,63 +21,145 @@
  */
 
 #include <babeltrace/plugin/plugin-dev.h>
-#include <babeltrace/component/component.h>
-#include <babeltrace/component/component-sink.h>
-#include <babeltrace/component/notification/iterator.h>
-#include <babeltrace/component/notification/notification.h>
+#include <babeltrace/graph/connection.h>
+#include <babeltrace/graph/component.h>
+#include <babeltrace/graph/private-component.h>
+#include <babeltrace/graph/private-component-sink.h>
+#include <babeltrace/graph/private-port.h>
+#include <babeltrace/graph/port.h>
+#include <babeltrace/graph/private-connection.h>
+#include <babeltrace/graph/component-sink.h>
+#include <babeltrace/graph/notification-iterator.h>
+#include <babeltrace/graph/notification.h>
+#include <babeltrace/babeltrace-internal.h>
+#include <plugins-common.h>
 #include <assert.h>
+#include "dummy.h"
 
-enum bt_component_status dummy_consume(struct bt_component *component)
+static
+void destroy_private_dummy_data(struct dummy *dummy)
+{
+       if (dummy->iterators) {
+               g_ptr_array_free(dummy->iterators, TRUE);
+       }
+       g_free(dummy);
+}
+
+void dummy_finalize(struct bt_private_component *component)
+{
+       struct dummy *dummy;
+
+       assert(component);
+       dummy = bt_private_component_get_user_data(component);
+       assert(dummy);
+       destroy_private_dummy_data(dummy);
+}
+
+enum bt_component_status dummy_init(struct bt_private_component *component,
+               struct bt_value *params, UNUSED_VAR void *init_method_data)
 {
        enum bt_component_status ret;
+       struct dummy *dummy = g_new0(struct dummy, 1);
+
+       if (!dummy) {
+               ret = BT_COMPONENT_STATUS_NOMEM;
+               goto end;
+       }
+
+       ret = bt_private_component_sink_add_input_private_port(component,
+               "in", NULL, NULL);
+       if (ret != BT_COMPONENT_STATUS_OK) {
+               goto end;
+       }
+
+       dummy->iterators = g_ptr_array_new_with_free_func(
+                       (GDestroyNotify) bt_put);
+       if (!dummy->iterators) {
+               ret = BT_COMPONENT_STATUS_NOMEM;
+               goto end;
+       }
+
+       ret = bt_private_component_set_user_data(component, dummy);
+       if (ret != BT_COMPONENT_STATUS_OK) {
+               goto error;
+       }
+end:
+       return ret;
+error:
+       destroy_private_dummy_data(dummy);
+       return ret;
+}
+
+void dummy_port_connected(
+               struct bt_private_component *component,
+               struct bt_private_port *self_port,
+               struct bt_port *other_port)
+{
+       struct dummy *dummy;
+       struct bt_notification_iterator *iterator;
+       struct bt_private_connection *connection;
+       enum bt_connection_status conn_status;
+
+       dummy = bt_private_component_get_user_data(component);
+       assert(dummy);
+       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) {
+               dummy->error = true;
+               goto end;
+       }
+
+       g_ptr_array_add(dummy->iterators, iterator);
+
+end:
+       bt_put(connection);
+}
+
+enum bt_component_status dummy_consume(struct bt_private_component *component)
+{
+       enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
        struct bt_notification *notif = NULL;
-       struct bt_notification_iterator *it = NULL;
-       unsigned int it_count;
        size_t i;
-       bool got_one = false;
+       struct dummy *dummy;
 
-       ret = bt_component_sink_get_input_count(component, &it_count);
-       assert(ret == 0);
+       dummy = bt_private_component_get_user_data(component);
+       assert(dummy);
 
-       for (i = 0; i < it_count; i++) {
+       if (unlikely(dummy->error)) {
+               ret = BT_COMPONENT_STATUS_ERROR;
+               goto end;
+       }
+
+       /* Consume one notification from each iterator. */
+       for (i = 0; i < dummy->iterators->len; i++) {
+               struct bt_notification_iterator *it;
                enum bt_notification_iterator_status it_ret;
 
-               ret = bt_component_sink_get_input_iterator(component, i, &it);
-               assert(ret == 0);
+               it = g_ptr_array_index(dummy->iterators, i);
+
                it_ret = bt_notification_iterator_next(it);
                switch (it_ret) {
                case BT_NOTIFICATION_ITERATOR_STATUS_ERROR:
                        ret = BT_COMPONENT_STATUS_ERROR;
                        goto end;
+               case BT_NOTIFICATION_ITERATOR_STATUS_AGAIN:
+                       ret = BT_COMPONENT_STATUS_AGAIN;
+                       goto end;
                case BT_NOTIFICATION_ITERATOR_STATUS_END:
-                       ret = BT_COMPONENT_STATUS_END;
-                       BT_PUT(it);
+                       g_ptr_array_remove_index(dummy->iterators, i);
+                       i--;
                        continue;
                default:
                        break;
                }
-
-               notif = bt_notification_iterator_get_notification(it);
-               if (!notif) {
-                       ret = BT_COMPONENT_STATUS_ERROR;
-                       goto end;
-               }
-
-               /*
-                * Dummy! I'm doing nothing with this notification,
-                * NOTHING.
-                */
-               got_one = true;
-               BT_PUT(it);
-               BT_PUT(notif);
        }
 
-       if (!got_one) {
+       if (dummy->iterators->len == 0) {
                ret = BT_COMPONENT_STATUS_END;
        }
-
 end:
-       bt_put(it);
        bt_put(notif);
        return ret;
 }