fix: panic when prepare write failed

deepflowio · Nov 20, 2024 · 3837e76 · 3837e76
1 parent b5ba3ff
commit 3837e76
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/server/ingester/pkg/ckwriter/ckwriter.go b/server/ingester/pkg/ckwriter/ckwriter.go
@@ -383,12 +383,12 @@ func (c *Cache) Write() error {
 	if c.batchSize == 0 {
 		return nil
 	}
-	var err error
 	conn := c.conns[c.writeCounter%c.connCount]
-	c.batch, err = conn.PrepareReuseBatch(context.Background(), c.prepare, c.batch)
+	batch, err := conn.PrepareReuseBatch(context.Background(), c.prepare, c.batch)
 	if err != nil {
 		return fmt.Errorf("prepare reuse batch item write block failed: %s", err)
 	}
+	c.batch = batch
 
 	err = c.batch.Send()
 	c.batch.Reset()