98 lines
2.2 KiB
Go
98 lines
2.2 KiB
Go
|
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()
|
|||
|
|
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
}
|