OpenFire源码学习之二十：在openfire中使用redis插件(下)

Redis插件实现

首先来看下插件目录：

RedisServicePlugin

源码清单：

import java.io.File;
import java.sql.Connection;
import java.sql.PreparedStatement;
import java.util.Collection;
import java.util.HashMap;
import java.util.Map;

import org.jivesoftware.database.DbConnectionManager;
import org.jivesoftware.openfire.XMPPServer;
import org.jivesoftware.openfire.container.Plugin;
import org.jivesoftware.openfire.container.PluginManager;
import org.jivesoftware.util.JiveGlobals;
import org.jivesoftware.util.PropertyEventDispatcher;
import org.jivesoftware.util.PropertyEventListener;
import org.jivesoftware.util.redis.enity.RedisConfig;
import org.jivesoftware.util.redis.enity.User;
import org.jivesoftware.util.redis.enity.UserVcard;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import redis.clients.jedis.Jedis;

public class RedisServicePlugin implements Plugin, PropertyEventListener{
	
	private static final Logger log = LoggerFactory.getLogger(RedisServicePlugin.class);
	
	private static final String OF_REDIS_CONFIG_DELETE = "delete from redisConfiguration";
	private static final String OF_REDIS_CONFIG_INSERT = "INSERT INTO redisConfiguration (name, ip, port, auto) VALUES (?, ?, ?, ?)";

	private XMPPServer server;
	private boolean enabled;
	
	@Override
	public void propertySet(String property, Map<String, Object> params) {
		if (property.equals("plugin.redis.enabled")) {
            this.enabled = Boolean.parseBoolean((String)params.get("value"));
        }
	}
......
	public boolean isEnabled() {
        return enabled;
    }
	
	public void setEnabled(boolean enabled) {
        this.enabled = enabled;
        JiveGlobals.setProperty("plugin.redis.enabled",  enabled ? "true" : "false");
    }

	@Override
	public void initializePlugin(PluginManager manager, File pluginDirectory) {
		enabled = JiveGlobals.getBooleanProperty("plugin.redis.enabled", true);
		server = XMPPServer.getInstance();
		PropertyEventDispatcher.addListener(this);
	}

	@Override
	public void destroyPlugin() {
		enabled = JiveGlobals.getBooleanProperty("plugin.redis.enabled", false);
		PropertyEventDispatcher.removeListener(this);
	}

	public void initializeAllData () {
		Collection<RedisConfig> crcfs = getAllRedisConfig();
		if (!crcfs.isEmpty()) {
			for (RedisConfig rcf : crcfs) {
				if ("REDIS_USER".equals(rcf.getName())) {
					setRedisByUserInfo();
				}
			}
		}
	}
	
	public void saveRedisConfig(Collection<RedisConfig> object) {
		PreparedStatement pstmt = null;
		Connection con = null;
		try {
			con = (Connection) DbConnectionManager.getConnection();
			if (object != null) {
				pstmt = con.prepareStatement(OF_REDIS_CONFIG_DELETE);
				pstmt.execute();
				pstmt = (PreparedStatement) con.prepareStatement(OF_REDIS_CONFIG_INSERT);
				for (RedisConfig cr : object) {
					pstmt.setString(1, cr.getName());
					pstmt.setString(2, cr.getIp());
					pstmt.setString(3, cr.getPort());
					pstmt.setString(4, cr.getAuto());
					pstmt.executeUpdate();
				}
			}
		} catch (Exception e) {
			 log.info(e.getMessage());
			 e.printStackTrace();
		}
		finally {
			 DbConnectionManager.closeConnection(pstmt, con);
		}
	}
	
	public Collection<RedisConfig> getAllRedisConfig() {
		if (server == null)
			server = XMPPServer.getInstance();
		return server.getJedisConfDao().getAllRedisConfig();
	}
	
	public void setRedisOfflineMessage() {
		XMPPServer.getInstance().getChatMessageJedisPoolManager().initDataByOfflineMessage();
	}
	
	public void destroyOfflineMessage() {
		Jedis jedis = server.getChatMessageJedisPoolManager().getJedis();
		jedis.flushDB();
		server.getChatMessageJedisPoolManager().returnRes(jedis);
	}
	
