初始化代码

2025-12-22 14:34:25 +08:00
parent c2c5ae2fdd
commit a77dbc743f
1510 changed files with 213008 additions and 0 deletions
--- a/vendor/topthink/think-swoole/src/Service.php
+++ b/vendor/topthink/think-swoole/src/Service.php
@@ -0,0 +1,87 @@
+<?php
+// +----------------------------------------------------------------------
+// | ThinkPHP [ WE CAN DO IT JUST THINK IT ]
+// +----------------------------------------------------------------------
+// | Copyright (c) 2006-2018 http://thinkphp.cn All rights reserved.
+// +----------------------------------------------------------------------
+// | Licensed ( http://www.apache.org/licenses/LICENSE-2.0 )
+// +----------------------------------------------------------------------
+// | Author: yunwuxin <448901948@qq.com>
+// +----------------------------------------------------------------------
+
+namespace think\swoole;
+
+use Swoole\Http\Server as HttpServer;
+use Swoole\Server;
+use Swoole\Websocket\Server as WebsocketServer;
+use think\Route;
+use think\swoole\command\RpcInterface;
+use think\swoole\command\Server as ServerCommand;
+use think\swoole\websocket\socketio\Controller;
+
+class Service extends \think\Service
+{
+    protected $isWebsocket = false;
+
+    /**
+     * @var HttpServer | WebsocketServer
+     */
+    protected static $server;
+
+    public function register()
+    {
+        $this->isWebsocket = $this->app->config->get('swoole.websocket.enable', false);
+
+        $this->app->bind(Server::class, function () {
+            if (is_null(static::$server)) {
+                $this->createSwooleServer();
+            }
+
+            return static::$server;
+        });
+
+        $this->app->bind("swoole.server", Server::class);
+
+        $this->app->bind(PidManager::class, function () {
+            return new PidManager($this->app->config->get("swoole.server.options.pid_file"));
+        });
+    }
+
+    public function boot()
+    {
+        $this->commands(ServerCommand::class, RpcInterface::class);
+
+        if ($this->isWebsocket) {
+            $this->registerRoutes(function (Route $route) {
+                $route->group(function () use ($route) {
+                    $route->get('socket.io/', '@upgrade');
+                    $route->post('socket.io/', '@reject');
+                })
+                    ->prefix(Controller::class)
+                    ->allowCrossDomain([
+                        'Access-Control-Allow-Credentials' => 'true',
+                        'X-XSS-Protection'                 => 0,
+                    ]);
+            });
+        }
+    }
+
+    /**
+     * Create swoole server.
+     */
+    protected function createSwooleServer()
+    {
+        $server     = $this->isWebsocket ? WebsocketServer::class : HttpServer::class;
+        $config     = $this->app->config;
+        $host       = $config->get('swoole.server.host');
+        $port       = $config->get('swoole.server.port');
+        $socketType = $config->get('swoole.server.socket_type', SWOOLE_SOCK_TCP);
+        $mode       = $config->get('swoole.server.mode', SWOOLE_PROCESS);
+
+        static::$server = new $server($host, $port, $mode, $socketType);
+
+        $options = $config->get('swoole.server.options');
+
+        static::$server->set($options);
+    }
+}