├── .gitignore
├── demo
    └── laravel
    │   ├── app
    │       ├── config
    │       │   └── websocket.php
    │       ├── Console
    │       │   ├── Commands
    │       │   │   └── ChatWebSocketCommand.php
    │       │   └── Kernel.php
    │       └── Server
    │       │   └── Chat
    │       │       └── ChatHandle.php
    │   └── public
    │       └── js
    │           └── push.js
├── src
    ├── Facades
    │   └── Queue.php
    ├── RedisSubscribe.php
    ├── Redis
    │   ├── RedisManagerCli.php
    │   └── RedisServiceProvider.php
    ├── Contracts
    │   └── Queue.php
    ├── Commands
    │   └── WebSocketCommand.php
    ├── RedisSubscribeCommand.php
    ├── Service
    │   └── PushService.php
    ├── WebSocket.php
    └── ServerHandle.php
├── composer.json
└── README.md


/.gitignore:
--------------------------------------------------------------------------------
1 | /.idea
2 | 


--------------------------------------------------------------------------------
/demo/laravel/app/config/websocket.php:
--------------------------------------------------------------------------------
1 | <?php
2 | return [
3 |     'host' =>'0.0.0.0',
4 | 
5 |     'port'=>9501,
6 | 
7 |     'task_worker_num'=>8
8 | ];


--------------------------------------------------------------------------------
/src/Facades/Queue.php:
--------------------------------------------------------------------------------
 1 | <?php
 2 | 
 3 | namespace webSocket\Facades;
 4 | 
 5 | use Illuminate\Support\Facades\Facade;
 6 | 
 7 | class Queue extends Facade
 8 | {
 9 |     protected static function getFacadeAccessor() {
10 |         return \webSocket\Contracts\Queue::class;
11 |     }
12 | }


--------------------------------------------------------------------------------
/composer.json:
--------------------------------------------------------------------------------
 1 | {
 2 |     "name": "laravel/webSocket",
 3 |     "description": "基于laravel+swoole的webSocket开发库",
 4 |     "keywords": ["webSocket","swoole", "laravel"],
 5 |     "license": "MIT",
 6 |     "authors": [
 7 |         {
 8 |             "name": "W+",
 9 |             "email": "907515459@qq.com"
10 |         }
11 |     ],
12 |     "require": {
13 |         "php": ">=5.5.9",
14 |         "laravel/framework": ">=5.4.0"
15 |     },
16 |     "autoload": {
17 |         "psr-4": {
18 |             "webSocket\\": "src/"
19 |         }
20 |     }
21 | }
22 | 


--------------------------------------------------------------------------------
/src/RedisSubscribe.php:
--------------------------------------------------------------------------------
 1 | <?php
 2 | /**
 3 |  * Created by PhpStorm.
 4 |  * User: pz
 5 |  * Date: 2017/6/11
 6 |  * Time: 21:10
 7 |  */
 8 | 
 9 | namespace webSocket;
10 | 
11 | 
12 | abstract class RedisSubscribe
13 | {
14 |    protected $data;
15 | 
16 |    protected $fd;
17 | 
18 |    protected $channel;
19 | 
20 |    function setData($channel,$data)
21 |    {
22 |        $this->channel = $channel;
23 |        $this->fd = $data['fd'];
24 |        $this->data = $data['data'];
25 |    }
26 |    
27 |    function reset(){
28 |       $this->fd = null;
29 |       $this->data = null;
30 |       $this->channel = null;
31 |    }
32 |    
33 |    abstract function handle();
34 | }


--------------------------------------------------------------------------------
/src/Redis/RedisManagerCli.php:
--------------------------------------------------------------------------------
 1 | <?php
 2 | 
 3 | namespace webSocket\Redis;
 4 | 
 5 | class RedisManagerCli extends \Illuminate\Redis\RedisManager
 6 | {
 7 | 
 8 |     /**
 9 |      * Get a Redis connection by name.
10 |      *
11 |      * @param  string|null  $name
12 |      * @return \Illuminate\Redis\Connections\Connection
13 |      */
14 |     public function connection($name = null)
15 |     {
16 |         $name = $name ?: 'default';
17 |         $pid_name =getmypid()."_".$name;
18 | 
19 |         if (isset($this->connections[$pid_name])) {
20 |             return $this->connections[$pid_name];
21 |         }
22 | 
23 |         return $this->connections[$pid_name] = $this->resolve($name);
24 |     }
25 | 
26 | }


--------------------------------------------------------------------------------
/src/Redis/RedisServiceProvider.php:
--------------------------------------------------------------------------------
 1 | <?php
 2 | /**
 3 |  * Created by PhpStorm.
 4 |  * User: WanZengchuang
 5 |  * Date: 2017/7/27
 6 |  * Time: 17:23
 7 |  */
 8 | 
 9 | namespace webSocket\Redis;
