Commit
This commit does not belong to any branch on this repository, and may belong to a fork outside of the repository.
feat(server/v2/stf): delayed marshalling of typed event (#22684)
(cherry picked from commit 7fa2356) # Conflicts: # server/v2/stf/core_event_service.go
- Loading branch information
1 parent
a595e3c
commit db69831
There are no files selected for viewing
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
This file contains bidirectional Unicode text that may be interpreted or compiled differently than what appears below. To review, open the file in an editor that reveals hidden Unicode characters.
Learn more about bidirectional Unicode characters
| Original file line number | Diff line number | Diff line change |
|---|---|---|
| @@ -0,0 +1,112 @@ | ||
| package stf | ||
|
|
||
| import ( | ||
| "bytes" | ||
| "context" | ||
| "encoding/json" | ||
| "maps" | ||
| "slices" | ||
|
|
||
| "github.com/cosmos/gogoproto/jsonpb" | ||
| gogoproto "github.com/cosmos/gogoproto/proto" | ||
|
|
||
| "cosmossdk.io/core/event" | ||
| "cosmossdk.io/core/transaction" | ||
| ) | ||
|
|
||
| func NewEventService() event.Service { | ||
| return eventService{} | ||
| } | ||
|
|
||
| type eventService struct{} | ||
|
|
||
| // EventManager implements event.Service. | ||
| func (eventService) EventManager(ctx context.Context) event.Manager { | ||
| exCtx, err := getExecutionCtxFromContext(ctx) | ||
|
Check failure on line 25 in server/v2/stf/core_event_service.go
|
||
| if err != nil { | ||
| panic(err) | ||
| } | ||
|
|
||
| return &eventManager{exCtx} | ||
| } | ||
|
|
||
| var _ event.Manager = (*eventManager)(nil) | ||
|
|
||
| type eventManager struct { | ||
| executionContext *executionContext | ||
|
Check failure on line 36 in server/v2/stf/core_event_service.go
|
||
| } | ||
|
|
||
| // Emit emits an typed event that is defined in the protobuf file. | ||
| // In the future these events will be added to consensus. | ||
| func (em *eventManager) Emit(tev transaction.Msg) error { | ||
| ev := event.Event{ | ||
| Type: gogoproto.MessageName(tev), | ||
| Attributes: func() ([]event.Attribute, error) { | ||
| outerEvent, err := TypedEventToEvent(tev) | ||
| if err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| return outerEvent.Attributes() | ||
| }, | ||
| Data: func() (json.RawMessage, error) { | ||
| buf := new(bytes.Buffer) | ||
| jm := &jsonpb.Marshaler{OrigName: true, EmitDefaults: true, AnyResolver: nil} | ||
| if err := jm.Marshal(buf, tev); err != nil { | ||
| return nil, err | ||
| } | ||
|
|
||
| return buf.Bytes(), nil | ||
| }, | ||
| } | ||
|
|
||
| em.executionContext.events = append(em.executionContext.events, ev) | ||
| return nil | ||
| } | ||
|
|
||
| // EmitKV emits a key value pair event. | ||
| func (em *eventManager) EmitKV(eventType string, attrs ...event.Attribute) error { | ||
| ev := event.Event{ | ||
| Type: eventType, | ||
| Attributes: func() ([]event.Attribute, error) { | ||
| return attrs, nil | ||
| }, | ||
| Data: func() (json.RawMessage, error) { | ||
| return json.Marshal(attrs) | ||
| }, | ||
| } | ||
|
|
||
| em.executionContext.events = append(em.executionContext.events, ev) | ||
| return nil | ||
| } | ||
|
|
||
| // TypedEventToEvent takes typed event and converts to Event object | ||
| func TypedEventToEvent(tev transaction.Msg) (event.Event, error) { | ||
| evtType := gogoproto.MessageName(tev) | ||
| buf := new(bytes.Buffer) | ||
| jm := &jsonpb.Marshaler{OrigName: true, EmitDefaults: true, AnyResolver: nil} | ||
| if err := jm.Marshal(buf, tev); err != nil { | ||
| return event.Event{}, err | ||
| } | ||
|
|
||
| var attrMap map[string]json.RawMessage | ||
| if err := json.Unmarshal(buf.Bytes(), &attrMap); err != nil { | ||
| return event.Event{}, err | ||
| } | ||
|
|
||
| // sort the keys to ensure the order is always the same | ||
| keys := slices.Sorted(maps.Keys(attrMap)) | ||
| attrs := make([]event.Attribute, 0, len(attrMap)) | ||
| for _, k := range keys { | ||
| v := attrMap[k] | ||
| attrs = append(attrs, event.Attribute{ | ||
| Key: k, | ||
| Value: string(v), | ||
| }) | ||
| } | ||
|
|
||
| return event.Event{ | ||
| Type: evtType, | ||
| Attributes: func() ([]event.Attribute, error) { return attrs, nil }, | ||
| }, nil | ||
| } | ||
Oops, something went wrong.