从0实现基于Linux socket聊天室-增加数据库sqlite功能-5

一. 调整目录结构

为了方便编译，现在我们将前面文章的代码结构做如下调整。

root@ubuntu:/mnt/hgfs/code/chat# tree .
.
├── chat_client
│   ├── include
│   ├── Makefile
│   ├── obj
│   │   └── Makefile
│   └── src
│       ├── client.c
│       └── Makefile
├── chat.h
├── chat_server
│   ├── bin
│   │   └── server
│   ├── data
│   ├── include
│   ├── Makefile
│   ├── obj
│   │   └── server.o
│   └── src
│       ├── Makefile
│       └── server.c
└── gcc.sh

10 directories, 15 files
最终增加了数据的文件目录如下：

peng@ubuntu:/mnt/hgfs/code/chat-sqlite$ tree .
.
├── chat_client
│   ├── include
│   ├── Makefile
│   ├── obj
│   │   └── Makefile
│   └── src
│       ├── client.c
│       └── Makefile
├── chat.h
├── chat_server
│   ├── data
│   ├── include
│   │   └── data.h
│   ├── Makefile
│   ├── obj
│   │   └── Makefile
│   └── src
│       ├── data.c
│       ├── Makefile
│       └── server.c
├── clean.sh
├── gcc.sh
├── user.db
└── 解压密码.txt

9 directories, 17 files
clean.sh 用于清除临时文件gcc.sh 用于编译整个工程

服务端代码放置到chat_server目录下；客户端代码放置到chat_client目录下；

数据库相关代码放在chat_server/data下。

chat.h是所有客户端和服务器都会用到的头文件，所以放置在根目录下。

后续增加功能后，新增的头文件和C文件分别添加到对应工程目录的include和src目录下即可。

二、 设计数据库表

我们之前维护的所有客户端的信息是用一个全局数组，并且没有保存功能，现在我们要把所有客户端的信息全部保存到数据库中。

数据库存储的目录

chat_server/data
数据库名：

user.db
存储用户信息的表名：

user
表user格式如下：

名称	属性	说明
name	TEXT PRIMARY KEY	用户名，不能重复
passwd	TEXT NOT NULL	密码
fd	INT NOT NULL	套接字描述符：-1表示不在线，>0表示在线
regist	INT NOT NULL	用户名是否注册：-1没有注册，1注册

三、 主要功能操作的语句及函数

数据库操作最重要的就是语句，下面讲解针对不同的功能对应的实现语句

1 创建表user

CREATE TABLE IF NOT EXISTS user(name TEXT PRIMARY KEY  NOT NULL,passwd TEXT NOT NULL,fd INT NOT NULL,regist INT  NOT NULL);


2 增加一个用户

客户端发送注册请求后，服务器端注册用户信息到数据库中

数据库操作语句如下：

insert into user values('一口Linux', '123456',-1, 1)
功能函数如下：

int db_add_user(char name[],char passwd[])
功能：
增加一个用户，执行该函数前需要先判断该用户名是否存在

参数：
name：用户名
passwd：密码

返回值：
-1：失败
1：成功


3 判断用户是否在线

客户端发送登陆命令后，服务器通过该函数判断该用户是否已经登陆成功

数据库操作语句如下：

select fd from user where name='一口Linux'
功能函数如下：

int db_user_if_online(char *name,char *passwd)
功能：
判断用户是否在线，该函数主要根据fd的值来判断用户是否在线

参数：
name：用户名
passwd：密码  

返回值：
1：在线
-1：不在线
-2：用户不存在


4 判断某个用户名是否注册

用户发送注册命令，服务器需要判断该用户名是否已经被注册过

数据库操作语句如下：

select regist from user where name='一口Linux'
功能函数如下：

int db_user_if_reg(char *name)
功能：
判断某个用户名是否注册过

参数：
name：用户名

返回值：
 1：注册过
-1：没有注册过


5 判断用户名密码是否正确

用户发送登陆命令，需要判断用户名密码是否正确

数据库操作语句如下：

select * from user where name='一口Linux' and passwd='123456'
功能函数如下：

int db_user_pwd_corrct(char *name,char* passwd)
功能：
判断客户端发送的用户名密码是否正确

参数：
name：用户名
passwd：密码

返回值：
 1：正确
-1：用户名或者密码不正确


6 用户上线、下线

用户登陆成功后，或者发送下线申请，或者异常掉线，需要更新数据库的状态。

数据库操作语句如下：

UPDATE  user set fd=-1 where name='一口Linux'
fd的值是套接字描述符，下线设置为-1，上线设置为对应的>0的值

功能函数如下：

int db_user_on_off(int fd,char *name,unsigned int on_off) 
功能：
更新数据库中用户的fd字段

参数：
fd：套接字描述符
name：用户名
on_off：上线还是下线

返回值：
 1：正确
-1：失败


7. 显示在线用户

用户发送显示在线用户命令后，服务器从数据库当中查找所有在线用户，并将姓名循环发送给客户端

int db_list_online_user(int fd)


四、运行结果

编译

./gcc.sh

1.服务器启动

./server 9999
端口号设定为9999

2. 客户端注册

客户端启动

./client 127.0.0.1 9999
选择1  注册，输入用户名密码即可。

3. 用户登录

输入选项2，输入刚才注册的用户名密码，如果不一致会提示错误登录成功：

4. 注册登录其他几个用户

注册并登录新的用户111、222、333

5. 公聊

选择选项3，即进入公聊， 用户yikou向所有用户说：hello！可见所有用户均收到信息。

6. 私聊

用户yikou向用户111发送信息：由下图可知，只有用户111收到该信息，其他用户均没有收到信息。

7. 显示在线用户

8. 查看最终数据库信息

五、代码说明

为方便读者学习增加数据库和去掉数据之间的差别，

用git维护版本。

切换到没有数据库的版本，执行下面命令即可。

git reset --hard  597330ae0a183c9db8f68b7c9f60df94f8965778
要切回有数据库的版本执行下面的命令：

git reset --hard 10bfbfaf2d09ae895313273c960ecfd84663f9fd
使用数据库后，数据的存储管理更加方便，数据类型更易于扩充， 逻辑关系也更加清晰。

版权申明：内容来源网络，版权归原创者所有。除非无法确认，我们都会标明作者及出处，如有侵权烦请告知，我们会立即删除并表示歉意。谢谢!