Wait for all messages to be received in subscribe publish sample

Fixes #73
aws · Oct 12, 2017 · 9b84c7b · 9b84c7b
1 parent b5fc155
commit 9b84c7b
Showing 1 changed file with 7 additions and 0 deletions.
diff --git a/samples/linux/subscribe_publish_sample/subscribe_publish_sample.c b/samples/linux/subscribe_publish_sample/subscribe_publish_sample.c
@@ -240,6 +240,10 @@ int main(int argc, char **argv) {
 			publishCount--;
 		}
 
+		if(publishCount == 0) {
+			break;
+		}
+
 		sprintf(cPayload, "%s : %d ", "hello from SDK QOS1", i++);
 		paramsQOS1.payloadLen = strlen(cPayload);
 		rc = aws_iot_mqtt_publish(&client, "sdkTest/sub", 11, &paramsQOS1);
@@ -252,6 +256,9 @@ int main(int argc, char **argv) {
 		}
 	}
 
+	// Wait for all the messages to be received
+	aws_iot_mqtt_yield(&client, 100);
+
 	if(SUCCESS != rc) {
 		IOT_ERROR("An error occurred in the loop.\n");
 	} else {