移除不需要的系统
This commit is contained in:
liwei1dao
parent
ce0cc9385f
commit
601eabaa7e
@ -1,63 +0,0 @@
|
||||
package proto
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"reflect"
|
||||
)
|
||||
|
||||
type (
|
||||
IMessage interface {
|
||||
GetComId() uint16
|
||||
GetMsgId() uint16
|
||||
GetMsgLen() uint32
|
||||
GetBuffer() []byte
|
||||
ToStriing() string
|
||||
}
|
||||
IMessageFactory interface {
|
||||
SetMessageConfig(MsgProtoType ProtoType, IsUseBigEndian bool)
|
||||
DecodeMessageBybufio(r *bufio.Reader) (message IMessage, err error)
|
||||
DecodeMessageBybytes(buffer []byte) (message IMessage, err error)
|
||||
EncodeToMesage(comId uint16, msgId uint16, msg interface{}) (message IMessage)
|
||||
EncodeToByte(message IMessage) (buffer []byte)
|
||||
RpcEncodeMessage(d interface{}) ([]byte, error)
|
||||
RpcDecodeMessage(dataType reflect.Type, d []byte) (interface{}, error)
|
||||
}
|
||||
ISys interface {
|
||||
DecodeMessageBybufio(r *bufio.Reader) (message IMessage, err error)
|
||||
DecodeMessageBybytes(buffer []byte) (message IMessage, err error)
|
||||
EncodeToMesage(comId uint16, msgId uint16, msg interface{}) (message IMessage)
|
||||
EncodeToByte(message IMessage) (buffer []byte)
|
||||
ByteDecodeToStruct(t reflect.Type, d []byte) (interface{}, error)
|
||||
}
|
||||
)
|
||||
|
||||
var (
|
||||
defsys ISys
|
||||
)
|
||||
|
||||
func OnInit(config map[string]interface{}, option ...Option) (err error) {
|
||||
defsys, err = newSys(newOptions(config, option...))
|
||||
return
|
||||
}
|
||||
|
||||
func NewSys(option ...Option) (sys ISys, err error) {
|
||||
sys, err = newSys(newOptionsByOption(option...))
|
||||
return
|
||||
}
|
||||
|
||||
func DecodeMessageBybufio(r *bufio.Reader) (IMessage, error) {
|
||||
return defsys.DecodeMessageBybufio(r)
|
||||
}
|
||||
func DecodeMessageBybytes(buffer []byte) (msg IMessage, err error) {
|
||||
return defsys.DecodeMessageBybytes(buffer)
|
||||
}
|
||||
func EncodeToMesage(comId uint16, msgId uint16, msg interface{}) (message IMessage) {
|
||||
return defsys.EncodeToMesage(comId, msgId, msg)
|
||||
}
|
||||
func EncodeToByte(message IMessage) (buffer []byte) {
|
||||
return defsys.EncodeToByte(message)
|
||||
}
|
||||
|
||||
func ByteDecodeToStruct(t reflect.Type, d []byte) (interface{}, error) {
|
||||
return defsys.ByteDecodeToStruct(t, d)
|
||||
}
|
||||
@ -1,218 +0,0 @@
|
||||
package proto
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"encoding/binary"
|
||||
"encoding/json"
|
||||
"fmt"
|
||||
"io"
|
||||
"reflect"
|
||||
|
||||
"go_dreamfactory/lego/sys/rpc"
|
||||
|
||||
"github.com/golang/protobuf/proto"
|
||||
)
|
||||
|
||||
//默认消息体
|
||||
type DefMessage struct {
|
||||
ComId uint16 //主Id
|
||||
MsgId uint16 //次Id
|
||||
MsgLen uint32 //消息体长度
|
||||
Buffer []byte //消息体
|
||||
}
|
||||
|
||||
func (this *DefMessage) GetComId() uint16 {
|
||||
return this.ComId
|
||||
}
|
||||
|
||||
func (this *DefMessage) GetMsgId() uint16 {
|
||||
return this.MsgId
|
||||
}
|
||||
|
||||
func (this *DefMessage) GetMsgLen() uint32 {
|
||||
return this.MsgLen
|
||||
}
|
||||
|
||||
func (this *DefMessage) GetBuffer() []byte {
|
||||
return this.Buffer
|
||||
}
|
||||
|
||||
func (this *DefMessage) ToStriing() string {
|
||||
return fmt.Sprintf("ComId：%d MsgId:%d MsgLen:%d", this.ComId, this.MsgId, this.MsgLen)
|
||||
}
|
||||
|
||||
//默认消息工厂
|
||||
type DefMessageFactory struct {
|
||||
msgProtoType ProtoType //消息体类型
|
||||
isUseBigEndian bool //消息传输码
|
||||
msgheadsize uint32 //消息头大小
|
||||
msgmaxleng uint32 //消息体最大长度
|
||||
}
|
||||
|
||||
func (this *DefMessageFactory) SetMessageConfig(MsgProtoType ProtoType, IsUseBigEndian bool) {
|
||||
this.msgProtoType = MsgProtoType
|
||||
this.isUseBigEndian = IsUseBigEndian
|
||||
this.msgheadsize = 8
|
||||
this.msgmaxleng = 1024 * 1204 * 3
|
||||
//自己rpc消息解析
|
||||
rpc.OnRegisterRpcData(&DefMessage{}, this.RpcEncodeMessage, this.RpcDecodeMessage)
|
||||
}
|
||||
|
||||
func (this *DefMessageFactory) DecodeMessageBybufio(r *bufio.Reader) (message IMessage, err error) {
|
||||
msg := &DefMessage{}
|
||||
msg.ComId, err = this.readUInt16(r)
|
||||
if err != nil {
|
||||
return msg, err
|
||||
}
|
||||
msg.MsgId, err = this.readUInt16(r)
|
||||
if err != nil {
|
||||
return msg, err
|
||||
}
|
||||
msg.MsgLen, err = this.readUInt32(r)
|
||||
if err != nil {
|
||||
return msg, err
|
||||
}
|
||||
if msg.MsgLen > this.msgmaxleng {
|
||||
return nil, fmt.Errorf("DecodeMessageBybufio err msg.MsgLen:%d Super long", msg.MsgLen)
|
||||
}
|
||||
msg.Buffer = make([]byte, msg.MsgLen)
|
||||
_, err = io.ReadFull(r, msg.Buffer)
|
||||
return msg, err
|
||||
}
|
||||
|
||||
func (this *DefMessageFactory) DecodeMessageBybytes(buffer []byte) (message IMessage, err error) {
|
||||
if uint32(len(buffer)) >= this.msgheadsize {
|
||||
msg := &DefMessage{}
|
||||
if this.isUseBigEndian {
|
||||
msg.ComId = binary.BigEndian.Uint16(buffer[0:])
|
||||
msg.MsgId = binary.BigEndian.Uint16(buffer[2:])
|
||||
msg.MsgLen = binary.BigEndian.Uint32(buffer[4:])
|
||||
} else {
|
||||
msg.ComId = binary.LittleEndian.Uint16(buffer[0:])
|
||||
msg.MsgId = binary.LittleEndian.Uint16(buffer[2:])
|
||||
msg.MsgLen = binary.LittleEndian.Uint32(buffer[4:])
|
||||
}
|
||||
|
||||
if uint32(len(buffer)) >= msg.MsgLen+this.msgheadsize {
|
||||
msg.Buffer = buffer[this.msgheadsize : msg.MsgLen+this.msgheadsize]
|
||||
return msg, nil
|
||||
} else {
|
||||
return nil, fmt.Errorf("DecodeMessageBybytes err package:%v msg.MsgLen:%d", buffer, msg.MsgLen)
|
||||
}
|
||||
}
|
||||
return nil, fmt.Errorf("DecodeMessageBybytes err package:%v", buffer)
|
||||
}
|
||||
|
||||
func (this *DefMessageFactory) EncodeToMesage(comId uint16, msgId uint16, msg interface{}) (message IMessage) {
|
||||
defmessage := &DefMessage{
|
||||
ComId: uint16(comId),
|
||||
MsgId: uint16(msgId),
|
||||
}
|
||||
if this.msgProtoType == Proto_Buff {
|
||||
defmessage.Buffer, _ = proto.Marshal(msg.(proto.Message))
|
||||
} else {
|
||||
defmessage.Buffer, _ = json.Marshal(msg)
|
||||
}
|
||||
defmessage.MsgLen = uint32(len(defmessage.Buffer))
|
||||
return defmessage
|
||||
}
|
||||
|
||||
func (this *DefMessageFactory) EncodeToByte(message IMessage) (buffer []byte) {
|
||||
if this.isUseBigEndian {
|
||||
buffer := []byte{}
|
||||
_msg := make([]byte, 2)
|
||||
binary.BigEndian.PutUint16(_msg, message.GetComId())
|
||||
buffer = append(buffer, _msg...)
|
||||
_msg = make([]byte, 2)
|
||||
binary.BigEndian.PutUint16(_msg, message.GetMsgId())
|
||||
buffer = append(buffer, _msg...)
|
||||
_msg = make([]byte, 4)
|
||||
binary.BigEndian.PutUint32(_msg, message.GetMsgLen())
|
||||
buffer = append(buffer, _msg...)
|
||||
buffer = append(buffer, message.GetBuffer()...)
|
||||
return buffer
|
||||
} else {
|
||||
buffer := []byte{}
|
||||
_msg := make([]byte, 2)
|
||||
binary.LittleEndian.PutUint16(_msg, message.GetComId())
|
||||
buffer = append(buffer, _msg...)
|
||||
_msg = make([]byte, 2)
|
||||
binary.LittleEndian.PutUint16(_msg, message.GetMsgId())
|
||||
buffer = append(buffer, _msg...)
|
||||
_msg = make([]byte, 4)
|
||||
binary.LittleEndian.PutUint32(_msg, message.GetMsgLen())
|
||||
buffer = append(buffer, _msg...)
|
||||
buffer = append(buffer, message.GetBuffer()...)
|
||||
return buffer
|
||||
}
|
||||
}
|
||||
|
||||
func (this *DefMessageFactory) RpcEncodeMessage(d interface{}) ([]byte, error) {
|
||||
message := d.(IMessage)
|
||||
if this.isUseBigEndian {
|
||||
buffer := []byte{}
|
||||
_msg := make([]byte, 2)
|
||||
binary.BigEndian.PutUint16(_msg, message.GetComId())
|
||||
buffer = append(buffer, _msg...)
|
||||
_msg = make([]byte, 2)
|
||||
binary.BigEndian.PutUint16(_msg, message.GetMsgId())
|
||||
buffer = append(buffer, _msg...)
|
||||
_msg = make([]byte, 4)
|
||||
binary.BigEndian.PutUint32(_msg, message.GetMsgLen())
|
||||
buffer = append(buffer, _msg...)
|
||||
buffer = append(buffer, message.GetBuffer()...)
|
||||
return buffer, nil
|
||||
} else {
|
||||
buffer := []byte{}
|
||||
_msg := make([]byte, 2)
|
||||
binary.LittleEndian.PutUint16(_msg, message.GetComId())
|
||||
buffer = append(buffer, _msg...)
|
||||
_msg = make([]byte, 2)
|
||||
binary.LittleEndian.PutUint16(_msg, message.GetMsgId())
|
||||
buffer = append(buffer, _msg...)
|
||||
_msg = make([]byte, 4)
|
||||
binary.LittleEndian.PutUint32(_msg, message.GetMsgLen())
|
||||
buffer = append(buffer, _msg...)
|
||||
buffer = append(buffer, message.GetBuffer()...)
|
||||
return buffer, nil
|
||||
}
|
||||
}
|
||||
|
||||
func (this *DefMessageFactory) RpcDecodeMessage(dataType reflect.Type, d []byte) (interface{}, error) {
|
||||
return this.DecodeMessageBybytes(d)
|
||||
}
|
||||
|
||||
func (this *DefMessageFactory) readByte(r *bufio.Reader) (byte, error) {
|
||||
buf := make([]byte, 1)
|
||||
_, err := io.ReadFull(r, buf[:1])
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
return buf[0], nil
|
||||
}
|
||||
|
||||
func (this *DefMessageFactory) readUInt16(r *bufio.Reader) (uint16, error) {
|
||||
buf := make([]byte, 2)
|
||||
_, err := io.ReadFull(r, buf[:2])
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
if this.isUseBigEndian {
|
||||
return binary.BigEndian.Uint16(buf[:2]), nil
|
||||
} else {
|
||||
return binary.LittleEndian.Uint16(buf[:2]), nil
|
||||
}
|
||||
}
|
||||
|
||||
func (this *DefMessageFactory) readUInt32(r *bufio.Reader) (uint32, error) {
|
||||
buf := make([]byte, 4)
|
||||
_, err := io.ReadFull(r, buf[:4])
|
||||
if err != nil {
|
||||
return 0, err
|
||||
}
|
||||
if this.isUseBigEndian {
|
||||
return binary.BigEndian.Uint32(buf[:4]), nil
|
||||
} else {
|
||||
return binary.LittleEndian.Uint32(buf[:4]), nil
|
||||
}
|
||||
}
|
||||
@ -1,68 +0,0 @@
|
||||
package proto
|
||||
|
||||
import (
|
||||
"go_dreamfactory/lego/utils/mapstructure"
|
||||
)
|
||||
|
||||
type ProtoType uint8
|
||||
|
||||
const (
|
||||
Proto_Json ProtoType = 1
|
||||
Proto_Buff ProtoType = 2
|
||||
)
|
||||
|
||||
type Option func(*Options)
|
||||
type Options struct {
|
||||
MsgProtoType ProtoType
|
||||
IsUseBigEndian bool
|
||||
MessageFactory IMessageFactory
|
||||
}
|
||||
|
||||
func SetMsgProtoType(v ProtoType) Option {
|
||||
return func(o *Options) {
|
||||
o.MsgProtoType = v
|
||||
}
|
||||
}
|
||||
|
||||
func SetIsUseBigEndian(v bool) Option {
|
||||
return func(o *Options) {
|
||||
o.IsUseBigEndian = v
|
||||
}
|
||||
}
|
||||
|
||||
func SetMessageFactory(v IMessageFactory) Option {
|
||||
return func(o *Options) {
|
||||
o.MessageFactory = v
|
||||
}
|
||||
}
|
||||
|
||||
func newOptions(config map[string]interface{}, opts ...Option) Options {
|
||||
options := Options{
|
||||
MsgProtoType: Proto_Buff,
|
||||
IsUseBigEndian: false,
|
||||
}
|
||||
if config != nil {
|
||||
mapstructure.Decode(config, &options)
|
||||
}
|
||||
for _, o := range opts {
|
||||
o(&options)
|
||||
}
|
||||
if options.MessageFactory == nil {
|
||||
options.MessageFactory = new(DefMessageFactory)
|
||||
}
|
||||
return options
|
||||
}
|
||||
|
||||
func newOptionsByOption(opts ...Option) Options {
|
||||
options := Options{
|
||||
MsgProtoType: Proto_Buff,
|
||||
IsUseBigEndian: false,
|
||||
}
|
||||
for _, o := range opts {
|
||||
o(&options)
|
||||
}
|
||||
if options.MessageFactory == nil {
|
||||
options.MessageFactory = new(DefMessageFactory)
|
||||
}
|
||||
return options
|
||||
}
|
||||
@ -1,48 +0,0 @@
|
||||
package proto
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"encoding/json"
|
||||
"reflect"
|
||||
|
||||
"github.com/golang/protobuf/proto"
|
||||
)
|
||||
|
||||
func newSys(options Options) (sys *Proto, err error) {
|
||||
sys = &Proto{
|
||||
options: options,
|
||||
}
|
||||
options.MessageFactory.SetMessageConfig(options.MsgProtoType, options.IsUseBigEndian)
|
||||
return
|
||||
}
|
||||
|
||||
type Proto struct {
|
||||
options Options
|
||||
}
|
||||
|
||||
func (this *Proto) DecodeMessageBybufio(r *bufio.Reader) (message IMessage, err error) {
|
||||
return this.options.MessageFactory.DecodeMessageBybufio(r)
|
||||
}
|
||||
|
||||
func (this *Proto) DecodeMessageBybytes(buffer []byte) (message IMessage, err error) {
|
||||
return this.options.MessageFactory.DecodeMessageBybytes(buffer)
|
||||
}
|
||||
|
||||
func (this *Proto) EncodeToMesage(comId uint16, msgId uint16, msg interface{}) (message IMessage) {
|
||||
return this.options.MessageFactory.EncodeToMesage(comId, msgId, msg)
|
||||
}
|
||||
|
||||
func (this *Proto) EncodeToByte(message IMessage) (buffer []byte) {
|
||||
return this.options.MessageFactory.EncodeToByte(message)
|
||||
}
|
||||
|
||||
func (this *Proto) ByteDecodeToStruct(t reflect.Type, d []byte) (data interface{}, err error) {
|
||||
if this.options.MsgProtoType == Proto_Json {
|
||||
data = reflect.New(t.Elem()).Interface()
|
||||
err = json.Unmarshal(d, data)
|
||||
} else {
|
||||
data = reflect.New(t.Elem()).Interface()
|
||||
err = proto.UnmarshalMerge(d, data.(proto.Message))
|
||||
}
|
||||
return
|
||||
}
|
||||
Loading…
Reference in New Issue
Block a user