zx/app/webscoket/Manager.php

235 lines
7.2 KiB
PHP
Raw Blame History

This file contains ambiguous Unicode characters

This file contains Unicode characters that might be confused with other characters. If you think that this is intentional, you can safely ignore this warning. Use the Escape button to reveal them.

<?php
// +----------------------------------------------------------------------
// | CRMEB [ CRMEB赋能开发者助力企业发展 ]
// +----------------------------------------------------------------------
// | Copyright (c) 2016~2022 https://www.crmeb.com All rights reserved.
// +----------------------------------------------------------------------
// | Licensed CRMEB并不是自由软件未经许可不能去掉CRMEB相关版权
// +----------------------------------------------------------------------
// | Author: CRMEB Team <admin@crmeb.com>
// +----------------------------------------------------------------------
namespace app\webscoket;
use app\common\repositories\system\DataScreenRepository;
use app\webscoket\handler\AdminHandler;
use app\webscoket\handler\MerchantHandler;
use app\webscoket\handler\ServiceHandler;
use app\webscoket\handler\UserHandler;
use Swoole\Server;
use Swoole\Websocket\Frame;
use think\Config;
use think\Event;
use think\facade\Cache;
use think\Request;
use think\response\Json;
use think\swoole\Websocket;
use think\swoole\websocket\Room;
/**
* Class Manager
* @package app\webscoket
* @author xaboy
* @day 2020-04-29
*/
class Manager extends Websocket
{
/**
* @var \Swoole\WebSocket\Server
*/
protected $server;
/**
* @var Ping
*/
protected $pingService;
/**
* @var int
*/
protected $cache_timeout;
const USER_TYPE = ['admin', 'user', 'mer', 'ser'];
/**
* Manager constructor.
* @param Server $server
* @param Room $room
* @param Event $event
* @param Ping $ping
* @param Config $config
*/
public function __construct(Server $server, Room $room, Event $event, Ping $ping, Config $config)
{
parent::__construct($server, $room, $event);
$this->pingService = $ping;
$this->cache_timeout = (int)($config->get('swoole.websocket.ping_timeout', 60000) / 1000) + 2;
app()->bind('websocket_handler_admin', AdminHandler::class);
app()->bind('websocket_handler_user', UserHandler::class);
app()->bind('websocket_handler_mer', MerchantHandler::class);
app()->bind('websocket_handler_ser', ServiceHandler::class);
}
/**
* @param int $fd
* @param Request $request
* @return mixed
* @author xaboy
* @day 2020-05-06
*/
public function onOpen($fd, Request $request)
{
$type = $request->get('type');
$token = $request->get('token');
if (!$token || !in_array($type, self::USER_TYPE)) {
return $this->server->close($fd);
}
try {
$data = $this->exec($type, 'login', compact('fd', 'request', 'token'))->getData();
} catch (\Exception $e) {
return $this->server->close($fd);
}
if (!isset($data['status']) || $data['status'] != 200 || !($data['data']['uid'] ?? null)) {
return $this->server->close($fd);
}
$type = array_search($type, self::USER_TYPE);
$this->login($type, $fd, $data['data']);
$this->pingService->createPing($fd, time(), $this->cache_timeout);
return $this->send($fd, app('json')->message('ping', ['now' => time()]));
}
public function login($type, $fd, $data)
{
$key = '_ws_' . $type;
Cache::set('_ws_f_' . $fd, [
'type' => $type,
'uid' => $data['uid'],
'fd' => $fd,
'payload' => $data['payload'] ?? null,
'mer_id' => $data['mer_id'] ?? null
], 3600);
if (isset($data['mer_id'])) {
$groupKey = $key . '_group' . $data['mer_id'];
Cache::sadd($groupKey, $fd);
Cache::expire($groupKey, 3600);
} else {
Cache::sadd($key, $fd);
}
$this->refresh($type, $fd, $data['uid']);
}
public function refresh($type, $fd, $uid)
{
$key = '_ws_' . $type;
Cache::expire($key, 3600);
Cache::expire($key . $uid, 3600);
Cache::expire('_ws_f_' . $fd, 3600);
}
public function logout($type, $fd)
{
$data = Cache::get('_ws_f_' . $fd);
$key = '_ws_' . $type;
Cache::srem($key, $fd);
if ($data) {
Cache::delete('_ws_f_' . $fd);
Cache::srem($key . $data['uid'], $fd);
if (($data['mer_id'] ?? null) !== null) {
$groupKey = $key . '_group' . $data['mer_id'];
Cache::srem($groupKey, $fd);
}
}
}
public static function merFd($merId)
{
return Cache::smembers('_ws_2_group' . $merId) ?: [];
}
public static function userFd($type, $uid = '')
{
$key = '_ws_' . $type . $uid;
return Cache::smembers($key) ?: [];
}
/**
* @param $type
* @param $method
* @param $result
* @return null|Json
* @author xaboy
* @day 2020-05-06
*/
protected function exec($type, $method, $result)
{
$handler = app()->make('websocket_handler_' . $type);
if (!method_exists($handler, $method)) return null;
/** @var Json $response */
return $handler->{$method}($result);
}
/**
* @param Frame $frame
* @return bool
* @author xaboy
* @day 2020-04-29
*/
public function onMessage(Frame $frame)
{
$info = Cache::get('_ws_f_' . $frame->fd);
$result = json_decode($frame->data, true) ?: [];
if (!isset($result['type']) || !$result['type']) return true;
if (!$info) return true;
$this->refresh($info['type'], $frame->fd, $info['uid']);
if (($info['mer_id'] ?? null) !== null) {
$groupKey = '_ws_' . $info['type'] . '_group' . $info['mer_id'];
Cache::expire($groupKey, 3600);
}
if ($result['type'] == 'ping') {
return $this->send($frame->fd, app('json')->message('ping', ['now' => time() ]));
}
if ($result['type'] == 'data_status') {
$k = env('APP_KEY','merchant').'_data_status';
Cache::set($k, time(),60);
$data = app()->make(DataScreenRepository::class)->data_screen();
return $this->send($frame->fd, app('json')->message('data_screen', $data));
}
$data = $result['data'] ?? [];
$frame->uid = $info['uid'];
$frame->payload = $info['payload'];
/** @var Json $response */
$response = $this->exec(self::USER_TYPE[$info['type']], $result['type'], compact('data', 'frame', 'info'));
if ($response) return $this->send($frame->fd, $response);
return true;
}
protected function send($fd, Json $json)
{
$this->pingService->createPing($fd, time(), $this->cache_timeout);
if ($this->server->isEstablished($fd) && $this->server->exist($fd)) {
$this->server->push($fd, json_encode($json->getData()));
}
return true;
}
/**
* @param int $fd
* @param int $reactorId
* @author xaboy
* @day 2020-04-29
*/
public function onClose($fd, $reactorId)
{
$data = Cache::get('_ws_f_' . $fd);
if ($data) {
$this->logout($data['type'], $fd);
$this->exec(self::USER_TYPE[$data['type']], 'close', $data);
}
$this->pingService->removePing($fd);
}
}