移除不需要的系统
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