Mina airQQ聊天 服务端篇（二）
来源：互联网 发布：淘宝网今日特价女装 编辑：程序博客网 时间：2024/05/18 00:47
Mina聊天服务端实现思路：在用户登录的时候，连接服务端并且验证登录用户，如果成功，则将IoSession保存到map<账号，IoSession>中，并且通知该用户的好友上线，然
后再请求好友列表；若不成功，则断开连接。
自定义协议格式：包头+包体
包头（10字节）：包头长度（short）+ 消息类型（byte）+ 内容类型（byte） + 消息命令（short）+ 包体长度（int）
包体：JSON字符串
自定义编码解码：由于数据在网络传输过程中都是以二进制传输的，所以我们可以自定义自己的编码解码格式，详细实现代码可以看下面的
ChatServerDecode和ChatServerEncode
数据库（chat）：三张表 用户表（user），好友表（friends），分类表（category）
/*Navicat MySQL Data TransferSource Server : bufoonSource Server Version : 50527Source Host : localhost:3306Source Database : chatTarget Server Type : MYSQLTarget Server Version : 50527File Encoding : 65001Date: 2014-06-29 23:30:28*/SET FOREIGN_KEY_CHECKS=0;-- ------------------------------ Table structure for category-- ----------------------------DROP TABLE IF EXISTS `category`;CREATE TABLE `category` ( `id` int(11) NOT NULL AUTO_INCREMENT, `user_id` int(11) DEFAULT NULL, `name` varchar(20) COLLATE utf8_unicode_ci DEFAULT NULL, `create_time` datetime DEFAULT NULL, PRIMARY KEY (`id`), KEY `FK_CATEGOFY_USER_ID` (`user_id`), CONSTRAINT `FK_CATEGOFY_USER_ID` FOREIGN KEY (`user_id`) REFERENCES `user` (`id`) ON DELETE CASCADE) ENGINE=InnoDB AUTO_INCREMENT=6 DEFAULT CHARSET=utf8 COLLATE=utf8_unicode_ci;-- ------------------------------ Records of category-- ----------------------------INSERT INTO `category` VALUES ('1', '1', '我的好友', '2014-06-29 19:00:25');INSERT INTO `category` VALUES ('3', '2', '我的好友', '2014-06-29 19:00:55');INSERT INTO `category` VALUES ('4', '3', '我的好友', '2014-06-29 19:01:00');INSERT INTO `category` VALUES ('5', '1', '同学', '2014-06-29 20:39:04');-- ------------------------------ Table structure for friends-- ----------------------------DROP TABLE IF EXISTS `friends`;CREATE TABLE `friends` ( `id` int(11) NOT NULL AUTO_INCREMENT, `user_id` int(11) DEFAULT NULL, `friend_id` int(11) DEFAULT NULL, `category_id` int(11) DEFAULT NULL, `create_time` datetime DEFAULT NULL, PRIMARY KEY (`id`), KEY `FK_FRIENDS_USER_ID` (`user_id`), KEY `FK_FRIENDS_CATEGORY_ID` (`category_id`), KEY `FK_FRIENDS_FUSER_ID` (`friend_id`), CONSTRAINT `FK_FRIENDS_CATEGORY_ID` FOREIGN KEY (`category_id`) REFERENCES `category` (`id`) ON DELETE CASCADE, CONSTRAINT `FK_FRIENDS_FUSER_ID` FOREIGN KEY (`friend_id`) REFERENCES `user` (`id`) ON DELETE CASCADE, CONSTRAINT `FK_FRIENDS_USER_ID` FOREIGN KEY (`user_id`) REFERENCES `user` (`id`) ON DELETE CASCADE) ENGINE=InnoDB AUTO_INCREMENT=5 DEFAULT CHARSET=utf8 COLLATE=utf8_unicode_ci;-- ------------------------------ Records of friends-- ----------------------------INSERT INTO `friends` VALUES ('1', '1', '2', '1', '2014-06-21 23:35:16');INSERT INTO `friends` VALUES ('2', '1', '3', '1', '2014-06-21 23:35:22');INSERT INTO `friends` VALUES ('3', '2', '1', '3', '2014-06-22 02:09:24');INSERT INTO `friends` VALUES ('4', '3', '1', '4', '2014-06-22 02:09:29');-- ------------------------------ Table structure for user-- ----------------------------DROP TABLE IF EXISTS `user`;CREATE TABLE `user` ( `id` int(11) NOT NULL AUTO_INCREMENT, `name` varchar(20) COLLATE utf8_unicode_ci DEFAULT NULL, `user_num` varchar(15) COLLATE utf8_unicode_ci DEFAULT NULL, `password` varchar(20) COLLATE utf8_unicode_ci DEFAULT NULL, `regist_time` datetime DEFAULT NULL, `sex` varchar(2) COLLATE utf8_unicode_ci DEFAULT NULL, `signature` varchar(300) COLLATE utf8_unicode_ci DEFAULT NULL, `head_pic_path` varchar(200) COLLATE utf8_unicode_ci DEFAULT NULL, `is_online` varchar(1) COLLATE utf8_unicode_ci DEFAULT NULL, PRIMARY KEY (`id`)) ENGINE=InnoDB AUTO_INCREMENT=4 DEFAULT CHARSET=utf8 COLLATE=utf8_unicode_ci;-- ------------------------------ Records of user-- ----------------------------INSERT INTO `user` VALUES ('1', '张三', '12345', '12345', '2014-06-20 23:32:26', '男', null, null, '0');INSERT INTO `user` VALUES ('2', '李四', '123456', '123456', '2014-06-20 23:32:31', '女', null, null, '1');INSERT INTO `user` VALUES ('3', '王二', '1234567', '1234567', '2014-06-21 11:29:41', '男', null, null, '1');项目目录结构：
需要的jar包：
ChatServer.java
package com.bufoon.main;import java.io.IOException;import java.net.InetSocketAddress;import org.apache.mina.core.filterchain.DefaultIoFilterChainBuilder;import org.apache.mina.core.session.IdleStatus;import org.apache.mina.filter.codec.ProtocolCodecFilter;import org.apache.mina.filter.keepalive.KeepAliveFilter;import org.apache.mina.filter.keepalive.KeepAliveMessageFactory;import org.apache.mina.filter.keepalive.KeepAliveRequestTimeoutHandler;import org.apache.mina.filter.logging.LogLevel;import org.apache.mina.filter.logging.LoggingFilter;import org.apache.mina.transport.socket.SocketAcceptor;import org.apache.mina.transport.socket.nio.NioSocketAcceptor;import com.bufoon.codeFactory.ChatServerCodecFactory;import com.bufoon.handle.ChatServerHandle;import com.bufoon.handle.KeepAliveMessageFactoryImpl;import com.bufoon.handle.KeepAliveRequestTimeoutHandlerImpl;public class ChatServer {private static final int PORT = 7073; //30秒后超时 private static final int IDELTIMEOUT = 30; //15秒发送一次心跳包 private static final int HEARTBEATRATE = 15; private static SocketAcceptor acceptor; public static SocketAcceptor getAcceptor(){ if(null==acceptor){ // 创建非阻塞的server端的Socket连接 acceptor = new NioSocketAcceptor(); } return acceptor; } public static boolean serverStart() { DefaultIoFilterChainBuilder filterChain = getAcceptor().getFilterChain(); // 添加编码过滤器 处理乱码、编码问题 filterChain.addLast("codec", new ProtocolCodecFilter(new ChatServerCodecFactory())); LoggingFilter loggingFilter = new LoggingFilter(); loggingFilter.setMessageReceivedLogLevel(LogLevel.INFO); loggingFilter.setMessageSentLogLevel(LogLevel.INFO); // 添加日志过滤器 filterChain.addLast("loger", loggingFilter); // 设置核心消息业务处理器 getAcceptor().setHandler(new ChatServerHandle()); KeepAliveMessageFactory heartBeatFactory = new KeepAliveMessageFactoryImpl(); KeepAliveRequestTimeoutHandler heartBeatHandler = new KeepAliveRequestTimeoutHandlerImpl(); KeepAliveFilter heartBeat = new KeepAliveFilter(heartBeatFactory,IdleStatus.BOTH_IDLE, heartBeatHandler); // 是否回发 heartBeat.setForwardEvent(true); // 发送频率 heartBeat.setRequestInterval(HEARTBEATRATE); // getAcceptor().getFilterChain().addLast("heartbeat", heartBeat); getAcceptor().getSessionConfig().setBothIdleTime(30); // 设置session配置，30秒内无操作进入空闲状态 getAcceptor().getSessionConfig().setIdleTime(IdleStatus.BOTH_IDLE, IDELTIMEOUT); try { // 绑定端口7033 getAcceptor().bind(new InetSocketAddress(PORT)); return true; } catch (IOException e) { e.printStackTrace(); } return false; } public static void main(String[] args) { ChatServer.serverStart();System.out.println("服务器启动...");}}ChatServerHandle.java
package com.bufoon.handle;import java.sql.ResultSet;import java.util.HashMap;import java.util.List;import java.util.Map;import net.sf.json.JSONArray;import net.sf.json.JSONObject;import org.apache.mina.core.service.IoHandlerAdapter;import org.apache.mina.core.session.IdleStatus;import org.apache.mina.core.session.IoSession;import org.apache.mina.example.chat.ChatProtocolHandler;import org.slf4j.Logger;import org.slf4j.LoggerFactory;import com.bufoon.model.CategoryModel;import com.bufoon.model.FriendsModel;import com.bufoon.model.LoginModel;import com.bufoon.model.PackageHead;import com.bufoon.model.SendModel;import com.bufoon.model.UserModel;import com.bufoon.util.BaseDAO;import com.bufoon.util.DBUtil;import com.bufoon.util.MessageType;public class ChatServerHandle extends IoHandlerAdapter {private final static Logger logger = LoggerFactory.getLogger(ChatProtocolHandler.class);public static Map<String, IoSession> sessionMap = new HashMap<String, IoSession>();@Overridepublic void sessionCreated(IoSession session) throws Exception {logger.info("创建连接");}@Overridepublic void sessionOpened(IoSession session) throws Exception {logger.info("打开连接");}@Overridepublic void sessionClosed(IoSession session) throws Exception {logger.info("关闭连接");String userNum = (String) session.getAttribute("userNum");String sql = "update user set is_online=1 where user_num like'" + userNum + "'";DBUtil.getInstance().executeUpdate(sql);sessionMap.remove(userNum);//修改下线，通知String userSql = "select * from user where user_num like '" + userNum + "'";UserModel userModel = BaseDAO.getInstance().findUserModel(userSql);String friendListSql = "select * from friends where user_id=" + userModel.getId();List<FriendsModel> list = BaseDAO.getInstance().findFriendsList(friendListSql);for (FriendsModel friendsModel : list) {String fUserSql = "select * from user where id=" + friendsModel.getFriendId();UserModel userModel2 = BaseDAO.getInstance().findUserModel(fUserSql);IoSession is = sessionMap.get(userModel2.getUserNum());if (is != null) {PackageHead ph = new PackageHead();Map<String, Object> object = new HashMap<String, Object>(); object.put("userNum", userModel2.getUserNum()); object.put("status", 1); //下线 String content = JSONObject.fromObject(object).toString(); ph.setPackageHeadLength(10); ph.setMessageCommand(MessageType.USER_ON_OFF_LINE_NOTICE); ph.setContentType(MessageType.CONTENT_TYPE_OBJECT); ph.setMessageType(MessageType.MESSAGE_TYPE_PUSH); ph.setPackageBodyLength(content.getBytes().length); ph.setPackageBodyContent(content);is.write(ph);}}}@Overridepublic void sessionIdle(IoSession session, IdleStatus status)throws Exception {logger.info("进入空闲");}@Overridepublic void exceptionCaught(IoSession session, Throwable cause)throws Exception {logger.warn("异常.", cause); session.close(true);}@Overridepublic void messageReceived(IoSession session, Object message)throws Exception {System.out.println(message);PackageHead ph = (PackageHead) message;System.out.println("还有没有这个session：" + sessionMap.size());switch (ph.getMessageCommand()) {case MessageType.LOGIN_VERIFY: //登录请求LoginModel loginModel = (LoginModel)JSONObject.toBean(JSONObject.fromObject(ph.getPackageBodyContent()), LoginModel.class);String username = loginModel.getUsername();IoSession is = sessionMap.get(username);Map<String, Object> map = new HashMap<String, Object>();if (!DBUtil.getInstance().isExit("user", "user_num", username)) {System.out.println("用户不存在");map.put("status", 1);map.put("info", "用户不存在");map.put("username", username);} else if (is != null && is.isConnected()) {System.out.println("用户已登录");map.put("status", 2);map.put("info", "用户已登录");map.put("username", username);} else if (!DBUtil.getInstance().isExit("user", new String[]{"user_num", "password"}, new Object[]{username, loginModel.getPassword()})) {System.out.println("用户密码错误");map.put("status", 3);map.put("info", "用户密码错误");map.put("username", username);} else if (DBUtil.getInstance().isExit("user", new String[]{"user_num", "password"}, new Object[]{username, loginModel.getPassword()})) {String sql = "select * from user where user_num like '" + username + "'";ResultSet rs = DBUtil.getInstance().executeQuery(sql);UserModel vo = null;while (rs.next()) {vo = new UserModel();vo.setId(rs.getInt("id"));vo.setUsername(rs.getString("name"));vo.setUserNum(rs.getString("user_num"));vo.setPassword(rs.getString("password"));vo.setSex(rs.getString("sex"));vo.setSignature(rs.getString("signature"));vo.setIsOnline(rs.getInt("is_online"));break;} if (vo != null) {map.put("username", vo.getUsername());}map.put("status", 0);map.put("info", "成功");map.put("userNum", username);map.put("userVO", vo);session.setAttribute("userNum", username);sessionMap.put(username, session);}String onLinesql = "update user set is_online=0 where user_num like'" + username + "'";DBUtil.getInstance().executeUpdate(onLinesql);String content = JSONObject.fromObject(map).toString(); ph.setMessageCommand(MessageType.LOGIN_VERIFY_ACK); ph.setContentType(MessageType.CONTENT_TYPE_OBJECT); ph.setMessageType(MessageType.MESSAGE_TYPE_REQUEST); ph.setPackageBodyLength(content.getBytes().length); ph.setPackageBodyContent(content); session.write(ph); String friendSql = "select * from user where user_num like '" + username + "'"; ResultSet rs1 = DBUtil.getInstance().executeQuery(friendSql); Map<String, Object> object = new HashMap<String, Object>(); object.put("userNum", username); object.put("status", 0); content = JSONObject.fromObject(object).toString(); ph.setMessageCommand(MessageType.USER_ON_OFF_LINE_NOTICE); ph.setContentType(MessageType.CONTENT_TYPE_OBJECT); ph.setMessageType(MessageType.MESSAGE_TYPE_PUSH); ph.setPackageBodyLength(content.getBytes().length); ph.setPackageBodyContent(content); while (rs1.next()) { String sql1 = "select * from friends where user_id=" + rs1.getInt("id"); ResultSet rs2 = DBUtil.getInstance().executeQuery(sql1); while(rs2.next()){ String sql2 = "select * from user where id=" + rs2.getInt("friend_id"); ResultSet rs3 = DBUtil.getInstance().executeQuery(sql2); while(rs3.next()){ IoSession iso = sessionMap.get(rs3.getString("user_num")); if(iso != null){ iso.write(ph); } } } }break;case MessageType.FRIEND_LIST: //好友列表请求JSONObject obj = JSONObject.fromObject(ph.getPackageBodyContent());String userId = obj.getString("userId");String friendListSql = "select * from friends where user_id=" + userId;List<FriendsModel> friendList = BaseDAO.getInstance().findFriendsList(friendListSql);String categorySql = "select * from category where user_id=" + userId;List<CategoryModel> categoryList = BaseDAO.getInstance().findCategoryList(categorySql);for (CategoryModel categoryModel : categoryList) {for (FriendsModel friendModel : friendList) {if (categoryModel.getId() == friendModel.getCategoryId()) {String userSql = "select * from user where id=" + friendModel.getFriendId();categoryModel.getList().add(BaseDAO.getInstance().findUserModel(userSql));}}}String friends = JSONArray.fromObject(categoryList).toString();System.out.println("frends:" + friends); ph.setMessageCommand(MessageType.FRIEND_LIST_ACK); ph.setContentType(MessageType.CONTENT_TYPE_ARRAY); ph.setMessageType(MessageType.MESSAGE_TYPE_REQUEST); ph.setPackageBodyLength(friends.getBytes().length); ph.setPackageBodyContent(friends); session.write(ph);break;case MessageType.SEND_MESSAGE: //消息发送SendModel sendModel = (SendModel)JSONObject.toBean(JSONObject.fromObject(ph.getPackageBodyContent()), SendModel.class);ph.setMessageType(MessageType.SEND_MESSAGE_ACK);session.write(ph);ph.setMessageCommand(MessageType.SEND_MESSAGE_ACK_NOTICE);String sendStr = JSONObject.fromObject(sendModel).toString();ph.setPackageBodyLength(sendStr.getBytes().length);ph.setPackageBodyContent(sendStr);ph.setMessageType(MessageType.MESSAGE_TYPE_PUSH);ph.setContentType(MessageType.CONTENT_TYPE_OBJECT);sessionMap.get(sendModel.getReceiverNum()).write(ph);break;//查找好友//注册//添加好友}}@Overridepublic void messageSent(IoSession session, Object message) throws Exception {logger.info("发送消息: " + message);}}ChatServerCodecFactory.java
package com.bufoon.codeFactory;import java.nio.charset.Charset;import org.apache.mina.core.session.IoSession;import org.apache.mina.filter.codec.ProtocolCodecFactory;import org.apache.mina.filter.codec.ProtocolDecoder;import org.apache.mina.filter.codec.ProtocolEncoder;public class ChatServerCodecFactory implements ProtocolCodecFactory{private static final Charset charset = Charset.forName("UTF-8");@Overridepublic ProtocolEncoder getEncoder(IoSession session) throws Exception {return new ChatServerEncode(charset);}@Overridepublic ProtocolDecoder getDecoder(IoSession session) throws Exception {return new ChatServerDecode(charset);}}ChatServerDecode.javapackage com.bufoon.codeFactory;import java.nio.ByteOrder;import java.nio.charset.Charset;import org.apache.mina.core.buffer.IoBuffer;import org.apache.mina.core.session.AttributeKey;import org.apache.mina.core.session.IoSession;import org.apache.mina.filter.codec.ProtocolDecoder;import org.apache.mina.filter.codec.ProtocolDecoderOutput;import com.bufoon.model.PackageHead;public class ChatServerDecode implements ProtocolDecoder {private final AttributeKey CONTEXT = new AttributeKey(getClass(), "context");private final Charset charset;private int maxPackLength = 100;public ChatServerDecode() {this(Charset.defaultCharset());}public ChatServerDecode(Charset charset) {this.charset = charset;}public int getMaxLineLength() {return maxPackLength;}public void setMaxLineLength(int maxLineLength) {if (maxLineLength <= 0) {throw new IllegalArgumentException("maxLineLength: "+ maxLineLength);}this.maxPackLength = maxLineLength;}private ChatContext getContext(IoSession session) {ChatContext ctx;ctx = (ChatContext) session.getAttribute(CONTEXT);if (ctx == null) {ctx = new ChatContext(charset);session.setAttribute(CONTEXT, ctx);}return ctx;}@Overridepublic void decode(IoSession session, IoBuffer in, ProtocolDecoderOutput out)throws Exception {final int packHeadLength = 10;// 先获取上次的处理上下文，其中可能有未处理完的数据in.order(ByteOrder.LITTLE_ENDIAN);ChatContext ctx = getContext(session);// 先把当前buffer中的数据追加到Context的buffer当中ctx.append(in);// 把position指向0位置，把limit指向原来的position位置IoBuffer buf = ctx.getBuffer();buf.flip();// 然后按数据包的协议进行读取if (buf.remaining() >= packHeadLength) {buf.mark();// 读取消息头部分PackageHead message = new PackageHead();message.setPackageHeadLength(buf.getShort());message.setMessageType(buf.get());message.setContentType(buf.get());message.setMessageCommand(buf.getShort());int bodyLen = buf.getInt();message.setPackageBodyLength(bodyLen);// 读取正常的消息包，并写入输出流中，以便IoHandler进行处理if (bodyLen > 0 && buf.remaining() >= bodyLen) { message.setPackageBodyContent(buf.getString(bodyLen, charset.newDecoder()));} else {//buf.clear();}out.write(message);}if (buf.hasRemaining()) {// 将数据移到buffer的最前面IoBuffer temp = IoBuffer.allocate(maxPackLength).setAutoExpand(true);temp.put(buf);temp.flip();buf.clear();buf.put(temp);} else {// 如果数据已经处理完毕，进行清空buf.clear();}}@Overridepublic void finishDecode(IoSession session, ProtocolDecoderOutput out)throws Exception {}@Overridepublic void dispose(IoSession session) throws Exception {ChatContext ctx = (ChatContext) session.getAttribute(CONTEXT);if (ctx != null) {session.removeAttribute(CONTEXT);}}}ChatServerEncode.java
package com.bufoon.codeFactory;import java.nio.ByteOrder;import java.nio.charset.Charset;import org.apache.mina.core.buffer.IoBuffer;import org.apache.mina.core.session.IoSession;import org.apache.mina.filter.codec.ProtocolEncoderAdapter;import org.apache.mina.filter.codec.ProtocolEncoderOutput;import com.bufoon.model.PackageHead;public class ChatServerEncode extends ProtocolEncoderAdapter {private Charset charset = null;public ChatServerEncode(Charset charset) {this.charset = charset;}@Overridepublic void encode(IoSession session, Object message,ProtocolEncoderOutput out) throws Exception {if (message instanceof PackageHead) {PackageHead ph = (PackageHead) message;IoBuffer buf = IoBuffer.allocate(ph.getPackageHeadLength() + ph.getPackageBodyLength());buf.order(ByteOrder.LITTLE_ENDIAN);//buf.setAutoExpand(true);buf.putShort((short) ph.getPackageHeadLength());buf.put((byte) ph.getMessageType());buf.put((byte) ph.getContentType());buf.putShort((short) ph.getMessageCommand());buf.putInt((int) ph.getPackageBodyLength());if (ph.getPackageBodyLength() > 0) {buf.putString(ph.getPackageBodyContent(), charset.newEncoder());}buf.flip();out.write(buf);out.flush();buf.free();}}}ChatContext.javapackage com.bufoon.codeFactory;import java.nio.ByteOrder;import java.nio.charset.Charset;import java.nio.charset.CharsetDecoder;import org.apache.mina.core.buffer.IoBuffer;public class ChatContext {private final CharsetDecoder decoder;private IoBuffer buf;private int matchCount = 0;private int overflowPosition = 0;public ChatContext(Charset charset) {decoder = charset.newDecoder();buf = IoBuffer.allocate(80).setAutoExpand(true);buf.order(ByteOrder.LITTLE_ENDIAN);}public CharsetDecoder getDecoder() {return decoder;}public IoBuffer getBuffer() {return buf;}public int getOverflowPosition() {return overflowPosition;}public int getMatchCount() {return matchCount;}public void setMatchCount(int matchCount) {this.matchCount = matchCount;}public void reset() {overflowPosition = 0;matchCount = 0;decoder.reset();}public void append(IoBuffer in) {getBuffer().put(in);}}MessageType.java
package com.bufoon.util;public class MessageType {/**登录验证请求消息类型**/public final static int LOGIN_VERIFY = 0x0000;/**登录验证响应消息类型**/public final static int LOGIN_VERIFY_ACK = 0x0001;/**心跳请求消息类型**/public final static int HEART_BEAT = 0x0002;/**心跳响应消息类型**/public final static int HEART_BEAT_ACK = 0x0003;/**好友列表请求消息类型**/public final static int FRIEND_LIST = 0x0004;/**好友列表响应消息类型**/public final static int FRIEND_LIST_ACK = 0x0005;/**发送消息请求**/public final static int SEND_MESSAGE = 0x0006;/**发送消息响应**/public final static int SEND_MESSAGE_ACK = 0x0007;/**发送消息通知响应**/public final static int SEND_MESSAGE_ACK_NOTICE = 0x1000;/**通知用户上下线**/public final static int USER_ON_OFF_LINE_NOTICE = 0X1001;/**包头大小**/public final static int HEAD_LENGTH = 10;/**返回的消息类型 0服务端推送**/public final static int MESSAGE_TYPE_PUSH = 0;/**返回的消息类型 1请求响应**/public final static int MESSAGE_TYPE_REQUEST = 1;/**返回的内容类型 0 JsonObject**/public final static int CONTENT_TYPE_OBJECT = 0;/**返回的内容类型 1 JsonArray**/public final static int CONTENT_TYPE_ARRAY = 1;}DBUtil.javapackage com.bufoon.util;import java.sql.Connection;import java.sql.DriverManager;import java.sql.PreparedStatement;import java.sql.ResultSet;import java.sql.SQLException;import java.sql.Statement;/** * 数据库 连接类 * @author AllenYe * */public class DBUtil {private static final DBUtil uniqueInstance = new DBUtil(); private DBUtil(){//加载mysql-jdbc桥接器：try{Class.forName("com.mysql.jdbc.Driver");}catch(ClassNotFoundException e){}}public static DBUtil getInstance() { return uniqueInstance; }//公共的connectionprivate Connection conn=null;private Connection getConnection() throws Exception{if(conn == null){//设置connection的url，账号，密码conn=DriverManager.getConnection("jdbc:mysql://localhost:3306/chat?useUnicode=true&characterEncoding=UTF-8","root","root");}return conn;}public ResultSet executeQuery(String sql){try {Statement statement = getConnection().createStatement(ResultSet.TYPE_SCROLL_SENSITIVE,ResultSet.CONCUR_UPDATABLE);ResultSet rs = statement.executeQuery(sql);return rs;} catch (SQLException e) {e.printStackTrace();} catch (Exception e) {e.printStackTrace();}return null;}public int executeUpdate(String sql){try {PreparedStatement statement = getConnection().prepareStatement(sql, ResultSet.TYPE_SCROLL_SENSITIVE, ResultSet.CONCUR_UPDATABLE);int result = statement.executeUpdate();return result;} catch (SQLException e) {e.printStackTrace();} catch (Exception e) {e.printStackTrace();}return -1;}public boolean isExit(String table, String column, String value){String sql = "select * from " + table + " where " + column + " like '" + value + "'";ResultSet rs = this.executeQuery(sql);boolean flag = false;try {while (rs.next()) {flag = true;break;}} catch (SQLException e) {e.printStackTrace();}return flag;}public boolean isExit(String table, String column[], Object value[]){StringBuffer sb = new StringBuffer("select * from " + table + " where 1=1");for(int i = 0; i < column.length; i++){sb.append(" and " + column[i] + " like '" + value[i] + "'");}ResultSet rs = this.executeQuery(sb.toString());boolean flag = false;try {while (rs.next()) {flag = true;break;}} catch (SQLException e) {e.printStackTrace();}return flag;}}BaseDAO.java
package com.bufoon.util;import java.sql.ResultSet;import java.sql.SQLException;import java.util.ArrayList;import java.util.List;import com.bufoon.model.CategoryModel;import com.bufoon.model.FriendsModel;import com.bufoon.model.UserModel;public class BaseDAO {private static final BaseDAO uniqueInstance = new BaseDAO(); private BaseDAO(){}public static BaseDAO getInstance() { return uniqueInstance; }public UserModel findUserModel(String sql){UserModel vo = null;try {ResultSet rs = DBUtil.getInstance().executeQuery(sql);while(rs.next()){vo = new UserModel();vo.setId(rs.getInt("id"));vo.setUsername(rs.getString("name"));vo.setUserNum(rs.getString("user_num"));vo.setPassword(rs.getString("password"));vo.setIsOnline(rs.getInt("is_online"));vo.setSignature(rs.getString("signature"));vo.setSex(rs.getString("sex"));}} catch (SQLException e) {e.printStackTrace();}return vo;}public List<FriendsModel> findFriendsList(String sql){List<FriendsModel> list = new ArrayList<FriendsModel>();try {ResultSet rs = DBUtil.getInstance().executeQuery(sql);FriendsModel vo = null;while(rs.next()){vo = new FriendsModel();vo.setId(rs.getInt("id"));vo.setUserId(rs.getInt("user_id"));vo.setFriendId(rs.getInt("friend_id"));vo.setCategoryId(rs.getInt("category_id"));vo.setCreateTime(Util.formatTime(rs.getDate("create_time")));list.add(vo);}} catch (SQLException e) {e.printStackTrace();}return list;}public List<CategoryModel> findCategoryList(String sql){List<CategoryModel> list = new ArrayList<CategoryModel>();CategoryModel vo = null;ResultSet rs = DBUtil.getInstance().executeQuery(sql);try {while(rs.next()){vo = new CategoryModel();vo.setId(rs.getInt("id"));vo.setUserId(rs.getInt("user_id"));vo.setName(rs.getString("name"));vo.setCreateTime(Util.formatTime(rs.getDate("create_time")));list.add(vo);}} catch (SQLException e) {e.printStackTrace();}return list;}}PackageHead.java
package com.bufoon.model;import java.io.Serializable;public class PackageHead implements Serializable {private static final long serialVersionUID = 3965541808116510722L;private int id; //idprivate int packageHeadLength; //包头长度 short 2个字节 长度为10private int messageType; //消息类型 byte 1字节private int contentType; //内容类型 1字节 private int messageCommand; //消息命令 short 2字节private int packageBodyLength; //包体长度 消息int 4字节private String packageBodyContent; //包体内容 大小 为packageBodyLengthpublic int getId() {return id;}public void setId(int id) {this.id = id;}public int getPackageHeadLength() {return packageHeadLength;}public void setPackageHeadLength(int packageHeadLength) {this.packageHeadLength = packageHeadLength;}public int getMessageType() {return messageType;}public void setMessageType(int messageType) {this.messageType = messageType;}public int getContentType() {return contentType;}public void setContentType(int contentType) {this.contentType = contentType;}public int getMessageCommand() {return messageCommand;}public void setMessageCommand(int messageCommand) {this.messageCommand = messageCommand;}public int getPackageBodyLength() {return packageBodyLength;}public void setPackageBodyLength(int packageBodyLength) {this.packageBodyLength = packageBodyLength;}public String getPackageBodyContent() {return packageBodyContent;}public void setPackageBodyContent(String packageBodyContent) {this.packageBodyContent = packageBodyContent;}@Overridepublic String toString() {return "Messeage is: command=" + getMessageCommand() + ", type=" + getMessageType() + ", contentLength=" + getPackageBodyLength() + ", content=" + getPackageBodyContent();}} 0 0
- Mina airQQ聊天 服务端篇（二）
- Mina airQQ聊天 服务端篇（二)
- Mina airQQ聊天开门见山篇（一）
- Mina airQQ聊天 客户端篇（三）
- Mina框架实现最简单的服务端（二）
- IM聊天进阶(三)IM>openfire>mina>mina搭建服务端和客户端 实现简单点对点聊天
- Mina学习（一）：mina实现简单服务端与客户端
- Apache Mina（二）
- Apache Mina（二）
- Mina学习（二）
- Apache Mina（二）
- Apache Mina（二）
- Apache Mina（二）
- Apache Mina（二）
- Mina初识（二）
- Apache Mina（二）
- Mina 简单案例（客户端与服务端）
- Mina入门教程（二）----Spring4 集成Mina
- DB2恢复一例 SQL0928N
- 关于numpy的安装
- 程序猿笑话，放松一下。
- 第三章 【创建型设计模式】
- DB2 SQL0902C 一例
- Mina airQQ聊天 服务端篇（二）
- Light OJ 1174
- javasrcipt函数
- XML中的方法总结
- ArcGIS10.2安装中需要注意的问题
- 对大学的思考：一流大学应培养什么样的精英
- GWT里面的CSS样式单
- DB2 SQL20321N 一例
- JAVA学习之Object类-equals方法的覆写
