用Swoole快速搭建一個(gè)推送服務(wù)器

Moxmi 發(fā)布于2019-06-28 13:21 / 3073人閱讀

摘要：環(huán)境擴(kuò)展需求實(shí)現(xiàn)簡單的廣播支持身份驗(yàn)證源碼廢話少說，上代碼使用實(shí)例就會以守護(hù)進(jìn)程方式運(yùn)行客戶端代碼搭建過程非常方便。實(shí)現(xiàn)的比較簡陋，但是可以滿足一般的需求了。支持身份驗(yàn)證功能，驗(yàn)證不通過會自動(dòng)踢掉。

環(huán)境

php5.6+

swoole擴(kuò)展

需求

實(shí)現(xiàn)簡單的廣播, 支持身份驗(yàn)證

源碼

廢話少說，上代碼.

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
";
            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;
    }


}

使用實(shí)例

 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  #就會以守護(hù)進(jìn)程方式運(yùn)行

客戶端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);
};

搭建過程非常方便。實(shí)現(xiàn)的比較簡陋，但是可以滿足一般的需求了。支持身份驗(yàn)證功能，驗(yàn)證不通過會自動(dòng)踢掉。