openfire聊天记录插件

其他 2018-12-05 08:51:30 阅读次数: 0 
package com.sqj.openfire.chat.logs;


import java.io.File;
import java.util.Date;
import java.util.List;
import java.util.Map;
import java.util.UUID;

import org.apache.commons.lang.StringUtils;
import org.dom4j.DocumentFactory;
import org.dom4j.Element;
import org.jivesoftware.openfire.OfflineMessageStore;
import org.jivesoftware.openfire.XMPPServer;
import org.jivesoftware.openfire.container.Plugin;
import org.jivesoftware.openfire.container.PluginManager;
import org.jivesoftware.openfire.interceptor.InterceptorManager;
import org.jivesoftware.openfire.interceptor.PacketInterceptor;
import org.jivesoftware.openfire.interceptor.PacketRejectedException;
import org.jivesoftware.openfire.session.Session;
import org.jivesoftware.openfire.user.UserManager;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.xmpp.packet.IQ;
import org.xmpp.packet.JID;
import org.xmpp.packet.Message;
import org.xmpp.packet.Packet;
import org.xmpp.packet.Presence;

import com.sqj.openfire.chat.logs.entity.Chat;
import com.sqj.openfire.chat.logs.entity.XmlToMap;

/** <b>function:</b> 聊天记录插件  
 *  @author shm 
 *  @createDate 2016-3-14
 *  @version 1.0  
 */
public class ChatLogsPlugin implements PacketInterceptor, Plugin {
      private static final Logger log = LoggerFactory.getLogger(ChatLogsPlugin.class);
      private static PluginManager pluginManager;
      private static DbChatLogsManager logsManager;
      private InterceptorManager interceptorManager;
    
      public ChatLogsPlugin()
      {
        this.interceptorManager = InterceptorManager.getInstance();
        logsManager = DbChatLogsManager.getInstance();
      }

      /**     
       *  * <b>function:</b> 拦截消息核心方法，Packet就是拦截消息对象      
       *  * @author shm      
       *  * @createDate 2016-3-14
       *  */    
      @Override
      public void interceptPacket(Packet packet, Session session, boolean incoming, boolean processed) throws PacketRejectedException {
          if (session != null) {             
              //debug(packet, incoming, processed, session); 
          }
          JID recipient = packet.getTo();
          if (recipient != null) {   
              String username = recipient.getNode();    
              // 广播消息或是不存在/没注册的用户.             
              if (username == null || !UserManager.getInstance().isRegisteredUser(recipient)) {   
                  return;         
              } else if (!XMPPServer.getInstance().getServerInfo().getXMPPDomain().equals(recipient.getDomain())) {  
                  // 非当前openfire服务器信息                
                  return;             
              } else if ("".equals(recipient.getResource())) {         
              }       
          } 
          this.doAction(packet, incoming, processed, session);    
     }
      
