Makes calling consumer.close() multiple times safe. Throws a

johnistan · johnistan · commit d5a046a7bf4c · 2017-10-03T13:44:40.000Z
RuntimeError not a segfaults
diff --git a/confluent_kafka/src/Consumer.c b/confluent_kafka/src/Consumer.c
@@ -550,17 +550,24 @@ static PyObject *Consumer_poll (Handle *self, PyObject *args,
 
 
 static PyObject *Consumer_close (Handle *self, PyObject *ignore) {
-	CallState cs;
 
-	CallState_begin(self, &cs);
+	if (!self->rk) {
+            PyErr_SetString(PyExc_RuntimeError,
+                            "Consumer already closed");
+            return NULL;
+    }
 
-	rd_kafka_consumer_close(self->rk);
+    CallState cs;
 
-	rd_kafka_destroy(self->rk);
-	self->rk = NULL;
+    CallState_begin(self, &cs);
 
-        if (!CallState_end(self, &cs))
-                return NULL;
+    rd_kafka_consumer_close(self->rk);
+
+    rd_kafka_destroy(self->rk);
+    self->rk = NULL;
+
+    if (!CallState_end(self, &cs))
+            return NULL;
 
 	Py_RETURN_NONE;
 }
diff --git a/tests/test_Consumer.py b/tests/test_Consumer.py
@@ -158,3 +158,22 @@ def poll(self, somearg):
     sc = SubConsumer({"group.id": "test", "session.timeout.ms": "90"})
     sc.poll("astring")
     sc.close()
+
+
+def test_multiple_close_throw_exception():
+    """ Calling Consumer.close() multiple times should throw Runtime Exception
+    """
+    c = Consumer({'group.id': 'test',
+                  'enable.auto.commit': True,
+                  'enable.auto.offset.store': False,
+                  'socket.timeout.ms': 50,
+                  'session.timeout.ms': 100})
+
+    c.subscribe(["test"])
+
+    c.unsubscribe()
+    c.close()
+
+    with pytest.raises(RuntimeError) as ex:
+        c.close()
+        assert 'Consumer already closed' == str(ex.value)
