* SOFTWARE.
*/
-#include <babeltrace/plugin/plugin-dev.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 <babeltrace2/babeltrace.h>
+#include <babeltrace2/babeltrace-internal.h>
#include <plugins-common.h>
-#include <assert.h>
+#include <babeltrace2/assert-internal.h>
#include "dummy.h"
+static
+const char * const in_port_name = "in";
+
void destroy_private_dummy_data(struct dummy *dummy)
{
- bt_put(dummy->notif_iter);
+ bt_self_component_port_input_message_iterator_put_ref(dummy->msg_iter);
g_free(dummy);
+
}
-void dummy_finalize(struct bt_private_component *component)
+BT_HIDDEN
+void dummy_finalize(bt_self_component_sink *comp)
{
struct dummy *dummy;
- assert(component);
- dummy = bt_private_component_get_user_data(component);
- assert(dummy);
+ BT_ASSERT(comp);
+ dummy = bt_self_component_get_data(
+ bt_self_component_sink_as_self_component(comp));
+ BT_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)
+BT_HIDDEN
+bt_self_component_status dummy_init(
+ bt_self_component_sink *component,
+ const bt_value *params,
+ UNUSED_VAR void *init_method_data)
{
- enum bt_component_status ret;
+ bt_self_component_status ret;
struct dummy *dummy = g_new0(struct dummy, 1);
if (!dummy) {
- ret = BT_COMPONENT_STATUS_NOMEM;
+ ret = BT_SELF_COMPONENT_STATUS_NOMEM;
goto end;
}
- 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;
- }
-
- ret = bt_private_component_set_user_data(component, dummy);
- if (ret != BT_COMPONENT_STATUS_OK) {
+ if (ret != BT_SELF_COMPONENT_STATUS_OK) {
goto error;
}
+ bt_self_component_set_data(
+ bt_self_component_sink_as_self_component(component), dummy);
goto end;
error:
return ret;
}
-void dummy_port_connected(
- struct bt_private_component *component,
- struct bt_private_port *self_port,
- struct bt_port *other_port)
+BT_HIDDEN
+bt_self_component_status dummy_graph_is_configured(
+ bt_self_component_sink *comp)
{
+ bt_self_component_status status = BT_SELF_COMPONENT_STATUS_OK;
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;
+ bt_self_component_port_input_message_iterator *iterator;
+
+ dummy = bt_self_component_get_data(
+ bt_self_component_sink_as_self_component(comp));
+ BT_ASSERT(dummy);
+ iterator = bt_self_component_port_input_message_iterator_create(
+ bt_self_component_sink_borrow_input_port_by_name(comp,
+ in_port_name));
+ if (!iterator) {
+ status = BT_SELF_COMPONENT_STATUS_NOMEM;
goto end;
}
- BT_MOVE(dummy->notif_iter, iterator);
+ BT_SELF_COMPONENT_PORT_INPUT_MESSAGE_ITERATOR_MOVE_REF(
+ dummy->msg_iter, iterator);
end:
- bt_put(connection);
+ return status;
}
-enum bt_component_status dummy_consume(struct bt_private_component *component)
+BT_HIDDEN
+bt_self_component_status dummy_consume(
+ bt_self_component_sink *component)
{
- enum bt_component_status ret = BT_COMPONENT_STATUS_OK;
- struct bt_notification *notif = NULL;
+ bt_self_component_status ret = BT_SELF_COMPONENT_STATUS_OK;
+ bt_message_array_const msgs;
+ uint64_t count;
struct dummy *dummy;
- enum bt_notification_iterator_status it_ret;
-
- dummy = bt_private_component_get_user_data(component);
- assert(dummy);
+ bt_message_iterator_status it_ret;
+ uint64_t i;
- if (unlikely(dummy->error)) {
- ret = BT_COMPONENT_STATUS_ERROR;
- goto end;
- }
+ dummy = bt_self_component_get_data(
+ bt_self_component_sink_as_self_component(component));
+ BT_ASSERT(dummy);
- if (unlikely(!dummy->notif_iter)) {
- ret = BT_COMPONENT_STATUS_END;
+ if (unlikely(!dummy->msg_iter)) {
+ ret = BT_SELF_COMPONENT_STATUS_END;
goto end;
}
- /* Consume one notification */
- it_ret = bt_notification_iterator_next(dummy->notif_iter);
+ /* Consume one message */
+ it_ret = bt_self_component_port_input_message_iterator_next(
+ dummy->msg_iter, &msgs, &count);
switch (it_ret) {
- case BT_NOTIFICATION_ITERATOR_STATUS_ERROR:
- ret = BT_COMPONENT_STATUS_ERROR;
+ case BT_MESSAGE_ITERATOR_STATUS_OK:
+ ret = BT_SELF_COMPONENT_STATUS_OK;
+
+ for (i = 0; i < count; i++) {
+ bt_message_put_ref(msgs[i]);
+ }
+
+ break;
+ case BT_MESSAGE_ITERATOR_STATUS_AGAIN:
+ ret = BT_SELF_COMPONENT_STATUS_AGAIN;
+ goto end;
+ case BT_MESSAGE_ITERATOR_STATUS_END:
+ ret = BT_SELF_COMPONENT_STATUS_END;
goto end;
- case BT_NOTIFICATION_ITERATOR_STATUS_AGAIN:
- ret = BT_COMPONENT_STATUS_AGAIN;
+ case BT_MESSAGE_ITERATOR_STATUS_ERROR:
+ ret = BT_SELF_COMPONENT_STATUS_ERROR;
goto end;
- case BT_NOTIFICATION_ITERATOR_STATUS_END:
- ret = BT_COMPONENT_STATUS_END;
+ case BT_MESSAGE_ITERATOR_STATUS_NOMEM:
+ ret = BT_SELF_COMPONENT_STATUS_NOMEM;
goto end;
default:
break;
}
end:
- bt_put(notif);
return ret;
}
projects / babeltrace.git / blobdiff