text.pretty: handle AGAIN status

Signed-off-by: Philippe Proulx <eeppeliteloop@gmail.com>
Signed-off-by: Jérémie Galarneau <jeremie.galarneau@efficios.com>

diff --git a/plugins/text/pretty/pretty.c b/plugins/text/pretty/pretty.c
index 21272d41204ac92276e62543d9ff0bc67c9e435b..88fc521d60cd3562d054687721cfefaca6cc7b0a 100644 (file)
--- a/plugins/text/pretty/pretty.c
+++ b/plugins/text/pretty/pretty.c
@@ -186,13 +186,20 @@ enum bt_component_status pretty_consume(struct bt_private_component *component)
                ret = BT_COMPONENT_STATUS_END;
                BT_PUT(pretty->input_iterator);
                goto end;
-       default:
+       case BT_NOTIFICATION_ITERATOR_STATUS_AGAIN:
+               ret = BT_COMPONENT_STATUS_AGAIN;
+               goto end;
+       case BT_NOTIFICATION_ITERATOR_STATUS_OK:
                break;
+       default:
+               ret = BT_COMPONENT_STATUS_ERROR;
+               goto end;
        }
 
        notification = bt_notification_iterator_get_notification(it);
        assert(notification);
        ret = handle_notification(pretty, notification);
+
 end:
        bt_put(notification);
        return ret;