Add "port connected" component class method

[babeltrace.git] / lib / graph / component-class.c

diff --git a/lib/graph/component-class.c b/lib/graph/component-class.c
index 4585b1f3078153a83575ef5ffa35d9df01499936..28aa04c51cca340b548ab1cde507574c590a8ba2 100644 (file)
--- a/lib/graph/component-class.c
+++ b/lib/graph/component-class.c
@@ -256,6 +256,23 @@ end:
        return ret;
 }
 
+int bt_component_class_set_port_connected_method(
+               struct bt_component_class *component_class,
+               bt_component_class_port_connected_method method)
+{
+       int ret = 0;
+
+       if (!component_class || component_class->frozen || !method) {
+               ret = -1;
+               goto end;
+       }
+
+       component_class->methods.port_connected = method;
+
+end:
+       return ret;
+}
+
 int bt_component_class_set_port_disconnected_method(
                struct bt_component_class *component_class,
                bt_component_class_port_disconnected_method method)