源代码下载链接:http://download.csdn.net/detail/sky453589103/9514686
如果有什么问题,欢迎留言。
我设计的时候是先从服务器端开始设计的,服务器端先给出接口,然后客户端针对接口编程。
在说服务器的代码之前先说说数据库的设计吧。
数据的设计是简单的,只有两个表:
create database SimpleChat; CREATE TABLE IF NOT EXISTS Account ( username varchar(50) NOT NULL, userpassword varchar(20) NOT NULL, sex ENUM('male', 'female') NOT NULL, status ENUM('on', 'off', 'leave', 'busy') NOT NULL, constraint A_PK primary key(username) ); CREATE TABLE IF NOT EXISTS Friend( user_1 varchar(50) NOT NULL, user_2 varchar(50) NOT NULL, constraint F_PK primary key (user_1, user_2), constraint F_FK1 foreign key (user_1) references Account(username), constraint F_FK2 foreign key (user_2) references Account(username) );
只有这两个表,表中的字段还不多。很方便学习。关于怎么用java连接数据库的,请看代码张的Main.java文件和SimpleChatServer.java中的SimpleChatServer类的构造函数。
在源代码中,服务器端的项目名称是SimpleChatServer,
服务器端功能模块包括下面几个:
1 注册模块;
2 登陆模块;
3 注销模块;
4 发送模块;
5 添加朋友模块;
6 删除朋友模块;
7 获取朋友列表模块;
8 输入检测模块;
上面这几个模块基本对应着一个函数,除了输入检测。输入检测只是简单的检测输入的用户名或者密码中有没空格,分号等字符,算是简单的拦截一下SQL注入吧。
值得注意的是,我在SimpleChatServer中设置了一个成员变量userBuf,这个变量用来存放的是已经登陆成功的用户已经对用的Socket,因为不可能每次需要查看一个用户是否登陆就要去查数据库,这样会影响效率。但这个做法也会有一个缺点,就是会增大程序需要的内存。
按惯例也是拿出一个函数来说一说就可以了。大部分的实现是还很相似的,只是细节不相同。下面就注册这个函数来说一下吧。
public ResponseMessage SignUp(String name, String password, Sex sex) { String sexValue = ""; String statusValue = "off"; if (Sex.MALE == sex) { sexValue = "male"; } else if (Sex.FEMALE == sex) { sexValue = "female"; } ResponseMessage response = new ResponseMessage(0, "Success"); try { if (username.matches("[; ,]") || password.matches("[; ,]")) { response.setCode(1); response.setDescription("username or password has illigal charactor!"); return response; } final String sql = "insert into Account values(\'" + name + "\', \'" + password + "\', \'" + sexValue + "\', \'" + statusValue + "\');"; Statement statement = conn.createStatement(); statement.executeUpdate(sql); } catch (SQLException e) { response.setCode(1); response.setDescription(e.getMessage()); } return response; }
需要重点关注的其实也就下面三句:
final String sql = "insert into Account values(\'" + name + "\', \'" + password + "\', \'" + sexValue + "\', \'" + statusValue + "\');"; Statement statement = conn.createStatement(); statement.executeUpdate(sql);
第一条语句是设定一条数据库的语法。
第二条是在已有的数据连接上创建一个Statement,通过Statement来执行sql语句。
第三条语句是执行sql语句,因为是insert,所有执行的是executeUpdate方法。如果想要查看类似select语句的例子,那请看获取朋友列表的函数。
最后要说明的是,Server提供了添加在线用户的接口和删除在线用户的接口。
public void AddUser(String name, Socket s) { userBuf.put(name, s); } public void RemoveUser(String name) { userBuf.remove(name); }
其实这两个函数只是吧userBuf中的响应的键值对给删除了。不会访问数据库。写这两个方法的原因是为了以后如果要改变userBuf的表现形式,可以不用改变客户端的代码。