blob: d6f45dc9c555339ee7bcde11323bd13da7fb551c (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
|
package net
import (
"github.com/gorilla/websocket"
)
type wsConn struct {
conn *websocket.Conn
}
func newWSConn(conn *websocket.Conn) *wsConn {
return &wsConn{conn: conn}
}
func (w *wsConn) ReadMessage() (string, error) {
_, msg, err := w.conn.ReadMessage()
return string(msg), err
}
func (w *wsConn) Write(b []byte) (int, error) {
err := w.conn.WriteMessage(websocket.TextMessage, b)
return len(b), err
}
func (w *wsConn) Close() error {
return w.conn.Close()
}
func (w *wsConn) SetEcho(on bool) error {
if on {
return w.conn.WriteMessage(websocket.TextMessage, []byte("\x1b]10;0\x07"))
}
return w.conn.WriteMessage(websocket.TextMessage, []byte("\x1b]10;1\x07"))
}
|
