fixed: internalBatchSend issue(#100)

Signed-off-by: cyruslo <[email protected]>
openGemini · Jul 28, 2024 · a2222d4 · a2222d4
1 parent 41fbeca
commit a2222d4
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/opengemini/write.go b/opengemini/write.go
@@ -27,6 +27,11 @@ type sendBatchWithCB struct {
 	callback WriteCallback
 }
 
+func (s *sendBatchWithCB) Clear() {
+	s.point = nil
+	s.callback(fmt.Errorf("sendBatchWithCB timeout"))
+}
+
 type dbRp struct {
 	db string
 	rp string
@@ -129,6 +134,9 @@ func (c *client) internalBatchSend(ctx context.Context, database string, rp stri
 		select {
 		case <-ctx.Done():
 			ticker.Stop()
+			for record := range resource {
+				record.Clear()
+			}
 			return
 		case <-ticker.C:
 			needFlush = true