-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathclient.go
95 lines (81 loc) · 1.68 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
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
package gosocketio
import (
"github.com/deutschesoft-inc/golang-socketio/transport"
"strconv"
"time"
)
const (
webSocketProtocol = "ws://"
webSocketSecureProtocol = "wss://"
socketioUrl = "/socket.io/?EIO=3&transport=websocket"
)
/**
Socket.io client representation
*/
type Client struct {
methods
Channel
url string
}
/**
Get ws/wss url by host and port
*/
func GetUrl(host string, port int, secure bool, path string) string {
var prefix string
if secure {
prefix = webSocketSecureProtocol
} else {
prefix = webSocketProtocol
}
return prefix + host + ":" + strconv.Itoa(port) + path + socketioUrl
}
/**
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 Dial(url string, tr transport.Transport, reconnect bool) (*Client, error) {
c := &Client{}
c.initChannel()
c.initMethods()
var err error
c.conn, err = tr.Connect(url)
if err != nil {
return nil, err
}
go inLoop(&c.Channel, &c.methods)
go outLoop(&c.Channel, &c.methods)
go pinger(&c.Channel)
if reconnect {
c.On(OnDisconnection, func(channel *Channel, msg interface{}) {
Redial(c)
})
}
return c, nil
}
/**
Close client connection
*/
func (c *Client) Close() {
closeChannel(&c.Channel, &c.methods)
}
/**
Reconnect to client
*/
func Redial(c *Client) {
var err error
tr := transport.GetDefaultWebsocketTransport()
c.initChannel()
for {
c.conn, err = tr.Connect(c.url)
if err == nil {
break
} else {
time.Sleep(time.Second)
}
}
go inLoop(&c.Channel, &c.methods)
go outLoop(&c.Channel, &c.methods)
go pinger(&c.Channel)
}