10 | 
11 | use Illuminate\Support\Arr;
12 | class RedisServiceProvider extends \Illuminate\Redis\RedisServiceProvider
13 | {
14 | 
15 |     /**
16 |      * Register the service provider.
17 |      *
18 |      * @return void
19 |      */
20 |     public function register()
21 |     {
22 |         $this->app->singleton('redis', function ($app) {
23 |             $config = $app->make('config')->get('database.redis');
24 | 
25 | 			return new RedisManagerCli($app,Arr::pull($config, 'client', 'predis'), $config);
26 |         });
27 | 
28 |         $this->app->bind('redis.connection', function ($app) {
29 |             return $app['redis']->connection();
30 |         });
31 |     }
32 | 
33 | }


--------------------------------------------------------------------------------
/demo/laravel/app/Console/Commands/ChatWebSocketCommand.php:
--------------------------------------------------------------------------------
 1 | <?php
 2 | namespace App\Console\Commands;
 3 | 
 4 | use App\Server\Chat\ChatHandle;
 5 | use webSocket\Commands\WebSocketCommand;
 6 | 
 7 | /**
 8 |  * 负责绑定启动命令、绑定webSocket的处理程序。
 9 |  * Class ChatWebSocketCommand
10 |  * @package App\Console\Commands
11 |  */
12 | class ChatWebSocketCommand extends WebSocketCommand
13 | {
14 |     /**
15 |      * artisan 启动命令
16 |      * @var string
17 |      */
18 |     protected $signature = 'Chat:Start';
19 | 
20 |     /**
21 |      * The console command description.
22 |      *
23 |      * @var string
24 |      */
25 |     protected $description = 'chat Server';
26 | 
27 |     /**
28 |      * 设置websocket处理程序
29 |      * @return mixed
30 |      */
31 |     protected function getBindClass()
32 |     {
33 |         return ChatHandle::class;
34 |     }
35 | 
36 |     /**
37 |      * 获取本命令启动的websoket读取的配置文件，默认为websocket.php文件
38 |      * @return string
39 |      */
40 |     protected function getConfig(){
41 |         return 'websocket';
42 |     }
43 | 
44 | 
45 | }


--------------------------------------------------------------------------------
/demo/laravel/app/Console/Kernel.php:
--------------------------------------------------------------------------------
 1 | <?php
 2 | 
 3 | namespace App\Console;
 4 | 
 5 | 
 6 | use App\Console\Commands\ChatLogCommand;
 7 | use App\Console\Commands\ChatWebSocketCommand;
 8 | use App\Console\Commands\PacketRobSubscribeCommand;
 9 | use App\Console\Commands\RobotChatCommand;
10 | use Illuminate\Console\Scheduling\Schedule;
11 | use Illuminate\Foundation\Console\Kernel as ConsoleKernel;
12 | 
13 | class Kernel extends ConsoleKernel
14 | {
15 |     /**
16 |      * The Artisan commands provided by your application.
17 |      *
18 |      * @var array
19 |      */
20 |     protected $commands = [
21 |       ChatWebSocketCommand::class
22 |     ];
23 | 
24 |     /**
25 |      * Define the application's command schedule.
26 |      *
27 |      * @param  \Illuminate\Console\Scheduling\Schedule  $schedule
28 |      * @return void
29 |      */
30 |     protected function schedule(Schedule $schedule)
31 |     {
32 |         // $schedule->command('inspire')
33 |         //          ->hourly();
34 |     }
35 | 
36 |     /**
37 |      * Register the Closure based commands for the application.
38 |      *
39 |      * @return void
40 |      */
41 |     protected function commands()
42 |     {
43 |         require base_path('routes/console.php');
44 |     }
45 | }
46 | 


--------------------------------------------------------------------------------
/src/Contracts/Queue.php:
--------------------------------------------------------------------------------
 1 | <?php
 2 | 
 3 | namespace webSocket\Contracts;
 4 | 
 5 | use Illuminate\Support\Facades\Redis;
 6 | 
 7 | class Queue
 8 | {
 9 |     /**
10 |      * 发布服务
11 |      * @param $channel
12 |      * @param $data
13 |      */
14 |      public function push($channel,$data){
15 |         Redis::lpush($channel,is_string($data)?$data:json_encode($data));
16 |     }
17 | 
18 |     /**
19 |      * 阻塞获取最后一个元素
20 |      * @param $channel
21 |      * @return mixed
22 |      */
23 |      public function brpop($channel){
24 |         while(true){
25 |             if($value = Redis::brpop($channel, 10)){
26 |                 return $value;
27 |             }
28 |         }
29 |     }
30 | 
31 |     /**
32 |      * 阻塞获取第一个元素
33 |      * @param $channel
34 |      * @return mixed
35 |      */
36 |      public function blpop($channel){
37 |         while(true){
38 |             if($value = Redis::blpop($channel, 10)){
39 |                 return $value;
40 |             }
41 |         }
42 |     }
43 | 
44 |     /**
45 |      * 非阻塞获取第一个元素
46 |      * @param $channel
47 |      * @return mixed
48 |      */
49 |     public function lpop($channel){
50 |         return Redis::lpop($channel);
51 |     }
52 | 
53 |     /**
54 |      * 清空队列
55 |      * @param $channel
56 |      */
57 |     public function clean($channel){
58 |         Redis::ltrim($channel,1,0);
59 |     }
60 | }


