<?phpini_set('display_errors', 'on');
class chatClass {
 private $redis;
 //這個(gè)變量模擬用戶當(dāng)前狀態(tài)，是否登錄，是否可查看
 public $checkUserReadable = false;
 //構(gòu)造函數(shù)鏈接redis數(shù)據(jù)庫(kù)
 public function __construct() {
  $this -> redis = new Redis();
  $this -> redis -> connect('127.0.0.1', '6379');
  $this -> redis -> auth('***cnblogs.com/handle');
 }
 /*
 發(fā)送消息時(shí)保存聊天記錄
 * 這里用的redis存儲(chǔ)是list數(shù)據(jù)類型
 * 兩個(gè)人的聊天用一個(gè)list保存
 *
 * @from 消息發(fā)送者id
 * @to 消息接受者id
 * @meassage 消息內(nèi)容
 *
 * 返回值，當(dāng)前聊天的總聊天記錄數(shù)
 */
 public function setChatRecord($from, $to, $message) {
  $data = array('from' => $from, 'to' => $to, 'message' => $message, 'sent' => time()/*, 'recd' => 0*/);
  $value = json_encode($data);
  //生成json字符串
  $keyName = 'rec:' . $this -> getRecKeyName($from, $to);
  //echo $keyName;
  $res = $this -> redis -> lPush($keyName, $value);
  if (!$this -> checkUserReadable) {//消息接受者無(wú)法立刻查看時(shí)，將消息設(shè)置為未讀
   $this -> cacheUnreadMsg($from, $to);
  }
  return $res;
 }
 /*
 * 獲取聊天記錄
 * @from 消息發(fā)送者id
 * @to 消息接受者id
 * @num 獲取的數(shù)量
 *
 * 返回值，指定長(zhǎng)度的包含聊天記錄的數(shù)組
 */
 public function getChatRecord($from, $to, $num) {
  $keyName = 'rec:' . $this -> getRecKeyName($from, $to);
  //echo $keyName;
  $recList = $this -> redis -> lRange($keyName, 0, (int)($num));
  return $recList;
 }
 /*
 * 當(dāng)用戶上線時(shí)，或點(diǎn)開(kāi)聊天框時(shí)，獲取未讀消息的數(shù)目
 * @user 用戶id
 *
 * 返回值，一個(gè)所有當(dāng)前用戶未讀的消息的發(fā)送者和數(shù)組
 * 數(shù)組格式為‘消息發(fā)送者id'＝>‘未讀消息數(shù)目'
 *
 */
 public function getUnreadMsgCount($user) {
  return $this -> redis -> hGetAll('unread_' . $user);
 }
 /*
 * 獲取未讀消息的內(nèi)容
 * 通過(guò)未讀消息數(shù)目，在列表中取得最新的相應(yīng)消息即為未讀
 * @from 消息發(fā)送者id
 * @to 消息接受者id
 *
 * 返回值，包括所有未讀消息內(nèi)容的數(shù)組
 *
 *
 */
 public function getUnreadMsg($from, $to) {
  $countArr = $this -> getUnreadMsgCount($to);
  $count = $countArr[$from];
  $keyName = 'rec:' . $this -> getRecKeyName($from, $to);
  return $this -> redis -> lRange($keyName, 0, (int)($count));
 }
 /*
 * 將消息設(shè)為已讀
 * 當(dāng)一個(gè)用戶打開(kāi)另一個(gè)用戶的聊天框時(shí)，將所有未讀消息設(shè)為已讀
 * 清楚未讀消息中的緩存
 * @from 消息發(fā)送者id
 * @to 消息接受者id
 *
 * 返回值，成功將未讀消息設(shè)為已讀則返回true,沒(méi)有未讀消息則返回false
 */
 public function setUnreadToRead($from, $to) {
  $res = $this -> redis -> hDel('unread_' . $to, $from);
  return (bool)$res;
 }
 /*
 * 當(dāng)用戶不在線時(shí)，或者當(dāng)前沒(méi)有立刻接收消息時(shí)，緩存未讀消息,將未讀消息的數(shù)目和發(fā)送者信息存到一個(gè)與接受者關(guān)聯(lián)的hash數(shù)據(jù)中
 *
 * @from 發(fā)送消息的用戶id
 * @to 接收消息的用戶id
 *
 * 返回值，當(dāng)前兩個(gè)用戶聊天中的未讀消息
 *
 */
 private function cacheUnreadMsg($from, $to) {
  return $this -> redis -> hIncrBy('unread_' . $to, $from, 1);
 }
 /*生成聊天記錄的鍵名，即按大小規(guī)則將兩個(gè)數(shù)字排序
 * @from 消息發(fā)送者id
 * @to 消息接受者id
 *
 *
 */
 private function getRecKeyName($from, $to) {
  return ($from > $to) ? $to . '_' . $from : $from . '_' . $to;
 }
}
/*
* 下面為測(cè)試用的代碼 ，偽造數(shù)據(jù)模擬場(chǎng)景
* 假定有兩個(gè)用戶id為2和3 ，2 向 3 發(fā)送消息
*
$chat = new chatClass();
$chat -> checkUserReadable = true;
for ($i = 0; $i < 20; $i++) {
$chat -> setChatRecord('2', '3', 'message_' . $i);
}
echo 'get 20 chat records</br>';
$arr = $chat -> getChatRecord('2', '3', 20);
for ($j = 0; $j < count($arr); $j++) {
echo $arr[$j] . '</br>';
}
$chat -> checkUserReadable = false;
for ($m = 0; $m < 5; $m++) {
$chat -> setChatRecord('2', '3', 'message_' . $m);
}
echo "</br>";
$umsg_1 = $chat -> getUnreadMsgCount(3);
echo "Unread message counts ";
echo "</br>";
print_r($umsg_1);
echo "Unread message content </br> ";
$umsgContent = $chat -> getUnreadMsg(2, 3);
for ($n = 0; $n < count($umsgContent); $n++) {
echo $arr[$n] . '</br>';
}
echo "</br>";
$chat -> setUnreadToRead(2, 3);
$umsg_2 = $chat -> getUnreadMsgCount(3);
echo "</br>";
echo "Unread message counts ";
echo "</br>";
print_r($umsg_2);
*
*/
?>