ctx

internal/msgtransfer/modify_msg_handler.go

@@ -66,7 +66,7 @@ func (mmc *ModifyMsgConsumerHandler) ModifyMsg(ctx context.Context, cMsg *sarama
 		if !isReactionFromCache {
 			continue
 		}
-		mcontext.SetOperationID(ctx, operationID)
+		ctx = mcontext.SetOperationID(ctx, operationID)
 		if msgDataToMQ.MsgData.ContentType == constant.ReactionMessageModifier {
 			notification := &apistruct.ReactionMessageModifierNotification{}
 			if err := json.Unmarshal(msgDataToMQ.MsgData.Content, notification); err != nil {

internal/tools/msg_test.go

@@ -66,7 +66,7 @@ func TestDeleteMongoMsgAndResetRedisSeq(t *testing.T) {
 	mongoClient := mgo.GetDatabase().Collection(unRelationTb.MsgDocModel{}.TableName())
 
 	ctx := context.Background()
-	mcontext.SetOperationID(ctx, operationID)
+	ctx = mcontext.SetOperationID(ctx, operationID)
 	testUID1 := "test_del_id1"
 	_, err = mongoClient.DeleteOne(ctx, bson.M{"uid": testUID1 + ":" + strconv.Itoa(0)})
 	if err != nil {

pkg/common/mcontext/ctx.go

@@ -20,20 +20,19 @@ func WithTriggerIDContext(ctx context.Context, triggerID string) context.Context
 func NewCtx(operationID string) context.Context {
 	c := context.Background()
 	ctx := context.WithValue(c, constant.OperationID, operationID)
-	SetOperationID(ctx, operationID)
+	return SetOperationID(ctx, operationID)
-	return ctx
 }
 
-func SetOperationID(ctx context.Context, operationID string) {
+func SetOperationID(ctx context.Context, operationID string) context.Context {
-	ctx = context.WithValue(ctx, constant.OperationID, operationID)
+	return context.WithValue(ctx, constant.OperationID, operationID)
 }
 
-func SetOpUserID(ctx context.Context, opUserID string) {
+func SetOpUserID(ctx context.Context, opUserID string) context.Context {
-	ctx = context.WithValue(ctx, constant.OpUserID, opUserID)
+	return context.WithValue(ctx, constant.OpUserID, opUserID)
 }
 
-func SetConnID(ctx context.Context, connID string) {
+func SetConnID(ctx context.Context, connID string) context.Context {
-	ctx = context.WithValue(ctx, constant.ConnID, connID)
+	return context.WithValue(ctx, constant.ConnID, connID)
 }
 
 func GetOperationID(ctx context.Context) string {