wip: nnet

main
NoahLan 2 years ago
parent 26ad30fd23
commit 52eea69cd0

@ -1,6 +1,7 @@
package protocol package protocol
import ( import (
"encoding/json"
"git.noahlan.cn/noahlan/nnet/core" "git.noahlan.cn/noahlan/nnet/core"
"git.noahlan.cn/noahlan/nnet/entity" "git.noahlan.cn/noahlan/nnet/entity"
"git.noahlan.cn/noahlan/nnet/middleware" "git.noahlan.cn/noahlan/nnet/middleware"
@ -9,26 +10,62 @@ import (
"time" "time"
) )
type NNetConfig struct { 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"`
func WithNNetProtocol( // 路由
handshakeValidator func([]byte) error, Routes map[string]uint16 `json:"routes"` // route map to code
heartbeatInterval time.Duration, Codes map[uint16]string `json:"codes"` // code map to route
) []core.RunOption {
if handshakeValidator == nil { // 透传信息
handshakeValidator = func(bytes []byte) error { return nil } Payload interface{} `json:"payload,optional,omitempty"`
} }
packer := NewNNetPacker() )
hbd, err := packer.Pack(Handshake, nil)
nlog.Must(err)
return []core.RunOption{ func WithNNetProtocol(config NNetConfig) []core.RunOption {
WithNNetPipeline(handshakeValidator), 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.WithRouter(NewNNetRouter()),
core.WithPacker(func() packet.Packer { return NewNNetPacker() }), core.WithPacker(func() packet.Packer { return NewNNetPacker() }),
middleware.WithHeartbeat(heartbeatInterval, func(_ entity.NetworkEntity) []byte { }
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 hbd
}), }))
} }
return opts
} }

@ -1,44 +1,20 @@
package protocol package protocol
import ( import (
"encoding/json" "errors"
"fmt" "fmt"
"git.noahlan.cn/noahlan/nnet/core" "git.noahlan.cn/noahlan/nnet/core"
"git.noahlan.cn/noahlan/nnet/entity" "git.noahlan.cn/noahlan/nnet/entity"
"git.noahlan.cn/noahlan/ntools-go/core/nlog" "git.noahlan.cn/noahlan/ntools-go/core/nlog"
"time"
) )
type ( type (
handshakeData struct { HandshakeValidatorFunc func([]byte) error
Version string `json:"version"` // 客户端版本号,服务器以此判断是否合适与客户端通信 HandshakeAckBuilderFunc func() (interface{}, error)
Type string `json:"type"` // 客户端类型,与客户端版本号一起来确定客户端是否合适
// 透传信息
Payload interface{} `json:"payload,optional,omitempty"`
}
handshakeAckData struct {
Heartbeat int64 `json:"heartbeat"` // 心跳间隔,单位秒 0表示不需要心跳
// 路由
Routes map[string]uint16 `json:"routes"` // route map to code
Codes map[uint16]string `json:"codes"` // code map to route
// 服务端支持的body部分消息传输协议
//Protocol string `json:"protocol,options=[plain,json,protobuf]"` // plain/json/protobuf
// 透传信息
Payload interface{} `json:"payload,optional,omitempty"`
}
) )
func WithNNetPipeline(heartbeatInterval time.Duration, handshakeValidator func([]byte) error) core.RunOption { func WithNNetPipeline(ackDataBuilder HandshakeAckBuilderFunc, validator HandshakeValidatorFunc) core.RunOption {
handshakeAck := &handshakeAckData{}
data, err := json.Marshal(handshakeAck)
nlog.Must(err)
packer := NewNNetPacker() packer := NewNNetPacker()
hrd, _ := packer.Pack(Handshake, data)
return func(server *core.Server) { return func(server *core.Server) {
server.Pipeline().Inbound().PushFront(func(entity entity.NetworkEntity, v interface{}) error { server.Pipeline().Inbound().PushFront(func(entity entity.NetworkEntity, v interface{}) error {
pkg, ok := v.(*NNetPacket) pkg, ok := v.(*NNetPacket)
@ -49,9 +25,13 @@ func WithNNetPipeline(heartbeatInterval time.Duration, handshakeValidator func([
switch pkg.PacketType { switch pkg.PacketType {
case Handshake: case Handshake:
if err := handshakeValidator(pkg.Data); err != nil { if err := validator(pkg.Data); err != nil {
return err return err
} }
data, err := ackDataBuilder()
nlog.Must(err)
hrd, _ := packer.Pack(Handshake, data)
if err := entity.SendBytes(hrd); err != nil { if err := entity.SendBytes(hrd); err != nil {
return err return err
} }

Loading…
Cancel
Save