311 lines
9.1 KiB
PHP
311 lines
9.1 KiB
PHP
<?php
|
|
|
|
namespace app\common\utils;
|
|
|
|
use GatewayWorker\Lib\Gateway;
|
|
use app\common\utils\WsResponse;
|
|
use app\common\service\MessageNotificationsService;
|
|
|
|
class WsEvents
|
|
{
|
|
|
|
private $registerAddress = '127.0.0.1:1236';
|
|
protected $clientId;
|
|
protected $eventData;
|
|
|
|
public function __construct()
|
|
{
|
|
Gateway::$registerAddress = $this->registerAddress;
|
|
}
|
|
|
|
/**
|
|
* 监听消息接收
|
|
*/
|
|
public function message($client_id, $message)
|
|
{
|
|
$this->clientId = $client_id;
|
|
|
|
$payload = json_decode($message, true);
|
|
$event = $payload['event'];
|
|
$this->eventData = $payload['data'] ?? [];
|
|
$traceId = $payload['trace_id'] ?? null;
|
|
|
|
$method = $this->eventToMethod($event);
|
|
|
|
if (method_exists($this, $method)) {
|
|
try {
|
|
$this->$method($traceId);
|
|
} catch (\Throwable $th) {
|
|
Gateway::sendToClient($client_id, WsResponse::error(
|
|
$event,
|
|
500,
|
|
$th->getMessage(),
|
|
$traceId
|
|
));
|
|
}
|
|
} else {
|
|
Gateway::sendToClient($client_id, WsResponse::error(
|
|
$event,
|
|
404,
|
|
"Event handler not found: $method",
|
|
$traceId
|
|
));
|
|
}
|
|
}
|
|
|
|
protected function eventToMethod(string $event): string
|
|
{
|
|
$parts = explode('.', $event);
|
|
$method = array_shift($parts);
|
|
foreach ($parts as $part) {
|
|
$method .= ucfirst($part);
|
|
}
|
|
return $method;
|
|
}
|
|
|
|
|
|
// ================= 系统事件 =================
|
|
/**
|
|
* 认证
|
|
*/
|
|
public function auth($client_id, $payload)
|
|
{
|
|
try {
|
|
// 检查来源
|
|
if (!isset($payload['server']['HTTP_ORIGIN'])) {
|
|
flg('[x] [TOKEN认证] 未定义HTTP_ORIGIN', false);
|
|
Gateway::sendToClient($client_id, WsResponse::error(
|
|
'system.auth',
|
|
1001,
|
|
'[x] [TOKEN认证] 未定义HTTP_ORIGIN'
|
|
));
|
|
Gateway::closeClient($client_id);
|
|
return;
|
|
}
|
|
|
|
// 连接是否满足条件
|
|
if (getenv('ENVIRONMENT') === 'master' && $payload['server']['HTTP_ORIGIN'] === getenv('MAIN_DOMAIN')) {
|
|
flg('[x] [TOKEN认证] HTTP_ORIGIN不满足条件', false);
|
|
Gateway::sendToClient($client_id, WsResponse::error(
|
|
'system.auth',
|
|
1001,
|
|
'[x] [TOKEN认证] HTTP_ORIGIN不满足条件'
|
|
));
|
|
Gateway::closeClient($client_id);
|
|
return;
|
|
}
|
|
|
|
// 签名检查
|
|
if (!isset($payload['get']['token'])) {
|
|
flg('[x] [TOKEN认证] 未携带TOKEN参数', false);
|
|
Gateway::sendToClient($client_id, WsResponse::error(
|
|
'system.auth',
|
|
1001,
|
|
'[x] [TOKEN认证] 未携带TOKEN参数'
|
|
));
|
|
Gateway::closeClient($client_id);
|
|
return;
|
|
}
|
|
|
|
// 解析TOKEN
|
|
$user = (new Jwt())->verifyToken($payload['get']['token']);
|
|
|
|
// 绑定用户ID
|
|
Gateway::bindUid($client_id, $user->uid);
|
|
|
|
// 绑定SESSION
|
|
$_SESSION['uid'] = $user->uid;
|
|
$_SESSION['name'] = $user->uname;
|
|
|
|
// 响应
|
|
Gateway::sendToClient($client_id, WsResponse::success(
|
|
'system.auth',
|
|
['uid' => $user->uid]
|
|
));
|
|
} catch (\Throwable $th) {
|
|
flg('[x] [TOKEN认证] TOKEN认证失败：'.$th->getMessage(), false);
|
|
Gateway::sendToClient($client_id, WsResponse::error(
|
|
'system.auth',
|
|
1001,
|
|
'[x] [TOKEN认证] TOKEN认证失败：'.$th->getMessage()
|
|
));
|
|
Gateway::closeClient($client_id);
|
|
return;
|
|
}
|
|
}
|
|
|
|
public function logout(string $client_id)
|
|
{
|
|
if ($_SESSION !== null) {
|
|
var_dump($_SESSION);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* 系统心跳
|
|
*
|
|
* @param string|null $traceId
|
|
*/
|
|
public function systemPing(?string $traceId)
|
|
{
|
|
Gateway::sendToClient($this->clientId, WsResponse::success(
|
|
'system.pong',
|
|
['time' => time()],
|
|
$traceId
|
|
));
|
|
}
|
|
|
|
|
|
public function systemConnect(?string $traceId)
|
|
{
|
|
$num = (new \app\common\utils\approval\service\ApprovalFlowService())->getInitTodoCount($_SESSION['uid']);
|
|
$this->todoCounts($_SESSION['uid'], $num);
|
|
|
|
Gateway::sendToClient($this->clientId, WsResponse::success(
|
|
'system.connect.receive',
|
|
['time' => time()],
|
|
$traceId
|
|
));
|
|
}
|
|
|
|
// ================= 消息通知事件 =================
|
|
/**
|
|
* 管理端主动推送通知
|
|
*/
|
|
public function notifyPush(?string $traceId)
|
|
{
|
|
$title = $this->eventData['title'] ?? null;
|
|
$content = $this->eventData['content'] ?? null;
|
|
$toUids = $this->eventData['to_uids'] ?? [];
|
|
|
|
if (!$title || !$content) {
|
|
Gateway::sendToClient($this->clientId, WsResponse::error(
|
|
'notify.push.result',
|
|
400,
|
|
'标题和内容不能为空',
|
|
$traceId
|
|
));
|
|
return;
|
|
}
|
|
|
|
// 写消息
|
|
$msgService = new MessageNotificationsService();
|
|
$msgId = $msgService->createMessage([
|
|
'sender_id' => $this->eventData['sender_id'] ?? 0,
|
|
'sender_type' => $this->eventData['sender_type'] ?? 'system',
|
|
'title' => $title,
|
|
'content' => $content,
|
|
'scene' => $this->eventData['scene'] ?? 0,
|
|
'recipient' => $this->eventData['to_uids'] ? implode(',', $this->eventData['to_uids']) : '',
|
|
'push_method' => isset($this->eventData['push_method']) ? $this->eventData['push_method'] : 0
|
|
]);
|
|
|
|
// 写用户状态
|
|
$uids = !empty($toUids) ? $toUids : $msgService->getAllUserIds();
|
|
$msgService->addUserStatus($msgId, $uids);
|
|
|
|
$payload = WsResponse::success('notify.receive', [
|
|
'msg_id' => $msgId,
|
|
'title' => $title,
|
|
'content' => $content,
|
|
'created_at' => time()
|
|
], $traceId);
|
|
|
|
if (!empty($toUids)) {
|
|
// 推送给指定用户
|
|
foreach ($toUids as $uid) {
|
|
Gateway::sendToUid($uid, $payload);
|
|
}
|
|
} else {
|
|
// 推送给全体在线用户
|
|
Gateway::sendToAll($payload);
|
|
}
|
|
|
|
// 回复发送者确认
|
|
Gateway::sendToClient($this->clientId, WsResponse::success(
|
|
'notify.push.result',
|
|
['msg_id' => $msgId],
|
|
$traceId,
|
|
'消息发送成功'
|
|
));
|
|
}
|
|
|
|
/**
|
|
* 客户端标记通知已读
|
|
*/
|
|
public function notifyRead(?string $traceId)
|
|
{
|
|
$msgIds = $this->eventData['msg_ids'];
|
|
if (!is_array($msgIds) && $msgIds !== 'all') {
|
|
Gateway::sendToClient($this->clientId, WsResponse::error(
|
|
'notify.read.result',
|
|
400,
|
|
'消息ID不能为空',
|
|
$traceId
|
|
));
|
|
return;
|
|
}
|
|
|
|
// 更新消息为已读
|
|
$msgService = new MessageNotificationsService();
|
|
$msgService->markAsRead($msgIds, Gateway::getUidByClientId($this->clientId));
|
|
|
|
Gateway::sendToClient($this->clientId, WsResponse::success(
|
|
'notify.read.result',
|
|
['msg_ids' => $msgIds],
|
|
$traceId
|
|
));
|
|
}
|
|
|
|
|
|
/*
|
|
init.pull
|
|
approval.todo.counts 审批待办数量
|
|
|
|
*/
|
|
|
|
// ================= 主动推送 =================
|
|
|
|
public function pushWebsocket(string $event, int|string $uid, array $data)
|
|
{
|
|
return match (gettype($uid)) {
|
|
'integer' => Gateway::sendToUid($uid, WsResponse::success($event, $data)),
|
|
'string' => Gateway::sendToClient($uid, WsResponse::success($event, $data)),
|
|
default => throw new \InvalidArgumentException('UID must be integer or string'),
|
|
};
|
|
}
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
// ================= 临时业务处理 =================
|
|
|
|
public function todoCounts(int $userId, int $num = 0)
|
|
{
|
|
$ly = [
|
|
1 => 186,
|
|
3 => 224,
|
|
];
|
|
|
|
if ($num === 0) {
|
|
$num = (new \app\common\utils\approval\service\ApprovalFlowService())->getInitTodoCount($userId);
|
|
}
|
|
|
|
$systemIds = \app\common\model\UserSystems::where('user_id', $userId)->pluck('system_id')->toArray();
|
|
if (count($systemIds) > 0) {
|
|
foreach ($systemIds as $systemId) {
|
|
if (isset($ly[$systemId])) {
|
|
$chain = \app\common\service\MenuBadgeService::buildBadgeChain(menuId: $ly[$systemId], leafCount: $num);
|
|
$this->pushWebsocket('approval.todo.counts', $userId, $chain);
|
|
}
|
|
}
|
|
}
|
|
}
|
|
} |