live-gateway/douyu/codec.go

package douyu

import (
	"bytes"
	"fmt"
	"github.com/pkg/errors"
	"live-gateway/douyu/stt"
	"live-gateway/ws"
)

type CodecDouyu struct {
}

func NewCodecDouyu() ws.Codec {
	return &CodecDouyu{}
}

// Encode encodes TypedData to douyu.WsEntry
func (c *CodecDouyu) Encode(v interface{}) (interface{}, error) {
	data, ok := v.(TypedData)
	if !ok {
		return nil, errors.New("[Codec-Douyu] 写入值类型必须实现 douyu.TypedData 接口")
	}

	bf := bytes.NewBuffer([]byte{})
	bf.WriteString(fmt.Sprintf("type@=%s/", data.DataType()))

	sttData, err := stt.Marshal(v)
	if err != nil {
		return nil, err
	}
	bf.Write(sttData)

	resp := &WsEntry{
		data:    bf.Bytes(),
		msgType: TypeMessageToServer,
	}
	//logger.SLog.Debugf("发送消息: %s", string(resp.data))
	return resp, nil
}

func (c *CodecDouyu) Decode(customEntry interface{}) (interface{}, error) {
	// 处理data
	entry, ok := customEntry.(*WsEntry)
	if !ok {
		return nil, errors.New(fmt.Sprintf("[Codec-Douyu] 写入值类型必须为%T", WsEntry{}))
	}
	var typed struct {
		Type string `stt:"type"`
	}
	err := stt.Unmarshal(entry.data, &typed)
	if err != nil {
		return nil, errors.New("[Codec-Douyu] 获取type字段失败")
	}
	entry.dataType = typed.Type
	return entry, nil
}