71 lines
1.6 KiB
Go
71 lines
1.6 KiB
Go
|
package frame
|
||
|
|
|
||
|
|
import (
|
||
|
|
coder "git.hpds.cc/Component/mq_coder"
|
||
|
|
)
|
||
|
|
|
||
|
|
// BackFlowFrame is a coder encoded bytes
|
||
|
|
// It's used to receive stream function processed result
|
||
|
|
type BackFlowFrame struct {
|
||
|
|
Tag byte
|
||
|
|
Carriage []byte
|
||
|
|
}
|
||
|
|
|
||
|
|
// NewBackFlowFrame creates a new BackFlowFrame with a given tag and carriage
|
||
|
|
func NewBackFlowFrame(tag byte, carriage []byte) *BackFlowFrame {
|
||
|
|
return &BackFlowFrame{
|
||
|
|
Tag: tag,
|
||
|
|
Carriage: carriage,
|
||
|
|
}
|
||
|
|
}
|
||
|
|
|
||
|
|
// Type gets the type of Frame.
|
||
|
|
func (f *BackFlowFrame) Type() Type {
|
||
|
|
return TagOfBackFlowFrame
|
||
|
|
}
|
||
|
|
|
||
|
|
// SetCarriage sets the user's raw data
|
||
|
|
func (f *BackFlowFrame) SetCarriage(buf []byte) *BackFlowFrame {
|
||
|
|
f.Carriage = buf
|
||
|
|
return f
|
||
|
|
}
|
||
|
|
|
||
|
|
// Encode to coder encoded bytes
|
||
|
|
func (f *BackFlowFrame) Encode() []byte {
|
||
|
|
carriage := coder.NewPrimitivePacketEncoder(f.Tag)
|
||
|
|
carriage.SetBytesValue(f.Carriage)
|
||
|
|
|
||
|
|
node := coder.NewNodePacketEncoder(byte(TagOfBackFlowFrame))
|
||
|
|
node.AddPrimitivePacket(carriage)
|
||
|
|
|
||
|
|
return node.Encode()
|
||
|
|
}
|
||
|
|
|
||
|
|
// GetDataTag return the Tag of user's data
|
||
|
|
func (f *BackFlowFrame) GetDataTag() byte {
|
||
|
|
return f.Tag
|
||
|
|
}
|
||
|
|
|
||
|
|
// GetCarriage return data
|
||
|
|
func (f *BackFlowFrame) GetCarriage() []byte {
|
||
|
|
return f.Carriage
|
||
|
|
}
|
||
|
|
|
||
|
|
// DecodeToBackFlowFrame decodes coder encoded bytes to BackFlowFrame
|
||
|
|
func DecodeToBackFlowFrame(buf []byte) (*BackFlowFrame, error) {
|
||
|
|
nodeBlock := coder.NodePacket{}
|
||
|
|
_, err := coder.DecodeToNodePacket(buf, &nodeBlock)
|
||
|
|
if err != nil {
|
||
|
|
return nil, err
|
||
|
|
}
|
||
|
|
|
||
|
|
payload := &BackFlowFrame{}
|
||
|
|
for _, v := range nodeBlock.PrimitivePackets {
|
||
|
|
payload.Tag = v.SeqId()
|
||
|
|
payload.Carriage = v.GetValBuf()
|
||
|
|
break
|
||
|
|
}
|
||
|
|
|
||
|
|
return payload, nil
|
||
|
|
}
|