Merge pull request #10 from takaomag/spike_deliver_msg_at_exit

Add rd_kafka_poll() at exit. See issue #9 for detail.
brg-liuwei · Oct 31, 2016 · 54531ba · 54531ba
2 parents 17676b1 + cab7427
commit 54531ba
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/ngx_http_kafka_module.c b/ngx_http_kafka_module.c
@@ -382,6 +382,12 @@ void ngx_http_kafka_exit_worker(ngx_cycle_t *cycle)
 
     main_conf = ngx_http_cycle_get_module_main_conf(cycle, ngx_http_kafka_module);
 
+    rd_kafka_poll(main_conf->rk, 0);
+
+    while (rd_kafka_outq_len(main_conf->rk) > 0) {
+        rd_kafka_poll(main_conf->rk, 100);
+    }
+
     // TODO: rd_kafka_topic_destroy(each loc conf rkt );
     rd_kafka_destroy(main_conf->rk);
 }