Mina airQQ聊天 服务端篇(二)
Mina聊天服务端实现思路:在用户登录的时候。连接服务端而且验证登录用户,假设成功,则将IoSession保存到map<账号,IoSession>中,而且通知该用户的好友上线,然
后再请求好友列表;若不成功,则断开连接。
自己定义协议格式:包头+包体
包头(10字节):包头长度(short)+ 消息类型(byte)+ 内容类型(byte) + 消息命令(short)+ 包体长度(int)
包体:JSON字符串
自己定义编码解码:因为数据在网络传输过程中都是以二进制传输的,所以我们能够自己定义自己的编码解码格式。具体实现代码能够看以下的
ChatServerDecode和ChatServerEncode
数据库(chat):三张表 用户表(user)。好友表(friends),分类表(category)
/* Navicat MySQL Data Transfer Source Server : bufoon Source Server Version : 50527 Source Host : localhost:3306 Source Database : chat Target Server Type : MYSQL Target Server Version : 50527 File Encoding : 65001 Date: 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>(); @Override public void sessionCreated(IoSession session) throws Exception { logger.info("创建连接"); } @Override public void sessionOpened(IoSession session) throws Exception { logger.info("打开连接"); } @Override public 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); } } } @Override public void sessionIdle(IoSession session, IdleStatus status) throws Exception { logger.info("进入空暇"); } @Override public void exceptionCaught(IoSession session, Throwable cause) throws Exception { logger.warn("异常.", cause); session.close(true); } @Override public 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; //查找好友 //注冊 //加入好友 } } @Override public 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"); @Override public ProtocolEncoder getEncoder(IoSession session) throws Exception { return new ChatServerEncode(charset); } @Override public ProtocolDecoder getDecoder(IoSession session) throws Exception { return new ChatServerDecode(charset); } }ChatServerDecode.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.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; } @Override public 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(); } } @Override public void finishDecode(IoSession session, ProtocolDecoderOutput out) throws Exception { } @Override public 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; } @Override public 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.java
package 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.java
package 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; } //公共的connection private Connection conn=null; private Connection getConnection() throws Exception { if(conn == null) { //设置connection的url,账号,password 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; //id private int packageHeadLength; //包头长度 short 2个字节 长度为10 private int messageType; //消息类型 byte 1字节 private int contentType; //内容类型 1字节 private int messageCommand; //消息命令 short 2字节 private int packageBodyLength; //包体长度 消息int 4字节 private String packageBodyContent; //包体内容 大小 为packageBodyLength public 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; } @Override public String toString() { return "Messeage is: command=" + getMessageCommand() + ", type=" + getMessageType() + ", contentLength=" + getPackageBodyLength() + ", content=" + getPackageBodyContent(); } }