socket-chat.js
2.58 KB
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
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
/**
* socket 聊天
*
* @author: liqi <qi.li@yoho.cn>
* @date: 2016/11/8
*/
var times,
param,
socket,
server,
servers,
options,
connectId,
conversationMessage;
function _randomServer() {
return servers[new Date().getTime() % servers.length];
}
/**
* socket初始化
* @param opts
*/
function socketInit(opts) {
options = opts;
servers = options.servers;
server = _randomServer();
conversationMessage = options.conversationMessage;
param = JSON.stringify(conversationMessage);
/**
* 连接
* @returns {WebSocket}
*/
function socketConnect() {
var socketIns = new WebSocket(server + '/im?param=' + param);
socketIns.onmessage = options.onMessage || function() {
console.log('received msg');
};
socketIns.onopen = options.onOpen || function() {
console.log('websocket is open');
};
socketIns.onerror = options.onError || function(err) {
console.log(err);
};
return socketIns;
}
if (window.WebSocket) {
setTimeout(function() {
times = 1;
// socket对象存在或者正在链接
if(socket || socket && socket.readyState === WebSocket.CONNECTING) {
return;
}
socket = socketConnect();
connectId = setInterval(function() {
if (socket.readyState !== WebSocket.OPEN) {
if (times < 3) {
socket.close();
socket = socketConnect();
times++;
} else {
clearInterval(connectId);
socket = null;
// 连接失败回调
options.connectFailCb();
}
} else {
socket.onclose = options.onClose || function() {
console.log('websocket is closed');
};
clearInterval(connectId);
}
}, 5000);
}, 1000);
}
}
/**
* 发送消息
* @param msg 消息体
*/
function sendMsg(msg) {
if (!window.WebSocket) {
return;
}
if (socket.readyState === WebSocket.OPEN) {
socket.send(JSON.stringify(msg));
} else {
options.socketClosedCb();
}
}
/**
* 关闭socket
*/
function clearSocket() {
if (socket) {
socket = null;
}
}
module.exports = {
socket: socket,
init: socketInit,
clear: clearSocket,
send: sendMsg
};