fix: avoid assigning partitions if consumer is not running (#60)

Co-authored-by: Miguel Ángel Casado Hernández <miguelangel.casado@orange.com>
silviucpp · Jun 28, 2023 · 011b00b · 011b00b
1 parent 87b35ae
commit 011b00b
Showing 1 changed file with 6 additions and 0 deletions.
diff --git a/c_src/erlkaf_consumer.cc b/c_src/erlkaf_consumer.cc
@@ -103,6 +103,12 @@ ERL_NIF_TERM partition_list_to_nif(ErlNifEnv* env, enif_consumer* consumer, rd_k
 
 void assign_partitions(ErlNifEnv* env, enif_consumer* consumer, rd_kafka_t *rk, rd_kafka_topic_partition_list_t *partitions)
 {
+    if(!consumer->running)
+    {
+        rd_kafka_assign(rk, NULL);
+        return;
+    }
+
     rd_kafka_resp_err_t response = rd_kafka_assign(rk, partitions);
 
     if(response != RD_KAFKA_RESP_ERR_NO_ERROR)