码迷,mamicode.com
首页 > 其他好文 > 详细

简易聊天程序教程(二)服务器的基本功能

时间:2016-05-12 21:02:30      阅读:183      评论:0      收藏:0      [点我收藏+]

标签:

源代码下载链接: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的表现形式,可以不用改变客户端的代码。

简易聊天程序教程(二)服务器的基本功能

标签:

原文地址:http://blog.csdn.net/sky453589103/article/details/51351803

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!