lib: add "borrow" functions where "get" functions exist

[babeltrace.git] / include / babeltrace / graph / notification-iterator.h

diff --git a/include/babeltrace/graph/notification-iterator.h b/include/babeltrace/graph/notification-iterator.h
index 4d94e22c4a2ce9998ea961a08b2c77c4329b5ffc..a062f2e4d581142a8853c0feec7a36c110449a23 100644 (file)
--- a/include/babeltrace/graph/notification-iterator.h
+++ b/include/babeltrace/graph/notification-iterator.h
@@ -27,6 +27,9 @@
  * SOFTWARE.
  */
 
+/* For bt_get() */
+#include <babeltrace/ref.h>
+
 #include <stdint.h>
 
 #ifdef __cplusplus
@@ -58,6 +61,9 @@ enum bt_notification_iterator_status {
        BT_NOTIFICATION_ITERATOR_STATUS_UNSUPPORTED = -2,
 };
 
+extern struct bt_notification *bt_notification_iterator_borrow_notification(
+               struct bt_notification_iterator *iterator);
+
 /**
  * Get current notification at iterator's position.
  *
@@ -69,8 +75,12 @@ enum bt_notification_iterator_status {
  *
  * @see bt_put()
  */
-extern struct bt_notification *bt_notification_iterator_get_notification(
-               struct bt_notification_iterator *iterator);
+static inline
+struct bt_notification *bt_notification_iterator_get_notification(
+               struct bt_notification_iterator *iterator)
+{
+       return bt_get(bt_notification_iterator_borrow_notification(iterator));
+}
 
 /**
  * Advance the iterator's position forward.