使用 JWT 身份验证重新连接断开的 socket.io

Question

我正在使用 JWT 对 socket.io 进行身份验证。这是客户端代码：

const socket = io.connect(
            'ws://123.456.789.0:8500',
            {
                query: {
                    token: jwtToken
                },
                transports: ['websocket', 'polling']
            }
        );

现在，jwtToken 会在某个时候过期，并且与 服务器 的连接断开：

...
io.on('connection', async (socket) => {
    const isExpired = await checkExpireDateEveryMinute(socket.exp, 5);

    if (isExpired) {
        socket.disconnect();
    }

    await socket.on('query', async (data) => {

        ...

    });
});

使用新生成的 JWT 令牌从客户端重新连接的正确方法是什么？在这方面找不到太多资源。谢谢。

Answer 1

我认为另一种方法是使用 socket.io 的中间件功能，如果您使用 Error 对象调用 next()，它将阻止 on("connection") 回调触发。它还将错误消息传输到前端。您可以通过这种方式发送一个新的令牌，然后重新连接。

这是一个例子：

async function authenticate(socket,next){
    const isExpired = await checkExpireDateEveryMinute(socket.exp, 5);

    if(isExpired){
        const newToken = getNewToken()//Get a new token...
        next(new Error(newToken))//This will cause an error in your frontend, and pass the new token.
    }else{
        next();
    }
}
io.use(authenticate)//Register the middleware.

我必须指出一点：你在 socket.on(query,...) 上做“等待”，这是没有意义的。

至于你的前端，它可能看起来像这样：

function initSocket(token, url) {

  const socket = io.connect(url, {//Init the socket.
    query: {
      token
    }
  });

  return socket;
}

let socket = initSocket('someToken', 'http://localhost:8000')

socket.on('error', async (error) => {
  if(error === 'TOKEN_EXPIRED'){
    const newToken = await getToken();
    socket.close();
    socket = initSocket(newToken, 'http://localhost:8000')
  }
})

请注意，您需要重新注册您的套接字的所有事件。