You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
38 lines
768 B
38 lines
768 B
2 years ago
|
package msggateway
|
||
|
|
|
||
|
|
import (
|
||
|
|
"bytes"
|
||
|
|
"encoding/gob"
|
||
|
|
"github.com/OpenIMSDK/Open-IM-Server/pkg/utils"
|
||
|
|
)
|
||
|
|
|
||
|
|
type Encoder interface {
|
||
|
|
Encode(data interface{}) ([]byte, error)
|
||
|
|
Decode(encodeData []byte, decodeData interface{}) error
|
||
|
|
}
|
||
|
|
|
||
|
|
type GobEncoder struct {
|
||
|
|
}
|
||
|
|
|
||
|
|
func NewGobEncoder() *GobEncoder {
|
||
|
|
return &GobEncoder{}
|
||
|
|
}
|
||
|
|
func (g *GobEncoder) Encode(data interface{}) ([]byte, error) {
|
||
|
|
buff := bytes.Buffer{}
|
||
|
|
enc := gob.NewEncoder(&buff)
|
||
|
|
err := enc.Encode(data)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
return buff.Bytes(), nil
|
||
|
|
}
|
||
|
|
func (g *GobEncoder) Decode(encodeData []byte, decodeData interface{}) error {
|
||
|
|
buff := bytes.NewBuffer(encodeData)
|
||
|
|
dec := gob.NewDecoder(buff)
|
||
|
|
err := dec.Decode(decodeData)
|
||
|
|
if err != nil {
|
||
|
|
return utils.Wrap(err, "")
|
||
|
|
}
|
||
|
|
return nil
|
||
|
|
}
|