	public void setRedisUserSearch() {
		XMPPServer.getInstance().getRedisSearchManager().createSearchAllUserData();
	}
	
	public void destroyUserSearch() {
		Jedis jedis = server.getRedisSearchManager().getSearchJedis();
		jedis.flushDB();
		XMPPServer.getInstance().getUserJedis().returnRes(jedis);
	}
	
	public void destroyRedisByUserInfo () {
		Jedis jedis = server.getUserJedis().getJedis();
		Map<String, String > user = jedis.hgetAll("OFUSER:admin");
		jedis.flushAll();
		jedis.hmset("OFUSER:admin", user);
		XMPPServer.getInstance().getUserJedis().returnRes(jedis);
	}
	
	public void setRedisByUserInfo () {
		Jedis jedis = server.getUserJedis().getJedis();
		int i = 1;
		
		Collection<User> users = server.getUserJedis().getAllUser();
		if (!users.isEmpty()) {
			for (User user : users) {
				Map<String,String> map = new HashMap<String, String>();
				map.put("PASSWORD", user.getPassword());
				if (user.getName() != null )
					map.put("NAME", user.getName());
				if (user.getMoblie() != null )
					map.put("MOBILE", user.getMoblie().trim());
				if (user.getEmail() != null)
					map.put("EMAIL", user.getEmail().trim());
				map.put("CREATIONDATE",user.getCreationDate().trim());
				map.put("MODIFICATIONDATE",user.getModificationDate().trim());
				jedis.hmset("OFUSER:" + user.getUsername(), map);
				
				System.out.println("The " + i++ + "user generation=> username:" + user.getUsername() + ", name:" + user.getName());
			}
			System.out.println("the end:user base info generated...");
			i = 1;
			
			for (User user : users) {
				if (user.getMoblie() != null && !"".equals(user.getMoblie())) {
					jedis.set("MOBILE:" + user.getMoblie(), user.getUsername());
					System.out.println("The" + i++ + "user,mobile relation generation=> mobile:" + user.getMoblie() + ", username:" + user.getUsername());
				}
			}
			System.out.println("the end:user and mobile relation generated...");
			i = 1;
		}
		
		Collection<UserVcard> uservcards = server.getUserJedis().getUserVcard();
		if (!uservcards.isEmpty()) {
			for (UserVcard uv : uservcards) {
				if (uv.getVcard() != null && !"".equals(uv.getVcard())) {
					jedis.set("OFVCARD:" + uv.getUsername(), uv.getVcard());
					System.out.println("The" + i++ + "user vcard generation=> username:" + uv.getUsername());
				}
			}
			i = 1;
			System.out.println("the end:user vcard generated...");
 		}
		
		server.getUserJedis().returnRes(jedis);
	}
}

RedisUserInfoOpt：

import java.util.HashMap;
import java.util.Map;

import org.jivesoftware.openfire.XMPPServer;
import org.jivesoftware.util.redis.enity.User;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;

import redis.clients.jedis.Jedis;

public class RedisUserInfoOpt {

	private static final Logger Log = LoggerFactory.getLogger(RedisUserInfoOpt.class);
	
	public void setBaseUserInfo(User user) {
		if (user != null) {
			Jedis jedis = XMPPServer.getInstance().getUserJedis().getJedis();
			
			try {
				Map<String,String> map = new HashMap<String, String>();
				map.put("PASSWORD", user.getPassword());
				if (user.getName() != null )
					map.put("NAME", user.getName());
				if (user.getMoblie() != null )
					map.put("MOBILE", user.getMoblie().trim());
				if (user.getEmail() != null)
					map.put("EMAIL", user.getEmail().trim());
				map.put("CREATIONDATE",user.getCreationDate().trim());
				map.put("MODIFICATIONDATE",user.getModificationDate().trim());
				jedis.hmset("OFUSER:" + user.getUsername(), map);
				Log.info("user generation=> username:" + user.getUsername() + ", name:" + user.getName());
			} finally{
				XMPPServer.getInstance().getUserJedis().returnRes(jedis);
			}
		}
	}
	