--------------------------------------------------------------------------------
/demo/laravel/public/js/push.js:
--------------------------------------------------------------------------------
 1 | function createChat() {
 2 |     this.ws = new WebSocket("ws://127.0.0.1:9501");
 3 | 
 4 |     this.wsOpen();
 5 |     this.wsMessage();
 6 |     this.wsOnclose();
 7 |     this.wsOnerror();
 8 | }
 9 | 
10 | createChat.prototype = {
11 |     wsSend : function(data){
12 |         this.ws.send(data);
13 | 
14 |         console.log(this.tt);
15 |     },
16 |     wsOpen : function (){
17 |         var the = this;
18 | 
19 |         this.ws.onopen = function( event ){
20 |             console.log("ws is open ");
21 | 
22 |             /*setTimeout(function () {
23 |                 the.addMessage("async",1);
24 |             },1000);*/
25 | 
26 |             /*a.forEach(function (member) {
27 |                 member.addMessage("publish",1);
28 |             })*/
29 |         }
30 |     },
31 |     addMessage:function ($type,$data) {
32 |         var a =  {
33 |             'type':$type,
34 |             'data':$data
35 |         };
36 |         this.wsSend(JSON.stringify(a));
37 |     },
38 |     wsMessage : function(){
39 |         this.ws.onmessage=function(event){
40 |             //var d =JSON.parse(event.data);
41 |             console.log(event.data);
42 |         }
43 |     },
44 |     wsOnclose : function(){
45 |         var the = this;
46 |         this.ws.onclose = function(event){
47 | 
48 |         }
49 |     },
50 |     wsOnerror : function(){
51 |         this.ws.onerror = function(event){
52 |             alert("ws open error");
53 |         }
54 |     }
55 | };
56 | 
57 | new createChat()
58 | 
59 | /*
60 | var a = new Array();
61 | a.push(new createChat());
62 | a.push(new createChat());
63 | */
64 | 


--------------------------------------------------------------------------------
/README.md:
--------------------------------------------------------------------------------
 1 | # laravel-websocket
 2 | 
 3 | 基于 laravel + swoole +redis的WebSocket开发库。适合开发中小型webSocket程序。
 4 | 
 5 | 无需关注webSocket的细节，专注业务。
 6 | 
 7 | 可同步、异步处理请求，发送数据。
 8 | 
 9 | ## 安装方法
10 | 1. 安装laravel.
11 | 
12 | 2. 添加websocket库: composer require lactone/laravel-websocket
13 | 
14 | 3. 添加laravel command类，继承 webSocket\Commands\WebSocketCommand.
15 | 
16 | ## Demo
17 | 在目录/demo/laravel中是一个简单的Demo程序。
18 | 
19 | ### 使用方法
20 | 
21 | 1. 将Demo目录中的文件覆盖到laravel程序中。
22 | 2. 配置redis。
23 | 3. 打开app/config/app.php,将默认的Redis服务提供器修改为websocket的Redis服务提供器，将Illuminate\Redis\RedisServiceProvider替换为\webSocket\Redis\RedisServiceProvider
24 | 4. 打开app/config/websocket.php 文件，修改websocket的监听ip、端口和工作进程数量。
25 | 5. 命令行执行:**php artisan Chat:Start** 开启websocket服务。
26 | 
27 | ### 文件说明
28 | 
29 |     App\Console\Commands\ChatWebSocketCommand.php 负责绑定启动命令、设置websocket的配置和绑定webSocket处理程序。
30 |     
31 |     App\Server\Chat\ChatHandle.php 负责处理ws业务。当用户发送消息:{type:$type,data:$data}给服务器的时候，系统会自动调用message_$type方法。
32 | 
33 |     public/js/push.js 客户端发送消息程序
34 | 
35 | ## 基本概念
36 | 
37 | 当客户端与服务端建立webSocket连接后，服务端会生成一个fd,该fd表示当前链接，通过该fd服务端可以发送消息给客户端。
38 |     
39 | ### 数据交互
40 | 
41 |     客户端发送给服务端结构如下：{type:type,data:data}
42 |     其中如果type为login,服务端接受程序信息后会调用message_login方法，处理信息。
43 |     
44 | ### 常用API
45 |    
46 | #### webSocket\Service\PushService
47 | 
48 |     PushService::pushToAllAsync($data) 发送信息给所有客户端
49 |     PushService::pushToFdAsync($fd,$data) 发送消息给某个客户端
50 |     PushService::pushToAllOutMeAsync($fd,$data) 发送消息给除开fd的其他客户端
51 |     PushService::getAllFdsFromStore() 获取所有在线的链接标识
52 |     PushService::isLive($fd) 客户端是否在线
53 |     PushService::success($action,$message = array()) 反馈成功的数据结构
54 |     PushService::error($action,$message = array()) 反馈失败的数据结构
55 | 
56 |     


