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

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