Introduce bt_plugin and bt_component_class

[babeltrace.git] / plugins / iterator.c

diff --git a/plugins/iterator.c b/plugins/iterator.c
index 66f918c8dbe629a266fd14380c5e915943d970fa..cfa143a30ef79bfd43d12adebb2f6a5fabe6fa32 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_ref *ref)
 {
-       
+       struct bt_notification_iterator *iterator;
+
+       if (!ref) {
+               return;
+       }
+
+       iterator = container_of(ref, struct bt_notification_iterator,
+               ref);
+       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;
        }
 
@@ -53,7 +64,7 @@ struct bt_notification_iterator *bt_notification_iterator_create(
                goto end;
        }
 
-       bt_ctf_ref_init(&iterator->ref_count);
+       bt_ref_init(&iterator->ref, bt_notification_iterator_destroy);
 end:
        return iterator;
 }
@@ -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;
@@ -78,7 +90,7 @@ void bt_notification_iterator_get(struct bt_notification_iterator *iterator)
                return;
        }
 
-       bt_ctf_ref_get(&iterator->ref_count);
+       bt_ref_get(&iterator->ref);
 }
 
 void bt_notification_iterator_put(struct bt_notification_iterator *iterator)
@@ -87,5 +99,21 @@ void bt_notification_iterator_put(struct bt_notification_iterator *iterator)
                return;
        }
 
-       bt_ctf_ref_put(&iterator->ref_count, bt_notification_iterator_destroy);
+       bt_ref_put(&iterator->ref);
+}
+
+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;
 }