|
|
package protocol
|
|
|
|
|
|
import (
|
|
|
"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
|
|
|
}
|
|
|
|
|
|
HandshakeReq struct {
|
|
|
Version string `json:"version"` // 客户端版本，服务器以此判断是否合适与客户端通信
|
|
|
Type string `json:"type"` // 客户端类型，与客户端版本号一起来确定客户端是否合适
|
|
|
ClientId string `json:"clientId"` // 客户端ID，服务器以此取值
|
|
|
ClientSecret string `json:"clientSecret"` // 客户端密钥，服务器以此判定客户端是否可用
|
|
|
|
|
|
// 透传信息
|
|
|
Payload interface{} `json:"payload,optional,omitempty"`
|
|
|
}
|
|
|
|
|
|
HandshakeResp struct {
|
|
|
// 心跳间隔，单位秒 0表示不需要心跳
|
|
|
Heartbeat int64 `json:"heartbeat"`
|
|
|
|
|
|
*RouteMap
|
|
|
|
|
|
// 透传信息
|
|
|
Payload interface{} `json:"payload,optional,omitempty"`
|
|
|
}
|
|
|
)
|
|
|
|
|
|
var routeMap *RouteMap
|
|
|
|
|
|
func init() {
|
|
|
routeMap = &RouteMap{
|
|
|
Routes: make(map[string]uint16),
|
|
|
Codes: make(map[uint16]string),
|
|
|
}
|
|
|
}
|
|
|
|
|
|
func WithNNetClientProtocol(onReady OnReadyFunc) []core.RunOption {
|
|
|
opts := []core.RunOption{
|
|
|
WithNNetClientPipeline(onReady),
|
|
|
core.WithRouter(NewNNetRouter()),
|
|
|
core.WithPacker(func() packet.Packer { return NewNNetPacker() }),
|
|
|
}
|
|
|
|
|
|
return opts
|
|
|
}
|
|
|
|
|
|
func WithNNetProtocol(config NNetConfig) []core.RunOption {
|
|
|
if config.HandshakeValidator == nil {
|
|
|
config.HandshakeValidator = func(data *HandshakeReq) error {
|
|
|
return nil
|
|
|
}
|
|
|
}
|
|
|
handshakeAckData := &HandshakeResp{
|
|
|
Heartbeat: int64(config.HeartbeatInterval.Seconds()),
|
|
|
RouteMap: routeMap,
|
|
|
}
|
|
|
|
|
|
opts := []core.RunOption{
|
|
|
WithNNetPipeline(handshakeAckData, config.HandshakeValidator),
|
|
|
core.WithRouter(NewNNetRouter()),
|
|
|
core.WithPacker(func() packet.Packer { return NewNNetPacker() }),
|
|
|
}
|
|
|
|
|
|
if config.HeartbeatInterval.Seconds() > 0 {
|
|
|
packer := NewNNetPacker()
|
|
|
hbd, err := packer.Pack(Heartbeat, nil)
|
|
|
nlog.Must(err)
|
|
|
|
|
|
opts = append(opts, middleware.WithHeartbeat(config.HeartbeatInterval, func(_ entity.NetworkEntity) []byte {
|
|
|
return hbd
|
|
|
}))
|
|
|
}
|
|
|
|
|
|
return opts
|
|
|
}
|