consumer: added close! with timeout argument

pyr · Jan 3, 2022 · 7629c0a · 7629c0a
1 parent b106ae9
commit 7629c0a
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/kinsky/client.clj b/src/kinsky/client.clj
@@ -533,6 +533,10 @@
        GenericDriver
        (close! [this]
          (.close consumer))
+       (close! [this timeout]
+         (if (nil? timeout)
+           (.close consumer)
+           (.close consumer (long timeout) TimeUnit/MILLISECONDS)))
        MetadataDriver
        (partitions-for [this topic]
          (mapv partition-info->data (.partitionsFor consumer topic)))