--------------------------------------------------------------------------------
/src/Commands/WebSocketCommand.php:
--------------------------------------------------------------------------------
 1 | <?php
 2 | 
 3 | namespace webSocket\Commands;
 4 | 
 5 | use webSocket\WebSocket;
 6 | use webSocket\ServerHandle;
 7 | use webSocket\Contracts\Queue;
 8 | use Illuminate\Console\Command;
 9 | 
10 | abstract class WebSocketCommand extends Command
11 | {
12 |     /**
13 |      * The name and signature of the console command.
14 |      *
15 |      * @var string
16 |      */
17 |     protected $signature = 'WebSocket:Start';
18 | 
19 |     /**
20 |      * The console command description.
21 |      *
22 |      * @var string
23 |      */
24 |     protected $description = 'WebSocket Server';
25 | 
26 |     /**
27 |      * Create a new command instance.
28 |      *
29 |      * @return void
30 |      */
31 |     public function __construct()
32 |     {
33 |         parent::__construct();
34 |     }
35 | 
36 |     /**
37 |      * Execute the console command.
38 |      *
39 |      * @return mixed
40 |      */
41 |     public function handle()
42 |     {
43 |         ini_set('default_socket_timeout', -1);
44 | 
45 |         $webSocketConfig =  app('config')->get($this->getConfig());
46 |         if(!$webSocketConfig){
47 |             throw new \Exception("请设置webSocket配置!");
48 |         }
49 | 
50 |         $webSocket = new WebSocket($webSocketConfig,$this->laravel);
51 |         if(!$bindClass = $this->getBindClass()){
52 |             throw new \Exception("请配置需要运行的webSocket类");
53 |         }
54 |         $webSocket->bind(ServerHandle::class, $bindClass);
55 | 
56 |         $webSocket->handle();
57 |     }
58 | 
59 |     protected function getConfig(){
60 |         return 'websocket';
61 |     }
62 | 
63 |     protected function getBindClass(){
64 |         throw new \Exception("请设置 Handle class 名称！");
65 |     }
66 | }
67 | 


--------------------------------------------------------------------------------
/src/RedisSubscribeCommand.php:
--------------------------------------------------------------------------------
 1 | <?php
 2 | 
 3 | namespace webSocket;
 4 | use Illuminate\Console\Command;
 5 | use Illuminate\Support\Facades\Redis;
 6 | 
 7 | abstract class RedisSubscribeCommand extends Command
 8 | {
 9 |     /**
10 |      * The name and signature of the console command.
11 |      *
12 |      * @var string
13 |      */
14 |     protected $signature = 'RedisSubscribe:Command';
15 | 
16 |     /**
17 |      * The console command description.
18 |      *
19 |      * @var string
20 |      */
21 |     protected $description = 'RedisSubscribe:Command';
22 | 
23 |     protected $channel = '';
24 | 
25 |     protected $handler = '';
26 | 
27 |     /**
28 |      * Create a new command instance.
29 |      *
30 |      * @return void
31 |      */
32 |     public function __construct()
33 |     {
34 |         ini_set('default_socket_timeout', -1);
35 | 
36 |         parent::__construct();
37 | 
38 |         if(!$this->channel){
39 |             throw new \Exception("not channel!");
40 |         }
41 | 
42 |         if(!$this->handler){
43 |             throw new \Exception("not handler!");
44 |         }
45 |     }
46 | 
47 |     /**
48 |      * Execute the console command.
49 |      *
50 |      * @return mixed
51 |      */
52 |     public function handle()
53 |     {
54 |         $this->laravel->bind(RedisSubscribe::class,$this->handler);
55 |         $instance = $this->laravel->make(RedisSubscribe::class);
56 | 
57 |         while(true) {
58 |             $value = Redis::brpop($this->channel, 0);
59 | 
60 |             $channel = $value[0];
61 |             $data    = $value[1];
62 | 
63 |             $instance->reset();
64 |             $instance->setData($channel,json_decode($data,true));
65 |             $instance->handle();
66 |         }
67 |     }
68 | }


--------------------------------------------------------------------------------
/demo/laravel/app/Server/Chat/ChatHandle.php:
--------------------------------------------------------------------------------
 1 | <?php
 2 | 
 3 | namespace App\Server\Chat;
 4 | 
 5 | use webSocket\Service\PushService;
 6 | use webSocket\ServerHandle;
 7 | use Illuminate\Support\Facades\Log;
 8 | 
 9 | class ChatHandle extends ServerHandle
