forked from leeonsoft/zerver
/
websocket.go
74 lines (60 loc) · 1.29 KB
/
websocket.go
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
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
package zerver
import (
"io"
"net/http"
"net/url"
"time"
"github.com/cosiner/gohper/unsafe2"
websocket "github.com/cosiner/zerver_websocket"
)
type (
WsConn interface {
io.ReadWriteCloser
Env
Vars() *ReqVars
WriteString(string) (int, error)
SetDeadline(t time.Time) error
SetReadDeadline(t time.Time) error
SetWriteDeadline(t time.Time) error
RemoteAddr() string
URL() *url.URL
}
wsConn struct {
Env
vars *ReqVars
*websocket.Conn
request *http.Request
}
WsHandlerFunc func(WsConn)
WsHandler interface {
Component
Handle(WsConn)
}
)
func newWsConn(e Env, conn *websocket.Conn, vars *ReqVars) *wsConn {
return &wsConn{
Env: e,
Conn: conn,
vars: vars,
request: conn.Request(),
}
}
func (c *wsConn) Vars() *ReqVars {
return c.vars
}
func (c *wsConn) WriteString(s string) (int, error) {
return c.Write(unsafe2.Bytes(s))
}
func (c *wsConn) URL() *url.URL {
return c.request.URL
}
func (c *wsConn) RemoteAddr() string {
return c.request.RemoteAddr
}
// UserAgent return user's agent identify
func (c *wsConn) UserAgent() string {
return c.request.Header.Get(HEADER_USERAGENT)
}
func (WsHandlerFunc) Init(Env) error { return nil }
func (fn WsHandlerFunc) Handle(conn WsConn) { fn(conn) }
func (WsHandlerFunc) Destroy() {}