hyperf-chat/app/Controller/WebSocketController.php

<?php
declare(strict_types=1);

namespace App\Controller;

use Hyperf\Di\Annotation\Inject;
use Hyperf\Contract\OnCloseInterface;
use Hyperf\Contract\OnMessageInterface;
use Hyperf\Contract\OnOpenInterface;
use Phper666\JWTAuth\JWT;
use Swoole\Http\Request;
use Swoole\Websocket\Frame;
use Hyperf\Amqp\Producer;
use Swoole\Http\Response;
use Swoole\WebSocket\Server;
use App\Traits\WebSocketTrait;
use App\Service\SocketFDService;
use App\Service\MessageHandleService;
use App\Service\SocketRoomService;
use App\Model\Group\UsersGroupMember;
use App\Amqp\Producer\ChatMessageProducer;

/**
 * Class WebSocketController
 * @package App\Controller
 */
class WebSocketController implements OnMessageInterface, OnOpenInterface, OnCloseInterface
{
    use WebSocketTrait;

    /**
     * @Inject
     * @var JWT
     */
    private $jwt;

    /**
     * @Inject
     * @var Producer
     */
    private $producer;

    /**
     * @inject
     * @var SocketFDService
     */
    private $socketFDService;

    /**
     * @inject
     * @var SocketRoomService
     */
    private $socketRoomService;

    /**
     * @inject
     * @var MessageHandleService
     */
    private $messageHandleService;

    const EVENTS = [
        'event_talk' => 'onTalk',
        'event_keyboard' => 'onKeyboard',
    ];

    /**
     * 连接创建成功回调事件
     *
     * @param Response|Server $server
     * @param Request $request
     */
    public function onOpen($server, Request $request): void
    {
        $token = $request->get['token'] ?? '';
        $userInfo = $this->jwt->getParserData($token);
        stdout_log()->notice("用户连接信息 : user_id:{$userInfo['user_id']} | fd:{$request->fd} 时间：" . date('Y-m-d H:i:s'));

        // 判断是否存在异地登录
        $isOnline = $this->socketFDService->isOnlineAll(intval($userInfo['user_id']));

        // 绑定fd与用户关系
        $this->socketFDService->bindRelation($request->fd, $userInfo['user_id']);

        // 加入群聊
        $groupIds = UsersGroupMember::getUserGroupIds($userInfo['user_id']);
        foreach ($groupIds as $group_id) {
            $this->socketRoomService->addRoomMember($userInfo['user_id'], $group_id);
        }

        if (!$isOnline) {
            // 推送消息至队列
            $this->producer->produce(
                new ChatMessageProducer('event_online_status', [
                    'user_id' => $userInfo['user_id'],
                    'status' => 1,
                    'notify' => '好友上线通知...'
                ])
            );
        }
    }

    /**
     * 消息接收回调事件
     *
     * @param Response|Server $server
     * @param Frame $frame
     */
    public function onMessage($server, Frame $frame): void
    {
        // 判断是否为心跳检测
        if ($frame->data == 'PING') return;

        [$event, $data] = array_values(json_decode($frame->data, true));

        if (isset(self::EVENTS[$event])) {
            $this->messageHandleService->{self::EVENTS[$event]}($server, $frame, $data);
            return;
        }
    }

    /**
     * 连接创建成功回调事件
     *
     * @param Response|\Swoole\Server $server
     * @param int $fd
     * @param int $reactorId
     */
    public function onClose($server, int $fd, int $reactorId): void
    {
        $user_id = $this->socketFDService->findFdUserId($fd);

        stdout_log()->notice("客户端FD:{$fd} 已关闭连接 ，用户ID为【{$user_id}】，关闭时间：" . date('Y-m-d H:i:s'));

        // 解除fd关系
        $this->socketFDService->removeRelation($fd);

        // 判断是否存在异地登录
        $isOnline = $this->socketFDService->isOnlineAll(intval($user_id));
        // ... 不存在异地登录，推送下线通知消息
        if (!$isOnline) {
            // 推送消息至队列
            $this->producer->produce(
                new ChatMessageProducer('event_online_status', [
                    'user_id' => $user_id,
                    'status' => 0,
                    'notify' => '好友离线通知通知...'
                ])
            );
        }
    }
}