Redis实现多人多聊天室

本文可作为redis in aciton第六章的读书笔记

功能特性

支持多人多聊天室

设计原理

左边的一个数据域,代表两个聊天室,聊天室id分别是827,729

在聊天室827里,有2个人,分别是jason22,jeff24他们分别已经阅读过聊天室内的id为5和6的消息


右边的一个数据域,代表了用户在不同的聊天室,jason22参与了827与729聊天室,在这两个聊天室里,他分别阅读到了id为5和id为10的消息


另外827聊天室内id为5的消息与729聊天室内id为5的消息不一样。


同时还有三个域
msgs:chatid
这是一个zset,有序集合,member是消息体,score是消息id
代表的是某个聊天室内已经发出的消息
另外 这里面存的是有用的消息,已经被所有人都阅读的消息就会被删除


ids:chatid
是一个String型的数据,里面放的是最新的消息的编号(发消息时,自增这个字段,即可获得最新的值)


ids:chat:
是一个String型的数据,里面放的是最新的聊天室的编号(创建聊天室时,自增这个字段)

代码

OK 开始看代码
 

   public String createChat(Jedis conn, String sender, Set recipients, String message) {
	//启动的时候redis里是没有ids:chat:这个键的
	//自增之后返回1
        String chatId = String.valueOf(conn.incr("ids:chat:"));
        return createChat(conn, sender, recipients, message, chatId);
    }


     /**
     * 
     * @param conn
     * @param sender 发送消息的人
     * @param recipients 接受消息的人
     * @param message 待发送的消息
     * @param chatId 聊天室的编号
     * @return
     */
    public String createChat( Jedis conn, String sender, 
    		Set recipients, String message, String chatId){
		//自己发的消息 自己也能接受到
        recipients.add(sender);


        Transaction trans = conn.multi();
        for (String recipient : recipients){
	    //聊天室的成员 最开始时 都阅读的是0号信息
            trans.zadd("chat:" + chatId, 0, recipient);
	    //记录每个人参加的聊天室
            trans.zadd("seen:" + recipient, 0, chatId);
        }
        trans.exec();


        return sendMessage(conn, chatId, sender, message);
    }


    public String sendMessage(Jedis conn, String chatId, String sender, String message) {
    	
    	//锁住聊天室 为啥? 人员变动了咋办
    	//这个acquireLock见上一章
        String identifier = acquireLock(conn, "chat:" + chatId);
        if (identifier == null){
            throw new RuntimeException("Couldn't get the lock");
        }
        try {
        	//给要发布的消息设定一个最新的编号  第一次时 返回的是1
            long messageId = conn.incr("ids:" + chatId);
            HashMap values = new HashMap();
            values.put("id", messageId);
            values.put("ts", System.currentTimeMillis());
            values.put("sender", sender);
            values.put("message", message);
            String packed = new Gson().toJson(values);
            
            //某个聊天室的消息列表
            //最旧的消息----消息json 
            //默认的zset是按照score的值从小到大排序
            conn.zadd("msgs:" + chatId, messageId, packed);
        }finally{
            releaseLock(conn, "chat:" + chatId, identifier);
        }
        return chatId;
    }

 发消息现在就OK了,剩下的就是用户去拉取未读的消息了。这个比较麻烦,恩,相当的麻烦
 

 @SuppressWarnings("unchecked")
    public List fetchPendingMessages(Jedis conn, String recipient) {
    	
    	//获得用户在各个聊天室 已经看到的最新消息的id
    	//有几个聊天室 seenSet的size就是几
        Set seenSet = conn.zrangeWithScores("seen:" + recipient, 0, -1);
        List seenList = new ArrayList(seenSet);


        Transaction trans = conn.multi();
        for (Tuple tuple : seenList){
            String chatId = tuple.getElement();
            int seenId = (int)tuple.getScore();
            //获取每个聊天室里  未读的所有消息
            //min 和 max 可以是 -inf 和 +inf
            trans.zrangeByScore("msgs:" + chatId, String.valueOf(seenId + 1), "inf");
        }
        //我参加了几个聊天室 results的长度就是几
        List