用swoole和websocket开发简单聊天室

用swoole和websocket开发简单聊天室

首先，我想说下写代码的一些习惯，第一，任何可配置的参数或变量都要写到一个config文件中。第二，代码中一定要有日志记录和完善的报错并记录报错。言归正传，swoole应该是每个phper必须要了解的，它号称重新定义了php。此聊天室利用了swoole高并发并且异步非阻塞的特点提高了程序的性能。

首先，定义一个 swoole_lock 和 swoole_websocket_server ，并且配置参数，具体参数详情可以去swoole官网查看。

 class WebSocketServer {
        private $server;
        private $addr = '';
        private $port = '';
        private $users = array();  //保存连接的用户，fd=>nickname的形式保存
        private $lock;
        public $time;


        public function __construct($addr, $port)
        {
            $this->addr = $addr;
            $this->port = $port;
            $this->time = 1;
        }

        public function start()
        {
            $this->lock = new swoole_lock(SWOOLE_MUTEX);
            $this->server = new swoole_websocket_server($this->addr, $this->port);
            $this->server->set(array(
                'daemonize' => 0,
                'worker_num' => 4,
                'task_worker_num' => 10,
                'max_request' => 1000,
                'log_file' => ROOT_PATH . 'storage\\logs\\swoole.log'
            ));

            $this->server->on('open', array($this, 'onOpen'));
            $this->server->on('message', array($this, 'onMessage'));
            $this->server->on('task', array($this, 'onTask'));
            $this->server->on('finish', array($this, 'onFinish'));
            $this->server->on('close', array($this, 'onClose'));

            $this->server->start();
        }

        public function onOpen($server, $request)
        {
            $message = array(
                'remote_addr' => $request->server['remote_addr'],
                'request_time' => date('Y-m-d H:i:s', $request->server['request_time'])
            );
            write_log($message);
        }

        public function onMessage($server, $frame)
        {
            $data = json_decode($frame->data);
            //这里进来用users把全部用户fd做key扔进数组里面后面返回信息，后面每次触发通过fd来做用户标识,每个进程都是独立的，直接不能相互获取，只能放到其他地方。例如文件，或者缓存
            switch ($data->type) {
                case 'init':
                case 'INIT':
                    $this->users[$frame->fd] = $data->message;
                    $message = '欢迎' . $data->message . '加入了聊天室';
                    $response = array(
                        'type' => 1,    // 1代表系统消息，2代表用户聊天
                        'message' => $message
                    );
                    break;
                case 'chat':
                case 'CHAT':
                    $this->time++;
                    echo $this->time;  //每个进程都是独立的，直接不能相互获取，只能放到其他地方。例如文件，或者缓存
                    $message = $data->message;
                    $response = array(
                        'type' => 2,    // 1代表系统消息，2代表用户聊天
                        'username' => $this->users[$frame->fd],
                        'message' => $message
                    );
                    break;
                default:
                    return false;
            }
            $this->server->task($response);
        }

        public function onTask($server, $task_id, $from_id, $message)
        {
            foreach ($this->server->connections as $fd) {
                $this->server->push($fd, json_encode($message));
            }
            $server->finish( 'Task' . $task_id . 'Finished' . PHP_EOL);
        }

        public function onFinish($server, $task_id, $data)
        {
            write_log( $data );
        }

        public function onClose($server, $fd)
        {
            $username = $this->users[$fd];
            // 释放客户端，利用锁进行同步
            $this->lock->lock();
            unset($this->users[$fd]);
            $this->lock->unlock();

            if( $username ) {
                $response = array(
                    'type' => 1,    // 1代表系统消息，2代表用户聊天
                    'message' => $username . '离开了聊天室'
                );
                $this->server->task($response);
            }


            write_log( $fd . ' disconnected');
        }

    }

服务端完了，下面就是客户端，很简单，只需要用websocket链接就ok！

        // websocket
        let address = 'ws://<?php echo CLIENT_CONNECT_ADDR . ':' . CLIENT_CONNECT_PORT ?>';
        address = 'ws://127.0.0.1:9601';
        let webSocket = new WebSocket(address);
        webSocket.onerror = function (event) {
            alert('服务器连接错误，请稍后重试');
        };
        webSocket.onopen = function (event) {
            if(!sessionStorage.getItem('username')) {
                setName();
            }else {
                username = sessionStorage.getItem('username')
                webSocket.send(JSON.stringify({
                    'message': username,
                    'type': 'init'
                }));
            }
        };
        webSocket.onmessage = function (event) {
            console.log(event);
            let data = JSON.parse(event.data);
            if (data.type == 1) {
                $('#chat-list2').append('<li class="ui-border-tb"><span class="username">系统消息:</span><span class="message">' + data.message + '</span></li>');
            } else if (data.type == 2) {
                $('#chat-list2').append('<li class="ui-border-tb"><span class="username">' + data.username + ':</span><span class="message">' + data.message + '</span></li>');
            }

        };
        webSocket.onclose = function (event) {
            alert('散了吧，服务器都关了');
        };

详细代码可以去我的github下载