72 lines
1.5 KiB
PHP
72 lines
1.5 KiB
PHP
<?php
|
|
|
|
namespace think\worker\websocket;
|
|
|
|
use think\Event;
|
|
use think\Request;
|
|
use think\worker\contract\websocket\HandlerInterface;
|
|
use think\worker\websocket\Event as WsEvent;
|
|
|
|
class Handler implements HandlerInterface
|
|
{
|
|
protected $event;
|
|
|
|
public function __construct(Event $event)
|
|
{
|
|
$this->event = $event;
|
|
}
|
|
|
|
/**
|
|
* "onOpen" listener.
|
|
*
|
|
* @param Request $request
|
|
*/
|
|
public function onOpen(Request $request)
|
|
{
|
|
$this->event->trigger('worker.websocket.Open', $request);
|
|
}
|
|
|
|
/**
|
|
* "onMessage" listener.
|
|
*
|
|
* @param Frame $frame
|
|
*/
|
|
public function onMessage(Frame $frame)
|
|
{
|
|
$this->event->trigger('worker.websocket.Message', $frame);
|
|
|
|
$event = $this->decode($frame->data);
|
|
if ($event) {
|
|
$this->event->trigger('worker.websocket.Event', $event);
|
|
}
|
|
}
|
|
|
|
/**
|
|
* "onClose" listener.
|
|
*/
|
|
public function onClose()
|
|
{
|
|
$this->event->trigger('worker.websocket.Close');
|
|
}
|
|
|
|
protected function decode($payload)
|
|
{
|
|
$data = json_decode($payload, true);
|
|
if (!empty($data['type'])) {
|
|
return new WsEvent($data['type'], $data['data'] ?? null);
|
|
}
|
|
return null;
|
|
}
|
|
|
|
public function encodeMessage($message)
|
|
{
|
|
if ($message instanceof WsEvent) {
|
|
return json_encode([
|
|
'type' => $message->type,
|
|
'data' => $message->data,
|
|
]);
|
|
}
|
|
return $message;
|
|
}
|
|
}
|