欢迎您访问程序员文章站本站旨在为大家提供分享程序员计算机编程知识!
您现在的位置是: 首页  >  IT编程

php redis 处理websocket聊天记录

程序员文章站 2022-07-03 12:04:24
欢迎反馈交流 ......
  1 <?php
  2 ini_set('display_errors', 'on');
  3 
  4 class chatClass {
  5     private $redis;
  6 
  7     //这个变量模拟用户当前状态,是否登录,是否可查看
  8     public $checkUserReadable = false;
  9 
 10     //构造函数链接redis数据库
 11     public function __construct() {
 12         $this -> redis = new Redis();
 13         $this -> redis -> connect('127.0.0.1', '6379');
 14         $this -> redis -> auth('***cnblogs.com/handle');
 15     }
 16 
 17     /*
 18     发送消息时保存聊天记录
 19     * 这里用的redis存储是list数据类型
 20     * 两个人的聊天用一个list保存
 21     *
 22     * @from 消息发送者id
 23     * @to 消息接受者id
 24     * @meassage 消息内容
 25     *
 26     * 返回值,当前聊天的总聊天记录数
 27     */
 28     public function setChatRecord($from, $to, $message) {
 29         $data = array('from' => $from, 'to' => $to, 'message' => $message, 'sent' => time()/*, 'recd' => 0*/);
 30         $value = json_encode($data);
 31         //生成json字符串
 32         $keyName = 'rec:' . $this -> getRecKeyName($from, $to);
 33         //echo $keyName;
 34         $res = $this -> redis -> lPush($keyName, $value);
 35         if (!$this -> checkUserReadable) {//消息接受者无法立刻查看时,将消息设置为未读
 36             $this -> cacheUnreadMsg($from, $to);
 37         }
 38         return $res;
 39     }
 40 
 41     /*
 42     * 获取聊天记录
 43     * @from 消息发送者id
 44     * @to 消息接受者id
 45     * @num 获取的数量
 46     *
 47     * 返回值,指定长度的包含聊天记录的数组
 48     */
 49     public function getChatRecord($from, $to, $num) {
 50         $keyName = 'rec:' . $this -> getRecKeyName($from, $to);
 51         //echo $keyName;
 52         $recList = $this -> redis -> lRange($keyName, 0, (int)($num));
 53         return $recList;
 54     }
 55 
 56     /*
 57     * 当用户上线时,或点开聊天框时,获取未读消息的数目
 58     * @user 用户id
 59     *
 60     * 返回值,一个所有当前用户未读的消息的发送者和数组
 61     * 数组格式为‘消息发送者id’=>‘未读消息数目’
 62     *
 63     */
 64     public function getUnreadMsgCount($user) {
 65         return $this -> redis -> hGetAll('unread_' . $user);
 66     }
 67 
 68     /*
 69     * 获取未读消息的内容
 70     * 通过未读消息数目,在列表中取得最新的相应消息即为未读
 71     * @from 消息发送者id
 72     * @to 消息接受者id
 73     *
 74     * 返回值,包括所有未读消息内容的数组
 75     *
 76     *
 77     */
 78     public function getUnreadMsg($from, $to) {
 79         $countArr = $this -> getUnreadMsgCount($to);
 80         $count = $countArr[$from];
 81         $keyName = 'rec:' . $this -> getRecKeyName($from, $to);
 82         return $this -> redis -> lRange($keyName, 0, (int)($count));
 83     }
 84 
 85     /*
 86     * 将消息设为已读
 87     * 当一个用户打开另一个用户的聊天框时,将所有未读消息设为已读
 88     * 清楚未读消息中的缓存
 89     * @from 消息发送者id
 90     * @to 消息接受者id
 91     *
 92     * 返回值,成功将未读消息设为已读则返回true,没有未读消息则返回false
 93     */
 94 
 95     public function setUnreadToRead($from, $to) {
 96         $res = $this -> redis -> hDel('unread_' . $to, $from);
 97         return (bool)$res;
 98     }
 99 
100     /*
101     * 当用户不在线时,或者当前没有立刻接收消息时,缓存未读消息,将未读消息的数目和发送者信息存到一个与接受者关联的hash数据中
102     *
103     * @from 发送消息的用户id
104     * @to 接收消息的用户id
105     *
106     * 返回值,当前两个用户聊天中的未读消息
107     *
108     */
109     private function cacheUnreadMsg($from, $to) {
110         return $this -> redis -> hIncrBy('unread_' . $to, $from, 1);
111     }
112 
113     /*生成聊天记录的键名,即按大小规则将两个数字排序
114     * @from 消息发送者id
115     * @to 消息接受者id
116     *
117     *
118     */
119     private function getRecKeyName($from, $to) {
120         return ($from > $to) ? $to . '_' . $from : $from . '_' . $to;
121     }
122 
123 }
124 
125 /*
126 * 下面为测试用的代码 ,伪造数据模拟场景
127 * 假定有两个用户id为2和3 ,2 向 3 发送消息
128 *
129 
130 $chat = new chatClass();
131 
132 $chat -> checkUserReadable = true;
133 for ($i = 0; $i < 20; $i++) {
134 $chat -> setChatRecord('2', '3', 'message_' . $i);
135 }
136 
137 echo 'get 20 chat records</br>';
138 $arr = $chat -> getChatRecord('2', '3', 20);
139 for ($j = 0; $j < count($arr); $j++) {
140 echo $arr[$j] . '</br>';
141 }
142 
143 $chat -> checkUserReadable = false;
144 
145 for ($m = 0; $m < 5; $m++) {
146 $chat -> setChatRecord('2', '3', 'message_' . $m);
147 }
148 
149 echo "</br>";
150 $umsg_1 = $chat -> getUnreadMsgCount(3);
151 echo "Unread message counts ";
152 echo "</br>";
153 print_r($umsg_1);
154 echo "Unread message content </br> ";
155 $umsgContent = $chat -> getUnreadMsg(2, 3);
156 for ($n = 0; $n < count($umsgContent); $n++) {
157 echo $arr[$n] . '</br>';
158 }
159 echo "</br>";
160 $chat -> setUnreadToRead(2, 3);
161 $umsg_2 = $chat -> getUnreadMsgCount(3);
162 echo "</br>";
163 echo "Unread message counts ";
164 echo "</br>";
165 print_r($umsg_2);
166 *
167 */
168 ?>

欢迎反馈交流