      /**     
       * <b>function:</b> 执行保存/分析聊天记录动作      
       * @author shm     
       * @createDate 2016-3-14
       * @param packet 数据包  
       * @param incoming true表示发送方      
       * @param session 当前用户session     
       */     
      private void doAction(Packet packet, boolean incoming, boolean processed, Session session) { 
          Packet copyPacket = packet.createCopy(); 
          if (packet instanceof Message) { 
              Message message = (Message) copyPacket;  
              // 一对一聊天，单人模式    
              if (message.getType() == Message.Type.chat) {
                  log.info("单人聊天信息：{}", message.toXML());
                  //debug("单人聊天信息：" + message.toXML()); 
                  // 程序执行中；是否为结束或返回状态（是否是当前session用户发送消息）    
                  if (processed || !incoming) { 
                      return;            
                  }         
                  
                  java.sql.Timestamp date = new java.sql.Timestamp(new Date().getTime());
                  
                  //在原有的扩展xml中添加发送日期子节点
                  Element chatInfoElement = message.getElement().element("chatinfo");
                  Element timeElement = DocumentFactory.getInstance().createDocument().addElement("sendtime");
                  timeElement.setText(String.valueOf(date.getTime()));
                  chatInfoElement.add(timeElement);
                  
                  //logsManager.add(this.get(packet, incoming, session));       
                  logsManager.addByMap(this.getMsgInfo(packet, incoming, session, date)); 
                  
                  //保存到离线消息表，客户端收到后调用删除离线消息功能，这样可确保即使网络突然掉线或不好的情况下消息丢失的问题
                  OfflineMessageStore offlineMessageStore = new OfflineMessageStore();
                  offlineMessageStore.addMessage(message);
                 
                  //消息回执
                  Message receiptMessage = new Message();
                  receiptMessage.setTo(message.getFrom());
                  receiptMessage.setType(Message.Type.normal);
                  Element received = receiptMessage.addChildElement("received", "urn:xmpp:receipts");
                  received.addAttribute("id", message.getID());
                  received.addAttribute("type", "normal");
                  log.info("回执内容：" + receiptMessage);
                  
                  // 判断接受者是否在线，2代表离线状态，1代表在线状态，0代表用戶不存在
                  //注意一定要修改Constant类中的url,否则url错误或导致消息发送很慢，因为url不对时查找用户是否在线会浪费很多时间
                  /*if (IsOnLineUtils.IsUserOnLine(message.getTo()) == 1) {
                      received.addAttribute("status", "1");
                  } else if (IsOnLineUtils.IsUserOnLine(message.getTo()) == 0) {
                      received.addAttribute("status", "0");
                  } else if (IsOnLineUtils.IsUserOnLine(message.getTo()) == 2) {
                      received.addAttribute("status", "2");
                  }*/
                  
                  try {
                      XMPPServer.getInstance().getPacketDeliverer().deliver(receiptMessage);
                      log.info("服务端回执成功！");
                  } catch (Exception e) {
                      e.printStackTrace();
                  }
                  
              // 群聊天，多人模式           
              } else if (message.getType() ==  Message.Type.groupchat) {  
                  List<?> els = message.getElement().elements("x");  
                  if (els != null && !els.isEmpty()) {             
                      log.info("群聊天信息：{}", message.toXML());    
                      debug("群聊天信息：" + message.toXML());         
                  } else {                   
                      log.info("群系统信息：{}", message.toXML());  
                        debug("群系统信息：" + message.toXML());        
                  }                           
              // 其他信息           
              }else {     
                  log.info("其他信息：{}", message.toXML()); 
                  //debug("其他信息：" + message.toXML());   
              }      
          } else if (packet instanceof IQ) { 
              IQ iq = (IQ) copyPacket;       
              if (iq.getType() == IQ.Type.set && iq.getChildElement() != null && "session".equals(iq.getChildElement().getName())) {
                  log.info("用户登录成功：{}", iq.toXML());           
                  //debug("用户登录成功：" + iq.toXML());         
              }      
          } else if (packet instanceof Presence) {  
              Presence presence = (Presence) copyPacket;     
              if (presence.getType() == Presence.Type.unavailable) {   
                  log.info("用户退出服务器成功：{}", presence.toXML());             
                  //debug("用户退出服务器成功：" + presence.toXML());            
              }       
         }    
     }
      
/**      
       * <b>function:</b> 创建一个聊天记录实体对象，并设置相关数据      
       * @author shm     
       * @createDate 2016-3-14
       * @param packet 数据包     
       * @param incoming 如果为ture就表明是发送者     
       * @param session 当前用户session      
       * @return 聊天实体  
   */    
    @SuppressWarnings("unchecked")
    private Chat get(Packet packet, boolean incoming, Session session) {        
          Message message = (Message) packet;        
          JID jid = session.getAddress();         
         
         Map<String, Object> map = XmlToMap.xmlToHashMap(message.toXML());
         Chat chat = new Chat();
         if(map.containsKey("zid") && !"null".equals(map.get("zid")) && !StringUtils.isEmpty(map.get("zid")+"")){
             chat.setId(map.get("zid")+"");
         }else{
             chat.setId(UUID.randomUUID().toString().replace("-", ""));
         }
         chat.setCaseId(map.get("caseid")+"");
         chat.setContent(message.getBody());
         chat.setCreator(jid.getNode());
         chat.setSender(jid.getNode());
         chat.setModifier(jid.getNode());
         chat.setSendee(message.getTo().getNode());
         chat.setIdentity(map.get("identity")+"");
         chat.setzType(map.get("ztype")+"");
         chat.setSenderId(map.get("senderid")+"");
         chat.setSendeeId(map.get("sendeeid")+"");
         
         if(map.get("msglength") != null && !"".equals(map.get("msglength")))
         {
             chat.setMsgLength(Integer.parseInt(map.get("msglength")+""));
         }
         return chat;    
      }
      
