parent
a2ed3090e7
commit
dfbc5cbd63
@ -0,0 +1,40 @@
|
|||||||
|
package core
|
||||||
|
|
||||||
|
type (
|
||||||
|
LifetimeHandler func(conn *Connection)
|
||||||
|
|
||||||
|
lifetime struct {
|
||||||
|
onOpen []LifetimeHandler
|
||||||
|
onClosed []LifetimeHandler
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
var Lifetime = &lifetime{}
|
||||||
|
|
||||||
|
func (lt *lifetime) OnClosed(h LifetimeHandler) {
|
||||||
|
lt.onClosed = append(lt.onClosed, h)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (lt *lifetime) OnOpen(h LifetimeHandler) {
|
||||||
|
lt.onOpen = append(lt.onOpen, h)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (lt *lifetime) Open(conn *Connection) {
|
||||||
|
if len(lt.onOpen) <= 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, handler := range lt.onOpen {
|
||||||
|
handler(conn)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (lt *lifetime) Close(conn *Connection) {
|
||||||
|
if len(lt.onClosed) <= 0 {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
|
||||||
|
for _, handler := range lt.onClosed {
|
||||||
|
handler(conn)
|
||||||
|
}
|
||||||
|
}
|
@ -0,0 +1,49 @@
|
|||||||
|
package core
|
||||||
|
|
||||||
|
import (
|
||||||
|
"errors"
|
||||||
|
"git.noahlan.cn/northlan/nnet/internal/log"
|
||||||
|
"git.noahlan.cn/northlan/nnet/internal/packet"
|
||||||
|
)
|
||||||
|
|
||||||
|
type nNetRouter struct {
|
||||||
|
handlers map[string]Handler
|
||||||
|
notFound Handler
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewRouter() Router {
|
||||||
|
return &nNetRouter{
|
||||||
|
handlers: make(map[string]Handler),
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *nNetRouter) Handle(conn *Connection, p packet.IPacket) {
|
||||||
|
pkg, ok := p.(*packet.Packet)
|
||||||
|
if !ok {
|
||||||
|
log.Error(packet.ErrWrongMessage)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
handler, ok := r.handlers[pkg.Header.Route]
|
||||||
|
if !ok {
|
||||||
|
if r.notFound == nil {
|
||||||
|
log.Error("message handler not found")
|
||||||
|
return
|
||||||
|
}
|
||||||
|
r.notFound.Handle(conn, p)
|
||||||
|
return
|
||||||
|
}
|
||||||
|
handler.Handle(conn, p)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *nNetRouter) Register(matches interface{}, handler Handler) error {
|
||||||
|
route, ok := matches.(string)
|
||||||
|
if !ok {
|
||||||
|
return errors.New("the type of matches must be string")
|
||||||
|
}
|
||||||
|
r.handlers[route] = handler
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (r *nNetRouter) SetNotFoundHandler(handler Handler) {
|
||||||
|
r.notFound = handler
|
||||||
|
}
|
@ -0,0 +1,9 @@
|
|||||||
|
package core
|
||||||
|
|
||||||
|
import "git.noahlan.cn/northlan/nnet/internal/packet"
|
||||||
|
|
||||||
|
type (
|
||||||
|
Processor interface {
|
||||||
|
Process(conn *Connection, packet packet.IPacket) error
|
||||||
|
}
|
||||||
|
)
|
@ -0,0 +1,72 @@
|
|||||||
|
package core
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/json"
|
||||||
|
"errors"
|
||||||
|
"fmt"
|
||||||
|
"git.noahlan.cn/northlan/nnet/internal/log"
|
||||||
|
"git.noahlan.cn/northlan/nnet/internal/packet"
|
||||||
|
"time"
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
hrd []byte // handshake response data
|
||||||
|
hbd []byte // heartbeat packet data
|
||||||
|
)
|
||||||
|
|
||||||
|
type NNetProcessor struct {
|
||||||
|
}
|
||||||
|
|
||||||
|
func NewNNetProcessor() *NNetProcessor {
|
||||||
|
// TODO custom hrd hbd
|
||||||
|
data, _ := json.Marshal(map[string]interface{}{
|
||||||
|
"code": 200,
|
||||||
|
"sys": map[string]float64{"heartbeat": time.Second.Seconds()},
|
||||||
|
})
|
||||||
|
packer := packet.NewNNetPacker()
|
||||||
|
hrd, _ = packer.Pack(packet.Handshake, data)
|
||||||
|
|
||||||
|
return &NNetProcessor{}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (n *NNetProcessor) Process(conn *Connection, p packet.IPacket) error {
|
||||||
|
h, ok := p.(*packet.Packet)
|
||||||
|
if !ok {
|
||||||
|
return packet.ErrWrongPacketType
|
||||||
|
}
|
||||||
|
|
||||||
|
switch h.PacketType {
|
||||||
|
case packet.Handshake:
|
||||||
|
// TODO validate handshake
|
||||||
|
if err := conn.SendBytes(hrd); err != nil {
|
||||||
|
return err
|
||||||
|
}
|
||||||
|
|
||||||
|
conn.SetStatus(StatusPrepare)
|
||||||
|
log.Debugf("connection handshake Id=%d, Remote=%s", conn.ID(), conn.Conn().RemoteAddr())
|
||||||
|
case packet.HandshakeAck:
|
||||||
|
conn.SetStatus(StatusPending)
|
||||||
|
log.Debugf("receive handshake ACK Id=%d, Remote=%s", conn.ID(), conn.Conn().RemoteAddr())
|
||||||
|
case packet.Heartbeat:
|
||||||
|
// Expected
|
||||||
|
case packet.Data:
|
||||||
|
if conn.Status() < StatusPending {
|
||||||
|
return errors.New(fmt.Sprintf("receive data on socket which not yet ACK, session will be closed immediately, remote=%s",
|
||||||
|
conn.Conn().RemoteAddr()))
|
||||||
|
}
|
||||||
|
conn.SetStatus(StatusWorking)
|
||||||
|
|
||||||
|
var lastMid uint64
|
||||||
|
switch h.MsgType {
|
||||||
|
case packet.Request:
|
||||||
|
lastMid = h.ID
|
||||||
|
case packet.Notify:
|
||||||
|
lastMid = 0
|
||||||
|
default:
|
||||||
|
return fmt.Errorf("Invalid message type: %s ", h.MsgType.String())
|
||||||
|
}
|
||||||
|
conn.SetLastMID(lastMid)
|
||||||
|
}
|
||||||
|
conn.SetLastHeartbeatAt(time.Now().Unix())
|
||||||
|
return nil
|
||||||
|
}
|
@ -0,0 +1,54 @@
|
|||||||
|
package core
|
||||||
|
|
||||||
|
type (
|
||||||
|
Middleware func(next HandlerFunc) HandlerFunc
|
||||||
|
|
||||||
|
Route struct {
|
||||||
|
Matches interface{} // 用于匹配的关键字段
|
||||||
|
Handler HandlerFunc // 处理方法
|
||||||
|
}
|
||||||
|
|
||||||
|
Router interface {
|
||||||
|
Handler
|
||||||
|
Register(matches interface{}, handler Handler) error
|
||||||
|
SetNotFoundHandler(handler Handler)
|
||||||
|
}
|
||||||
|
|
||||||
|
Constructor func(Handler) Handler
|
||||||
|
)
|
||||||
|
|
||||||
|
type Chain struct {
|
||||||
|
constructors []Constructor
|
||||||
|
}
|
||||||
|
|
||||||
|
func newChain(constructors ...Constructor) Chain {
|
||||||
|
return Chain{append(([]Constructor)(nil), constructors...)}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c Chain) Then(h Handler) Handler {
|
||||||
|
// TODO nil
|
||||||
|
|
||||||
|
for i := range c.constructors {
|
||||||
|
h = c.constructors[len(c.constructors)-1-i](h)
|
||||||
|
}
|
||||||
|
return h
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c Chain) ThenFunc(fn HandlerFunc) Handler {
|
||||||
|
if fn == nil {
|
||||||
|
return c.Then(nil)
|
||||||
|
}
|
||||||
|
return c.Then(fn)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c Chain) Append(constructors ...Constructor) Chain {
|
||||||
|
newCons := make([]Constructor, 0, len(c.constructors)+len(constructors))
|
||||||
|
newCons = append(newCons, c.constructors...)
|
||||||
|
newCons = append(newCons, constructors...)
|
||||||
|
|
||||||
|
return Chain{newCons}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (c Chain) Extend(chain Chain) Chain {
|
||||||
|
return c.Append(chain.constructors...)
|
||||||
|
}
|
@ -0,0 +1,15 @@
|
|||||||
|
package core
|
||||||
|
|
||||||
|
import "git.noahlan.cn/northlan/nnet/internal/packet"
|
||||||
|
|
||||||
|
type (
|
||||||
|
Handler interface {
|
||||||
|
Handle(conn *Connection, pkg packet.IPacket)
|
||||||
|
}
|
||||||
|
|
||||||
|
HandlerFunc func(conn *Connection, pkg packet.IPacket)
|
||||||
|
)
|
||||||
|
|
||||||
|
func (f HandlerFunc) Handle(conn *Connection, pkg packet.IPacket) {
|
||||||
|
f(conn, pkg)
|
||||||
|
}
|
@ -0,0 +1,247 @@
|
|||||||
|
package packet
|
||||||
|
|
||||||
|
import (
|
||||||
|
"bytes"
|
||||||
|
"encoding/binary"
|
||||||
|
"errors"
|
||||||
|
)
|
||||||
|
|
||||||
|
type NNetPacker struct {
|
||||||
|
buf *bytes.Buffer
|
||||||
|
size int // 最近一次 length
|
||||||
|
typ byte // 最近一次 packet type
|
||||||
|
flag byte // 最近一次 flag
|
||||||
|
}
|
||||||
|
|
||||||
|
// packer constants.
|
||||||
|
const (
|
||||||
|
headLength = 5
|
||||||
|
maxPacketSize = 64 * 1024
|
||||||
|
|
||||||
|
msgRouteCompressMask = 0x01 // 0000 0001 last bit
|
||||||
|
msgTypeMask = 0x07 // 0000 0111 1-3 bit (需要>>)
|
||||||
|
msgRouteLengthMask = 0xFF // 1111 1111 last 8 bit
|
||||||
|
msgHeadLength = 0x02 // 0000 0010 2 bit
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
ErrPacketSizeExceed = errors.New("packer: packet size exceed")
|
||||||
|
ErrWrongMessageType = errors.New("wrong message type")
|
||||||
|
ErrRouteInfoNotFound = errors.New("route info not found in dictionary")
|
||||||
|
ErrWrongMessage = errors.New("wrong message")
|
||||||
|
|
||||||
|
// ErrWrongPacketType represents a wrong packet type.
|
||||||
|
ErrWrongPacketType = errors.New("wrong packet type")
|
||||||
|
)
|
||||||
|
|
||||||
|
var (
|
||||||
|
routes = make(map[string]uint16) // route map to code
|
||||||
|
codes = make(map[uint16]string) // code map to route
|
||||||
|
)
|
||||||
|
|
||||||
|
func NewNNetPacker() *NNetPacker {
|
||||||
|
p := &NNetPacker{
|
||||||
|
buf: bytes.NewBuffer(nil),
|
||||||
|
}
|
||||||
|
p.resetFlags()
|
||||||
|
return p
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *NNetPacker) resetFlags() {
|
||||||
|
d.size = -1
|
||||||
|
d.typ = byte(Unknown)
|
||||||
|
d.flag = 0x00
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *NNetPacker) routable(t MsgType) bool {
|
||||||
|
return t == Request || t == Notify || t == Push
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *NNetPacker) invalidType(t MsgType) bool {
|
||||||
|
return t < Request || t > Push
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *NNetPacker) Pack(header interface{}, data []byte) ([]byte, error) {
|
||||||
|
h, ok := header.(Header)
|
||||||
|
if !ok {
|
||||||
|
return nil, ErrWrongPacketType
|
||||||
|
}
|
||||||
|
typ := h.PacketType
|
||||||
|
|
||||||
|
if typ < Handshake || typ > Kick {
|
||||||
|
return nil, ErrWrongPacketType
|
||||||
|
}
|
||||||
|
|
||||||
|
if d.invalidType(h.MsgType) {
|
||||||
|
return nil, ErrWrongMessageType
|
||||||
|
}
|
||||||
|
|
||||||
|
buf := make([]byte, 0)
|
||||||
|
|
||||||
|
// packet type
|
||||||
|
buf = append(buf, byte(h.PacketType))
|
||||||
|
|
||||||
|
// length
|
||||||
|
buf = append(buf, d.intToBytes(uint32(len(data)))...)
|
||||||
|
|
||||||
|
// flag
|
||||||
|
flag := byte(h.MsgType << 1) // 编译器提示,此处 byte 转换不能删
|
||||||
|
code, compressed := routes[h.Route]
|
||||||
|
if compressed {
|
||||||
|
flag |= msgRouteCompressMask
|
||||||
|
}
|
||||||
|
buf = append(buf, flag)
|
||||||
|
|
||||||
|
// msg id
|
||||||
|
if h.MsgType == Request || h.MsgType == Response {
|
||||||
|
n := h.ID
|
||||||
|
// variant length encode
|
||||||
|
for {
|
||||||
|
b := byte(n % 128)
|
||||||
|
n >>= 7
|
||||||
|
if n != 0 {
|
||||||
|
buf = append(buf, b+128)
|
||||||
|
} else {
|
||||||
|
buf = append(buf, b)
|
||||||
|
break
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// route
|
||||||
|
if d.routable(h.MsgType) {
|
||||||
|
if compressed {
|
||||||
|
buf = append(buf, byte((code>>8)&0xFF))
|
||||||
|
buf = append(buf, byte(code&0xFF))
|
||||||
|
} else {
|
||||||
|
buf = append(buf, byte(len(h.Route)))
|
||||||
|
buf = append(buf, []byte(h.Route)...)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
// body
|
||||||
|
buf = append(buf, data...)
|
||||||
|
return buf, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Encode packet data length to bytes(Big end)
|
||||||
|
func (d *NNetPacker) intToBytes(n uint32) []byte {
|
||||||
|
buf := make([]byte, 3)
|
||||||
|
buf[0] = byte((n >> 16) & 0xFF)
|
||||||
|
buf[1] = byte((n >> 8) & 0xFF)
|
||||||
|
buf[2] = byte(n & 0xFF)
|
||||||
|
return buf
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *NNetPacker) Unpack(data []byte) ([]IPacket, error) {
|
||||||
|
d.buf.Write(data) // copy
|
||||||
|
|
||||||
|
var (
|
||||||
|
packets []IPacket
|
||||||
|
err error
|
||||||
|
)
|
||||||
|
|
||||||
|
// 检查包长度
|
||||||
|
if d.buf.Len() < headLength {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
|
||||||
|
// 第一次拆包
|
||||||
|
if d.size < 0 {
|
||||||
|
if err = d.readHeader(); err != nil {
|
||||||
|
return nil, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
for d.size <= d.buf.Len() {
|
||||||
|
// 读取
|
||||||
|
p := newPacket(Type(d.typ))
|
||||||
|
p.MsgType = MsgType((d.flag >> 1) & msgTypeMask)
|
||||||
|
|
||||||
|
if d.invalidType(p.MsgType) {
|
||||||
|
return nil, ErrWrongMessageType
|
||||||
|
}
|
||||||
|
if p.MsgType == Request || p.MsgType == Response {
|
||||||
|
id := uint64(0)
|
||||||
|
// little end byte order
|
||||||
|
// WARNING: must be stored in 64 bits integer
|
||||||
|
// variant length encode
|
||||||
|
c := 0
|
||||||
|
for {
|
||||||
|
b, err := d.buf.ReadByte()
|
||||||
|
if err != nil {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
id += uint64(b&0x7F) << uint64(7*c)
|
||||||
|
if b < 128 {
|
||||||
|
break
|
||||||
|
}
|
||||||
|
c++
|
||||||
|
}
|
||||||
|
p.ID = id
|
||||||
|
}
|
||||||
|
if d.routable(p.MsgType) {
|
||||||
|
if d.flag&msgRouteCompressMask == 1 {
|
||||||
|
p.compressed = true
|
||||||
|
code := binary.BigEndian.Uint16(d.buf.Next(2))
|
||||||
|
route, ok := codes[code]
|
||||||
|
if !ok {
|
||||||
|
return nil, ErrRouteInfoNotFound
|
||||||
|
}
|
||||||
|
p.Route = route
|
||||||
|
} else {
|
||||||
|
p.compressed = false
|
||||||
|
rl, _ := d.buf.ReadByte()
|
||||||
|
if int(rl) > d.buf.Len() {
|
||||||
|
return nil, ErrWrongMessage
|
||||||
|
}
|
||||||
|
p.Route = string(d.buf.Next(int(rl)))
|
||||||
|
}
|
||||||
|
}
|
||||||
|
p.Length = uint32(d.size)
|
||||||
|
p.Data = d.buf.Next(d.size)
|
||||||
|
|
||||||
|
packets = append(packets, p)
|
||||||
|
|
||||||
|
// 剩余数据不满足至少一个数据帧,重置数据帧长度
|
||||||
|
// 数据缓存内存 保留至 下一次进入本方法以继续拆包
|
||||||
|
if d.buf.Len() < headLength {
|
||||||
|
d.resetFlags()
|
||||||
|
break
|
||||||
|
}
|
||||||
|
// 读取下一个包 next
|
||||||
|
if err = d.readHeader(); err != nil {
|
||||||
|
return packets, err
|
||||||
|
}
|
||||||
|
}
|
||||||
|
if packets == nil || len(packets) <= 0 {
|
||||||
|
d.resetFlags()
|
||||||
|
d.buf.Reset()
|
||||||
|
}
|
||||||
|
return packets, nil
|
||||||
|
}
|
||||||
|
|
||||||
|
func (d *NNetPacker) readHeader() error {
|
||||||
|
header := d.buf.Next(headLength)
|
||||||
|
d.typ = header[0]
|
||||||
|
if d.typ < Handshake || d.typ > Kick {
|
||||||
|
return ErrWrongPacketType
|
||||||
|
}
|
||||||
|
d.size = d.bytesToInt(header[1 : len(header)-1])
|
||||||
|
d.flag = header[len(header)-1]
|
||||||
|
|
||||||
|
// 最大包限定
|
||||||
|
if d.size > maxPacketSize {
|
||||||
|
return ErrPacketSizeExceed
|
||||||
|
}
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
|
||||||
|
// Decode packet data length byte to int(Big end)
|
||||||
|
func (d *NNetPacker) bytesToInt(b []byte) int {
|
||||||
|
result := 0
|
||||||
|
for _, v := range b {
|
||||||
|
result = result<<8 + int(v)
|
||||||
|
}
|
||||||
|
return result
|
||||||
|
}
|
@ -0,0 +1,98 @@
|
|||||||
|
package packet
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/hex"
|
||||||
|
"fmt"
|
||||||
|
"testing"
|
||||||
|
)
|
||||||
|
|
||||||
|
func TestPacker(t *testing.T) {
|
||||||
|
p := NewNNetPacker()
|
||||||
|
|
||||||
|
body := []byte("")
|
||||||
|
header := Header{
|
||||||
|
PacketType: Handshake,
|
||||||
|
Length: uint32(len(body)),
|
||||||
|
MessageHeader: MessageHeader{
|
||||||
|
MsgType: Request,
|
||||||
|
ID: 1,
|
||||||
|
Route: "",
|
||||||
|
compressed: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
pack, err := p.Pack(header, body)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fmt.Println(hex.EncodeToString(pack))
|
||||||
|
|
||||||
|
// handshake ack
|
||||||
|
body = []byte("")
|
||||||
|
header = Header{
|
||||||
|
PacketType: HandshakeAck,
|
||||||
|
Length: uint32(len(body)),
|
||||||
|
MessageHeader: MessageHeader{
|
||||||
|
MsgType: Response,
|
||||||
|
ID: 1,
|
||||||
|
Route: "",
|
||||||
|
compressed: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
pack, err = p.Pack(header, body)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fmt.Println(hex.EncodeToString(pack))
|
||||||
|
|
||||||
|
// data
|
||||||
|
body = []byte("123")
|
||||||
|
header = Header{
|
||||||
|
PacketType: Data,
|
||||||
|
Length: uint32(len(body)),
|
||||||
|
MessageHeader: MessageHeader{
|
||||||
|
MsgType: Request,
|
||||||
|
ID: 2,
|
||||||
|
Route: "",
|
||||||
|
compressed: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
pack, err = p.Pack(header, body)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fmt.Println(hex.EncodeToString(pack))
|
||||||
|
|
||||||
|
// data -> route: test -> handler
|
||||||
|
// data
|
||||||
|
body = []byte("ni hao")
|
||||||
|
header = Header{
|
||||||
|
PacketType: Data,
|
||||||
|
Length: uint32(len(body)),
|
||||||
|
MessageHeader: MessageHeader{
|
||||||
|
MsgType: Request,
|
||||||
|
ID: 3,
|
||||||
|
Route: "test",
|
||||||
|
compressed: false,
|
||||||
|
},
|
||||||
|
}
|
||||||
|
|
||||||
|
pack, err = p.Pack(header, body)
|
||||||
|
if err != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
fmt.Println(hex.EncodeToString(pack))
|
||||||
|
}
|
||||||
|
|
||||||
|
func TestUnPack(t *testing.T) {
|
||||||
|
data := []byte{0x04, 0x00, 0x00, 0x23, 0x04, 0x03, 0xE6, 0x9C, 0x8D, 0xE5, 0x8A, 0xA1, 0xE5, 0x99, 0xA8, 0xE6, 0x8E, 0xA5, 0xE6, 0x94, 0xB6, 0xE5, 0x88, 0xB0, 0xE6, 0x95, 0xB0, 0xE6, 0x8D, 0xAE, 0xE4, 0xB8, 0xBA, 0x3A, 0x20, 0x6E, 0x69, 0x20, 0x68, 0x61, 0x6F}
|
||||||
|
p := NewNNetPacker()
|
||||||
|
|
||||||
|
unPacked, err := p.Unpack(data)
|
||||||
|
if err != nil {
|
||||||
|
panic(err)
|
||||||
|
}
|
||||||
|
fmt.Println(unPacked)
|
||||||
|
}
|
@ -0,0 +1,19 @@
|
|||||||
|
package packet
|
||||||
|
|
||||||
|
type (
|
||||||
|
// IPacket 数据帧
|
||||||
|
IPacket interface {
|
||||||
|
GetHeader() interface{} // 数据帧头部 Header
|
||||||
|
GetLen() uint32 // 数据帧长度 4bytes - 32bit 占位,根据实际情况进行转换
|
||||||
|
GetBody() []byte // 数据 Body
|
||||||
|
}
|
||||||
|
|
||||||
|
// Packer 数据帧 封包/解包
|
||||||
|
Packer interface {
|
||||||
|
// Pack 封包,将原始数据构造为二进制流数据帧
|
||||||
|
Pack(header interface{}, data []byte) ([]byte, error)
|
||||||
|
|
||||||
|
// Unpack 解包
|
||||||
|
Unpack(data []byte) ([]IPacket, error)
|
||||||
|
}
|
||||||
|
)
|
@ -0,0 +1,94 @@
|
|||||||
|
package packet
|
||||||
|
|
||||||
|
import (
|
||||||
|
"encoding/hex"
|
||||||
|
"fmt"
|
||||||
|
)
|
||||||
|
|
||||||
|
// Type 数据帧类型,如:握手,心跳,数据 等
|
||||||
|
type Type byte
|
||||||
|
|
||||||
|
const (
|
||||||
|
// Unknown 未知包类型,无意义
|
||||||
|
Unknown Type = iota
|
||||||
|
|
||||||
|
// Handshake 握手数据(服务端主动发起)
|
||||||
|
Handshake = 0x01
|
||||||
|
|
||||||
|
// HandshakeAck 握手回复(客户端回复)
|
||||||
|
HandshakeAck = 0x02
|
||||||
|
|
||||||
|
// Heartbeat 心跳(服务端发起)
|
||||||
|
Heartbeat = 0x03
|
||||||
|
|
||||||
|
// Data 数据传输
|
||||||
|
Data = 0x04
|
||||||
|
|
||||||
|
// Kick 服务端主动断开连接
|
||||||
|
Kick = 0x05
|
||||||
|
)
|
||||||
|
|
||||||
|
type MsgType byte
|
||||||
|
|
||||||
|
// Message types
|
||||||
|
const (
|
||||||
|
Request MsgType = 0x00
|
||||||
|
Notify = 0x01
|
||||||
|
Response = 0x02
|
||||||
|
Push = 0x03
|
||||||
|
)
|
||||||
|
|
||||||
|
var msgTypes = map[MsgType]string{
|
||||||
|
Request: "Request",
|
||||||
|
Notify: "Notify",
|
||||||
|
Response: "Response",
|
||||||
|
Push: "Push",
|
||||||
|
}
|
||||||
|
|
||||||
|
func (t MsgType) String() string {
|
||||||
|
return msgTypes[t]
|
||||||
|
}
|
||||||
|
|
||||||
|
type (
|
||||||
|
Header struct {
|
||||||
|
PacketType Type // 数据帧 类型
|
||||||
|
Length uint32 // 数据长度
|
||||||
|
MessageHeader // 消息头
|
||||||
|
}
|
||||||
|
MessageHeader struct {
|
||||||
|
MsgType MsgType // message type (flag)
|
||||||
|
ID uint64 // unique id, zero while notify mode
|
||||||
|
Route string // route for locating service
|
||||||
|
compressed bool // if message compressed
|
||||||
|
}
|
||||||
|
Packet struct {
|
||||||
|
Header
|
||||||
|
Data []byte // 原始数据
|
||||||
|
}
|
||||||
|
)
|
||||||
|
|
||||||
|
func newPacket(typ Type) *Packet {
|
||||||
|
return &Packet{
|
||||||
|
Header: Header{
|
||||||
|
PacketType: typ,
|
||||||
|
MessageHeader: MessageHeader{},
|
||||||
|
},
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Packet) GetHeader() interface{} {
|
||||||
|
return p.Header
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Packet) GetLen() uint32 {
|
||||||
|
return p.Length
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Packet) GetBody() []byte {
|
||||||
|
return p.Data
|
||||||
|
}
|
||||||
|
|
||||||
|
func (p *Packet) String() string {
|
||||||
|
return fmt.Sprintf("Packet[Type: %d, Len: %d] Message[Type: %s, ID: %d, Route: %s, Compressed: %v] BodyStr: [%s], BodyHex: [%s]",
|
||||||
|
p.PacketType, p.Length, p.MsgType, p.ID, p.Route, p.compressed, string(p.Data), hex.EncodeToString(p.Data))
|
||||||
|
}
|
@ -0,0 +1,27 @@
|
|||||||
|
package pool
|
||||||
|
|
||||||
|
import "github.com/panjf2000/ants/v2"
|
||||||
|
|
||||||
|
var _pool *pool
|
||||||
|
|
||||||
|
type pool struct {
|
||||||
|
connPool *ants.Pool
|
||||||
|
workerPool *ants.Pool
|
||||||
|
}
|
||||||
|
|
||||||
|
func InitPool(size int) {
|
||||||
|
p := &pool{}
|
||||||
|
|
||||||
|
p.connPool, _ = ants.NewPool(size, ants.WithNonblocking(true))
|
||||||
|
p.workerPool, _ = ants.NewPool(size*2, ants.WithNonblocking(true))
|
||||||
|
|
||||||
|
_pool = p
|
||||||
|
}
|
||||||
|
|
||||||
|
func SubmitConn(h func()) error {
|
||||||
|
return _pool.connPool.Submit(h)
|
||||||
|
}
|
||||||
|
|
||||||
|
func SubmitWorker(h func()) error {
|
||||||
|
return _pool.workerPool.Submit(h)
|
||||||
|
}
|
@ -1,146 +0,0 @@
|
|||||||
package message
|
|
||||||
|
|
||||||
import (
|
|
||||||
"encoding/binary"
|
|
||||||
"errors"
|
|
||||||
)
|
|
||||||
|
|
||||||
var _ Codec = (*NNetCodec)(nil)
|
|
||||||
|
|
||||||
const (
|
|
||||||
msgRouteCompressMask = 0x01 // 0000 0001 last bit
|
|
||||||
msgTypeMask = 0x07 // 0000 0111 1-3 bit (需要>>)
|
|
||||||
msgRouteLengthMask = 0xFF // 1111 1111 last 8 bit
|
|
||||||
msgHeadLength = 0x02 // 0000 0010 2 bit
|
|
||||||
)
|
|
||||||
|
|
||||||
// Errors that could be occurred in message codec
|
|
||||||
var (
|
|
||||||
ErrWrongMessageType = errors.New("wrong message type")
|
|
||||||
ErrInvalidMessage = errors.New("invalid message")
|
|
||||||
ErrRouteInfoNotFound = errors.New("route info not found in dictionary")
|
|
||||||
ErrWrongMessage = errors.New("wrong message")
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
routes = make(map[string]uint16) // route map to code
|
|
||||||
codes = make(map[uint16]string) // code map to route
|
|
||||||
)
|
|
||||||
|
|
||||||
type NNetCodec struct{}
|
|
||||||
|
|
||||||
func (n *NNetCodec) routable(t Type) bool {
|
|
||||||
return t == Request || t == Notify || t == Push
|
|
||||||
}
|
|
||||||
|
|
||||||
func (n *NNetCodec) invalidType(t Type) bool {
|
|
||||||
return t < Request || t > Push
|
|
||||||
}
|
|
||||||
|
|
||||||
func (n *NNetCodec) Encode(v interface{}) ([]byte, error) {
|
|
||||||
m, ok := v.(*Message)
|
|
||||||
if !ok {
|
|
||||||
return nil, ErrWrongMessageType
|
|
||||||
}
|
|
||||||
if n.invalidType(m.Type) {
|
|
||||||
return nil, ErrWrongMessageType
|
|
||||||
}
|
|
||||||
buf := make([]byte, 0)
|
|
||||||
flag := byte(m.Type << 1) // 编译器提示,此处 byte 转换不能删
|
|
||||||
|
|
||||||
code, compressed := routes[m.Route]
|
|
||||||
if compressed {
|
|
||||||
flag |= msgRouteCompressMask
|
|
||||||
}
|
|
||||||
buf = append(buf, flag)
|
|
||||||
|
|
||||||
if m.Type == Request || m.Type == Response {
|
|
||||||
n := m.ID
|
|
||||||
// variant length encode
|
|
||||||
for {
|
|
||||||
b := byte(n % 128)
|
|
||||||
n >>= 7
|
|
||||||
if n != 0 {
|
|
||||||
buf = append(buf, b+128)
|
|
||||||
} else {
|
|
||||||
buf = append(buf, b)
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if n.routable(m.Type) {
|
|
||||||
if compressed {
|
|
||||||
buf = append(buf, byte((code>>8)&0xFF))
|
|
||||||
buf = append(buf, byte(code&0xFF))
|
|
||||||
} else {
|
|
||||||
buf = append(buf, byte(len(m.Route)))
|
|
||||||
buf = append(buf, []byte(m.Route)...)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
buf = append(buf, m.Data...)
|
|
||||||
return buf, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (n *NNetCodec) Decode(data []byte) (interface{}, error) {
|
|
||||||
if len(data) < msgHeadLength {
|
|
||||||
return nil, ErrInvalidMessage
|
|
||||||
}
|
|
||||||
m := New()
|
|
||||||
flag := data[0]
|
|
||||||
offset := 1
|
|
||||||
m.Type = Type((flag >> 1) & msgTypeMask) // 编译器提示,此处Type转换不能删
|
|
||||||
|
|
||||||
if n.invalidType(m.Type) {
|
|
||||||
return nil, ErrWrongMessageType
|
|
||||||
}
|
|
||||||
|
|
||||||
if m.Type == Request || m.Type == Response {
|
|
||||||
id := uint64(0)
|
|
||||||
// little end byte order
|
|
||||||
// WARNING: must can be stored in 64 bits integer
|
|
||||||
// variant length encode
|
|
||||||
for i := offset; i < len(data); i++ {
|
|
||||||
b := data[i]
|
|
||||||
id += uint64(b&0x7F) << uint64(7*(i-offset))
|
|
||||||
if b < 128 {
|
|
||||||
offset = i + 1
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
m.ID = id
|
|
||||||
}
|
|
||||||
|
|
||||||
if offset >= len(data) {
|
|
||||||
return nil, ErrWrongMessage
|
|
||||||
}
|
|
||||||
|
|
||||||
if n.routable(m.Type) {
|
|
||||||
if flag&msgRouteCompressMask == 1 {
|
|
||||||
m.compressed = true
|
|
||||||
code := binary.BigEndian.Uint16(data[offset:(offset + 2)])
|
|
||||||
route, ok := codes[code]
|
|
||||||
if !ok {
|
|
||||||
return nil, ErrRouteInfoNotFound
|
|
||||||
}
|
|
||||||
m.Route = route
|
|
||||||
offset += 2
|
|
||||||
} else {
|
|
||||||
m.compressed = false
|
|
||||||
rl := data[offset]
|
|
||||||
offset++
|
|
||||||
if offset+int(rl) > len(data) {
|
|
||||||
return nil, ErrWrongMessage
|
|
||||||
}
|
|
||||||
m.Route = string(data[offset:(offset + int(rl))])
|
|
||||||
offset += int(rl)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
if offset > len(data) {
|
|
||||||
return nil, ErrWrongMessage
|
|
||||||
}
|
|
||||||
m.Data = data[offset:]
|
|
||||||
return m, nil
|
|
||||||
}
|
|
@ -1,11 +0,0 @@
|
|||||||
package message
|
|
||||||
|
|
||||||
type (
|
|
||||||
// Codec 消息编解码器
|
|
||||||
Codec interface {
|
|
||||||
// Encode 编码
|
|
||||||
Encode(v interface{}) ([]byte, error)
|
|
||||||
// Decode 解码
|
|
||||||
Decode(data []byte) (interface{}, error)
|
|
||||||
}
|
|
||||||
)
|
|
@ -1,46 +0,0 @@
|
|||||||
package message
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Type represents the type of message, which could be Request/Notify/Response/Push
|
|
||||||
type Type byte
|
|
||||||
|
|
||||||
// Message types
|
|
||||||
const (
|
|
||||||
Request Type = 0x00
|
|
||||||
Notify = 0x01
|
|
||||||
Response = 0x02
|
|
||||||
Push = 0x03
|
|
||||||
)
|
|
||||||
|
|
||||||
var types = map[Type]string{
|
|
||||||
Request: "Request",
|
|
||||||
Notify: "Notify",
|
|
||||||
Response: "Response",
|
|
||||||
Push: "Push",
|
|
||||||
}
|
|
||||||
|
|
||||||
func (t Type) String() string {
|
|
||||||
return types[t]
|
|
||||||
}
|
|
||||||
|
|
||||||
// Message represents an unmarshaler message or a message which to be marshaled
|
|
||||||
type Message struct {
|
|
||||||
Type Type // message type (flag)
|
|
||||||
ID uint64 // unique id, zero while notify mode
|
|
||||||
Route string // route for locating service
|
|
||||||
Data []byte // payload
|
|
||||||
compressed bool // if message compressed
|
|
||||||
}
|
|
||||||
|
|
||||||
// New returns a new message instance
|
|
||||||
func New() *Message {
|
|
||||||
return &Message{}
|
|
||||||
}
|
|
||||||
|
|
||||||
// String, implementation of fmt.Stringer interface
|
|
||||||
func (m *Message) String() string {
|
|
||||||
return fmt.Sprintf("%s %s (%dbytes)", types[m.Type], m.Route, len(m.Data))
|
|
||||||
}
|
|
@ -1,31 +0,0 @@
|
|||||||
package nface
|
|
||||||
|
|
||||||
import "net"
|
|
||||||
|
|
||||||
const (
|
|
||||||
// StatusStart 开始阶段
|
|
||||||
StatusStart int32 = iota + 1
|
|
||||||
// StatusPrepare 准备阶段
|
|
||||||
StatusPrepare
|
|
||||||
// StatusWorking 工作阶段
|
|
||||||
StatusWorking
|
|
||||||
// StatusClosed 连接关闭
|
|
||||||
StatusClosed
|
|
||||||
)
|
|
||||||
|
|
||||||
type IConnection interface {
|
|
||||||
// Server 获取Server实例
|
|
||||||
Server() IServer
|
|
||||||
// Status 获取连接状态
|
|
||||||
Status() int32
|
|
||||||
// SetStatus 设置连接状态
|
|
||||||
SetStatus(s int32)
|
|
||||||
// Conn 获取底层网络连接
|
|
||||||
Conn() net.Conn
|
|
||||||
// ID 获取连接ID
|
|
||||||
ID() int64
|
|
||||||
// Session 获取当前连接绑定的Session
|
|
||||||
Session() ISession
|
|
||||||
// Close 关闭连接
|
|
||||||
Close() error
|
|
||||||
}
|
|
@ -1,5 +0,0 @@
|
|||||||
package nface
|
|
||||||
|
|
||||||
// IRouter 路由接口
|
|
||||||
type IRouter interface {
|
|
||||||
}
|
|
@ -1,4 +0,0 @@
|
|||||||
package nface
|
|
||||||
|
|
||||||
type IServer interface {
|
|
||||||
}
|
|
@ -1,187 +0,0 @@
|
|||||||
package nnet
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"git.noahlan.cn/northlan/nnet/log"
|
|
||||||
"git.noahlan.cn/northlan/nnet/nface"
|
|
||||||
"git.noahlan.cn/northlan/nnet/packet"
|
|
||||||
"git.noahlan.cn/northlan/nnet/pipeline"
|
|
||||||
"git.noahlan.cn/northlan/nnet/session"
|
|
||||||
"net"
|
|
||||||
"sync/atomic"
|
|
||||||
"time"
|
|
||||||
)
|
|
||||||
|
|
||||||
var (
|
|
||||||
_ nface.IConnection = (*Connection)(nil)
|
|
||||||
|
|
||||||
ErrCloseClosedSession = errors.New("close closed session")
|
|
||||||
// ErrBrokenPipe represents the low-level connection has broken.
|
|
||||||
ErrBrokenPipe = errors.New("broken low-level pipe")
|
|
||||||
// ErrBufferExceed indicates that the current session buffer is full and
|
|
||||||
// can not receive more data.
|
|
||||||
ErrBufferExceed = errors.New("session send buffer exceed")
|
|
||||||
)
|
|
||||||
|
|
||||||
type (
|
|
||||||
Connection struct {
|
|
||||||
session nface.ISession // Session
|
|
||||||
server *Server // Server 引用
|
|
||||||
|
|
||||||
conn net.Conn // low-level conn fd
|
|
||||||
status int32 // 连接状态
|
|
||||||
lastMid uint64 // 最近一次消息ID
|
|
||||||
lastHeartbeatAt int64 // 最近一次心跳时间
|
|
||||||
|
|
||||||
chDie chan struct{} // 停止通道
|
|
||||||
chSend chan pendingMessage // 消息发送通道
|
|
||||||
|
|
||||||
pipeline pipeline.Pipeline // 消息管道
|
|
||||||
}
|
|
||||||
|
|
||||||
pendingMessage struct {
|
|
||||||
typ interface{} // message type
|
|
||||||
route string // message route
|
|
||||||
mid uint64 // response message id
|
|
||||||
payload interface{} // payload
|
|
||||||
}
|
|
||||||
)
|
|
||||||
|
|
||||||
func newConnection(server *Server, conn net.Conn, pipeline pipeline.Pipeline) *Connection {
|
|
||||||
r := &Connection{
|
|
||||||
conn: conn,
|
|
||||||
server: server,
|
|
||||||
status: nface.StatusStart,
|
|
||||||
|
|
||||||
lastHeartbeatAt: time.Now().Unix(),
|
|
||||||
|
|
||||||
chDie: make(chan struct{}),
|
|
||||||
chSend: make(chan pendingMessage, 512),
|
|
||||||
|
|
||||||
pipeline: pipeline,
|
|
||||||
}
|
|
||||||
|
|
||||||
// binding session
|
|
||||||
r.session = session.New()
|
|
||||||
return r
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Connection) send(m pendingMessage) (err error) {
|
|
||||||
defer func() {
|
|
||||||
if e := recover(); e != nil {
|
|
||||||
err = ErrBrokenPipe
|
|
||||||
}
|
|
||||||
}()
|
|
||||||
r.chSend <- m
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Connection) Server() nface.IServer {
|
|
||||||
return r.server
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Connection) Status() int32 {
|
|
||||||
return atomic.LoadInt32(&r.status)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Connection) SetStatus(s int32) {
|
|
||||||
atomic.StoreInt32(&r.status, s)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Connection) Conn() net.Conn {
|
|
||||||
return r.conn
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Connection) ID() int64 {
|
|
||||||
return r.session.ID()
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Connection) setLastHeartbeatAt(t int64) {
|
|
||||||
atomic.StoreInt64(&r.lastHeartbeatAt, t)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Connection) Session() nface.ISession {
|
|
||||||
return r.session
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Connection) write() {
|
|
||||||
ticker := time.NewTicker(r.server.HeartbeatInterval)
|
|
||||||
|
|
||||||
chWrite := make(chan []byte, 1024)
|
|
||||||
|
|
||||||
defer func() {
|
|
||||||
ticker.Stop()
|
|
||||||
close(r.chSend)
|
|
||||||
close(chWrite)
|
|
||||||
_ = r.Close()
|
|
||||||
|
|
||||||
log.Debugf("Connection write goroutine exit, ConnID=%d, SessionUID=%d", r.ID(), r.session.UID())
|
|
||||||
}()
|
|
||||||
|
|
||||||
for {
|
|
||||||
select {
|
|
||||||
case <-ticker.C:
|
|
||||||
// TODO heartbeat enable control
|
|
||||||
deadline := time.Now().Add(-2 * r.server.HeartbeatInterval).Unix()
|
|
||||||
if atomic.LoadInt64(&r.lastHeartbeatAt) < deadline {
|
|
||||||
log.Debugf("Session heartbeat timeout, LastTime=%d, Deadline=%d", atomic.LoadInt64(&r.lastHeartbeatAt), deadline)
|
|
||||||
return
|
|
||||||
}
|
|
||||||
// TODO heartbeat data
|
|
||||||
chWrite <- []byte{}
|
|
||||||
case data := <-r.chSend:
|
|
||||||
// marshal packet body (data)
|
|
||||||
payload, err := r.server.Serializer.Marshal(data.payload)
|
|
||||||
if err != nil {
|
|
||||||
log.Errorf("message body marshal err: %v", err)
|
|
||||||
break
|
|
||||||
}
|
|
||||||
// TODO new message and pipeline
|
|
||||||
|
|
||||||
if pipe := r.pipeline; pipe != nil {
|
|
||||||
err := pipe.Outbound().Process(r)
|
|
||||||
if err != nil {
|
|
||||||
log.Errorf("broken pipeline err: %s", err.Error())
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
// TODO encode message ? message processor ?
|
|
||||||
|
|
||||||
// packet pack data
|
|
||||||
p, err := r.server.Packer.Pack(packet.Data, payload)
|
|
||||||
if err != nil {
|
|
||||||
log.Error(err.Error())
|
|
||||||
break
|
|
||||||
}
|
|
||||||
chWrite <- p
|
|
||||||
case data := <-chWrite:
|
|
||||||
// 回写数据
|
|
||||||
if _, err := r.conn.Write(data); err != nil {
|
|
||||||
log.Error(err.Error())
|
|
||||||
return
|
|
||||||
}
|
|
||||||
case <-r.chDie: // connection close signal
|
|
||||||
return
|
|
||||||
case <-r.server.DieChan: // application quit signal
|
|
||||||
return
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (r *Connection) Close() error {
|
|
||||||
if r.Status() == nface.StatusClosed {
|
|
||||||
return ErrCloseClosedSession
|
|
||||||
}
|
|
||||||
r.SetStatus(nface.StatusClosed)
|
|
||||||
|
|
||||||
log.Debugf("close connection, ID: %d", r.ID())
|
|
||||||
|
|
||||||
select {
|
|
||||||
case <-r.chDie:
|
|
||||||
default:
|
|
||||||
close(r.chDie)
|
|
||||||
// TODO lifetime
|
|
||||||
}
|
|
||||||
return r.conn.Close()
|
|
||||||
}
|
|
@ -1,27 +0,0 @@
|
|||||||
package nnet
|
|
||||||
|
|
||||||
import "github.com/panjf2000/ants/v2"
|
|
||||||
|
|
||||||
var pool *Pool
|
|
||||||
|
|
||||||
type Pool struct {
|
|
||||||
connPool *ants.Pool
|
|
||||||
workerPool *ants.Pool
|
|
||||||
}
|
|
||||||
|
|
||||||
func initPool(size int) {
|
|
||||||
p := &Pool{}
|
|
||||||
|
|
||||||
p.connPool, _ = ants.NewPool(size, ants.WithNonblocking(true))
|
|
||||||
p.workerPool, _ = ants.NewPool(size*2, ants.WithNonblocking(true))
|
|
||||||
|
|
||||||
pool = p
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *Pool) SubmitConn(h func()) error {
|
|
||||||
return p.connPool.Submit(h)
|
|
||||||
}
|
|
||||||
|
|
||||||
func (p *Pool) SubmitWorker(h func()) error {
|
|
||||||
return p.workerPool.Submit(h)
|
|
||||||
}
|
|
@ -1,9 +0,0 @@
|
|||||||
package nnet
|
|
||||||
|
|
||||||
import "testing"
|
|
||||||
|
|
||||||
func TestServer(t *testing.T) {
|
|
||||||
server := NewServer("tcp4", ":22112")
|
|
||||||
|
|
||||||
server.Serve()
|
|
||||||
}
|
|
@ -1,25 +0,0 @@
|
|||||||
package packet
|
|
||||||
|
|
||||||
import (
|
|
||||||
"git.noahlan.cn/northlan/nnet/nface"
|
|
||||||
)
|
|
||||||
|
|
||||||
// Type 数据帧类型,如:握手,心跳,数据 等
|
|
||||||
type Type byte
|
|
||||||
|
|
||||||
type (
|
|
||||||
Packer interface {
|
|
||||||
// Pack 从原始raw bytes创建一个用于网络传输的 数据帧结构
|
|
||||||
Pack(typ Type, data []byte) ([]byte, error)
|
|
||||||
|
|
||||||
// Unpack 解包
|
|
||||||
Unpack(data []byte) ([]interface{}, error)
|
|
||||||
}
|
|
||||||
|
|
||||||
// Processor 数据帧处理器,拆包之后的处理
|
|
||||||
Processor interface {
|
|
||||||
// ProcessPacket 单个数据包处理方法
|
|
||||||
// packet 为实际数据包,是 packet.Packer 的Unpack方法拆包出来的数据指针
|
|
||||||
ProcessPacket(conn nface.IConnection, packet interface{}) error
|
|
||||||
}
|
|
||||||
)
|
|
@ -1,123 +0,0 @@
|
|||||||
package packet
|
|
||||||
|
|
||||||
import (
|
|
||||||
"bytes"
|
|
||||||
"errors"
|
|
||||||
)
|
|
||||||
|
|
||||||
var _ Packer = (*NNetPacker)(nil)
|
|
||||||
|
|
||||||
type NNetPacker struct {
|
|
||||||
buf *bytes.Buffer
|
|
||||||
size int // 最近一次 长度
|
|
||||||
typ byte // 最近一次 数据帧类型
|
|
||||||
}
|
|
||||||
|
|
||||||
// Codec constants.
|
|
||||||
const (
|
|
||||||
headLength = 4
|
|
||||||
maxPacketSize = 64 * 1024
|
|
||||||
)
|
|
||||||
|
|
||||||
var ErrPacketSizeExceed = errors.New("codec: packet size exceed")
|
|
||||||
|
|
||||||
func NewNNetPacker() Packer {
|
|
||||||
return &NNetPacker{
|
|
||||||
buf: bytes.NewBuffer(nil),
|
|
||||||
size: -1,
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *NNetPacker) Pack(typ Type, data []byte) ([]byte, error) {
|
|
||||||
if typ < Handshake || typ > Kick {
|
|
||||||
return nil, ErrWrongPacketType
|
|
||||||
}
|
|
||||||
|
|
||||||
p := &Packet{Type: typ, Length: uint32(len(data))}
|
|
||||||
buf := make([]byte, p.Length+headLength)
|
|
||||||
|
|
||||||
// header
|
|
||||||
buf[0] = byte(p.Type)
|
|
||||||
copy(buf[1:headLength], d.intToBytes(p.Length))
|
|
||||||
|
|
||||||
// body
|
|
||||||
copy(buf[headLength:], data)
|
|
||||||
|
|
||||||
return buf, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Encode packet data length to bytes(Big end)
|
|
||||||
func (d *NNetPacker) intToBytes(n uint32) []byte {
|
|
||||||
buf := make([]byte, 3)
|
|
||||||
buf[0] = byte((n >> 16) & 0xFF)
|
|
||||||
buf[1] = byte((n >> 8) & 0xFF)
|
|
||||||
buf[2] = byte(n & 0xFF)
|
|
||||||
return buf
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *NNetPacker) Unpack(data []byte) ([]interface{}, error) {
|
|
||||||
d.buf.Write(data) // copy
|
|
||||||
|
|
||||||
var (
|
|
||||||
packets []interface{}
|
|
||||||
err error
|
|
||||||
)
|
|
||||||
|
|
||||||
// 检查包长度
|
|
||||||
if d.buf.Len() < headLength {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
|
|
||||||
// 第一次拆包
|
|
||||||
if d.size < 0 {
|
|
||||||
if err = d.readHeader(); err != nil {
|
|
||||||
return nil, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
for d.size <= d.buf.Len() {
|
|
||||||
// 读取
|
|
||||||
p := &Packet{
|
|
||||||
Type: Type(d.typ),
|
|
||||||
Length: uint32(d.size),
|
|
||||||
Data: d.buf.Next(d.size),
|
|
||||||
}
|
|
||||||
packets = append(packets, p)
|
|
||||||
|
|
||||||
// 剩余数据不满足至少一个数据帧,重置数据帧长度
|
|
||||||
// 数据缓存内存 保留至 下一次进入本方法以继续拆包
|
|
||||||
if d.buf.Len() < headLength {
|
|
||||||
d.size = -1
|
|
||||||
break
|
|
||||||
}
|
|
||||||
// 读取下一个包 next
|
|
||||||
if err = d.readHeader(); err != nil {
|
|
||||||
return packets, err
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return packets, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *NNetPacker) readHeader() error {
|
|
||||||
header := d.buf.Next(headLength)
|
|
||||||
d.typ = header[0]
|
|
||||||
if d.typ < Handshake || d.typ > Kick {
|
|
||||||
return ErrWrongPacketType
|
|
||||||
}
|
|
||||||
d.size = d.bytesToInt(header[1:])
|
|
||||||
|
|
||||||
// 最大包限定
|
|
||||||
if d.size > maxPacketSize {
|
|
||||||
return ErrPacketSizeExceed
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
// Decode packet data length byte to int(Big end)
|
|
||||||
func (d *NNetPacker) bytesToInt(b []byte) int {
|
|
||||||
result := 0
|
|
||||||
for _, v := range b {
|
|
||||||
result = result<<8 + int(v)
|
|
||||||
}
|
|
||||||
return result
|
|
||||||
}
|
|
@ -1,40 +0,0 @@
|
|||||||
package packet
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
)
|
|
||||||
|
|
||||||
const (
|
|
||||||
// Default 默认,暂无意义
|
|
||||||
Default Type = iota
|
|
||||||
|
|
||||||
// Handshake 握手数据(服务端主动发起)
|
|
||||||
Handshake = 0x01
|
|
||||||
|
|
||||||
// HandshakeAck 握手回复(客户端回复)
|
|
||||||
HandshakeAck = 0x02
|
|
||||||
|
|
||||||
// Heartbeat 心跳(服务端发起)
|
|
||||||
Heartbeat = 0x03
|
|
||||||
|
|
||||||
// Data 数据传输
|
|
||||||
Data = 0x04
|
|
||||||
|
|
||||||
// Kick 服务端主动断开连接
|
|
||||||
Kick = 0x05
|
|
||||||
)
|
|
||||||
|
|
||||||
// ErrWrongPacketType represents a wrong packet type.
|
|
||||||
var ErrWrongPacketType = errors.New("wrong packet type")
|
|
||||||
|
|
||||||
type Packet struct {
|
|
||||||
Type Type // 数据帧 类型
|
|
||||||
Length uint32 // 数据长度
|
|
||||||
Data []byte // 原始数据
|
|
||||||
}
|
|
||||||
|
|
||||||
func New() *Packet {
|
|
||||||
return &Packet{
|
|
||||||
Type: Default,
|
|
||||||
}
|
|
||||||
}
|
|
@ -1,40 +0,0 @@
|
|||||||
package packet
|
|
||||||
|
|
||||||
import (
|
|
||||||
"fmt"
|
|
||||||
"git.noahlan.cn/northlan/nnet/log"
|
|
||||||
"git.noahlan.cn/northlan/nnet/nface"
|
|
||||||
)
|
|
||||||
|
|
||||||
type NNetProcessor struct{}
|
|
||||||
|
|
||||||
func NewNNetProcessor() Processor {
|
|
||||||
return &NNetProcessor{}
|
|
||||||
}
|
|
||||||
|
|
||||||
func (d *NNetProcessor) ProcessPacket(conn nface.IConnection, packet interface{}) error {
|
|
||||||
p := packet.(*Packet)
|
|
||||||
switch p.Type {
|
|
||||||
case Handshake:
|
|
||||||
// TODO validate handshake
|
|
||||||
if _, err := conn.Conn().Write([]byte{}); err != nil {
|
|
||||||
return err
|
|
||||||
}
|
|
||||||
conn.SetStatus(nface.StatusPrepare)
|
|
||||||
log.Debugf("Connection handshake Id=%d, Remote=%s", conn.ID(), conn.Conn().RemoteAddr())
|
|
||||||
case HandshakeAck:
|
|
||||||
conn.SetStatus(nface.StatusWorking)
|
|
||||||
log.Debugf("Receive handshake ACK Id=%d, Remote=%s", conn.ID(), conn.Conn().RemoteAddr())
|
|
||||||
case Data:
|
|
||||||
if conn.Status() < nface.StatusWorking {
|
|
||||||
return fmt.Errorf("receive data on socket which not yet ACK, session will be closed immediately, remote=%s",
|
|
||||||
conn.Conn().RemoteAddr())
|
|
||||||
}
|
|
||||||
|
|
||||||
// TODO message data 处理
|
|
||||||
case Heartbeat:
|
|
||||||
// expected
|
|
||||||
}
|
|
||||||
|
|
||||||
return nil
|
|
||||||
}
|
|
@ -0,0 +1,203 @@
|
|||||||
|
// Code generated by protoc-gen-go. DO NOT EDIT.
|
||||||
|
// versions:
|
||||||
|
// protoc-gen-go v1.28.0
|
||||||
|
// protoc v3.19.4
|
||||||
|
// source: test.proto
|
||||||
|
|
||||||
|
package testdata
|
||||||
|
|
||||||
|
import (
|
||||||
|
protoreflect "google.golang.org/protobuf/reflect/protoreflect"
|
||||||
|
protoimpl "google.golang.org/protobuf/runtime/protoimpl"
|
||||||
|
reflect "reflect"
|
||||||
|
sync "sync"
|
||||||
|
)
|
||||||
|
|
||||||
|
const (
|
||||||
|
// Verify that this generated code is sufficiently up-to-date.
|
||||||
|
_ = protoimpl.EnforceVersion(20 - protoimpl.MinVersion)
|
||||||
|
// Verify that runtime/protoimpl is sufficiently up-to-date.
|
||||||
|
_ = protoimpl.EnforceVersion(protoimpl.MaxVersion - 20)
|
||||||
|
)
|
||||||
|
|
||||||
|
type Ping struct {
|
||||||
|
state protoimpl.MessageState
|
||||||
|
sizeCache protoimpl.SizeCache
|
||||||
|
unknownFields protoimpl.UnknownFields
|
||||||
|
|
||||||
|
Content string `protobuf:"bytes,1,opt,name=Content,proto3" json:"Content,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *Ping) Reset() {
|
||||||
|
*x = Ping{}
|
||||||
|
if protoimpl.UnsafeEnabled {
|
||||||
|
mi := &file_test_proto_msgTypes[0]
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *Ping) String() string {
|
||||||
|
return protoimpl.X.MessageStringOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*Ping) ProtoMessage() {}
|
||||||
|
|
||||||
|
func (x *Ping) ProtoReflect() protoreflect.Message {
|
||||||
|
mi := &file_test_proto_msgTypes[0]
|
||||||
|
if protoimpl.UnsafeEnabled && x != nil {
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
if ms.LoadMessageInfo() == nil {
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
return ms
|
||||||
|
}
|
||||||
|
return mi.MessageOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Deprecated: Use Ping.ProtoReflect.Descriptor instead.
|
||||||
|
func (*Ping) Descriptor() ([]byte, []int) {
|
||||||
|
return file_test_proto_rawDescGZIP(), []int{0}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *Ping) GetContent() string {
|
||||||
|
if x != nil {
|
||||||
|
return x.Content
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
type Pong struct {
|
||||||
|
state protoimpl.MessageState
|
||||||
|
sizeCache protoimpl.SizeCache
|
||||||
|
unknownFields protoimpl.UnknownFields
|
||||||
|
|
||||||
|
Content string `protobuf:"bytes,1,opt,name=Content,proto3" json:"Content,omitempty"`
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *Pong) Reset() {
|
||||||
|
*x = Pong{}
|
||||||
|
if protoimpl.UnsafeEnabled {
|
||||||
|
mi := &file_test_proto_msgTypes[1]
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *Pong) String() string {
|
||||||
|
return protoimpl.X.MessageStringOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
func (*Pong) ProtoMessage() {}
|
||||||
|
|
||||||
|
func (x *Pong) ProtoReflect() protoreflect.Message {
|
||||||
|
mi := &file_test_proto_msgTypes[1]
|
||||||
|
if protoimpl.UnsafeEnabled && x != nil {
|
||||||
|
ms := protoimpl.X.MessageStateOf(protoimpl.Pointer(x))
|
||||||
|
if ms.LoadMessageInfo() == nil {
|
||||||
|
ms.StoreMessageInfo(mi)
|
||||||
|
}
|
||||||
|
return ms
|
||||||
|
}
|
||||||
|
return mi.MessageOf(x)
|
||||||
|
}
|
||||||
|
|
||||||
|
// Deprecated: Use Pong.ProtoReflect.Descriptor instead.
|
||||||
|
func (*Pong) Descriptor() ([]byte, []int) {
|
||||||
|
return file_test_proto_rawDescGZIP(), []int{1}
|
||||||
|
}
|
||||||
|
|
||||||
|
func (x *Pong) GetContent() string {
|
||||||
|
if x != nil {
|
||||||
|
return x.Content
|
||||||
|
}
|
||||||
|
return ""
|
||||||
|
}
|
||||||
|
|
||||||
|
var File_test_proto protoreflect.FileDescriptor
|
||||||
|
|
||||||
|
var file_test_proto_rawDesc = []byte{
|
||||||
|
0x0a, 0x0a, 0x74, 0x65, 0x73, 0x74, 0x2e, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x12, 0x08, 0x74, 0x65,
|
||||||
|
0x73, 0x74, 0x64, 0x61, 0x74, 0x61, 0x22, 0x20, 0x0a, 0x04, 0x50, 0x69, 0x6e, 0x67, 0x12, 0x18,
|
||||||
|
0x0a, 0x07, 0x43, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x18, 0x01, 0x20, 0x01, 0x28, 0x09, 0x52,
|
||||||
|
0x07, 0x43, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x22, 0x20, 0x0a, 0x04, 0x50, 0x6f, 0x6e, 0x67,
|
||||||
|
0x12, 0x18, 0x0a, 0x07, 0x43, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x18, 0x01, 0x20, 0x01, 0x28,
|
||||||
|
0x09, 0x52, 0x07, 0x43, 0x6f, 0x6e, 0x74, 0x65, 0x6e, 0x74, 0x42, 0x0b, 0x5a, 0x09, 0x2f, 0x74,
|
||||||
|
0x65, 0x73, 0x74, 0x64, 0x61, 0x74, 0x61, 0x62, 0x06, 0x70, 0x72, 0x6f, 0x74, 0x6f, 0x33,
|
||||||
|
}
|
||||||
|
|
||||||
|
var (
|
||||||
|
file_test_proto_rawDescOnce sync.Once
|
||||||
|
file_test_proto_rawDescData = file_test_proto_rawDesc
|
||||||
|
)
|
||||||
|
|
||||||
|
func file_test_proto_rawDescGZIP() []byte {
|
||||||
|
file_test_proto_rawDescOnce.Do(func() {
|
||||||
|
file_test_proto_rawDescData = protoimpl.X.CompressGZIP(file_test_proto_rawDescData)
|
||||||
|
})
|
||||||
|
return file_test_proto_rawDescData
|
||||||
|
}
|
||||||
|
|
||||||
|
var file_test_proto_msgTypes = make([]protoimpl.MessageInfo, 2)
|
||||||
|
var file_test_proto_goTypes = []interface{}{
|
||||||
|
(*Ping)(nil), // 0: testdata.Ping
|
||||||
|
(*Pong)(nil), // 1: testdata.Pong
|
||||||
|
}
|
||||||
|
var file_test_proto_depIdxs = []int32{
|
||||||
|
0, // [0:0] is the sub-list for method output_type
|
||||||
|
0, // [0:0] is the sub-list for method input_type
|
||||||
|
0, // [0:0] is the sub-list for extension type_name
|
||||||
|
0, // [0:0] is the sub-list for extension extendee
|
||||||
|
0, // [0:0] is the sub-list for field type_name
|
||||||
|
}
|
||||||
|
|
||||||
|
func init() { file_test_proto_init() }
|
||||||
|
func file_test_proto_init() {
|
||||||
|
if File_test_proto != nil {
|
||||||
|
return
|
||||||
|
}
|
||||||
|
if !protoimpl.UnsafeEnabled {
|
||||||
|
file_test_proto_msgTypes[0].Exporter = func(v interface{}, i int) interface{} {
|
||||||
|
switch v := v.(*Ping); i {
|
||||||
|
case 0:
|
||||||
|
return &v.state
|
||||||
|
case 1:
|
||||||
|
return &v.sizeCache
|
||||||
|
case 2:
|
||||||
|
return &v.unknownFields
|
||||||
|
default:
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
file_test_proto_msgTypes[1].Exporter = func(v interface{}, i int) interface{} {
|
||||||
|
switch v := v.(*Pong); i {
|
||||||
|
case 0:
|
||||||
|
return &v.state
|
||||||
|
case 1:
|
||||||
|
return &v.sizeCache
|
||||||
|
case 2:
|
||||||
|
return &v.unknownFields
|
||||||
|
default:
|
||||||
|
return nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
}
|
||||||
|
type x struct{}
|
||||||
|
out := protoimpl.TypeBuilder{
|
||||||
|
File: protoimpl.DescBuilder{
|
||||||
|
GoPackagePath: reflect.TypeOf(x{}).PkgPath(),
|
||||||
|
RawDescriptor: file_test_proto_rawDesc,
|
||||||
|
NumEnums: 0,
|
||||||
|
NumMessages: 2,
|
||||||
|
NumExtensions: 0,
|
||||||
|
NumServices: 0,
|
||||||
|
},
|
||||||
|
GoTypes: file_test_proto_goTypes,
|
||||||
|
DependencyIndexes: file_test_proto_depIdxs,
|
||||||
|
MessageInfos: file_test_proto_msgTypes,
|
||||||
|
}.Build()
|
||||||
|
File_test_proto = out.File
|
||||||
|
file_test_proto_rawDesc = nil
|
||||||
|
file_test_proto_goTypes = nil
|
||||||
|
file_test_proto_depIdxs = nil
|
||||||
|
}
|
Loading…
Reference in New Issue