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 ?>
歡迎反饋交流