hyperf-chat/app/Service/MessageHandleService.php

<?php

namespace App\Service;

use Hyperf\Di\Annotation\Inject;
use Swoole\Http\Response;
use Swoole\WebSocket\Frame;
use Swoole\WebSocket\Server;
use App\Amqp\Producer\ChatMessageProducer;
use App\Model\Chat\ChatRecord;
use App\Model\Group\Group;
use App\Model\UsersFriend;
use App\Cache\LastMessage;
use App\Cache\UnreadTalk;

class MessageHandleService
{
    /**
     * @inject
     * @var SocketClientService
     */
    private $socketClientService;

    /**
     * 对话消息
     *
     * @param Response|Server $server
     * @param Frame           $frame
     * @param array|string    $data 解析后数据
     * @return void
     */
    public function onTalk($server, Frame $frame, $data)
    {
        $user_id = $this->socketClientService->findFdUserId($frame->fd);
        if ($user_id != $data['send_user']) {
            return;
        }

        // 验证消息类型 私聊|群聊
        if (!in_array($data['source_type'], [1, 2])) {
            return;
        }

        // 验证发送消息用户与接受消息用户之间是否存在好友或群聊关系(后期走缓存)
        if ($data['source_type'] == 1) {// 私信
            // 判断发送者和接受者是否是好友关系
            if (!UsersFriend::isFriend((int)$data['send_user'], (int)$data['receive_user'], true)) {
                return;
            }
        } else if ($data['source_type'] == 2) {// 群聊
            // 判断是否属于群成员
            if (!Group::isMember((int)$data['receive_user'], (int)$data['send_user'])) {
                return;
            }
        }

        $result = ChatRecord::create([
            'source'     => $data['source_type'],
            'msg_type'   => 1,
            'user_id'    => $data['send_user'],
            'receive_id' => $data['receive_user'],
            'content'    => htmlspecialchars($data['text_message']),
            'created_at' => date('Y-m-d H:i:s'),
        ]);

        if (!$result) return;

        // 判断是否私聊
        if ($result->source == 1) {
            // 设置好友消息未读数
            UnreadTalk::getInstance()->increment($result->user_id, $result->receive_id);
        }

        // 缓存最后一条聊天消息
        LastMessage::getInstance()->save($result->source, $result->user_id, $result->receive_id, [
            'text'       => mb_substr($result->content, 0, 30),
            'created_at' => $result->created_at
        ]);

        // 推送消息
        push_amqp(new ChatMessageProducer('event_talk', [
            'sender'    => intval($data['send_user']),     // 发送者ID
            'receive'   => intval($data['receive_user']),  // 接收者ID
            'source'    => intval($data['source_type']),   // 接收者类型[1:好友;2:群组;]
            'record_id' => $result->id
        ]));
    }

    /**
     * 键盘输入消息
     *
     * @param Response|Server $server
     * @param Frame           $frame
     * @param array|string    $data 解析后数据
     */
    public function onKeyboard($server, Frame $frame, $data)
    {
        push_amqp(new ChatMessageProducer('event_keyboard', [
            'send_user'    => intval($data['send_user']),     // 发送者ID
            'receive_user' => intval($data['receive_user']),  // 接收者ID
        ]));
    }
}