10 | {
11 | 
12 |     /**
13 |      * 对客户端发送的登录消息进行处理
14 |      * @param $fd
15 |      * @param $data
16 |      * @return array
17 |      */
18 |     public  function message_login($fd,$data){
19 |         PushService::pushToAllOutMeAsync($fd,$this->results("login",1,[
20 |             'message' => "{$fd} 用户登录成功!"
21 |         ]));
22 |     }
23 | 
24 |     /**
25 |      * 异步消息请求处理
26 |      * @param $fd
27 |      * @param $data
28 |      */
29 |     public function message_Async($fd,$data){
30 | 
31 |         PushService::pushToAllOutMeAsync($fd,$this->results("async",1,[
32 |             'message' => "Async 消息发送成功"
33 |         ]));
34 | 
35 |         /*PushService::pushToAllAsync($this->results("async1",1,[
36 |             'message' => "Async 任务发送成功1"
37 |         ]));*/
38 | 
39 |     }
40 | 
41 |     /**
42 |      * 客户端发送订阅事件
43 |      * @param $fd
44 |      * @param $data
45 |      */
46 |     public function message_publish($fd,$data){
47 |         $data = is_string($data)?$data:json_encode($data);
48 | 
49 |         $this->publish("packet.rob",[
50 |             'fd'=>$fd,
51 |             'data'=>$data
52 |         ]);
53 |     }
54 | 
55 |     /**
56 |      * 对客户端的其它消息发送信息
57 |      * @param $fd
58 |      * @param $data
59 |      * @return array
60 |      */
61 |     public function message_chat($fd,$data){
62 |         return $this->results("chat",1,"发送成功");
63 |     }
64 | 
65 | 
66 |     /**
67 |      * 用户关闭了连接
68 |      * @param $fd
69 |      */
70 |     public function close($fd){
71 |         log:info($fd." is close !");
72 | 
73 |     }
74 | 
75 | }


--------------------------------------------------------------------------------
/src/Service/PushService.php:
--------------------------------------------------------------------------------
  1 | <?php
  2 | 
  3 | 
  4 | namespace webSocket\Service;
  5 | 
  6 | use Illuminate\Support\Facades\Log;
  7 | use Illuminate\Support\Facades\Redis;
  8 | use webSocket\Facades\Queue;
  9 | 
 10 | class PushService
 11 | {
 12 |     /**
 13 |      * 存储当前在线的fds
 14 |      */
 15 |     const Store_fds       = 'redis.set.fds';
 16 | 
 17 |     /**
 18 |      * 存储当前fds的长度
 19 |      */
 20 |     const Store_fds_length  = 'redis.key.fds.length';
 21 | 
 22 |     /**
 23 |      *   清空缓存数据
 24 |      */
 25 |     static public function clean(){
 26 |         if($values  = Redis::sscan(self::Store_fds,1)){
 27 |             foreach( $values as $key){
 28 |                 Redis::srem(self::Store_fds,$key);
 29 |             }
 30 |         }
 31 | 
 32 |         Log::info("clean push all info !");
 33 | 
 34 |         Redis::set(self::Store_fds_length,0);
 35 |     }
 36 | 
 37 |     /**
 38 |      * 清理fd下原来的数据
 39 |      * @param $fd
 40 |      */
 41 |     static public function cleanFdChannel($fd){
 42 |         Queue::clean(self::getFdChannel($fd));
 43 |     }
 44 | 
 45 |     /**
 46 |      * push
 47 |      * @param $fd
 48 |      * @param $data
 49 |      */
 50 |     static public function pushToFdAsync($fd,$data){
 51 |         Queue::push(self::getFdChannel($fd),$data);
 52 |     }
 53 | 
 54 |     /**
 55 |      * 发送消息给所有的fd
 56 |      * @param $data
 57 |      */
 58 |     static public function pushToAllAsync($data){
 59 |         if($fds = self::getAllFdsFromStore()){
 60 |             foreach($fds as $fd){
 61 |                 self::pushToFdAsync($fd,$data);
 62 |             }
 63 |         }
 64 |     }
 65 | 
 66 |     /**
 67 |      * 发送消息给所有的fd，不包括$fd
 68 |      * @param $fd
 69 |      * @param $data
 70 |      */
 71 |     static public function pushToAllOutMeAsync($fd,$data){
 72 |         if($fds = self::getAllFdsFromStore()){
 73 |             foreach($fds as $i){
 74 |                 if($i != $fd){
 75 |                     self::pushToFdAsync($i,$data);
 76 |                 }
 77 |             }
 78 |         }
 79 |     }
 80 | 
 81 |     /**
 82 |      * 获取从Redis得到的激活的连接数
 83 |      * @return mixed
 84 |      */
 85 |     static public function getAllFdsFromStore(){
 86 |         return Redis::smembers(self::Store_fds);
 87 |     }
 88 | 
 89 |     /**
 90 |      * 获取当前用户的channel
 91 |      * @param $fd
 92 |      * @return string
 93 |      */
 94 |     static public function getFdChannel($fd){
 95 |         return "webSocket.fd.".$fd;
 96 |     }
 97 | 
 98 |     /**
 99 |      * 将fd纳入push服务
100 |      * @param $fd
101 |      */
102 |     static public function login($fd){
103 |         Redis::sadd(PushService::Store_fds,$fd);
104 |         Redis::incr(PushService::Store_fds_length);
105 |     }
106 | 
107 |     /**
108 |      * fd推出push服务
109 |      * @param $fd
110 |      */
111 |     static public function out($fd){
112 |         Redis::srem(PushService::Store_fds,$fd);
113 |         Redis::decr(PushService::Store_fds_length);
114 |     }
115 | 
116 |     /**
117 |      * fd是否还是
118 |      * @param $fd
119 |      * @return bool
120 |      */
121 |     static public function isLive($fd){
122 |         return (bool)Redis::sismember(self::Store_fds,$fd);
123 |     }
124 | 
125 |     /**
126 |      * 反馈需要Push的数据
127 |      * @param $action
128 |      * @param $status
129 |      * @param array $message
130 |      * @return array
131 |      */
132 |     static public function results($action,$status,$message = array()){
133 |         return [
134 |             'action'=>$action,
135 |             'status'=>$status,
136 |             'message'=>$message
137 |         ];
138 |     }
139 | 
140 |     /**
141 |      * 执行成功消息
142 |      * @param $action
143 |      * @param array $message
144 |      * @return array
145 |      */
146 |     static public function success($action,$message = array()){
147 |         return [
148 |             'action'=>$action,
149 |             'status'=>1,
150 |             'message'=>$message
151 |         ];
152 |     }
153 | 
154 |     /**
155 |      * 执行失败消息
156 |      * @param $action
157 |      * @param array $message
158 |      * @return array
159 |      */
160 |     static public function error($action,$message = array()){
161 |         return [
162 |             'action'=>$action,
163 |             'status'=>0,
164 |             'message'=>$message
165 |         ];
166 |     }
167 | }


