blackteay/open-im-server
1
0
Fork 0
mirror of https://github.com/openimsdk/open-im-server.git synced 2025-08-23 06:02:34 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat: implement no gob encoder.

Browse Source
This commit is contained in:
Monet Lee 2024-09-02 17:47:33 +08:00
parent 8c7de0416b
commit 6890da44c9
1 changed files with 8 additions and 13 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
internal/msggateway/encoder.go
View File

@ -15,9 +15,6 @@
package msggateway package msggateway
import ( import (
"bytes"
"encoding/gob"
"github.com/openimsdk/tools/errs" "github.com/openimsdk/tools/errs"
) )
@ -33,19 +30,17 @@ func NewGobEncoder() *GobEncoder {
} }
func (g *GobEncoder) Encode(data any) ([]byte, error) { func (g *GobEncoder) Encode(data any) ([]byte, error) {
buff := bytes.Buffer{} if b, ok := data.([]byte); ok {
enc := gob.NewEncoder(&buff) return b, nil
if err := enc.Encode(data); err != nil {
return nil, errs.WrapMsg(err, "GobEncoder.Encode failed", "action", "encode")
} }
return buff.Bytes(), nil return nil, errs.New("Encoder.Encode failed", "action", "encode")
} }
func (g *GobEncoder) Decode(encodeData []byte, decodeData any) error { func (g *GobEncoder) Decode(encodeData []byte, decodeData any) error {
buff := bytes.NewBuffer(encodeData) if b, ok := decodeData.(*[]byte); ok {
dec := gob.NewDecoder(buff) *b = encodeData
if err := dec.Decode(decodeData); err != nil { return nil
return errs.WrapMsg(err, "GobEncoder.Decode failed", "action", "decode")
} }
return nil
return errs.New("Encoder.Decode failed", "action", "decode")
} }