forked from graarh/golang-socketio
-
Notifications
You must be signed in to change notification settings - Fork 0
/
client.go
74 lines (62 loc) · 1.22 KB
/
client.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 gosocketio
import (
"strconv"
"github.com/bigbluebutton-bot/golang-socketio/transport"
)
const (
webSocketProtocol = "ws://"
webSocketSecureProtocol = "wss://"
socketioUrl = "/socket.io/?EIO=3&transport=websocket"
)
/*
*
Socket.io client representation
*/
type Client struct {
methods
Channel
}
/*
*
Get ws/wss url by host and port
*/
func GetUrl(host string, port int, secure bool) string {
var prefix string
if secure {
prefix = webSocketSecureProtocol
} else {
prefix = webSocketProtocol
}
return prefix + host + ":" + strconv.Itoa(port) + socketioUrl
}
func NewClient() *Client {
c := &Client{}
c.initChannel()
c.initMethods()
return c
}
/*
*
connect to host and initialise socket.io protocol
The correct ws protocol url example:
ws://myserver.com/socket.io/?EIO=3&transport=websocket
You can use GetUrlByHost for generating correct url
*/
func (c *Client) Dial(url string, tr transport.Transport) error {
var err error
c.conn, err = tr.Connect(url)
if err != nil {
return err
}
go inLoop(&c.Channel, &c.methods)
go outLoop(&c.Channel, &c.methods)
go pinger(&c.Channel)
return nil
}
/*
*
Close client connection
*/
func (c *Client) Close() {
closeChannel(&c.Channel, &c.methods)
}