监听websocket✅

Author: flxxyz

README.md

@@ -1 +1,6 @@
-# swoole websocket and tcp and udp
+# swoole websocket and tcp and udp
+
+## 任务
+- [x] 监听websocket
+- [ ] 监听tcp
+- [ ] 监听udp

composer.lock

@@ -3,7 +3,9 @@
 require_once './vendor/autoload.php';
 
 $config = [
-    'websocket' => [
+    'timezone'            => 'Asia/Shanghai',
+    'tick_interval_timer' => 30, //秒
+    'websocket'           => [
         'enable'  => true,
         'host'    => '0.0.0.0',
         'port'    => '9000',
@@ -15,7 +17,7 @@
         ],
         'handler' => swoole_websocket_and_tcp_and_udp_test\webscoket::class,
     ],
-    'http'      => [
+    'http'                => [
         'enable'  => true,
         'host'    => '0.0.0.0',
         'port'    => '9001',
@@ -24,7 +26,7 @@
             'open_websocket_protocol' => true,
         ],
     ],
-    'tcp'       => [
+    'tcp'                 => [
         'enable'  => true,
         'host'    => '0.0.0.0',
         'port'    => '9002',
@@ -37,7 +39,7 @@
             'heartbeat_idle_time'      => 60,
         ],
     ],
-    'udp'       => [
+    'udp'                 => [
         'enable'  => true,
         'host'    => '0.0.0.0',
         'port'    => '9003',
@@ -52,4 +54,4 @@
 
 $server = new \swoole_websocket_and_tcp_and_udp\Server($config);
 
-$server;
+$server->run();

example.php

@@ -3,12 +3,20 @@
 namespace swoole_websocket_and_tcp_and_udp;
 
 
+use swoole_websocket_and_tcp_and_udp\common\Logger;
+use swoole_websocket_and_tcp_and_udp\common\ProcessTrait;
+
 class Server
 {
+    use ProcessTrait, Logger;
+
     protected $port;
 
     protected $config;
 
+    /**
+     * @var \swoole_websocket_server | \swoole_http_server
+     */
     protected $server;
 
     protected $primaryConfig = [];
@@ -18,14 +26,16 @@ class Server
     public function __construct($config)
     {
         $this->config = $config;
+        ini_set('date.timezone', $this->config['timezone']);
+
         if (isset($this->config['websocket'])) {
             $this->enableWebsocket = true;
             $this->primaryConfig = $this->config['websocket'];
-            $server = swoole_websocket_server::class;
+            $serverClass = \swoole_websocket_server::class;
         } else {
             if (isset($this->config['http'])) {
                 $this->primaryConfig = $this->config['http'];
-                $server = swoole_http_server::class;
+                $serverClass = \swoole_http_server::class;
             }
         }
 
@@ -34,13 +44,15 @@ public function __construct($config)
         $type = $this->primaryConfig['type'];
         $setting = $this->primaryConfig['setting'];
 
-        $this->server = new $server($host, $port, SWOOLE_PROCESS, $type);
+        Logger::info("开始监听端口 {$host}:{$port}");
+        $this->server = new $serverClass($host, $port, SWOOLE_PROCESS, $type);
         $this->server->set($setting);
 
 
         $this->bindBaseEvent();
         $this->bindHttpEvent();
         $this->bindTaskEvent();
+        $this->bindWebsocketEvent();
     }
 
     /**
@@ -55,6 +67,7 @@ protected function bindBaseEvent()
         $this->server->on('ManagerStop', [$this, 'ManagerStop']);
         $this->server->on('WorkerStart', [$this, 'WorkerStart']);
         $this->server->on('WorkerStop', [$this, 'WorkerStop']);
+        $this->server->on('WorkerExit', [$this, 'WorkerExit']);
         $this->server->on('WorkerError', [$this, 'WorkerError']);
         $this->server->on('PipeMessage', [$this, 'PipeMessage']);
     }
@@ -75,9 +88,9 @@ protected function bindWebsocketEvent()
         if ($this->enableWebsocket) {
             $handlerClass = $this->primaryConfig['handler'];
             $handler = new $handlerClass();
-            if (!($handler instanceof protocol\WebSocketHandlerInterface)) {
-                throw new \Exception(sprintf('%s 当前类不属于 interface %s',
-                    $handlerClass, protocol\WebSocketHandlerInterface::class));
+            if (!($handler instanceof protocol\WebsocketEvent)) {
+                throw new \Exception(sprintf('%s 当前类不属于 %s',
+                    $handlerClass, protocol\WebsocketEvent::class));
             }
 
             $eventHandler = function ($method, array $params) use ($handler) {
@@ -102,5 +115,110 @@ protected function bindWebsocketEvent()
         }
     }
 
+    public function start(\swoole_http_server $server)
+    {
+        foreach (spl_autoload_functions() as $function) {
+            spl_autoload_unregister($function);
+        }
+
+        $this->setProcessName('master process');
+
+        if (version_compare(swoole_version(), '1.10.4', '<')) {
+            file_put_contents($this->config['pid_file'], $server->master_pid);
+        }
+    }
+
+    public function shutdown(\swoole_http_server $server)
+    {
+
+    }
+
+    public function ManagerStart(\swoole_http_server $server)
+    {
+
+    }
+
+    public function ManagerStop(\swoole_http_server $server)
+    {
+
+    }
+
+    public function WorkerStart(\swoole_http_server $server, $worker_id)
+    {
+        if ($worker_id >= (swoole_cpu_num() * 2)) {
+            $process = 'task worker';
+        } else {
+            $process = 'worker';
+        }
+
+        $this->setProcessName(sprintf('%s process %d', $process,
+            $worker_id));
+
+        if (!$server->taskworker) {
+            $server->task(1);
+        }
+    }
+
+    public function WorkerStop(\swoole_http_server $server, $worker_id)
+    {
+
+    }
+
+    public function WorkerExit(\swoole_http_server $server, $worker_id)
+    {
+
+    }
+
+    public function WorkerError(
+        \swoole_http_server $server,
+        $worker_id,
+        $worker_pid,
+        $exit_code,
+        $signal
+    ) {
+        Logger::err(sprintf('worker[%d] error: exitCode=%s, signal=%s',
+            $worker_id,
+            $exit_code, $signal));
+    }
+
+    public function PipeMessage(
+        \swoole_http_server $server,
+        $src_worker_id,
+        $message
+    ) {
+
+    }
+
+    public function task(
+        \swoole_http_server $server,
+        $task_id,
+        $src_worker_id,
+        $data
+    ) {
+        if ($src_worker_id == 1) {
+            swoole_timer_tick($this->config['tick_interval_timer'] * 1000,
+                function ($timer_id) {
+                    Logger::debug('此处可定时清理任务');
+                });
+        }
+    }
+
+    public function finish(\swoole_http_server $server, $task_id, $data)
+    {
+
+    }
+
+    public function request(
+        \swoole_http_request $request,
+        \swoole_http_response $response
+    ) {
+
+    }
+
+    public function run()
+    {
+        Logger::info('运行服务...');
+        $this->server->start();
+    }
 
 }

src/Server.php

@@ -0,0 +1,31 @@
+<?php
+
+namespace swoole_websocket_and_tcp_and_udp\common;
+
+
+trait Logger
+{
+    static function info($msg)
+    {
+        $time = date('Y-m-d H:i:s');
+        echo '[' . $time . '] INFO ' . $msg . PHP_EOL;
+    }
+
+    static function debug($msg)
+    {
+        $time = date('Y-m-d H:i:s');
+        echo '[' . $time . '] DEBUG ' . $msg . PHP_EOL;
+    }
+
+    static function warn($msg)
+    {
+        $time = date('Y-m-d H:i:s');
+        echo '[' . $time . '] WARN ' . $msg . PHP_EOL;
+    }
+
+    static function err($msg)
+    {
+        $time = date('Y-m-d H:i:s');
+        echo '[' . $time . '] ERROR ' . $msg . PHP_EOL;
+    }
+}

src/common/Logger.php

@@ -0,0 +1,18 @@
+<?php
+
+namespace swoole_websocket_and_tcp_and_udp\common;
+
+
+trait ProcessTrait
+{
+    public function setProcessName($name)
+    {
+        if (PHP_OS == 'Darwin') {
+            return;
+        }
+
+        if (function_exists('\swoole_set_process_name')) {
+            \swoole_set_process_name($name);
+        }
+    }
+}

src/common/ProcessTrait.php

@@ -3,9 +3,46 @@
 namespace swoole_websocket_and_tcp_and_udp_test;
 
 
+use swoole_websocket_and_tcp_and_udp\common\Logger;
 use swoole_websocket_and_tcp_and_udp\protocol\WebsocketEvent;
 
 class webscoket extends WebsocketEvent
 {
+    use Logger;
 
+    public function __construct()
+    {
+    }
+
+    public function open(
+        \swoole_websocket_server $server,
+        \swoole_http_request $request
+    ) {
+        $fd = $request->fd;
+        $info = $server->connection_info($fd);
+        Logger::info("{$info['remote_ip']}:{$info['remote_port']}, Connect");
+    }
+
+    public function message(
+        \swoole_websocket_server $server,
+        \swoole_websocket_frame $frame
+    ) {
+        $fd = $frame->fd;
+        $data = $frame->data;
+        $info = $server->connection_info($fd);
+
+        if ($data == 'exit') {
+            $server->push($fd, '再见👋');
+            $server->close($fd);
+        }
+        $server->push($fd, '你好呀');
+
+        Logger::info("{$info['remote_ip']}:{$info['remote_port']}, Message [fd{$fd}]: {$data}");
+    }
+
+    public function close(\swoole_websocket_server $server, $fd)
+    {
+        $info = $server->connection_info($fd);
+        Logger::info("{$info['remote_ip']}:{$info['remote_port']}, Close");
+    }
 }