OpenFire源码学习之二十二：openfie对用户的优化(下)

用户名片

在预初始化中，贴出来用户名片的程序。这里也一样不在重复。

首先同样先修改系统属性：

provider.vcard.className

org.jivesoftware.util.redis.expand.RedisVCardProvider

然后需要修改VCardManager名片管理这个类。

RedisVCardProvider：

import redis.clients.jedis.Jedis;public class RedisVCardProvider implements VCardProvider {private static final Logger Log = LoggerFactory.getLogger(RedisVCardProvider.class);    private static final String DELETE_PROPERTIES =        "DELETE FROM ofVCard WHERE username=?";    private static final String UPDATE_PROPERTIES =        "UPDATE ofVCard SET vcard=? WHERE username=?";    private static final String INSERT_PROPERTY =        "INSERT INTO ofVCard (username, vcard) VALUES (?, ?)";private static final int POOL_SIZE = 10;private BlockingQueue<SAXReader> xmlReaders = new LinkedBlockingQueue<SAXReader>(POOL_SIZE);public RedisVCardProvider() {        super();        // Initialize the pool of sax readers        for (int i=0; i<POOL_SIZE; i++) {            SAXReader xmlReader = new SAXReader();            xmlReader.setEncoding("UTF-8");            xmlReaders.add(xmlReader);        }    }@Overridepublic Element loadVCard(String username) {synchronized (username.intern()) {Jedis jedis = XMPPServer.getInstance().getUserJedis().getJedis();            Element vCardElement = null;            SAXReader xmlReader = null;            try {                // Get a sax reader from the pool                xmlReader = xmlReaders.take();                String usercard = jedis.get("OFVCARD:" + username);                if (usercard == null || "".equals(usercard)) {                return XMPPServer.getInstance().getJedisConfDao().getVCardProvider().loadVCard(username);                }                vCardElement = xmlReader.read(new StringReader(usercard)).getRootElement();            }            catch (Exception e) {                Log.error("Error loading vCard of username: " + username, e);            }            finally {            if (xmlReader != null) {                    xmlReaders.add(xmlReader);                }            XMPPServer.getInstance().getUserJedis().returnRes(jedis);            }            return vCardElement;        }}@Overridepublic Element createVCard(String username, Element vCardElement)throws AlreadyExistsException {......}@Overridepublic Element updateVCard(String username, Element vCardElement)throws NotFoundException {......}@Overridepublic void deleteVCard(String username) {......}@Overridepublic boolean isReadOnly() {// TODO Auto-generated method stubreturn false;}}

VCardManager

/** * Manages VCard information for users. * * @author Matt Tucker */public class VCardManager extends BasicModule implements ServerFeaturesProvider {private static final Logger Log = LoggerFactory.getLogger(VCardManager.class);private VCardProvider provider;    private static VCardManager instance;        private EventHandler eventHandler;        private static HmThreadPool threadPool = new HmThreadPool(5);        public static VCardManager getInstance() {        return instance;    }        public static VCardProvider getProvider() {        return instance.provider;    }    public VCardManager() {super("VCard Manager");        //String cacheName = "VCard";        //vcardCache = CacheFactory.createCache(cacheName);        this.eventHandler = new EventHandler();        // Keeps the cache updated in case the vCard action was not performed by VCardManager        VCardEventDispatcher.addListener(new VCardListener() {            public void vCardCreated(String username, Element vCard) {                // Since the vCard could be created by the provider, add it to the cache.                //vcardCache.put(username, vCard);            Jedis jedis = XMPPServer.getInstance().getUserJedis().getJedis();try {jedis.set("OFVCARD:" + username, vCard.asXML());} finally {XMPPServer.getInstance().getUserJedis().returnRes(jedis);}            }            public void vCardUpdated(String username, Element vCard) {                // Since the vCard could be updated by the provider, update it to the cache.                //vcardCache.put(username, vCard);            vCardCreated(username, vCard);            }            public void vCardDeleted(String username, Element vCard) {                // Since the vCard could be delated by the provider, remove it to the cache.                //vcardCache.remove(username);            Jedis jedis = XMPPServer.getInstance().getUserJedis().getJedis();                        try {            jedis.del("OFVCARD:" + username);} finally {XMPPServer.getInstance().getUserJedis().returnRes(jedis);}            }        });}    public String getVCardProperty(String username, String name) {        ......    }    public void setVCard(String username, Element vCardElement) throws Exception {        boolean created = false;        boolean updated = false;        if (provider.isReadOnly()) {            throw new UnsupportedOperationException("VCard provider is read-only.");        }                //Element newvCard = null;    Jedis jedis = XMPPServer.getInstance().getUserJedis().getJedis();    try {    boolean exists = jedis.exists("OFVCARD:" + username);                if (exists) {        threadPool.execute(createTaskByUpdateVCard(provider, username, vCardElement));        updated = true;        }        else {        threadPool.execute(createTaskByCreateVCard(provider, username, vCardElement));        created = true;        }}finally {XMPPServer.getInstance().getUserJedis().returnRes(jedis);}        // Dispatch vCard events        if (created) {             // Alert listeners that a new vCard has been created            VCardEventDispatcher.dispatchVCardCreated(username, vCardElement);        } else if (updated) {            // Alert listeners that a vCard has been updated            VCardEventDispatcher.dispatchVCardUpdated(username, vCardElement);        }    }    private Runnable createTaskByCreateVCard(final VCardProvider provider, final String username, final Element vCardElement) {   return new Runnable() {   public void run() {try {provider.createVCard(username, vCardElement);} catch (AlreadyExistsException e) {Log.error("AlreadyExistsException: username=" + username + ", vCardElement=" + vCardElement);}}   };   } private Runnable createTaskByUpdateVCard(final VCardProvider provider, final String username, final Element vCardElement) {   return new Runnable() {   public void run() {try {provider.updateVCard(username, vCardElement);} catch (NotFoundException e) {Log.error("NotFoundException: username=" + username + ", vCardElement=" + vCardElement);}}   };   }    public void deleteVCard(String username) {        if (provider.isReadOnly()) {            throw new UnsupportedOperationException("VCard provider is read-only.");        }                final String vusername = username;        threadPool.execute(new Runnable() {@Overridepublic void run() {provider.deleteVCard(vusername);}});        VCardEventDispatcher.dispatchVCardDeleted(username, null);    }public Element getVCard(String username) {        Element vCardElement = getOrLoadVCard(username);        return vCardElement == null ? null : vCardElement.createCopy();    }private Element getOrLoadVCard(String username) {return provider.loadVCard(username);    }    @Overridepublic void initialize(XMPPServer server) {        ......    }    @Overridepublic void start() {        ......    }        @Overridepublic void stop() {        // Remove this module as a user event listener        UserEventDispatcher.removeListener(eventHandler);    }        public void reset() {        //vcardCache.clear();    }    @Overridepublic Iterator<String> getFeatures() {ArrayList<String> features = new ArrayList<String>();        features.add("redis-vcard-temp");        return features.iterator();}private class EventHandler extends UserEventAdapter {        @Overridepublic void userDeleting(User user, Map params) {            try {                deleteVCard(user.getUsername());            } catch (UnsupportedOperationException ue) { /* Do Nothing */ }        }    }    public UserCardEnity getUserCardByUserName (String username) {    Element element = getVCard(username);    UserCardEnity uce = new UserCardEnity();        if (element != null) {    String myName = element.elementText("MYNAME");    String sex = element.elementText("SEX");    String oname = element.elementText("ONAME");    String moblie = element.elementText("MOBILE");    String landline = element.elementText("LANDLINE");    String address = element.elementText("ADDRESS");    String workUnit = element.elementText("WORKUNIT");    String birthday = element.elementText("BIRTHDAY");    String photo = element.elementText("PHOTO");    String userType = element.elementText("USERTYPE");        uce.setMyName( myName);    uce.setSex(sex);    uce.setOname(oname);    uce.setMoblie( moblie) ;    uce.setLandline(landline);    uce.setAddress(address);    uce.setWorkUnit(workUnit) ;    uce.setBirthday(birthday);    uce.setPhoto(photo);        if (userType == null) {    uce.setUserType("");    }    else if (1 == Integer.valueOf(userType)) {    uce.setUserType("student");    }    else if (2 == Integer.valueOf(userType)) {    uce.setUserType("teacher");    }    else if (3 == Integer.valueOf(userType)) {    uce.setUserType("Guardian");    }    else if (4 == Integer.valueOf(userType)) {    uce.setUserType("edusun admin");    }    else if (5 == Integer.valueOf(userType)) {    uce.setUserType("Agents");    }    else {    uce.setUserType("Other locations");    }    }    return uce;    }}

用户名片就到这了。

用户搜索

搜索用户的时候，在openfire中都是重新查找关系数据库的。因为我们已经将用户预加载到了redis中。那么这里只需要对用户做一些分词存储检索即可。

本人在这里做了比较简单的分词处理。比如用户名，手机号码等。

首先要做的就是制作用户分词了。然后要做的就是需要修修改搜索的handler处理类。Openfire上提供了一个search搜索插件。查询消息最后提交给SearchPlugin这个类的handler方法。本人这里就不做描述。重点要说的就是如何在redis中分词存储。

客户端发送用户查找请求如下：

<iq id="E6l1b-43" to="[e-mail]240" type="get"><query xmlns="jabber:iq:search"></query></iq>

请看代码清单：

RedisSearchManager：

public class RedisSearchManager extends BasicModule{private static final Logger LOG = LoggerFactory.getLogger(RedisSearchManager.class);private static final Integer timeout = 1000*10;private static final int maxActive = 5000 * 10;private static final int maxIdle = 5;private static final long maxWait = (1000 * 100);private static JedisPool pool;private static XMPPServer loaclserver;private static JedisPoolConfig config;public RedisSearchManager() {super("Redis search manager");}private static JedisPoolConfig createConfig() {           ......}private static void createJedisPool() {......//创建连接池}private static synchronized void poolInit() {boolean enabled = JiveGlobals.getBooleanProperty("plugin.userservice.enabled");if (enabled) {if (pool == null)createJedisPool();}}public Jedis getSearchJedis() {if (pool == null)poolInit();Jedis jedis = pool.getResource();jedis.select(3);return jedis;}public void returnSearchJedis(Jedis jedis) {pool.returnResource(jedis);}@Overridepublic void initialize(XMPPServer server) {.....    }//创建用户的搜索关键词数据public void createSearchAllUserData() {Collection<User> userList = XMPPServer.getInstance().getUserJedis().getAllUser();for (User user : userList) {createSearchUserData(user);}System.out.println("Initialize the user search data is completed...(the end)");}   //这里根据string长度切分。按照“|”分割public void createSearchUserData(User user) {Jedis jedis = getSearchJedis();IndexWriter iw = new IndexWriter(jedis);String username = user.getUsername();String keyword = "";if (username.length() >= 4) {int index = 4;for (int i=0; i<=username.length()-4; i++) {String usernameKey = username.substring(0, index++) + "|";keyword += usernameKey;}}else {keyword = username + "|";}if (user.getName() != null && !"".equals(user.getName())) {keyword += user.getName() + "|";}if (user.getEmail() != null && !"".equals(user.getEmail())) {keyword += user.getEmail() + "|";}if (user.getMoblie() != null && !"".equals(user.getMoblie())) {keyword += user.getMoblie();}if ("|".equals(keyword.substring(keyword.length()-1))) {keyword = keyword.substring(0, keyword.length()-1);}iw.addIdAndIndexItem(username, keyword);iw.addNeedSortItem("USERNAME",username);        iw.addNeedSortItem("CREATIONDATE",user.getCreationDate() != null ? user.getCreationDate() : StringUtils.dateToMillis(new Date()));        iw.addNeedSortItem("MODIFICATIONDATE",user.getModificationDate() != null ? user.getModificationDate() : StringUtils.dateToMillis(new Date()));        iw.writer();        System.out.println("create user search data, id:" + username + ", keyword:" + keyword);        LOG.info("create user search data, id:" + username + ", keyword:" + keyword);        returnSearchJedis(jedis);} }

IndexWriter写入索引

public class IndexWriter {private Jedis jedis;private String id;private Map<String, String> items = new HashMap<String, String>();private String contentItems[];public IndexWriter(Jedis jedis) {if (!jedis.isConnected()) {jedis.connect();}this.jedis = jedis;}/** * @param id *            必须有 * @param content *            是分词程序切分后的内容，每个词中间必须用用“|”分隔，如：中国|中国人|2012 */public void addIdAndIndexItem(String id, String content) {this.id = id;contentItems = content.split("\\|");}public void addNeedSortItem(String name, String value) {items.put(name, value);}public void writer() {indexWriter();itemWriter();}private void indexWriter() {if (!id.equals("") && contentItems.length != 0) {for (int i = 0; i < contentItems.length; i++) {jedis.sadd(contentItems[i].trim(), id);}}}private void itemWriter() {if (items.size() != 0) {Iterator<Entry<String, String>> it = items.entrySet().iterator();while (it.hasNext()) {Entry<String, String> entry = (Entry<String, String>) it.next();jedis.set(entry.getKey().toString() + ":" + id, entry.getValue().toString());}}}}

IndexSearch：查找索引

public class IndexSearch {public static int ALPHA = 0;public static int DESC = 1;public static int ASC = 2;private Jedis jedis;private int limit = 100;private String itemName = null;private int pager = 0;public IndexSearch(Jedis jedis) {if (!jedis.isConnected()) {jedis.connect();}this.jedis = jedis;}private SortingParams getSP(String item, int sort) {SortingParams sp = new SortingParams();sp.limit(pager, limit);if (null == item || "".equals(item)) {switch (sort) {case 1:sp.desc();break;case 2:sp.asc();break;case 0:default:sp.alpha();break;}} else {switch (sort) {case 1:sp.by(itemName + ":*").desc();break;case 2:sp.by(itemName + ":*").asc();break;case 0:default:sp.by(itemName + ":*").alpha();break;}}return sp;}private List<String> isearch(int sort, String... query) {jedis.sinterstore("tempKey", query);return jedis.sort("tempKey", this.getSP(itemName, sort));}public List<String> search(String... query) {return this.isearch(0, query);}public List<String> search(int sort, String... query) {return this.isearch(sort, query);}public List<String> search(String itemName, int sort, String... query) {this.itemName = itemName;return this.isearch(sort, query);}public List<String> search(String itemName, int sort, int limit,String... query) {this.itemName = itemName;this.limit = limit;return this.isearch(sort, query);}public List<String> search(String itemName, int sort, int pager, int limit,String... query) {this.itemName = itemName;this.limit = limit;this.pager = pager;return this.isearch(sort, query);}}

Ok用户搜索这就只贴出一些比较关键性的代码。提供思路。代码贴多了搞了篇幅很长。