#include "common/common.h"
#include "lib/assert-pre.h"
#include "lib/object.h"
-#include <babeltrace2/graph/component-class-sink.h>
-#include <babeltrace2/graph/self-component-sink.h>
+#include <babeltrace2/graph/component-class.h>
#include <babeltrace2/graph/self-component-port.h>
-#include <babeltrace2/graph/self-component-port-input-message-iterator.h>
#include <babeltrace2/graph/self-component.h>
+#include <babeltrace2/graph/message-iterator.h>
#include <glib.h>
#include "component-class-sink-simple.h"
struct bt_component_class_sink *simple_comp_cls;
struct simple_sink_data {
- bt_self_component_port_input_message_iterator *msg_iter;
+ bt_message_iterator *msg_iter;
struct simple_sink_init_method_data init_method_data;
};
static
-enum bt_component_class_init_method_status simple_sink_init(
+enum bt_component_class_initialize_method_status simple_sink_init(
bt_self_component_sink *self_comp,
bt_self_component_sink_configuration *config,
const struct bt_value *params, void *init_method_data)
bt_self_component_sink *self_comp)
{
bt_component_class_sink_graph_is_configured_method_status status;
- bt_self_component_port_input_message_iterator_create_from_sink_component_status
+ bt_message_iterator_create_from_sink_component_status
msg_iter_status;
struct simple_sink_data *data = bt_self_component_get_data(
bt_self_component_sink_as_self_component(self_comp));
}
BT_ASSERT(data);
- msg_iter_status = bt_self_component_port_input_message_iterator_create_from_sink_component(
+ msg_iter_status = bt_message_iterator_create_from_sink_component(
self_comp, self_port, &data->msg_iter);
- if (msg_iter_status != BT_SELF_COMPONENT_PORT_INPUT_MESSAGE_ITERATOR_CREATE_FROM_SINK_COMPONENT_STATUS_OK) {
+ if (msg_iter_status != BT_MESSAGE_ITERATOR_CREATE_FROM_SINK_COMPONENT_STATUS_OK) {
BT_LIB_LOGE_APPEND_CAUSE(
"Cannot create input port message iterator: "
"%![comp-]+c, %![port-]+p", self_comp, self_port);
}
if (data->init_method_data.init_func) {
- bt_graph_simple_sink_component_init_func_status init_status;
+ bt_graph_simple_sink_component_initialize_func_status init_status;
/* Call user's initialization function */
init_status = data->init_method_data.init_func(data->msg_iter,
data->init_method_data.user_data);
- if (init_status != BT_GRAPH_SIMPLE_SINK_COMPONENT_INIT_FUNC_STATUS_OK) {
+ if (init_status != BT_GRAPH_SIMPLE_SINK_COMPONENT_INITIALIZE_FUNC_STATUS_OK) {
BT_LIB_LOGW_APPEND_CAUSE(
"Simple sink component's user's initialization function failed: "
"status=%s, %![comp-]+c, %![port-]+p",
struct simple_sink_data *data = bt_self_component_get_data(
bt_self_component_sink_as_self_component(self_comp));
- BT_ASSERT(data);
- BT_ASSERT(data->init_method_data.consume_func);
- BT_ASSERT(data->msg_iter);
+ BT_ASSERT_DBG(data);
+ BT_ASSERT_DBG(data->init_method_data.consume_func);
+ BT_ASSERT_DBG(data->msg_iter);
/* Call user's "consume" function */
status = data->init_method_data.consume_func(data->msg_iter,
data->init_method_data.user_data);
- if (status != BT_FUNC_STATUS_OK) {
+ if (status < 0) {
BT_LIB_LOGW_APPEND_CAUSE(
"Simple sink component's user's \"consume\" function failed: "
"status=%s, %![comp-]+c",
{
enum bt_component_class_set_method_status set_method_status;
+ BT_ASSERT_PRE_NO_ERROR();
+
if (simple_comp_cls) {
goto end;
}
goto end;
}
- set_method_status = bt_component_class_sink_set_init_method(
+ set_method_status = bt_component_class_sink_set_initialize_method(
simple_comp_cls, simple_sink_init);
BT_ASSERT(set_method_status == BT_FUNC_STATUS_OK);
set_method_status = bt_component_class_sink_set_finalize_method(