利用workerman实现在线聊天系统的用户认证与权限控制

31082023-10-11 13:01:12

作为一种高性能的PHP socket框架，workerman广泛应用于实时通信系统的开发。在开发在线聊天系统时，用户认证与权限控制是非常重要的环节。本文将介绍如何利用workerman完成用户认证与权限控制，并附上代码示例。

用户认证

用户认证是指验证用户的身份是否合法，在在线聊天系统中通常采用基于Token的认证机制。具体步骤如下：

步骤1：用户登录时，服务器生成一个Token，将Token发送给客户端保存。
步骤2：客户端在后续的请求中，将Token以HTTP Header的形式发送给服务器。
步骤3：服务器接收到请求时，验证Token的有效性。如果Token有效，则认为用户已登录，可以继续处理请求；否则，返回认证失败的错误信息。

下面是一个使用workerman实现用户认证的示例代码：

require_once __DIR__ . '/vendor/autoload.php';
 
use WorkermanWorker;
use WorkermanConnectionAsyncTcpConnection;
use WorkermanProtocolsHttpRequest;
use WorkermanProtocolsHttpResponse;
 
$worker = new Worker('http://0.0.0.0:8080');
 
$users = [
    'user1' => 'password1',
    'user2' => 'password2',
    // ...
];
 
$worker->onMessage = function ($connection, Request $request) use ($users) {
    $path = $request->path();
    if ($path === '/login') {
        $username = $request->post('username');
        $password = $request->post('password');
        if (!isset($users[$username]) || $users[$username] !== $password) {
            $response = new Response(401, ['Content-Type' => 'application/json'], json_encode(['error' => 'Invalid credentials']));
            $connection->send($response);
        } else {
            $token = generateToken();
            $response = new Response(200, ['Content-Type' => 'application/json'], json_encode(['token' => $token]));
            $connection->send($response);
        }
    } elseif (substr($path, 0, 7) === '/api/v1') {
        $token = $request->header('Authorization');
        if (!validateToken($token)) {
            $response = new Response(401, ['Content-Type' => 'application/json'], json_encode(['error' => 'Unauthorized']));
            $connection->send($response);
        } else {
            // 处理请求逻辑
        }
    } else {
        $response = new Response(404, ['Content-Type' => 'text/html'], 'Not found');
        $connection->send($response);
    }
};
 
Worker::runAll();
 
function generateToken()
{
    // 生成Token逻辑
}
 
function validateToken($token)
{
    // 验证Token逻辑
}

权限控制

权限控制是指控制用户对系统资源的访问权限，在在线聊天系统中通常采用角色与权限的方式进行权限控制。具体步骤如下：

步骤1：定义角色和权限列表，并将其存储在数据库中。
步骤2：用户登录后，服务器根据用户的角色，获取该角色对应的权限列表。
步骤3：服务器在处理请求时，根据请求所需的权限，判断用户是否具有执行该操作的权限。如果具有权限，则继续处理请求；否则，返回权限不足的错误信息。

下面是一个使用workerman实现权限控制的示例代码：

require_once __DIR__ . '/vendor/autoload.php';
 
use WorkermanWorker;
use WorkermanConnectionAsyncTcpConnection;
use WorkermanProtocolsHttpRequest;
use WorkermanProtocolsHttpResponse;
 
$worker = new Worker('http://0.0.0.0:8080');
 
$roles = [
    'admin' => ['create', 'read', 'update', 'delete'],
    'user' => ['read']
];
 
$worker->onMessage = function ($connection, Request $request) use ($roles) {
    $path = $request->path();
    $role = getUserRole(); // 根据Token获取用户角色
 
    if (!isset($roles[$role])) {
        $response = new Response(401, ['Content-Type' => 'application/json'], json_encode(['error' => 'Unauthorized']));
        $connection->send($response);
        return;
    }
 
    $allowedPermissions = $roles[$role];
    $requiredPermission = extractRequiredPermission($path); // 根据请求路径提取所需权限
 
    if (!in_array($requiredPermission, $allowedPermissions)) {
        $response = new Response(403, ['Content-Type' => 'application/json'], json_encode(['error' => 'Forbidden']));
        $connection->send($response);
        return;
    }
 
    // 处理请求逻辑
};
 
Worker::runAll();
 
function getUserRole()
{
    // 根据Token获取用户角色的逻辑
}
 
function extractRequiredPermission($path)
{
    // 从请求路径中提取所需权限的逻辑
}