package main import ( "encoding/json" "git.noahlan.cn/noahlan/nnet/config" "git.noahlan.cn/noahlan/nnet/core" "git.noahlan.cn/noahlan/nnet/entity" "git.noahlan.cn/noahlan/nnet/packet" "git.noahlan.cn/noahlan/nnet/protocol" "git.noahlan.cn/noahlan/ntools-go/core/nlog" "git.noahlan.cn/noahlan/ntools-go/core/pool" "math" "time" ) func runServer(addr string) { server := core.NewServer(config.EngineConf{ ServerConf: config.ServerConf{ Protocol: "tcp", Addr: addr, Name: "testServer", Mode: "dev", }, Pool: pool.Config{ PoolSize: math.MaxInt32, ExpiryDuration: time.Second, PreAlloc: false, MaxBlockingTasks: 0, Nonblocking: false, DisablePurge: false, }, }, protocol.WithNNetProtocol(protocol.NNetConfig{ HeartbeatInterval: 0, HandshakeValidator: nil, })...) server.AddRoutes([]core.Route{ { Matches: protocol.Match{ Route: "ping", Code: 1, }, Handler: func(et entity.NetworkEntity, pkg packet.IPacket) { nlog.Info("client ping, server pong -> ") err := et.Send(protocol.Header{ PacketType: protocol.Data, MessageHeader: protocol.MessageHeader{ MsgType: protocol.Request, ID: 1, Route: "pong", }, }, []byte("1")) nlog.Must(err) }, }, }) defer server.Stop() server.Start() } func runClient(addr string) (client *core.Client, et entity.NetworkEntity) { chReady := make(chan struct{}) client = core.NewClient(config.EngineConf{ Pool: pool.Config{ PoolSize: math.MaxInt32, ExpiryDuration: time.Second, PreAlloc: false, MaxBlockingTasks: 0, Nonblocking: false, DisablePurge: false, }, }, protocol.WithNNetClientProtocol(func() { chReady <- struct{}{} })...) client.AddRoutes([]core.Route{ { Matches: protocol.Match{ Route: "test.client", Code: 1, }, Handler: func(et entity.NetworkEntity, pkg packet.IPacket) { nlog.Info("client hahaha") }, }, }) et = client.Dial(addr) handshake, err := json.Marshal(&protocol.HandshakeReq{ Version: "1.0.0", Type: "test", ClientId: "a", ClientSecret: "a", Payload: map[string]string{ "pl": "test-data", }, }) nlog.Must(err) packer := protocol.NewNNetPacker(protocol.NewRouteMap()) hsd, err := packer.Pack(protocol.Header{ PacketType: protocol.Handshake, MessageHeader: protocol.MessageHeader{ MsgType: 0, ID: 0, Route: "", }, }, handshake) nlog.Must(err) err = et.SendBytes(hsd) nlog.Must(err) <-chReady return }