    @SuppressWarnings({ "unchecked", "rawtypes" })
    private Map getMsgInfo(Packet packet, boolean incoming, Session session, java.sql.Timestamp date) {        
       Message message = (Message) packet;        
       Map<String, Object> map = XmlToMap.xmlToHashMap(message.toXML());
       map.put("content", message.getBody());
       map.put("sendtime", date);
       log.info("tag：{}", map.get("tag"));             
       //debug("tag：" + map.get("tag"));            
       return map;
      }
    
      /**      
       * * <b>function:</b> 调试信息      
       * @author shm     
       * @createDate 2016-3-14
       * @param packet 数据包    
       * @param incoming 如果为ture就表明是发送者      
       * @param processed 执行      
       * @param session 当前用户session      
       */    
      private void debug(Packet packet, boolean incoming, boolean processed, Session session) {         
          //String info = "[ packetID: " + packet.getID() + ", to: " + packet.getTo() + ", from: " + packet.getFrom() + ", incoming: " + incoming + ", processed: " + processed + " ]";                 
          StringBuilder info = new StringBuilder();
          info.append("[ packetID: ").append(packet.getID()).append(", to: ").append(packet.getTo()).append(", from: ").append(packet.getFrom()).append(", incoming: ").append(incoming).append(", processed: ").append(processed).append(" ]");
          
          long timed = System.currentTimeMillis();         
          debug("################### start ###################" + timed);        
          debug("id:" + session.getStreamID() + ", address: " + session.getAddress());         
          debug("info: " + info);         debug("xml: " + packet.toXML());         
          debug("################### end #####################" + timed);                 
          log.info("id:" + session.getStreamID() + ", address: " + session.getAddress());         
          log.info("info: {}", info);        
          log.info("plugin Name: " + pluginManager.getName(this) + ", xml: " + packet.toXML());    
      }        
      
      private void debug(Object message) { 
          if (true) {             
              System.out.println(message);      
          }  
     }        
      
     @Override     
     public void destroyPlugin() { 
         interceptorManager.removeInterceptor(this);   
         debug("销毁聊天记录插件成功！");    
     }     
     
     @Override     
     public void initializePlugin(PluginManager manager, File pluginDirectory) { 
         interceptorManager.addInterceptor(this);         
         pluginManager = manager;                   
         debug("安装聊天记录插件成功！");     
     }
      
     
    /* // 判断接受者是否在线，2代表离线状态，1代表在线状态，0代表用戶不存在
     if (IsOnLineUtils.IsUserOnLine(message.getTo()) == 2) {
         // 离线時，向offline表写数据
         OfflineMessageStore offlineMessageStore = new OfflineMessageStore();
         offlineMessageStore.addMessage(message);
         // 向客户端发回执
         Message receiptMessage = new Message();
         receiptMessage.setTo(message.getFrom());
         receiptMessage.setType(Message.Type.normal);
         Element received = receiptMessage.addChildElement("received", "urn:xmpp:receipts");
         received.setAttributeValue("id", message.getID());
         System.out.println("0000000000回执内容" + receiptMessage);
         try {
             XMPPServer.getInstance().getPacketDeliverer().deliver(receiptMessage);
             System.out.println("服务端回执成功！");
         } catch (Exception e) {
             e.printStackTrace();
         }
     }*/
}

猜你喜欢

转载自www.cnblogs.com/tiancai/p/10068883.html
今日推荐
TIOBE 5 月榜单：Fortran “复活”进入 Top 10
GCC 14.1 发布
面壁智能发布 Eurux-8x22B 开源大模型 —— 堪称「理科状元」
开源日报 | 谷歌扶持鸿蒙上位；开源Rabbit R1；Docker加持的安卓手机；微软的焦虑和野心；海尔电器把开放平台关了
中国码农的“35岁魔咒”
蘭雅 CorelDRAW 插件 2024.5.1 国际劳动节版，免费下载
Arc Browser for Windows 1.0 正式 GA
90后程序员开发视频搬运软件、不到一年获利超 700 万，结局很刑！
周排行
Java自定义时间格式
同步整形电路
在开发中最最最常用的字符串的属性大集合
Linux 查看端口占用并杀掉
Java基础四：ArrayList
多线程之死锁就是这么简单
mysql 基础命令集
awk 命令详解
Centos6.3编译安装nginx+php步骤
OCR （Optical Character Recognition，光学字符识别）
每日归档
更多
2024-05-08(42)
2024-05-07(14)
2024-05-06(40)
2024-05-05(0)
2024-05-04(7)
2024-05-03(19)
2024-05-02(0)
2024-05-01(4)
2024-04-30(1)
2024-04-29(40)

代码天地 © 2018 codetd.com

境外服务器   最新电视剧2022