31 lines
443 B
Go
31 lines
443 B
Go
package common
|
|
|
|
import (
|
|
"net"
|
|
|
|
"github.com/gorilla/websocket"
|
|
)
|
|
|
|
func WsToTcpHandler(conn net.Conn, wsConn *websocket.Conn) {
|
|
for {
|
|
_, p, err := wsConn.ReadMessage()
|
|
if err != nil {
|
|
return
|
|
}
|
|
conn.Write(p)
|
|
}
|
|
}
|
|
|
|
func TcpToWsHandler(conn net.Conn, wsConn *websocket.Conn) {
|
|
buf := make([]byte, 1024)
|
|
for {
|
|
n, err := conn.Read(buf)
|
|
if err != nil {
|
|
return
|
|
}
|
|
|
|
wsConn.WriteMessage(websocket.BinaryMessage, buf[:n])
|
|
}
|
|
|
|
}
|