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

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

1. public function start() 
2. { 
3.    $this->lock = new swoole_lock(SWOOLE_MUTEX);             
4.   
5.    // 对文件或数组进行锁操作，已达到同步 
6.    $this->server = new swoole_websocket_server($this->addr, $this->port);   
7.   
8.    // swoole提供的Websocket Server 
9.    $this->server->set(array( 
10.       'daemonize' => 0, 
11.       'worker_num' => 4, 
12.       'task_worker_num' => 10, 
13.       'max_request' => 1000, 
14.       'log_file' => ROOT_PATH . 'storage\\logs\\swoole.log'   
15.   
16.     // swoole日志路径，必须是绝对路径 
17.    )); 
18.   
19.    $this->server->on('open', array($this, 'onOpen')); 
20.    $this->server->on('message', array($this, 'onMessage')); 
21.    $this->server->on('task', array($this, 'onTask')); 
22.    $this->server->on('finish', array($this, 'onFinish')); 
23.    $this->server->on('close', array($this, 'onClose')); 
24.    //phpfensi.com 
25.    // 启动服务 
26.    $this->server->start(); 
27. } 

当有客户端链接时，简单记录客户端的信息。

1. public function onOpen($server, $request) 
2.     { 
3.       $message = array( 
4.         'remote_addr' => $request->server['remote_addr'], 
5.         'request_time' => date('Y-m-d H:i:s', $request->server['request_time']) 
6.       ); 
7.       write_log($message); 
8.     } 

当有客户端发送信息时，对信息进行处理。

1. public function onMessage($server, $frame) 
2.     { 
3.       $data = json_decode($frame->data); 
4.   
5.       switch ($data->type) { 
6.         case 'init': 
7.         case 'INIT': 
8.           $this->users[$frame->fd] = $data->message;  、 
9.   
10.       // 记录每个链接的信息，同样不要尝试打印出来看，因为你只能看到自己的链接信息 
11.           $message = '欢迎' . $data->message . '加入了聊天室'; 
12.           $response = array( 
13.             'type' => 1,  // 1代表系统消息，2代表用户聊天 
14.             'message' => $message 
15.           ); 
16.           break; 
17.         case 'chat': 
18.         case 'CHAT': 
19.           $message = $data->message; 
20.           $response = array( 
21.             'type' => 2,  // 1代表系统消息，2代表用户聊天 
22.             'username' => $this->users[$frame->fd], 
23.             'message' => $message 
24.           ); 
25.           break; 
26.         default: 
27.           return false; 
28.       } 
29.          
30.        // 将信息交给task处理 
31.       $this->server->task($response); 
32.     } 
33.   
34.     public function onTask($server, $task_id, $from_id, $message) 
35.     { 
36.         //phpfensi.com 
37.   // 迭代所有的客户端链接，将消息推送过去。（如果你尝试将 $this->server->connections 打印出来，那么你会发现他是空的。但当时用 foreach 去循环时，它确实有用。） 
38.       foreach ($this->server->connections as $fd) { 
39.         $this->server->push($fd, json_encode($message)); 
40.       } 
41.       $server->finish( 'Task' . $task_id . 'Finished' . PHP_EOL); 
42.     } 

最后，当客户端断开链接时，利用锁机制，同步删除客户端信息，并记录日志。

1. public function onClose($server, $fd) 
2.     { 
3.       $username = $this->users[$fd]; 
4.       // 释放客户端，利用锁进行同步 
5.       $this->lock->lock(); 
6.       unset($this->users[$fd]); 
7.       $this->lock->unlock(); 
8.   
9.       if( $username ) { 
10.         $response = array( 
11.           'type' => 1,  // 1代表系统消息，2代表用户聊天 
12.           'message' => $username . '离开了聊天室' 
13.         ); 
14.         $this->server->task($response); 
15.       } 
16.   
17.   
18.       write_log( $fd . ' disconnected'); 
19.     } 

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

1. // websocket 
2.     let address = 'ws://<?php echo CLIENT_CONNECT_ADDR . ':' . CLIENT_CONNECT_PORT ?>'; 
3.     let webSocket = new WebSocket(address); 
4.     webSocket.onerror = function (event) { 
5.       alert('服务器连接错误，请稍后重试'); 
6.     }; 
7.     webSocket.onopen = function (event) { 
8.       if(!sessionStorage.getItem('username')) { 
9.         setName(); 
10.       }else { 
11.         username = sessionStorage.getItem('username') 
12.         webSocket.send(JSON.stringify({ 
13.           'message': username, 
14.           'type': 'init' 
15.         })); 
16.       } 
17.     }; 
18.     webSocket.onmessage = function (event) { 
19.       console.log(event); 
20.       let data = JSON.parse(event.data); 
21.       if (data.type == 1) { 
22.         $('#chat-list2').append('<li class="ui-border-tb"><span class="username">系统消息:</span><span class="message">' + data.message + '</span></li>'); 
23.       } else if (data.type == 2) { 
24.         $('#chat-list2').append('<li class="ui-border-tb"><span class="username">' + data.username + ':</span><span class="message">' + data.message + '</span></li>'); 
25.       } 
26.   
27.     }; 
28.     webSocket.onclose = function (event) { 
29.       alert('散了吧，服务器都关了'); 
30.     }; 

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

Tags: swoole websocket

分享到：