From b27dcfae957940e675bbf2bd4eaf8668f12a38c7 Mon Sep 17 00:00:00 2001
From: cyruslo <710058301@qq.com>
Date: Sun, 28 Jul 2024 23:57:26 +0800
Subject: [PATCH] fix: batch channel not close when client close issue(#100)

Signed-off-by: cyruslo <710058301@qq.com>
Signed-off-by: ZhangJian He <shoothzj@gmail.com>
Co-authored-by: ZhangJian He <shoothzj@gmail.com>
Signed-off-by: ZhangJian He <shoothzj@gmail.com>
---
 opengemini/client_impl.go | 4 ++++
 1 file changed, 4 insertions(+)

diff --git a/opengemini/client_impl.go b/opengemini/client_impl.go
index cd6d04f..ac79c02 100644
--- a/opengemini/client_impl.go
+++ b/opengemini/client_impl.go
@@ -76,6 +76,10 @@ func newClient(c *Config) (Client, error) {
 func (c *client) Close() error {
 	c.batchContextCancel()
 	c.dataChanMap.Range(func(key, value interface{}) bool {
+		cb, ok := value.(chan *sendBatchWithCB)
+		if ok {
+			close(cb)
+		}
 		c.dataChanMap.Delete(key)
 		return true
 	})