Add manual ID option

Signed-off-by: Congqi Xia <[email protected]>
milvus-io · May 8, 2024 · 4305e9d · 4305e9d
1 parent 9f3b941
commit 4305e9d
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/states/consume.go b/states/consume.go
@@ -25,6 +25,7 @@ type ConsumeParam struct {
 	Topic               string `name:"topic" default:"" desc:"topic to consume"`
 	ShardName           string `name:"shard_name" default:"" desc:"shard name(vchannel name) to filter with"`
 	Detail              bool   `name:"detail" default:"false" desc:"print msg detail"`
+	ManualID            int64  `name:"manual_id" default:"0" desc"manual id"`
 }
 
 func (s *InstanceState) ConsumeCommand(ctx context.Context, p *ConsumeParam) error {
@@ -49,6 +50,8 @@ func (s *InstanceState) ConsumeCommand(ctx context.Context, p *ConsumeParam) err
 				messageID = kafka.DeserializeKafkaID(checkpoint.GetMsgID())
 			}
 		}
+	case "manual":
+		messageID = kafka.DeserializeKafkaID(kafka.SerializeKafkaID(p.ManualID))
 	default:
 	}