LaconicNetwork/cosmos-sdk
10
0
Fork 0
Code Issues Pull Requests Packages Projects Releases 2 Wiki Activity

feat(runtime/v2): Add pre and post msg handler register (#21346)

Browse Source
This commit is contained in:
son trinh 2024-08-19 17:26:56 +07:00 committed by GitHub
parent c50c6b2549
commit 294b608022
No known key found for this signature in database
GPG Key ID: B5690EEEBB952194
2 changed files with 20 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

20
core/appmodule/v2/handlers.go
View File

@ -19,12 +19,12 @@ type (
// msg handler
type PreMsgRouter interface {
// Register will register a specific message handler hooking into the message with
// RegisterPreHandler will register a specific message handler hooking into the message with
// the provided name.
Register(msgName string, handler PreMsgHandler)
// RegisterGlobal will register a global message handler hooking into any message
RegisterPreHandler(msgName string, handler PreMsgHandler)
// RegisterGlobalPreHandler will register a global message handler hooking into any message
// being executed.
RegisterGlobal(handler PreMsgHandler)
RegisterGlobalPreHandler(handler PreMsgHandler)
}
type HasPreMsgHandlers interface {
@ -40,11 +40,15 @@ type HasMsgHandlers interface {
}
type PostMsgRouter interface {
// Register will register a specific message handler hooking after the execution of message with
// RegisterPostHandler will register a specific message handler hooking after the execution of message with
// the provided name.
Register(msgName string, handler PostMsgHandler)
// RegisterGlobal will register a global message handler hooking after the execution of any message.
RegisterGlobal(handler PreMsgHandler)
RegisterPostHandler(msgName string, handler PostMsgHandler)
// RegisterGlobalPostHandler will register a global message handler hooking after the execution of any message.
RegisterGlobalPostHandler(handler PostMsgHandler)
}
type HasPostMsgHandlers interface {
RegisterPostMsgHandlers(router PostMsgRouter)
}
// query handler

9
runtime/v2/manager.go
View File

@ -424,7 +424,14 @@ func (m *MM[T]) RegisterServices(app *App[T]) error {
}
}
// TODO: register pre and post msg
// register pre and post msg
if module, ok := module.(appmodulev2.HasPreMsgHandlers); ok {
module.RegisterPreMsgHandlers(app.msgRouterBuilder)
}
if module, ok := module.(appmodulev2.HasPostMsgHandlers); ok {
module.RegisterPostMsgHandlers(app.msgRouterBuilder)
}
}
return nil