	public void setUserMobile(User user) {
		if (user != null) {
			Jedis jedis = XMPPServer.getInstance().getUserJedis().getJedis();
			try {
				
			} finally {
				XMPPServer.getInstance().getUserJedis().returnRes(jedis);
			}
			
		}
	}
}

redis-service.jsp

<%@ page import="java.util.*,
                 org.jivesoftware.openfire.XMPPServer,
                 org.jivesoftware.util.*,
                 com.montnets.redis.plugin.RedisServicePlugin,
                 org.jivesoftware.util.redis.enity.RedisConfig"
    errorPage="error.jsp"
%>

<%@ taglib uri="http://java.sun.com/jstl/core_rt" prefix="c" %>
<%@ taglib uri="http://java.sun.com/jstl/fmt_rt" prefix="fmt" %>

<%-- Define Administration Bean --%>
<jsp:useBean id="admin" class="org.jivesoftware.util.WebManager"  />
<c:set var="admin" value="${admin.manager}" />
<% admin.init(request, response, session, application, out ); %>

<%  // Get parameters
    boolean save = request.getParameter("save") != null;
	String initdata = request.getParameter("initdata");
	String deldata = request.getParameter("deldata");
    boolean success = request.getParameter("success") != null;
    RedisServicePlugin rsp = new RedisServicePlugin();
    Collection<RedisConfig> rcList = rsp.getAllRedisConfig();
    
    // Handle a save
    Map errors = new HashMap();
    Collection<RedisConfig> rcColl = new ArrayList<RedisConfig>();
	
    if ("REDIS_USER".equals(initdata)) {
    	rsp.initializeAllData();
        response.sendRedirect("redis-service.jsp?success=true");
        return;
    }
    else if ("REDIS_USER".equals(deldata)) {
    	rsp.destroyRedisByUserInfo();
    	response.sendRedirect("redis-service.jsp?success=true");
        return;
    }
    ......
    if (save) {
    	String uname = request.getParameter("REDIS_USER");
    	String r_user_name = request.getParameter("REDIS_USER_IP").trim();
    	String r_user_port = request.getParameter("REDIS_USER_PORT").trim();
    	String r_user_auto = request.getParameter("REDIS_USER_AUTO").trim();
    	if (r_user_name == null || r_user_port == null) {
    		errors.put("uname", "Redis-User is error!");
    	}
    	else {
    		RedisConfig rc = new RedisConfig();
        	rc.setName(uname);
        	rc.setIp(r_user_name);
        	rc.setPort(r_user_port);
        	rc.setAuto(r_user_auto);
        	rcColl.add(rc);
    	}
    	......
    	String mmname = request.getParameter("REDIS_MUC_MESSAGE");
    	String r_mm_name = request.getParameter("REDIS_MUC_MESSAGE_IP").trim();
    	String r_mm_port = request.getParameter("REDIS_MUC_MESSAGE_PORT").trim();
    	String r_mm_auto = request.getParameter("REDIS_MUC_MESSAGE_AUTO").trim();
    	if (r_mm_name == "" || r_mm_port == "") {
    		errors.put("mname", "Redis-MUC MESSAGE is error!");
    	}
    	else {
    		RedisConfig rc = new RedisConfig();
        	rc.setName(mmname);
        	rc.setIp(r_mm_name);
        	rc.setPort(r_mm_port);
        	rc.setAuto(r_mm_auto);
        	rcColl.add(rc);
    	}
    	
    	String omname = request.getParameter("REDIS_OFFLINE_MESSAGE");
    	......
        if (errors.size() == 0) {
        	rsp.saveRedisConfig(rcColl);
            response.sendRedirect("redis-service.jsp?success=true");
            return;
        }
    }
    
%>

<html>
    <head>
        <title>Redis Service Properties</title>
        <meta name="pageID" content="redis-service"/>
    </head>
    <body>
