69 lines
1.2 KiB
Go
69 lines
1.2 KiB
Go
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
|
|
}
|