diff options
Diffstat (limited to 'internal/packet/encoders.go')
| -rw-r--r-- | internal/packet/encoders.go | 17 |
1 files changed, 9 insertions, 8 deletions
diff --git a/internal/packet/encoders.go b/internal/packet/encoders.go index f5841ce..760c11c 100644 --- a/internal/packet/encoders.go +++ b/internal/packet/encoders.go @@ -3,11 +3,12 @@ package packet import ( "encoding/json" - "github.com/kyren223/eko/pkg/assert" "github.com/vmihailenco/msgpack/v5" + + "github.com/kyren223/eko/pkg/assert" ) -type TypedMessage interface { +type Payload interface { Type() PacketType } @@ -29,24 +30,24 @@ func (e defaultPacketEncoder) Payload() []byte { return e.data } -func NewJsonEncoder(message TypedMessage) PacketEncoder { - data, err := json.Marshal(message) +func NewJsonEncoder(payload Payload) PacketEncoder { + data, err := json.Marshal(payload) assert.NoError(err, "encoding a message with JSON should never fail") return defaultPacketEncoder{ data: data, encoding: EncodingJson, - packetType: message.Type(), + packetType: payload.Type(), } } -func NewMsgPackEncoder(message TypedMessage) PacketEncoder { - data, err := msgpack.Marshal(message) +func NewMsgPackEncoder(payload Payload) PacketEncoder { + data, err := msgpack.Marshal(payload) assert.NoError(err, "encoding a message with msg pack should never fail") return defaultPacketEncoder{ data: data, encoding: EncodingMsgPack, - packetType: message.Type(), + packetType: payload.Type(), } } |
