package middleware import ( "context" "net" "time" "github.com/gogf/gf/v2/os/glog" "github.com/lxzan/gws" ) const PingInterval = 10 * time.Second type Handler struct { gws.BuiltinEventHandler port int target net.Conn } func (c *Handler) 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.RecvHandler(pack) // // client.OnReceiveBase(client, pack, length) // } // }(c.target, socket) // //err = <-errChan // if err != io.EOF { // log.Println("proxy error:", err) // } } func (c *Handler) OnPing(socket *gws.Conn, payload []byte) { _ = socket.SetDeadline(time.Now().Add(2 * PingInterval)) _ = socket.WritePong(nil) } func (c *Handler) OnPong(socket *gws.Conn, payload []byte) {} func (c *Handler) OnMessage(socket *gws.Conn, gwsmessage *gws.Message) { if c.target != nil { c.target.Write(gwsmessage.Bytes()) } //fmt.Println(gwsmessage.Bytes()) } func (c *Handler) OnClose(socket *gws.Conn, err error) { glog.Debug(context.Background(), "断开连接") if c.target != nil { c.target.Close() } } // RemoveSocket 移除WebSocket连接 func messeunpack(id int64, x []byte) { // json := gjson.ParseBytes(x) // json.ForEach(func(key, value gjson.Result) bool { // fmt.Println(key, value) // numeric := regexp.MustCompile(`\d`).MatchString(value.String()) // var ob int // if numeric { //如果正确 则使用常量 // } else { //如果错误,则使用变量,即从数据库查询常量定义值是多少 // } // work.User_script(uint(id), strings.Trim(message.Data.String(), " ")) // // switch s:=value.String() { // // case regexp.MustCompile(`\d`).MatchString(s): // // } // return true // }) // json.Get("1").String() // var ob int // numeric := regexp.MustCompile(`\d`).MatchString(json.Get("1").String()) // if numeric { //如果正确 则使用常量 // ob, _ = strconv.Atoi(json.Get("1").String()) // } else { //如果错误,则使用变量,即从数据库查询常量定义值是多少 // ob, _ = strconv.Atoi(json.Get("1").String()) // } // switch ob { // case 1: //发包函数 // work.User_script(uint(id), strings.Trim(message.Data.String(), " ")) // } // Go 1.9.7 onwards only. }