diff --git a/README.md b/README.md
index 2d7a0b6..ade366f 100644
--- a/README.md
+++ b/README.md
@@ -31,19 +31,21 @@ composer require "jianyan74/yii2-websocket"
  在 `console/config/main.php` 加入以下配置。（注意：配置在controllerMap里面）
  
  ```
- 'web-socket' => [
-     'class' => 'jianyan\websocket\WebSocketController',
-     'host' => '0.0.0.0',// 监听地址
-     'port' => 9501,// 监听端口
-     'config' => [// 标准的swoole配置项都可以再此加入
-         'daemonize' => false,// 守护进程执行
-         'ssl_cert_file' => '',
-         'ssl_key_file' => '',
-         'pid_file' => __DIR__ . '/../../backend/runtime/logs/server.pid',
-         'log_file' => __DIR__ . '/../../backend/runtime/logs/swoole.log',
-         'log_level' => 0,
-     ],
- ],
+// webSocket
+'web-socket' => [
+    'class' => 'jianyan\websocket\WebSocketController',
+    'server' => 'jianyan\websocket\WebSocketServer',
+    'host' => '0.0.0.0',// 监听地址
+    'port' => 9501,// 监听端口
+    'config' => [// 标准的swoole配置项都可以再此加入
+        'daemonize' => false,// 守护进程执行
+        'ssl_cert_file' => '',
+        'ssl_key_file' => '',
+        'pid_file' => __DIR__ . '/../../backend/runtime/logs/server.pid',
+        'log_file' => __DIR__ . '/../../backend/runtime/logs/swoole.log',
+        'log_level' => 0,
+    ],
+],
  ```
  
  ### 使用
diff --git a/src/WebSocketController.php b/WebSocketController.php
similarity index 94%
rename from src/WebSocketController.php
rename to WebSocketController.php
index a2fc46d..53c6e96 100644
--- a/src/WebSocketController.php
+++ b/WebSocketController.php
@@ -4,19 +4,23 @@
 use Yii;
 use yii\console\Controller;
 use yii\helpers\FileHelper;
-use jianyan\websocket\WebSocket;
 
 /**
- * WebSocket
  * 启动 php ./yii web-socket/start
  * 停止 php ./yii web-socket/stop
  * 重启 php ./yii web-socket/restart
- *
  * Class WebSocketController
- * @package console\controllers
+ * @package jianyan\websocket
  */
 class WebSocketController extends Controller
 {
+    /**
+     * 实例化服务
+     *
+     * @var
+     */
+    public $server;
+
     /**
      * 监听地址
      *
@@ -75,7 +79,7 @@ public function actionStart()
         $this->setPid();
 
         // 运行
-        $ws = new WebSocket($this->host, $this->port, $this->mode, $this->socketType, $this->config);
+        $ws = new $this->server($this->host, $this->port, $this->mode, $this->socketType, $this->config);
         $ws->run();
 
         $this->stdout("服务正在运行,监听 {$this->host}:{$this->port}" . PHP_EOL);
diff --git a/src/WebSocketServer.php b/WebSocketServer.php
similarity index 100%
rename from src/WebSocketServer.php
rename to WebSocketServer.php