初试NodeJs开启websocket服务, 并获取客户端访问ip地址

一、WebSocket原理

HTTP请求-响应协议
在理解WebSocket原理之前，我们需要了解HTTP请求-响应协议。HTTP是一种无状态的请求-响应协议，客户端通过发送HTTP请求到服务器，服务器接收并处理请求，并返回HTTP响应给客户端。但是，在传统的HTTP协议中，客户端只能发送请求，而服务器只能通过响应来处理客户端的请求。

WebSocket协议
WebSocket协议是在HTTP协议的基础上进行扩展的。在建立WebSocket连接时，客户端首先发送一个HTTP请求到服务器，并将Upgrade头部字段设置为"websocket"，表示希望升级到WebSocket协议。服务器接收到这个请求后，如果支持WebSocket协议，会返回一个状态码101 Switching Protocols的HTTP响应，并通过Upgrade头部字段将连接升级为WebSocket连接。

升级完成后，客户端和服务器之间的通信不再遵循HTTP请求-响应模式，而是通过WebSocket协议进行双向的实时通信。客户端和服务器可以直接发送消息给对方，不需要等待对方的请求。

二、创建服务

1. 安装ws 模块

npm install --save ws

2. 创建服务端

const WebSocket = require('ws')

// 创建websocket服务, 端口port为: 5001
var WebSocketServer = WebSocket.Server,
    wss = new WebSocketServer({ port: 5001 });

// 服务端连接
wss.on('connection', function(ws, req){
 
  //当用户发送消息时
  ws.on('message', function(message) {
    console.log("message: " + message);
    
  });

  //关闭socket连接时
  ws.on('close', function() {
    console.log("连接关闭");
  });

  // 连接发生错误
  ws.on("error", function(err) {
	console.log("client error", err);
  });
})

3. 根据需求添加其他功能

当用户连接到我们的服务, 我们需要知道是哪一个人连接到服务时, 需要给此次连接, 添加一个唯一标识. 这时候需要用到另一个模块uuid

安装uuid模块

npm install --save uuid

给用户创建标识, 并且识别该访问用户的ip地址 也一起保存.

const WebSocket = require('ws')

// 创建websocket服务, 端口port为: 5001
var WebSocketServer = WebSocket.Server,
    wss = new WebSocketServer({ port: 5001 });

// 引入uuid模块
const uuid = require('uuid')

//定义一个空数组, 存放客户端信息
let clients = [];

// 服务端连接
wss.on('connection', function(ws, req){
  // 保存访问用户客户端ip地址 
  let clientAddress = req.socket.remoteAddress.split("::ffff:").join("")

  // 客户端client_uuid随机生成
  let client_uuid = uuid.v4()

  // 将新连接的客户端push到clients数组中
  clients.push({
    "id": client_uuid,
    "clientAddress": clientAddress
  })

  //控制台打印连接的client_uuid
  console.log(clientAddress + '来了');

  //当用户发送消息时
  ws.on('message', function(message) {
    console.log("message: " + message);
  });

  //关闭socket连接时
  let closeSocket = function() {
      //遍历客户端
    for (let i = 0; i < clients.length; i++) {
      //如果客户端存在
      if (clients[i].id == client_uuid) {
        //服务广播消息
        console.log(clients[i].clientAddress + '走了');
        //客户端数组中删掉
        clients.splice(i, 1);
      }
    }
  }
  ws.on('close', function() {
      closeSocket();
  });

  process.on('SIGINT', function() {
      console.log("接收到SIGINT信号, 准备退出...");
      closeSocket('Server has disconnected');
      process.exit();
  });
})

注意: 在通过req.socket.remoteAddress的方法获取到ip地址时, 发现格式中带有::ffff:前缀, 例如:

::ffff:192.168.124.25, 这是什么地址呢?

::ffff:192.168.124.25 是一个IPv4-Mapped IPv6 Address。为了实现IPv4-IPv6互通，IPv4地址会嵌入IPv6地址中，此时地址常表示为：

X:X:X:X:X:X:d.d.d.d

地址前96位采用IPv6的冒分十六进制表示法，
地址后32位则使用IPv4的点分十进制表示法，

::ffff:192.168.1.1是0000:0000:0000:0000:0000:ffff:192.168.1.1的简写，本质是一个IPv6的地址

如果只想拿到ip地址, 直接将前缀删掉即可.

持续更新...