websocket如何实现自定义请求头的鉴权

要实现WebSocket自定义请求头的鉴权，可以在WebSocket客户端发送请求时添加自定义请求头，并在WebSocket服务端进行验证。

具体步骤如下：

1. 在WebSocket客户端中添加自定义请求头，例如：

const socket = new WebSocket('ws://localhost:8080', {
  headers: {
    'Authorization': 'Bearer YOUR_TOKEN_HERE'
  }
});

2. 在WebSocket服务端接收到客户端请求时，获取自定义请求头并进行验证，例如：

const httpServer = require('http').createServer();
const io = require('socket.io')(httpServer);

io.on('connection', (socket) => {
  const token = socket.handshake.headers.authorization.split(' ')[1];
  
  // 进行鉴权验证
  if (validateToken(token)) {
    // 鉴权通过，进行业务处理
    // ...
  } else {
    // 鉴权失败，关闭连接
    socket.close();
  }
});

httpServer.listen(8080);

其中`socket.handshake.headers`可以获取到WebSocket客户端发送的所有请求头，包括自定义请求头。我们可以从中获取`Authorization`请求头，并对其进行鉴权验证。

需要注意的是，WebSocket的握手阶段是基于HTTP协议的，因此我们可以使用HTTP请求头进行鉴权验证。同时，WebSocket服务端在进行鉴权验证时应该在`connection`事件中进行，而不是在`message`事件中进行。这是因为，在握手阶段鉴权失败时应该立即关闭连接，而在业务处理阶段鉴权失败时应该发送错误信息给客户端。