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

feat add afterSendFilters mock

Browse Source
This commit is contained in:
ltf 2022-02-28 16:12:13 +08:00
parent 7a6568a4fe
commit f1229be04c
2 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
cmd/rpc/open_im_msg/filters/mock.go
View File

@ -37,3 +37,16 @@ func MockBeforeSendFilter2(ctx *rpcChat.SendContext, pb *pbChat.SendMsgReq) (*pb
return nil, true, nil return nil, true, nil
} }
// MockAfterSendFilter is a mock handle that handles custom logic after send msg.
func MockAfterSendFilter(ctx *rpcChat.SendContext, pb *pbChat.SendMsgReq, res *pbChat.SendMsgResp) (*pbChat.SendMsgResp, bool, error) {
ctxKey := "test_key"
v, ok := ctx.Value(ctxKey).(bool)
if ok {
fmt.Printf("MockAfterSendFilter:%s selected from ctx,value is :%v\n", ctxKey, v)
}
fmt.Printf("MockAfterSendFilter trigger,contentType:%d\n", pb.MsgData.GetContentType())
return res, true, nil
}

2
cmd/rpc/open_im_msg/main.go
View File

@ -15,6 +15,8 @@ func main() {
// mock 注册发送前的拦截器 // mock 注册发送前的拦截器
rpcServer.UseBeforSendFilters(filters.MockBeforeSendFilter1, filters.MockBeforeSendFilter2) rpcServer.UseBeforSendFilters(filters.MockBeforeSendFilter1, filters.MockBeforeSendFilter2)
rpcServer.UseAfterSendFilters(filters.MockAfterSendFilter)
// //
rpcServer.Run() rpcServer.Run()
} }