XMPP即时通讯协议使用（二）——基于Smack相关操作

package com.test;

import java.util.ArrayList;
import java.util.Collection;
import java.util.Iterator;
import java.util.List;
import java.util.Map;
import java.util.Set;
import org.jivesoftware.smackx.pubsub.PayloadItem;
import org.jivesoftware.smack.chat.Chat;
import org.jivesoftware.smack.chat.ChatManager;
import org.jivesoftware.smack.chat.ChatManagerListener;
import org.jivesoftware.smack.chat.ChatMessageListener;
import org.jivesoftware.smack.packet.Message;
import org.jivesoftware.smack.roster.Roster;
import org.jivesoftware.smack.roster.RosterEntry;
import org.jivesoftware.smack.roster.RosterGroup;
import org.jivesoftware.smack.tcp.XMPPTCPConnection;
import org.jivesoftware.smack.tcp.XMPPTCPConnectionConfiguration;
import org.jivesoftware.smackx.iqregister.AccountManager;
import org.jivesoftware.smackx.pubsub.AccessModel;
import org.jivesoftware.smackx.pubsub.ConfigureForm;
import org.jivesoftware.smackx.pubsub.Item;
import org.jivesoftware.smackx.pubsub.LeafNode;
import org.jivesoftware.smackx.pubsub.PubSubManager;
import org.jivesoftware.smackx.pubsub.PublishModel;
import org.jivesoftware.smackx.pubsub.SimplePayload;
import org.jivesoftware.smackx.pubsub.SubscribeForm;
import org.jivesoftware.smackx.pubsub.Subscription;
import org.jivesoftware.smackx.vcardtemp.packet.VCard;
import org.jivesoftware.smackx.xdata.packet.DataForm;
import org.jxmpp.jid.EntityBareJid;
import org.jxmpp.jid.impl.JidCreate;
import org.jxmpp.jid.parts.Localpart;

public class SmarkUtil {

	private XMPPTCPConnection connection;

	private String userName;

	private String password;

	private String xmppDomain;

	private String serverName;

