/
nwsclient.go
100 lines (89 loc) · 2.37 KB
/
nwsclient.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
96
97
98
99
100
/**
*
* @author nghiatc
* @since Aug 8, 2018
*/
package nwsc
import (
"github.com/congnghia0609/ntc-gsuperwss/util"
"log"
"net/url"
"os"
"os/signal"
"time"
"github.com/gorilla/websocket"
uuid "github.com/satori/go.uuid"
)
// NWSClient struct
type NWSClient struct {
name string
interrupt chan os.Signal
done chan struct{}
url url.URL
conn *websocket.Conn
}
var mapInstanceWSC = make(map[string]*NWSClient)
// GetInstanceWSC Singleton NWSClient
func GetInstanceWSC(name string) *NWSClient {
log.Println("=================== NWSClient.GetInstanceWSC ===================")
return mapInstanceWSC[name]
}
// NewInstanceWSC new object NWSClient
func NewInstanceWSC(name string, scheme string, host string, path string) (*NWSClient, error) {
var nws *NWSClient
var err error
util.TCF{
Try: func() {
log.Println("=================== NWSClient.NewInstanceWSC ===================")
log.Printf("+++++++++ name: %s", name)
if len(name) <= 0 {
name = uuid.NewV4().String()
}
interrupt := make(chan os.Signal, 1)
signal.Notify(interrupt, os.Interrupt)
done := make(chan struct{})
url := url.URL{Scheme: scheme, Host: host, Path: path}
log.Printf("connecting to %s", url.String())
conn, _, err := websocket.DefaultDialer.Dial(url.String(), nil)
nws = &NWSClient{name: name, interrupt: interrupt, done: done, url: url, conn: conn}
if err != nil {
log.Println("[xxxxxx] ERROR:", err)
//log.Fatal("dial:", err)
}
mapInstanceWSC[name] = nws
log.Println("NewInstanceWSC done!!!")
},
Catch: func(e util.Exception) {
log.Printf("wsc.NewInstanceWSC Caught %v\n", e)
},
Finally: func() {
//log.Println("Finally...")
},
}.Do()
return nws, err
}
// Close NWSClient
func (nws *NWSClient) Close() {
if nws.conn != nil {
nws.conn.Close()
}
}
// Reconnect is method auto-reconnect to websocket server.
func (nws *NWSClient) Reconnect() {
// create ticket time for every 3 seconds
ticker := time.NewTicker(time.Duration(3) * time.Second)
var count = 1
for range ticker.C {
conn, _, err := websocket.DefaultDialer.Dial(nws.url.String(), nil)
// Create new websocket
log.Printf("\nRetry Connect [%s]: %d times\n", nws.url.String(), count)
count = count + 1
if err != nil {
log.Printf("Dial failed [%s]: %s\n\n", nws.url.String(), err.Error())
} else {
nws.conn.Close()
nws.conn = conn
break
}
}
}