hyperf-chat/app/Service/Message/ReceiveHandleService.php

<?php

namespace App\Service\Message;

use App\Cache\LastMessage;
use App\Cache\UnreadTalk;
use App\Constants\SocketConstants;
use App\Constants\TalkMsgType;
use App\Constants\TalkType;
use App\Model\Chat\TalkRecords;
use App\Model\Group\Group;
use App\Model\UsersFriend;
use App\Service\SocketClientService;
use App\Support\MessageProducer;
use Swoole\Http\Response;
use Swoole\WebSocket\Frame;
use Swoole\WebSocket\Server;

class ReceiveHandleService
{
    /**
     * @var SocketClientService
     */
    private $socketClientService;

    public function __construct(SocketClientService $clientService)
    {
        $this->socketClientService = $clientService;
    }

    /**
     * 对话消息
     *
     * @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['sender_id']) {
            return;
        }

        // 验证消息类型 私聊|群聊
        if (!in_array($data['talk_type'], TalkType::getTypes())) {
            return;
        }

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

        $result = TalkRecords::create([
            'talk_type'   => $data['talk_type'],
            'user_id'     => $data['sender_id'],
            'receiver_id' => $data['receiver_id'],
            'msg_type'    => TalkMsgType::TEXT_MESSAGE,
            'content'     => htmlspecialchars($data['text_message']),
            'created_at'  => date('Y-m-d H:i:s'),
            'updated_at'  => date('Y-m-d H:i:s'),
        ]);

        if (!$result) return;

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

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

        MessageProducer::publish(
            MessageProducer::create(SocketConstants::EVENT_TALK, [
                'sender_id'   => $result->user_id,
                'receiver_id' => $result->receiver_id,
                'talk_type'   => $result->talk_type,
                'record_id'   => $result->id
            ])
        );
    }

    /**
     * 键盘输入消息
     *
     * @param Response|Server $server
     * @param Frame           $frame
     * @param array|string    $data 解析后数据
     * @return false
     */
    public function onKeyboard($server, Frame $frame, $data)
    {
        MessageProducer::publish(
            MessageProducer::create(SocketConstants::EVENT_KEYBOARD, [
                'sender_id'   => intval($data['sender_id']),
                'receiver_id' => intval($data['receiver_id']),
            ])
        );
    }
}