From: Philippe Proulx <eeppeliteloop@gmail.com>
Date: Mon, 24 Apr 2017 20:29:39 +0000 (-0400)
Subject: text.pretty: handle AGAIN status
X-Git-Tag: v2.0.0-pre1~356
X-Git-Url: http://git.efficios.com/?a=commitdiff_plain;h=dfa6653cbc3357f3c5509cc8c6d2f0471e51e08f;p=babeltrace.git

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 21272d41..88fc521d 100644
--- 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;