layout: posttitle: Swoole 结合jwt auth实现登录聊天功能
subtitle: Swoole 结合jwt auth实现登录聊天功能
date: 2019-04-14
author: he xiaodong
header-img: img/default-post-bg.jpg
catalog: true
tags:
- Swoole
- jwt auth
- PHP
- 聊天室demo

整体逻辑：在聊天室页面，前期发送消息按钮不能点击，需要先登录，然后用户登录之后，在 login.php 方法验证用户信息，并使用 jwt 生成 token 并返回，前端收到 token 之后，可以选择保存到cookie中，也可以只在此页面短时间使用，在建立 websocket 链接的时候，带过去 token 信息，swoole 在 open 事件中收到 token, 解析 token, 如果验证失败则断开链接，成功则可以聊天。

个人写的 demo 在 swoole 文件夹下，在此文件夹下执行 composer require firebase/php-jwt ,安装 jwt 包，使用具体可以参考包文档。其他所需是 swoole.html, login.php, swoole.php 三个前后端页面，代码如下：

swoole.html

  1. <!DOCTYPE html>
  2. <html>
  3. <head>
  4.     <title>swoole chat room</title>
  5.     <meta charset="UTF-8">
  6.     <script type="text/javascript">
  7.         if(window.WebSocket){
  8.             // 端口和ip地址对应不要写错
  10.             var login = function () {
  11.                 var name = document.getElementById('name').value;
  12.                 var data = new FormData();
  13.                 data.append('name', name);
  15.                 var request = new XMLHttpRequest();
  16.                 request.open("post", 'login.php');
  17.                 request.send(data);
  18.                 request.onload = function() {
  19.                     if(request.readyState === 4 && request.status == 200) {
  20.                         var webSocket = new WebSocket("ws://127.0.0.1:9502?token=" + request.responseText);
  21.                         webSocket.onopen = function (event) {
  22.                             //webSocket.send("Hello,WebSocket!");
  23.                         };
  24.                         webSocket.onmessage = function (event) {
  25.                             var content = document.getElementById('content');
  26.                             content.innerHTML = content.innerHTML.concat('<p style="margin-left:20px;height:20px;line-height:20px;">'+event.data+'</p>');
  27.                         };
  29.                         // 处理发送事件
  31.                         var sendButton = document.getElementById('send');
  32.                         sendButton.disabled = false;
  33.                         sendButton.innerHTML  = '发送';
  34.                         sendButton.onclick = function(){
  35.                             var data = document.getElementById('message').value;
  36.                             webSocket.send(data);
  37.                         };
  38.                     } else {
  39.                         alert('error');
  40.                     }
  41.                 };
  43.             }
  44.         }else{
  45.             console.log("您的浏览器不支持WebSocket");
  46.         }
  47.     </script>
  48. </head>
  49. <body>
  50. <div style="width:600px;margin:0 auto;border:1px solid #ccc;">
  51.     <div style="height:40px">
  52.         <input type="text" id="name" style="margin-left:10px;height:25px;width:450px;">
  53.         <button onclick="login()" style="height:28px;width:75px;">登录</button>
  54.     </div>
  55.     <hr />
  56.     <div id="content" style="overflow-y:auto;height:300px;"></div>
  57.     <hr/>
  58.     <div style="height:40px">
  59.         <input type="text" id="message" style="margin-left:10px;height:25px;width:450px;">
  60.         <button id="send" style="height:28px;width:75px;" disabled="disabled">登录后可以发送</button>
  61.     </div>
  62. </div>
  63. </body>
  64. </html>

login.php

<?php
require __DIR__.'/vendor/autoload.php';
use Firebase\JWT\JWT;

// 前提检查用户信息，验证通过则返回数据，demo 跳过这步
// 文档地址： https://packagist.org/packages/firebase/php-jwt

$key = "swoole";
$token = array(
    "iss" => "localhost",
    "aud" => "localhost",
    "iat" => time(),
    "nbf" => time()  + 3600,
    "name" => $_POST['name']
);


$jwt = JWT::encode($token, $key);
echo $jwt;

swoole.php

<?php
require 'vendor/autoload.php';
use Firebase\JWT\JWT;
$server = new swoole_websocket_server("0.0.0.0", 9502);

$server->on('open', function ($server, $request) {
    $key = 'swoole';
    $token = $request->get['token'] ?? '';

    $decode = JWT::decode($token, $key, ['HS256']);
    if (!$decode) {
        $server->close($request->fd);
        return false;
    }

    $decodeArray = (array)$decode;
    $server->push($request->fd, "hello;" . $decodeArray['name'] . "\n");
});

$server->on('message', function (swoole_websocket_server $server, $request) {
    $server->push($request->fd, "hello");
});

$server->on('close', function ($server, $fd) {
    echo "client-{$fd} is closed\n";
    $server->close($fd);
});

$server->start();

© 原创文章，如遇到问题可以联系邮箱：hexiaodong1992@outlook.com

最后恰饭 阿里云全系列产品/短信包特惠购买 中小企业上云最佳选择 阿里云内部优惠券