blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-04-27 12:08:52 +08:00
Code Issues Packages Projects Releases Wiki Activity

ctx

Browse Source
This commit is contained in:
withchao 2023-03-23 10:39:03 +08:00
parent 3ea2128b93
commit 2564c2f06e
3 changed files with 9 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
internal/msgtransfer/modify_msg_handler.go
View File

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

2
internal/tools/msg_test.go
View File

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

15
pkg/common/mcontext/ctx.go
View File

@ -20,20 +20,19 @@ func WithTriggerIDContext(ctx context.Context, triggerID string) context.Context
func NewCtx(operationID string) context.Context { func NewCtx(operationID string) context.Context {
c := context.Background() c := context.Background()
ctx := context.WithValue(c, constant.OperationID, operationID) 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 { func GetOperationID(ctx context.Context) string {