#include <babeltrace/compiler.h>
#include <babeltrace/plugin/source-internal.h>
#include <babeltrace/plugin/component-internal.h>
+#include <babeltrace/plugin/notification/iterator.h>
+#include <babeltrace/plugin/notification/iterator-internal.h>
static
void bt_component_source_destroy(struct bt_component *component)
g_free(source);
}
-struct bt_component *bt_component_source_create(const char *name,
- void *private_data, bt_component_destroy_cb destroy_func,
- bt_component_source_iterator_create_cb iterator_create_cb)
+BT_HIDDEN
+struct bt_component *bt_component_source_create(const char *name)
{
struct bt_component_source *source = NULL;
enum bt_component_status ret;
-
- if (!iterator_create_cb) {
- goto end;
- }
-
source = g_new0(struct bt_component_source, 1);
if (!source) {
goto end;
}
- ret = bt_component_init(&source->parent, name, private_data,
- destroy_func, BT_COMPONENT_TYPE_SOURCE,
- bt_component_source_destroy);
+ ret = bt_component_init(&source->parent, name,
+ BT_COMPONENT_TYPE_SOURCE, bt_component_source_destroy);
if (ret != BT_COMPONENT_STATUS_OK) {
g_free(source);
source = NULL;
goto end;
}
-
- source->create_iterator = iterator_create_cb;
end:
return source ? &source->parent : NULL;
}
+
+struct bt_notification_iterator *bt_plugin_source_create_iterator(
+ struct bt_component *component)
+{
+ enum bt_component_status ret_component;
+ enum bt_notification_iterator_status ret_iterator;
+ struct bt_component_source *source;
+ struct bt_notification_iterator *iterator = NULL;
+
+ if (!component) {
+ goto end;
+ }
+
+ if (bt_component_get_type(component) != BT_COMPONENT_TYPE_SOURCE) {
+ goto end;
+ }
+
+ iterator = bt_notification_iterator_create(component);
+ if (!iterator) {
+ goto end;
+ }
+
+ source = container_of(component, struct bt_component_source, parent);
+ assert(source->init_iterator);
+ ret_component = source->init_iterator(component, iterator);
+ if (ret_component != BT_COMPONENT_STATUS_OK) {
+ goto error;
+ }
+
+ ret_iterator = bt_notification_iterator_validate(iterator);
+ if (ret_iterator != BT_NOTIFICATION_ITERATOR_STATUS_OK) {
+ goto error;
+ }
+end:
+ return iterator;
+error:
+ bt_notification_iterator_put(iterator);
+ return NULL;
+}