<%  if (!errors.isEmpty()) { %>

    <div class="jive-error">
    <table cellpadding="0" cellspacing="0" border="0">
    <tbody>
        <tr>
            <td class="jive-icon"><img src="images/error-16x16.gif" width="16" height="16" border="0" alt=""/></td>
            <td class="jive-icon-label">

            <% if (errors.get("uname") != null) { %>
                <%=errors.get("uname") %>
            <% } else if (errors.get("gname") != null) { %>
                <%=errors.get("gname") %>
            <% } else if (errors.get("rname") != null) { %>
                <%=errors.get("rname") %>
            <% } else if (errors.get("mname") != null) { %>
                <%=errors.get("mname") %>
            <% } else if (errors.get("mmname") != null) { %>
                <%=errors.get("mmname") %>
            <% } else if (errors.get("omname") != null) { %>
                <%=errors.get("omname") %>
            <% } else if (errors.get("cmname") != null) { %>
                <%=errors.get("cmname") %>
            <% } %>
            </td>
        </tr>
    </tbody>
    </table>
    </div><br>

<%  } %>

<p>
This is redis plugin
</p>

<%  if (success) { %>

    <div class="jive-success">
    <table cellpadding="0" cellspacing="0" border="0">
    <tbody>
        <tr><td class="jive-icon"><img src="images/success-16x16.gif" width="16" height="16" border="0"></td>
        <td class="jive-icon-label">
            redis service properties edited successfully.
        </td></tr>
    </tbody>
    </table>
    </div><br>
<% } %>

<form action="redis-service.jsp?save=true" method="post">

