-
Notifications
You must be signed in to change notification settings - Fork 88
/
client.go
78 lines (63 loc) · 1.38 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
package socket
import (
"fmt"
"strconv"
"github.com/replicatedhq/kots/pkg/socket/transport"
)
const (
webSocketProtocol = "ws://"
webSocketSecureProtocol = "wss://"
socketioUrl = "/socket.io/?EIO=3&transport=websocket&token=%s"
)
/**
Socket.io client representation
*/
type Client struct {
methods
Channel
}
/**
Get ws/wss url by host and port
*/
func GetUrl(host string, port int, auth string, secure bool) string {
var prefix string
if secure {
prefix = webSocketSecureProtocol
} else {
prefix = webSocketProtocol
}
u := fmt.Sprintf(socketioUrl, auth)
result := prefix + host + ":" + strconv.Itoa(port) + u
return result
}
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
}
// nspMsg := fmt.Sprintf("4%d%s", protocol.MessageTypeOpen, nsp)
// c.conn.WriteMessage(nspMsg)
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)
}