用Swoole快速搭建一个推送服务器

环境

• php5.6+
• swoole扩展

需求

实现简单的广播, 支持身份验证

源码

废话少说，上代码.

<?php namespace church;use ReflectionClass;use ReflectionException;use Swoole\Websocket\Server;class WebSocket{    public $host;    public $port;    public $config;    public $beforeAuthCallback = '';    public $afterAuthCallback = '';    public $beforeSendMsgCallback = '';     public $afterSendMsgCallback = '';    public $authenticator = '';    public $customers = [];    protected $server;    public function __construct($host = '0.0.0.0', $port = 9501, $config = [])    {        $this->host = $host;        $this->port = $port;            $this->config = $config;        $this->init();    }     public function init()    {        $this->server = new Server($this->host, $this->port);        !empty($this->config) && $this->server->set($this->config);        $this->beforeSendMsgCallback = function() {};        $this->afterSendMsgCallback = function() {};        $this->beforeAuthCallback = function() {};        $this->afterAuthCallback = function() {};        $this->authenticator = function() {};        $this->bindEvents();    }    public function bindEvents()    {        $this->server->on('open', [$this, 'onOpen']);        $this->server->on('message', [$this, 'onMessage']);        $this->server->on('close', [$this, 'onClose']);    }    public function run()    {       $this->server->start();     }    public function onOpen(Server $server, $request)    {        call_user_func($this->beforeAuthCallback, $server, $request);        if (!$this->auth($request)) {            return;        }        $this->customers[] = $request->fd;        call_user_func($this->afterAuthCallback, $server, $request);    }    public function onMessage(Server $server, $frame)    {        call_user_func($this->beforeSendMsgCallback, $server, $frame);        $this->broadcast($frame->fd, $frame->data);        call_user_func($this->afterSendMsgCallback, $server, $frame);    }    public function onClose($ser, $fd)    {        $key = array_search($fd, $this->customers);        unset($this->customers[$key]);    }    public function __call($method, $params)    {        $class_name = get_class($this->server);        $class = new ReflectionClass($class_name);        try {            $class->getMethod($method);        } catch (ReflectionException $e) {            echo "Method $method is not exists\n";            return;        }        call_user_func_array([$this->server, $method], $params);    }    public function broadcast($fromUser, $messages)    {        $users = array_diff($this->customers, [$fromUser]);        foreach ($users as $user)        {            $this->server->push($user, $messages);        }    }    private function auth($request)    {        if (($result = call_user_func($this->authenticator, $request)) === false) {            $this->server->push($request->fd, "auth failed");            $this->server->close($request->fd);        }        return $result;    }}

使用实例

<?php require './vendor/autoload.php';use church\WebSocket;$server = new WebSocket('0.0.0.0', 9501, [    'daemonize' => 1,]);$server->authenticator = function($request) use ($server) {    $get = $request->get;    return empty($get) || ($get['access_token'] == 'access_token');};$server->afterAuthCallback = function($ser, $request) use ($server) {    $server->push($request->fd, "auth successful"); };$server->run();

php index.php  #就会以守护进程方式运行

客户端js代码:

var websocket = new WebSocket("ws://192.168.2.106:9501?access_token=access_token");websocket.onopen=function(evt) {};websocket.onmessage = function(evt) {    console.log(evt);};websocket.onclose = function(evt) {  console.log(evt);};

搭建过程非常方便。实现的比较简陋，但是可以满足一般的需求了。支持身份验证功能，验证不通过会自动踢掉。