Move initialization of components to init functions

[babeltrace.git] / plugins / iterator.c

diff --git a/plugins/iterator.c b/plugins/iterator.c
index 66f918c8dbe629a266fd14380c5e915943d970fa..094c2ff010e99922ca503ef76c651942a4f794d1 100644 (file)
--- a/plugins/iterator.c
+++ b/plugins/iterator.c
@@ -33,9 +33,19 @@
 #include <babeltrace/plugin/notification/iterator-internal.h>
 
 static
-void bt_notification_iterator_destroy(struct bt_notification_iterator *iterator)
+void bt_notification_iterator_destroy(struct bt_ctf_ref *ref)
 {
-       
+       struct bt_notification_iterator *iterator;
+
+       if (!ref) {
+               return;
+       }
+
+       iterator = container_of(ref, struct bt_notification_iterator,
+               ref_count);
+       assert(iterator->user_destroy || !iterator->user_data);
+       iterator->user_destroy(iterator);
+       g_free(iterator);
 }
 
 BT_HIDDEN
@@ -44,7 +54,8 @@ struct bt_notification_iterator *bt_notification_iterator_create(
 {
        struct bt_notification_iterator *iterator = NULL;
 
-       if (!component || bt_component_get_type(component) != BT_COMPONENT_TYPE_SOURCE) {
+       if (!component ||
+               bt_component_get_type(component) != BT_COMPONENT_TYPE_SOURCE) {
                goto end;
        }
 
@@ -62,7 +73,8 @@ BT_HIDDEN
 enum bt_notification_iterator_status bt_notification_iterator_validate(
                struct bt_notification_iterator *iterator)
 {
-       enum bt_notification_iterator_status ret = BT_NOTIFICATION_ITERATOR_STATUS_OK;
+       enum bt_notification_iterator_status ret =
+               BT_NOTIFICATION_ITERATOR_STATUS_OK;
 
        if (!iterator || !iterator->get || !iterator->next) {
                ret = BT_NOTIFICATION_ITERATOR_STATUS_INVAL;
@@ -89,3 +101,19 @@ void bt_notification_iterator_put(struct bt_notification_iterator *iterator)
 
        bt_ctf_ref_put(&iterator->ref_count, bt_notification_iterator_destroy);
 }
+
+enum bt_notification_iterator_status bt_notification_iterator_set_get_cb(
+               struct bt_notification_iterator *iterator,
+               bt_notification_iterator_get_cb get)
+{
+       enum bt_notification_iterator_status ret;
+
+       if (!iterator || !get) {
+               ret = BT_NOTIFICATION_ITERATOR_STATUS_INVAL;
+               goto end;
+       }
+
+       
+end:
+       return ret;
+}