nodejs搭建一个webscoket服务器【转】

//引入ws模块

var WebSocket = require( 'ws' );

//创建websocket服务，端口port为：****

var WebSocketServer = WebSocket.Server,
wss =  new WebSocketServer({ port: 8180 });

//引入uuid模块

var uuid = require( 'uuid' );

//定义一个空数组，存放客户端的信息

var clients = [];

//定义发送消息方法wsSend

//参数为 type:类型

//client_uuid:随机生成的客户端id

//nickname:昵称

//message:消息

//clientcount:客户端个数

function wsSend(type, client_uuid, nickname, message, clientcount) {
//遍历客户端
for ( var i = 0; i < clients.length; i++) {
//&#x58F0;&#x660E;&#x5BA2;&#x6237;&#x7AEF;
var clientSocket = clients[i].ws;
if (clientSocket.readyState === WebSocket.OPEN) {
//&#x5BA2;&#x6237;&#x7AEF;&#x53D1;&#x9001;&#x5904;&#x7406;&#x8FC7;&#x7684;&#x4FE1;&#x606F;
clientSocket.send(JSON.stringify({
"type" : type,
"id" : client_uuid,
"nickname" : nickname,
"message" : message,
"clientcount" : clientcount,
}));
}
}

//声明客户端index默认为1

var clientIndex = 1;

//服务端连接

wss.on( 'connection' ,  function (ws) {
//客户端client_uuid随机生成
var client_uuid = uuid.v4();
//昵称为游客+客户端index
var nickname =  "游客" + clientIndex;
//client++
clientIndex += 1;
//将新连接的客户端push到clients数组中
clients.push({  "id" : client_uuid,  "ws" : ws,  "nickname" : nickname });
//控制台打印连接的client_uuid
console.log( 'client [%s] connected' , client_uuid);
//声明连接信息为 昵称+来了
// var connect_message = nickname + " 来了";
var connect_message =  " 来了" ;
//服务器广播信息 ***来了
wsSend( "notification" , client_uuid, nickname, connect_message, clients.length);
//当用户发送消息时
ws.on( 'message' ,  function (message) {
console.log( "message" , ArrayBufferUTF8ToStr(message));
// 用户输入"/nick"的话为重命名消息
if (message.indexOf( '/nick' ) === 0) {
var nickname_array = message.split( ' ' );
if (nickname_array.length >= 2) {
var old_nickname = nickname;
nickname = nickname_array[1];
var nickname_message =  "用户 " + old_nickname +  " 改名为： " + nickname;
wsSend( "nick_update" , client_uuid, nickname, nickname_message, clients.length);
}
}  //发送消息
else {
wsSend( "message" , client_uuid, nickname, message, clients.length);
}
});
//每个固定的时间 服务器给客户端发送消息
setInterval(() => {
if (ws.readyState == ws.OPEN){
let str =  "服务器send" +Math.floor(Math.random()*100)
var msg = Buffer.from( str);
console.log(str);
wsSend( "message" , client_uuid, nickname, msg, clients.length);
} else {
console.log( "还没有接通" )
}
}, 3000);
//关闭socket连接时
var closeSocket =  function (customMessage) {
console.log( "------关闭------" )
//遍历客户端
for ( var i = 0; i < clients.length; i++) {
//&#x5982;&#x679C;&#x5BA2;&#x6237;&#x7AEF;&#x5B58;&#x5728;
if (clients[i].id == client_uuid) {
// &#x58F0;&#x660E;&#x79BB;&#x5F00;&#x4FE1;&#x606F;
var disconnect_message;
if (customMessage) {
disconnect_message = customMessage;
}&#xA0; else {
disconnect_message = nickname +&#xA0; " &#x8D70;&#x4E86;" ;
}
//&#x5BA2;&#x6237;&#x7AEF;&#x6570;&#x7EC4;&#x4E2D;&#x5220;&#x6389;
clients.splice(i, 1);
//&#x670D;&#x52A1;&#x5E7F;&#x64AD;&#x6D88;&#x606F;
wsSend( "notification" , client_uuid, nickname, disconnect_message, clients.length);
}
}
}
ws.on( 'close' ,&#xA0; function () {
closeSocket();
});
process.on( 'SIGINT' ,&#xA0; function () {
console.log( "Closing things" );
closeSocket( 'Server has disconnected' );
process.exit();
});

});

function ArrayBufferUTF8ToStr(array) {
var out,i,len,c;
var char2,char3;
if (array  instanceof ArrayBuffer) {
array =  new Uint8Array(array);
}
out =  "" ;
len = array.length;
i = 0;
while (i < len) {
c = array[i++];
switch (c >> 4) {
case 0:&#xA0; case 1:&#xA0; case 2:&#xA0; case 3:&#xA0; case 4:&#xA0; case 5:&#xA0; case 6:&#xA0; case 7:
// 0xxxxxxx
out += String.fromCharCode(c);
break ;
case 12:&#xA0; case 13:
// 110x xxxx&#xA0;&#xA0; 10xx xxxx
char2 = array[i++];
out += String.fromCharCode(((c & 0x1F) << 6) | (char2 & 0x3F));
break ;
case 14:
// 1110 xxxx&#xA0; 10xx xxxx&#xA0; 10xx xxxx
char2 = array[i++];
char3 = array[i++];
out += String.fromCharCode(((c & 0x0F) << 12) |
((char2 & 0x3F) << 6) |
((char3 & 0x3F) << 0));
break ;
}
}
return out;

Original: https://www.cnblogs.com/mazhenyu/p/16375441.html
Author: 3D入魔
Title: nodejs搭建一个webscoket服务器【转】