	public SmarkUtil(String userName, String password, String xmppDomain, String serverName) {
		this.userName = userName;
		this.password = password;
		this.xmppDomain = xmppDomain;
		this.serverName = serverName;
		try {
			if (connection == null) {
				getConnection(userName, password, xmppDomain, serverName);
			}
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	/**
	 * 获取连接
	 *
	 * @param userName
	 * @param password
	 * @return
	 * @throws Exception
	 */
	public void getConnection(String userName, String password, String xmppDomain, String serverName) throws Exception {
		try {
			XMPPTCPConnectionConfiguration.Builder configBuilder = XMPPTCPConnectionConfiguration.builder();
			configBuilder.setUsernameAndPassword(userName, password);
			configBuilder.setXmppDomain(xmppDomain);
			configBuilder.setSecurityMode(XMPPTCPConnectionConfiguration.SecurityMode.disabled);

			connection = new XMPPTCPConnection(configBuilder.build());
			// 连接服务器
			connection.connect();
			// 登录服务器
			connection.login();
		} catch (Exception e) {
			e.printStackTrace();
		}
	}

	/**
	 * 创建一个新用户
	 *
	 * @param userName
	 *            用户名
	 * @param password
	 *            密码
	 * @param attr
	 *            用户资料
	 * @return
	 * @throws Exception
	 */
	public boolean registerAccount(String userName, String password, Map<String, String> attr) throws Exception {
		AccountManager manager = AccountManager.getInstance(connection);
		manager.sensitiveOperationOverInsecureConnection(true);
		Localpart l_username = Localpart.from(userName);
		if (attr == null) {
			manager.createAccount(l_username, password);
		} else {
			manager.createAccount(l_username, password, attr);
		}

		return true;
	}

	/**
	 * 修改当前登陆用户密码
	 *
	 * @param password
	 * @return
	 * @throws Exception
	 */
	public boolean changePassword(String password) throws Exception {
		AccountManager manager = AccountManager.getInstance(connection);
		manager.sensitiveOperationOverInsecureConnection(true);
		manager.changePassword(password);
		return true;
	}

	/**
	 * 删除当前登录用户
	 *
	 * @return
	 * @throws Exception
	 */
	public boolean deleteAccount() throws Exception {
		AccountManager manager = AccountManager.getInstance(connection);
		manager.sensitiveOperationOverInsecureConnection(true);
		manager.deleteAccount();
		return true;
	}

	/**
	 * 获取用户属性名称
	 *
	 * @return
	 * @throws Exception
	 */
	public List getAccountInfo() throws Exception {
		List<String> list = new ArrayList<String>();
		AccountManager manager = AccountManager.getInstance(connection);
		manager.sensitiveOperationOverInsecureConnection(true);
		Set<String> set = manager.getAccountAttributes();
		list.addAll(set);
		return list;
	}

	/**
	 * 获取所有组
	 *
	 * @return
	 * @throws Exception
	 */
	public List<RosterGroup> getGroups() throws Exception {
		List<RosterGroup> grouplist = new ArrayList<RosterGroup>();
		Roster roster = Roster.getInstanceFor(connection);
		Collection<RosterGroup> rosterGroup = roster.getGroups();
		Iterator<RosterGroup> i = rosterGroup.iterator();
		while (i.hasNext()) {
			grouplist.add(i.next());
		}
		return grouplist;
	}

	/**
	 * 添加分组
	 *
	 * @param groupName
	 * @return
	 * @throws Exception
	 */
	public boolean addGroup(String groupName) throws Exception {
		Roster roster = Roster.getInstanceFor(connection);
		roster.createGroup(groupName);
		return true;

	}

	/**
	 * 获取指定分组的好友
	 *
	 * @param groupName
	 * @return
	 * @throws Exception
	 */
	public List<RosterEntry> getEntriesByGroup(String groupName) throws Exception {
		Roster roster = Roster.getInstanceFor(connection);
		List<RosterEntry> Entrieslist = new ArrayList<RosterEntry>();
		RosterGroup rosterGroup = roster.getGroup(groupName);
		Collection<RosterEntry> rosterEntry = rosterGroup.getEntries();
		Iterator<RosterEntry> i = rosterEntry.iterator();
		while (i.hasNext()) {
			Entrieslist.add(i.next());
		}
		return Entrieslist;
	}

	/**
	 * 获取全部好友
	 *
	 * @return
	 * @throws Exception
	 */
	public List<RosterEntry> getAllEntries() throws Exception {
		Roster roster = Roster.getInstanceFor(connection);
		List<RosterEntry> Entrieslist = new ArrayList<RosterEntry>();
		Collection<RosterEntry> rosterEntry = roster.getEntries();
		Iterator<RosterEntry> i = rosterEntry.iterator();
		while (i.hasNext()) {
			Entrieslist.add(i.next());
		}
		return Entrieslist;
	}

	/**
	 * 获取用户VCard信息
	 *
	 * @param userName
	 * @return
	 * @throws Exception
	 */
	public VCard getUserVCard(String userName) throws Exception {
		VCard vcard = new VCard();
		EntityBareJid jid = JidCreate.entityBareFrom(userName + "@" + this.serverName);
		vcard.load(connection, jid);
		return vcard;
	}

	/**
	 * 添加好友 无分组
	 *
	 * @param userName
	 * @param name
	 * @return
	 * @throws Exception
	 */
	public boolean addUser(String userName, String name) throws Exception {
		Roster roster = Roster.getInstanceFor(connection);
		EntityBareJid jid = JidCreate.entityBareFrom(userName + "@" + this.serverName);
		roster.createEntry(jid, name, null);
		return true;

	}

	/**
	 * 添加好友 有分组
	 *
	 * @param userName
	 * @param name
	 * @param groupName
	 * @return
	 * @throws Exception
	 */
	public boolean addUser(String userName, String name, String groupName) throws Exception {
		Roster roster = Roster.getInstanceFor(connection);
		EntityBareJid jid = JidCreate.entityBareFrom(userName + "@" + this.serverName);
		roster.createEntry(jid, name, new String[] { groupName });
		return true;

	}

	/**
	 * 删除好友
	 *
	 * @param userName
	 * @return
	 * @throws Exception
	 */
	public boolean removeUser(String userName) throws Exception {
		Roster roster = Roster.getInstanceFor(connection);
		EntityBareJid jid = JidCreate.entityBareFrom(userName + "@" + this.serverName);
		RosterEntry entry = roster.getEntry(jid);
		System.out.println("删除好友：" + userName);
		System.out.println("User." + roster.getEntry(jid) == null);
		roster.removeEntry(entry);

		return true;

	}

	/**
	 * 创建发布订阅节点
	 *
	 * @param nodeId
	 * @return
	 * @throws Exception
	 */
	public boolean createPubSubNode(String nodeId) throws Exception {
		PubSubManager mgr = PubSubManager.getInstance(connection);
		// Create the node
		LeafNode leaf = mgr.createNode(nodeId);
		ConfigureForm form = new ConfigureForm(DataForm.Type.submit);
		form.setAccessModel(AccessModel.open);
		form.setDeliverPayloads(true);
		form.setNotifyRetract(true);
		form.setPersistentItems(true);
		form.setPublishModel(PublishModel.open);
		form.setMaxItems(10000000);// 设置最大的持久化消息数量

		leaf.sendConfigurationForm(form);
		return true;
	}

	/**
	 * 创建发布订阅节点
	 *
	 * @param nodeId
	 * @param title
	 * @return
	 * @throws Exception
	 */
	public boolean createPubSubNode(String nodeId, String title) throws Exception {
		PubSubManager mgr = PubSubManager.getInstance(connection);
		// Create the node
		LeafNode leaf = mgr.createNode(nodeId);
		ConfigureForm form = new ConfigureForm(DataForm.Type.submit);
		form.setAccessModel(AccessModel.open);
		form.setDeliverPayloads(true);
		form.setNotifyRetract(true);
		form.setPersistentItems(true);
		form.setPublishModel(PublishModel.open);
		form.setTitle(title);
		form.setBodyXSLT(nodeId);
		form.setMaxItems(10000000);// 设置最大的持久化消息数量
		form.setMaxPayloadSize(1024*12);//最大的有效载荷字节大小
		leaf.sendConfigurationForm(form);
		return true;
	}

	public boolean deletePubSubNode(String nodeId) throws Exception {
		PubSubManager mgr = PubSubManager.getInstance(connection);
		mgr.deleteNode(nodeId);
		return true;
	}

	/**
	 * 发布消息
	 *
	 * @param nodeId
	 *            主题ID
	 * @param eventId
	 *            事件ID
	 * @param messageType
	 *            消息类型：publish（发布）/receipt（回执）/state（状态）
	 * @param messageLevel
	 *            0/1/2
	 * @param messageSource
	 *            消息来源
	 * @param messageCount
	 *            消息数量
	 * @param packageCount
	 *            总包数
	 * @param packageNumber
	 *            当前包数
	 * @param createTime
	 *            创建时间 2018-06-07 09:43:06
	 * @param messageContent
	 *            消息内容
	 * @return
	 * @throws Exception
	 */
	public boolean publish(String nodeId, String eventId, String messageType, int messageLevel, String messageSource,
			int messageCount, int packageCount, int packageNumber, String createTime, String messageContent)
			throws Exception {

		if (messageContent.length() > 1024 * 10) {
			throw new Exception("消息内容长度超出1024*10，需要进行分包发布");
		}
		PubSubManager mgr = PubSubManager.getInstance(connection);
		LeafNode node = null;

		node = mgr.getNode(nodeId);

		StringBuffer xml = new StringBuffer();
		xml.append("<pubmessage xmlns='pub:message'>");
		xml.append("<nodeId>" + nodeId + "</nodeId>");
		xml.append("<eventId>" + eventId + "</eventId>");
		xml.append("<messageType>" + messageType + "</messageType>");
		xml.append("<messageLevel>" + messageLevel + "</messageLevel>");
		xml.append("<messageSource>" + messageSource + "</messageSource>");
		xml.append("<messageCount>" + messageCount + "</messageCount>");
		xml.append("<packageCount>" + packageCount + "</packageCount>");
		xml.append("<packageNumber>" + packageNumber + "</packageNumber>");
		xml.append("<createTime>" + createTime + "</createTime>");
		xml.append("<messageContent>" + messageContent + "</messageContent>");
		xml.append("</pubmessage>");

		SimplePayload payload = new SimplePayload("pubmessage", "pub:message", xml.toString().toLowerCase());
		PayloadItem<SimplePayload> item = new PayloadItem<SimplePayload>(System.currentTimeMillis() + "", payload);
		node.publish(item);
		return true;
	}

	/**
	 * 订阅主题
	 *
	 * @param nodeId
	 * @return
	 * @throws Exception
	 */
	public boolean subscribe(String nodeId) throws Exception {

		PubSubManager mgr = PubSubManager.getInstance(connection);

		// Get the node
		LeafNode node = mgr.getNode(nodeId);
		SubscribeForm subscriptionForm = new SubscribeForm(DataForm.Type.submit);
		subscriptionForm.setDeliverOn(true);
		subscriptionForm.setDigestFrequency(5000);
		subscriptionForm.setDigestOn(true);
		subscriptionForm.setIncludeBody(true);

		List<Subscription> subscriptions = node.getSubscriptions();

		boolean flag = true;
		for (Subscription s : subscriptions) {
			if (s.getJid().toLowerCase().equals(connection.getUser().asEntityBareJidString().toLowerCase())) {// 已订阅过
				flag = false;
				break;
			}
		}
		if (flag) {// 未订阅，开始订阅
			node.subscribe(userName + "@" + this.serverName, subscriptionForm);
		}
		return true;
	}

	/**
	 * 获取订阅的全部主题
	 *
	 * @return
	 * @throws Exception
	 */
	public List<Subscription> querySubscriptions() throws Exception {
		PubSubManager mgr = PubSubManager.getInstance(connection);
		List<Subscription> subs = mgr.getSubscriptions();
		return subs;
	}

	/**
	 * 获取订阅节点的配置信息
	 *
	 * @param nodeId
	 * @return
	 * @throws Exception
	 */
	public ConfigureForm getConfig(String nodeId) throws Exception {
		PubSubManager mgr = PubSubManager.getInstance(connection);
		LeafNode node = mgr.getNode(nodeId);
		ConfigureForm config = node.getNodeConfiguration();
		return config;
	}

	/**
	 * 获取订阅主题的全部历史消息
	 *
	 * @return
	 * @throws Exception
	 */
	public List<Item> queryHistoryMeassage() throws Exception {
		List<Item> result = new ArrayList<Item>();
		PubSubManager mgr = PubSubManager.getInstance(connection);
		List<Subscription> subs = mgr.getSubscriptions();
		if (subs != null && subs.size() > 0) {
			for (Subscription sub : subs) {
				String nodeId = sub.getNode();
				LeafNode node = mgr.getNode(nodeId);
				List<Item> list = node.getItems();
				result.addAll(list);
			}
		}

		/*
		 * for (Item item : result) { System.out.println(item.toXML()); }
		 */
		return result;
	}

	/**
	 * 获取指定主题的全部历史消息
	 *
	 * @return
	 * @throws Exception
	 */
	public List<Item> queryHistoryMeassage(String nodeId) throws Exception {
		List<Item> result = new ArrayList<Item>();
		PubSubManager mgr = PubSubManager.getInstance(connection);

		LeafNode node = mgr.getNode(nodeId);
		List<Item> list = node.getItems();
		result.addAll(list);

		/*
		 * for (Item item : result) { System.out.println(item.toXML()); }
		 */
		return result;
	}

	/**
	 * 获取指定主题指定数量的历史消息
	 *
	 * @param nodeId
	 * @param num
	 * @return
	 * @throws Exception
	 */
	public List<Item> queryHistoryMeassage(String nodeId, int num) throws Exception {
		List<Item> result = new ArrayList<Item>();
		PubSubManager mgr = PubSubManager.getInstance(connection);

		LeafNode node = mgr.getNode(nodeId);
		List<Item> list = node.getItems(num);
		result.addAll(list);

		/*
		 * for (Item item : result) { System.out.println(item.toXML()); }
		 */
		return result;
	}

	/**
	 * 向指定用户发送消息
	 *
	 * @param username
	 * @param message
	 * @throws Exception
	 */
	public void sendMessage(String username, String message) throws Exception {
		ChatManager chatManager = ChatManager.getInstanceFor(connection);
		EntityBareJid jid = JidCreate.entityBareFrom(username + "@" + serverName);
		Chat chat = chatManager.createChat(jid);
		Message newMessage = new Message();
		newMessage.setBody(message);
		chat.sendMessage(newMessage);
	}

	/**
	 * 添加聊天消息监听
	 *
	 * @param chatManagerListener
	 * @throws Exception
	 */
	public void addChatMessageListener(ChatManagerListener chatManagerListener) throws Exception {
		ChatManager chatManager = ChatManager.getInstanceFor(connection);
		chatManager.addChatListener(chatManagerListener);
	}

	/**
	 * 断开连接
	 */
	public void close() {
		connection.disconnect();
	}

}