Always evaluate BT_ASSERT(); add BT_ASSERT_DBG() for debug mode only

[babeltrace.git] / src / lib / graph / component-sink.c

diff --git a/src/lib/graph/component-sink.c b/src/lib/graph/component-sink.c
index b218144d145d863f3630843c21880a3b6d2e6d87..b848a019acba3932a7d054efdb326f5a4aedf1a1 100644 (file)
--- a/src/lib/graph/component-sink.c
+++ b/src/lib/graph/component-sink.c
@@ -26,6 +26,7 @@
 
 #include "common/assert.h"
 #include "lib/assert-pre.h"
+#include "lib/assert-post.h"
 #include "compat/compiler.h"
 #include <babeltrace2/value.h>
 #include <babeltrace2/graph/self-component-sink.h>
@@ -34,6 +35,7 @@
 
 #include "component-sink.h"
 #include "component.h"
+#include "graph.h"
 #include "lib/func-status.h"
 
 BT_HIDDEN
@@ -68,8 +70,8 @@ bt_component_sink_borrow_class_const(
 
        cls = component->parent.class;
 
-       BT_ASSERT(cls);
-       BT_ASSERT(cls->type == BT_COMPONENT_CLASS_TYPE_SINK);
+       BT_ASSERT_DBG(cls);
+       BT_ASSERT_DBG(cls->type == BT_COMPONENT_CLASS_TYPE_SINK);
 
        return (bt_component_class_sink *) cls;
 }
@@ -141,6 +143,16 @@ end:
        return status;
 }
 
+bt_bool bt_self_component_sink_is_interrupted(
+               const struct bt_self_component_sink *self_comp)
+{
+       struct bt_component *comp = (void *) self_comp;
+
+       BT_ASSERT_PRE_NON_NULL(comp, "Component");
+       return (bt_bool) bt_graph_is_interrupted(
+               bt_component_borrow_graph(comp));
+}
+
 void bt_component_sink_get_ref(
                const struct bt_component_sink *component_sink)
 {