<fieldset>
    <legend>Redis Service</legend>
    <div>
    <ul>
    <%if (rcList.size() < 1) { %>
        <label for="r_user">REDIS_USER:</label>
		<input type="hidden" id="r_user_name" name="REDIS_USER" value="REDIS_USER"/>
        <input type="text" id="r_user_ip" name="REDIS_USER_IP" size="20" value=""/>
		<label for="r_port">port:</label>
		<input type="text" id="r_user_port" name="REDIS_USER_PORT" size="5" value=""/>
		<label for="r_auto">auto:</label>
		<input type="text" name="REDIS_USER_AUTO" size="10" value=""/>
		<br><br>
		
		<label for="r_user">REDIS_GROUP:</label>
		<input type="hidden" id="r_group_name" name="REDIS_GROUP" value="REDIS_GROUP"/>
        <input type="text" id="r_group_ip" name="REDIS_GROUP_IP" size="20" value=""/>
        <label for="r_port">port:</label>
		<input type="text" id="r_group_port" name="REDIS_GROUP_PORT" size="5" value=""/>
		<label for="r_auto">auto:</label>
		<input type="text" name="REDIS_GROUP_AUTO" size="10" value=""/>
		<br><br>
		
		<label for="r_user">REDIS_ROSTER:</label>
		<input type="hidden" id="r_roster_name" name="REDIS_ROSTER" value="REDIS_ROSTER"/>
        <input type="text" id="r_roster_ip" name="REDIS_ROSTER_IP" size="20" value=""/>
        <label for="r_port">port:</label>
		<input type="text" id="r_roster_port" name="REDIS_ROSTER_PORT" size="5" value=""/>
		<label for="r_auto">auto:</label>
		<input type="text" name="REDIS_ROSTER_AUTO" size="10" value=""/>
		<br><br>
		
		<label for="r_user">REDIS_MUC:</label>
		<input type="hidden" id="r_muc_name" name="REDIS_MUC" value="REDIS_MUC"/>
        <input type="text" id="r_muc_ip" name="REDIS_MUC_IP" size="20" value=""/>
        <label for="r_port">port:</label>
		<input type="text" id="r_muc_port" name="REDIS_MUC_PORT" size="5" value=""/>
		<label for="r_auto">auto:</label>
		<input type="text" name="REDIS_MUC_AUTO" size="10" value=""/>
		<br><br>
		
		<label for="r_user">REDIS_MUC_MESSAGE:</label>
		<input type="hidden" id="r_mm_name" name="REDIS_MUC_MESSAGE" value="REDIS_MUC_MESSAGE"/>
        <input type="text" id="r_mm_ip" name="REDIS_MUC_MESSAGE_IP" size="20" value=""/>
        <label for="r_port">port:</label>
		<input type="text" id="r_mm_port" name="REDIS_MUC_MESSAGE_PORT" size="5" value=""/>
		<label for="r_auto">auto:</label>
		<input type="text" name="REDIS_MUC_MESSAGE_AUTO" size="10" value=""/>
		<br><br>
		
		<label for="r_user">REDIS_OFFLINE_MESSAGE:</label>
		<input type="hidden" id="r_om_name" name="REDIS_OFFLINE_MESSAGE" value="REDIS_OFFLINE_MESSAGE"/>
        <input type="text" id="r_om_ip" name="REDIS_OFFLINE_MESSAGE_IP" size="20" value=""/>
        <label for="r_port">port:</label>
		<input type="text" id="r_om_port" name="REDIS_OFFLINE_MESSAGE_PORT" size="5" value=""/>
		<label for="r_auto">auto:</label>
		<input type="text" name="REDIS_OFFLINE_MESSAGE_AUTO" size="10" value=""/>
		<br><br>
		
		<label for="r_user">REDIS_CHAT_MESSAGE:</label>
		<input type="hidden" id="r_cm_name" name="REDIS_CHAT_MESSAGE" value="REDIS_CHAT_MESSAGE"/>
        <input type="text" id="r_cm_ip" name="REDIS_CHAT_MESSAGE_IP" size="20" value=""/>
        <label for="r_port">port:</label>
		<input type="text" id="r_cm_port" name="REDIS_CHAT_MESSAGE_PORT" size="5" value=""/>
		<label for="r_auto">auto:</label>
		<input type="text" name="REDIS_CHAT_MESSAGE_AUTO" size="10" value=""/>
		<br><br>
		
		<label for="r_user">REDIS_SEARCH:</label>
		<input type="hidden" id="r_rs_name" name="REDIS_SEARCH" value="REDIS_SEARCH"/>
        <input type="text" id="r_rs_ip" name="REDIS_SEARCH_IP" size="20" value=""/>
        <label for="r_port">port:</label>
		<input type="text" id="r_rs_port" name="REDIS_SEARCH_PORT" size="5" value=""/>
		<label for="r_auto">auto:</label>
		<input type="text" name="REDIS_SEARCH_AUTO" size="10" value=""/>
		<br><br>
		
		<% } %>
		<% if (rcList.size() >= 1) { %>
			<% for (RedisConfig cr : rcList) { %>
				<label for="r_user"><%=cr.getName() %></label>
				<input type="hidden" name="<%=cr.getName() %>" value="<%=cr.getName() %>"/>
        		<input type="text" name="<%=cr.getName() %>_IP" size="20" value="<%=cr.getIp() %>"/>
        		<label for="r_port">port:</label>
				<input type="text" name="<%=cr.getName() %>_PORT" size="5" value="<%=cr.getPort() %>"/>
				<label for="r_auto">auto:</label>
				<input type="text" name="<%=cr.getName() %>_AUTO" size="10" value="<%=cr.getAuto() %>"/>
				<br>
				<input type="button" value="initialize data" onclick="window.location.href='redis-service.jsp?initdata=<%=cr.getName() %>'; return false;"/>
				<input type="button" value="destroy data" onclick="window.location.href='redis-service.jsp?deldata=<%=cr.getName() %>'; return false;"/>
				<br><br>
			<%} %>
		<% } %>
    </ul>
    </div>
</fieldset>

<br><br>

<input type="submit" value="Save Settings">
<input type="hidden" name="initdata" value=""/>
</form>

</body>
</html>

Plugin.xml

<?xml version="1.0" encoding="UTF-8"?>
<plugin>
    <class>com.montnets.redis.plugin.RedisServicePlugin</class>
    <name>Redis Service</name>
    <description>Redis distributed cache</description>
    <author>HuWenFeng</author>
    <version>1.3.3</version>
    <date>17/1/2014</date>
    <minServerVersion>3.8.2</minServerVersion>
    
    <adminconsole>		
        <tab id="tab-redis"  name="Redis"  url="redis-service.jsp">
            <sidebar id="sidebar-redis-settings"  name="Redis Service" >
                <item id="redis-service" name="Redis Service" url="redis-service.jsp"
                     description="Redis distributed cache" />
            </sidebar>
        </tab>
    </adminconsole>
</plugin>

这里只简单贴出了用户这类的相关数据的预加载。

下面开始具体讲讲基于Redis数据的几大分类。这里这里主要拿用户和消息这两大块具体来说说openfire基于redis的优化。