fix: batch channel not close when client close issue(#100)

Signed-off-by: cyruslo <[email protected]> Signed-off-by: ZhangJian He <[email protected]> Co-authored-by: ZhangJian He <[email protected]> Signed-off-by: ZhangJian He <[email protected]>
openGemini · Aug 4, 2024 · b27dcfa · b27dcfa
1 parent 13fba1e
commit b27dcfa
Showing 1 changed file with 4 additions and 0 deletions.
diff --git 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
 	})