--------------------------------------------------------------------------------
/src/WebSocket.php:
--------------------------------------------------------------------------------
  1 | <?php
  2 | 
  3 | namespace webSocket;
  4 | 
  5 | use Illuminate\Container\Container;
  6 | 
  7 | class WebSocket  extends Container
  8 | {
  9 | 
 10 |     private $server;
 11 | 
 12 |     /**
 13 |      * @var ServerHandle
 14 |      */
 15 |     private $handler;
 16 | 
 17 |     private $app;
 18 | 
 19 |     function __construct($config,$app)
 20 |     {
 21 | 
 22 |         $this->app = $app;
 23 | 
 24 |         $this->instance(WebSocket::class,$this);
 25 | 
 26 |         $this->server = new \swoole_websocket_server($config['host'],$config['port']);
 27 | 
 28 |         $this->server->set(array(
 29 |             'task_worker_num'     => $config['task_worker_num']
 30 |         ));
 31 | 
 32 |         $this->server->on("open",array($this,"onOpen"));
 33 |         $this->server->on("message",array($this,"onMessage"));
 34 |         $this->server->on("Task",array($this,"onTask"));
 35 |         $this->server->on("Finish",array($this,"onFinish"));
 36 |         $this->server->on("close",array($this,"onClose"));
 37 |     }
 38 | 
 39 |     function handle(){
 40 | 
 41 |         $this->handler = $this->make(ServerHandle::class);
 42 | 
 43 |         $this->server->start();
 44 |     }
 45 | 
 46 |     /**
 47 |      * 添加任务
 48 |      * @param $fd
 49 |      * @param $key
 50 |      * @param array $params
 51 |      */
 52 |     public function addTask($fd,$key,$params = []){
 53 |         $this->server->task(json_encode([
 54 |             'task'=>$key,
 55 |             'fd'  =>$fd,
 56 |             'data'=>$params
 57 |         ]));
 58 |     }
 59 | 
 60 |     /**
 61 |      * 当用户上线
 62 |      * @param $server
 63 |      * @param $request
 64 |      */
 65 |     public function onOpen( $server , $request){
 66 | 
 67 |         $this->handler->openBefore($request->fd);
 68 | 
 69 |         if($pushMsg = $this->handler->open($request->fd)){
 70 |             $this->push( $request->fd ,null, $pushMsg );
 71 |         }
 72 | 
 73 |         $this->handler->openAfter($request->fd);
 74 | 
 75 |     }
 76 | 
 77 |     /**
 78 |      * 但任务发送给服务器
 79 |      * @param $server
 80 |      * @param $task_id
 81 |      * @param $from_id
 82 |      * @param $data
 83 |      */
 84 |     public function onTask($server , $task_id , $from_id , $data){
 85 |         $data = json_decode($data,true);
 86 |         if($data['task'] == '__message__'){
 87 |             $pushMsg = $this->handler->message($data['fd'],$data['data']);
 88 |         }else{
 89 |             $pushMsg = $this->handler->task($data['task'],$data['fd'],$data['data']);
 90 |         }
 91 | 
 92 |         if($pushMsg){
 93 |             $this->push( $data['fd'] ,$task_id, $pushMsg );
 94 |         }
 95 |     }
 96 | 
 97 |     /**
 98 |      * 当信息发送过来
 99 |      * @param $server
100 |      * @param $frame
101 |      */
102 |     public function onMessage($server , $frame ){
103 |         $data = json_decode( $frame->data , true );
104 |         $this->addTask($frame->fd,'__message__',$data);
105 |     }
106 | 
107 |     /**
108 |      * 操作完成
109 |      * @param $server
110 |      * @param $task_id
111 |      * @param $data
112 |      * @return mixed
113 |      */
114 |     public function onFinish($server , $task_id , $data){
115 |         return $this->handler->finish($task_id,$data);
116 |     }
117 | 
118 |     /**
119 |      * 当系统结束
120 |      * @param $server
121 |      * @param $fd
122 |      * @return mixed
123 |      */
124 |     public function onClose($server , $fd){
125 | 
126 |         $this->handler->closeBefore($fd);
127 | 
128 |         $ret =  $this->handler->close($fd);
129 | 
130 |         $this->handler->closeAfter($fd);
131 | 
132 |         return $ret;
133 |     }
134 | 
135 |     /**
136 |      * 添加反馈的信息
137 |      * @param $fd
138 |      * @param null $task_id
139 |      * @param $data
140 |      */
141 |     public function push($fd,$task_id = null,$data){
142 |         $isLiveClient = false;
143 |         foreach($this->server->connections as $i){
144 |             if($fd == $i){
145 |                 $isLiveClient = true;
146 |                 break;
147 |             }
148 |         }
149 | 
150 |         if($isLiveClient){
151 |             try{
152 |                 $this->server->push( $fd , is_array($data)?json_encode($data):$data );
153 |             }catch (\ErrorException $e){
154 |                 Log::warning($fd." push data error!");
155 |             }
156 |         }
157 |     }
158 | 
159 |     /**
160 |      * 为所有用户发布发送消息
161 |      * @param $data
162 |      */
163 |     public function pushAll($data){
164 |         foreach($this->server->connections as $i){
165 |             $this->server->push( $i , is_array($data)?json_encode($data):$data  );
166 |         }
167 |     }
168 | 
169 |     /**
170 |      * 为所有用户push信息，除开fd
171 |      * @param $fd
172 |      * @param $data
173 |      */
174 |     public function pushToAllOutMe($fd,$data){
175 |         foreach($this->server->connections as $i){
176 |             if($i != $fd){
177 |                 $this->server->push( $i , is_array($data)?json_encode($data):$data );
178 |             }
179 |         }
180 |     }
181 | }


