增加是否强制提交参数

zeromicro · Dec 14, 2022 · a7e4d69 · a7e4d69
1 parent 0d3a3e5
commit a7e4d69
Show file tree

Hide file tree

Showing 2 changed files with 17 additions and 13 deletions.
diff --git a/kq/config.go b/kq/config.go
@@ -9,15 +9,16 @@ const (
 
 type KqConf struct {
 	service.ServiceConf
-	Brokers    []string
-	Group      string
-	Topic      string
-	Offset     string `json:",options=first|last,default=last"`
-	Conns      int    `json:",default=1"`
-	Consumers  int    `json:",default=8"`
-	Processors int    `json:",default=8"`
-	MinBytes   int    `json:",default=10240"`    // 10K
-	MaxBytes   int    `json:",default=10485760"` // 10M
-	Username   string `json:",optional"`
-	Password   string `json:",optional"`
+	Brokers     []string
+	Group       string
+	Topic       string
+	Offset      string `json:",options=first|last,default=last"`
+	Conns       int    `json:",default=1"`
+	Consumers   int    `json:",default=8"`
+	Processors  int    `json:",default=8"`
+	MinBytes    int    `json:",default=10240"`    // 10K
+	MaxBytes    int    `json:",default=10485760"` // 10M
+	Username    string `json:",optional"`
+	Password    string `json:",optional"`
+	ForceCommit bool   `json:",default=true"`
 }
diff --git a/kq/queue.go b/kq/queue.go
@@ -159,9 +159,12 @@ func (q *kafkaQueue) startConsumers() {
 			for msg := range q.channel {
 				if err := q.consumeOne(string(msg.Key), string(msg.Value)); err != nil {
 					logx.Errorf("Error on consuming: %s, error: %v", string(msg.Value), err)
-				} else {
-					q.consumer.CommitMessages(context.Background(), msg)
+					if q.c.ForceCommit {
+						q.consumer.CommitMessages(context.Background(), msg)
+					}
+					continue
 				}
+				q.consumer.CommitMessages(context.Background(), msg)
 			}
 		})
 	}