235 lines
7.2 KiB
PHP
235 lines
7.2 KiB
PHP
<?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);
|
||
}
|
||
}
|