--------------------------------------------------------------------------------
/src/ServerHandle.php:
--------------------------------------------------------------------------------
  1 | <?php
  2 | 
  3 | namespace webSocket;
  4 | 
  5 | use webSocket\Service\PushService;
  6 | use webSocket\Facades\Queue;
  7 | use Illuminate\Support\Facades\Log;
  8 | use Illuminate\Support\Facades\Redis;
  9 | 
 10 | class ServerHandle
 11 | {
 12 |     /**
 13 |      * 系统错误提示
 14 |      */
 15 |     const SYS_ERROR         = "SysError";
 16 | 
 17 |     /**
 18 |      * swoole websocket实例
 19 |      * @var bool|WebSocket
 20 |      */
 21 |     protected $server = false;
 22 | 
 23 |     /**
 24 |      * 用户监听是否有新数据时间间隔(毫秒)
 25 |      * @var int
 26 |      */
 27 |     protected $interval   = 200;
 28 | 
 29 |     /**
 30 |      * 测试环境下监听频道与处理程序之间的关系
 31 |      * @var array
 32 |      */
 33 |     protected $channels = [];
 34 |     
 35 |     public function __construct(WebSocket $webSocket)
 36 |     {
 37 |         PushService::clean();
 38 | 
 39 |         $this->server = $webSocket;
 40 |     }
 41 | 
 42 | 
 43 | 
 44 |     /**
 45 |      * 反馈给client的结果
 46 |      * @param $action 
 47 |      * @param $status
 48 |      * @param $message
 49 |      * @return array
 50 |      */
 51 |     public function results($action,$status,$message = array()){
 52 |         return PushService::results($action,$status,$message);
 53 |     }
 54 | 
 55 |     /**
 56 |      * 添加任务到Task进程
 57 |      * @param $fd
 58 |      * @param $key
 59 |      * @param array $params
 60 |      */
 61 |     public function pushTask($fd, $key, $params = []){
 62 |         return $this->server->addTask($fd,$key,$params);
 63 |     }
 64 | 
 65 | 
 66 |     /**
 67 |      * 发送消息给某fd
 68 |      * @param $fd
 69 |      * @param $data
 70 |      */
 71 |     public function pushToFd($fd,$data){
 72 |         $this->server->push($fd,null,$data);
 73 |     }
 74 | 
 75 |     /**
 76 |      * 发送消息给所有的fd
 77 |      * @param $data
 78 |      */
 79 |     public function pushToAll($data){
 80 |         $this->server->pushAll($data);
 81 |     }
 82 | 
 83 |     /**
 84 |      * 发送消息给所有的fd，不包括$fd
 85 |      * @param $fd
 86 |      * @param $data
 87 |      */
 88 |     public function pushToAllOutMe($fd,$data){
 89 |         $this->server->pushToAllOutMe($fd,$data);
 90 |     }
 91 | 
 92 |     /**
 93 |      * 调试模式下调试发布与订阅机制
 94 |      * @param $channel
 95 |      * @param $data
 96 |      */
 97 |     protected function call($channel,$data){
 98 |         foreach($this->channels as $c => $class){
 99 |             if($c == $channel){
100 |                 Log::info("{$c} channel call!");
101 | 
102 |                 /**
103 |                  * @var $Object RedisSubscribe
104 |                  */
105 |                 $Object = new $class($data);
106 |                 $Object->setData($channel,$data);
107 |                 $Object->handle();
108 | 
109 |             }
110 |         }
111 |     }
112 | 
113 |     /**
114 |      * 发布消息
115 |      * @param $channel
116 |      * @param string $data
117 |      */
118 |     protected function publish($channel,$data = ''){
119 |         if(env('APP_ENV') =='local'){
120 |             $this->call($channel,$data);
121 |         }else{
122 |             Queue::push($channel,$data);
123 |         }
124 |     }
125 | 
126 |     /**
127 |      * message分发
128 |      * @param $fd
129 |      * @param $data
130 |      * @return mixed
131 |      */
132 |     public function message($fd,$data){
133 |         if(isset($data['type'])){
134 |             $key = "message_".$data['type'];
135 |             if(method_exists($this,$key)){
136 |                 return $this->$key($fd,$data);
137 |             }
138 |         }
139 | 
140 |         Log::error("message no run ",$data);
141 |     }
142 | 
143 |     /**
144 |      * 任务分发
145 |      * @param $task
146 |      * @param $fd
147 |      * @param $data
148 |      * @return array
149 |      */
150 |     public function task($task,$fd,$data){
151 |         $key = "task_".$task;
152 |         if(method_exists($this,$key)){
153 |             return $this->$task($fd,$data);
154 |         }else{
155 |             return $this->results(self::SYS_ERROR,0,'无处理程序');
156 |         }
157 |     }
158 | 
159 |     /**
160 |      * 数据监听
161 |      * @param $fd
162 |      */
163 |     public function timer($fd){
164 |         $the = $this;
165 | 
166 |         \swoole_timer_after($this->interval,function () use ($fd,$the){
167 |             try{
168 |                // Log::info("{$fd}开始消费！");
169 | 
170 |                 while($value = Queue::lpop(PushService::getFdChannel($fd))){
171 |                     $this->pushToFd( $fd, $value );
172 |                 }
173 | 
174 |                 if(PushService::isLive($fd)){
175 |                     $the->timer($fd);
176 |                 }
177 | 
178 |             }catch (\Exception $e){
179 | 
180 |                 Log::info("timer error:".$e->getMessage());
181 | 
182 |                 $this->triggerClose($fd);
183 |             }
184 |         });
185 | 
186 |     }
187 | 
188 |     /**
189 |      * 链接打开之前执行的操作
190 |      * @param $fd
191 |      */
192 |     public function openBefore($fd){
193 |         PushService::cleanFdChannel($fd);
194 |     }
195 | 
196 |     /**
197 |      * 打开链接
198 |      * @param $fd
199 |      */
200 |     public function open($fd){
201 | 
202 |         $this->timer($fd);
203 |     }
204 | 
205 |     /**
206 |      * 打开链接之后
207 |      * @param $fd
208 |      */
209 |     public function openAfter($fd){
210 |         Log::info(" {$fd} is open! ");
211 | 
212 |         PushService::login($fd);
213 | 
214 |         $length = Redis::get(PushService::Store_fds_length);
215 |         Log::info("当前 fds length {$length}");
216 |     }
217 | 
218 |     /**
219 |      * 任务完成
220 |      * @param $task_id
221 |      * @param $data
222 |      */
223 |     public function finish($task_id,$data){
224 | 
225 |     }
226 | 
227 |     /**
228 |      * 链接关闭之前
229 |      * @param $fd
230 |      */
231 |     public function closeBefore($fd){
232 | 
233 |     }
234 | 
235 |     /**
236 |      * 链接关闭
237 |      * @param $fd
238 |      */
239 |     public function close($fd){
240 | 
241 |     }
242 | 
243 |     /**
244 |      * 关闭链接之后
245 |      * @param $fd
246 |      */
247 |     public function closeAfter($fd){
248 |         log:info($fd." is close !");
249 | 
250 |         PushService::out($fd);
251 |     }
252 | 
253 |     /**
254 |      * 触发关闭事件
255 |      * @param $fd
256 |      */
257 |     public function triggerClose($fd){
258 |         $this->openBefore($fd);
259 | 
260 |         $this->close($fd);
261 | 
262 |         $this->closeAfter($fd);
263 |     }
264 | }


--------------------------------------------------------------------------------