package middleware import ( "context" "encoding/json" "net" "time" "github.com/gogf/gf/v2/frame/g" "github.com/gogf/gf/v2/os/glog" "github.com/gogf/gf/v2/util/gconv" "github.com/lxzan/gws" ) type ServerHandler struct { gws.BuiltinEventHandler port int target net.Conn } func (c *ServerHandler) OnOpen(socket *gws.Conn) { // target, err := net.Dial("tcp", "127.0.0.1:"+gconv.String(c.port)) // if err != nil { // glog.Debug(context.Background(), "连接失败") // } // c.target = target // //errChan := make(chan error, 2) // if c.target == nil { // return // } // go func(conn net.Conn, socket *gws.Conn) { // reader := bufio.NewReader(conn) // LOOP: // for { // // select { // // default: // packlen, err := reader.Peek(4) // if err != nil { // socket.WriteClose(1000, nil) // break LOOP // } // length := int32(binary.BigEndian.Uint32(packlen)) // data := make([]byte, length) // io.ReadFull(reader, data) // //pack_Ver := data[4] //因为包体已经解析,所以这里直接取0 // // var pack = make([]byte, length) // socket.WriteMessage(gws.OpcodeBinary, data) // //t.event.RecvServerHandler(pack) // // client.OnReceiveBase(client, pack, length) // } // }(c.target, socket) // //err = <-errChan // if err != io.EOF { // log.Println("proxy error:", err) // } } func (c *ServerHandler) OnPing(socket *gws.Conn, payload []byte) { _ = socket.SetDeadline(time.Now().Add(2 * PingInterval)) _ = socket.WritePong(nil) } func (c *ServerHandler) OnPong(socket *gws.Conn, payload []byte) {} func (c *ServerHandler) OnMessage(socket *gws.Conn, gwsmessage *gws.Message) { //socket.WriteMessage(gwsmessage.Opcode, gwsmessage.Bytes()) for i := 0; i < 4; i++ { jsondata := g.Map{"type": 1, "message": "hel " + gconv.String(i)} json, _ := json.Marshal(jsondata) socket.WriteMessage(gwsmessage.Opcode, json) } jsondata := g.Map{"type": 0, "message": "hel"} json, _ := json.Marshal(jsondata) socket.WriteMessage(gwsmessage.Opcode, json) //fmt.Println(gwsmessage.Bytes()) } func (c *ServerHandler) OnClose(socket *gws.Conn, err error) { glog.Debug(context.Background(), "断开连接") if c.target != nil { c.target.Close() } }