|
|
package protocol
|
|
|
|
|
|
import (
|
|
|
"encoding/json"
|
|
|
"git.noahlan.cn/noahlan/nnet/core"
|
|
|
"git.noahlan.cn/noahlan/nnet/entity"
|
|
|
"git.noahlan.cn/noahlan/nnet/middleware"
|
|
|
"git.noahlan.cn/noahlan/nnet/packet"
|
|
|
"git.noahlan.cn/noahlan/ntools-go/core/nlog"
|
|
|
"time"
|
|
|
)
|
|
|
|
|
|
type (
|
|
|
NNetConfig struct {
|
|
|
HeartbeatInterval time.Duration
|
|
|
HandshakeValidator HandshakeValidatorFunc
|
|
|
HandshakeAckBuilder HandshakeAckBuilderFunc
|
|
|
}
|
|
|
|
|
|
handshakeData struct {
|
|
|
Version string `json:"version"` // 客户端版本，服务器以此判断是否合适与客户端通信
|
|
|
Type string `json:"type"` // 客户端类型，与客户端版本号一起来确定客户端是否合适
|
|
|
ClientId string `json:"clientId"` // 客户端ID，服务器以此取值
|
|
|
ClientSecret string `json:"clientSecret"` // 客户端密钥，服务器以此判定客户端是否可用
|
|
|
|
|
|
// 透传信息
|
|
|
Payload interface{} `json:"payload,optional,omitempty"`
|
|
|
}
|
|
|
|
|
|
HandshakeAckData struct {
|
|
|
// 心跳间隔，单位秒 0表示不需要心跳
|
|
|
Heartbeat int64 `json:"heartbeat"`
|
|
|
|
|
|
// 路由
|
|
|
Routes map[string]uint16 `json:"routes"` // route map to code
|
|
|
Codes map[uint16]string `json:"codes"` // code map to route
|
|
|
|
|
|
// 透传信息
|
|
|
Payload interface{} `json:"payload,optional,omitempty"`
|
|
|
}
|
|
|
)
|
|
|
|
|
|
func WithNNetProtocol(config NNetConfig) []core.RunOption {
|
|
|
if config.HandshakeValidator == nil {
|
|
|
config.HandshakeValidator = func(bytes []byte) error { return nil }
|
|
|
}
|
|
|
if config.HandshakeAckBuilder == nil {
|
|
|
config.HandshakeAckBuilder = func() ([]byte, error) {
|
|
|
defaultData := &HandshakeAckData{}
|
|
|
return json.Marshal(defaultData)
|
|
|
}
|
|
|
}
|
|
|
|
|
|
opts := []core.RunOption{
|
|
|
WithNNetPipeline(config.HandshakeAckBuilder, config.HandshakeValidator),
|
|
|
core.WithRouter(NewNNetRouter()),
|
|
|
core.WithPacker(func() packet.Packer { return NewNNetPacker() }),
|
|
|
}
|
|
|
|
|
|
if config.HeartbeatInterval.Seconds() > 0 {
|
|
|
packer := NewNNetPacker()
|
|
|
hbd, err := packer.Pack(Handshake, nil)
|
|
|
nlog.Must(err)
|
|
|
|
|
|
opts = append(opts, middleware.WithHeartbeat(config.HeartbeatInterval, func(_ entity.NetworkEntity) []byte {
|
|
|
return hbd
|
|
|
}))
|
|
|
}
|
|
|
|
|
|
return opts
|
|
|
}
|