Qt：基於QTcpSocket，QTcpServer編寫的多人網絡聊天室(TCP/IP協議) —— 服務器源碼

編譯環境：Ubuntu18.04，Qt5.11.2

QTcpSockServer.pro

TEMPLATE = app
CONFIG += console c++11
CONFIG += console precompile_header
CONFIG -= app_bundle
CONFIG += qt
QT += core gui network sql
greaterThan(QT_MAJOR_VERSION, 4): QT += widgets

HEADERS  += \
    CommonInc.h \
    Database.h \
    Structure.h \
    TcpServer.h \
    ServerSocket.h

SOURCES += \
        TcpServer.cpp \
        Database.cpp \
        main.cpp \
        ServerSocket.cpp

main.cpp

#include <QCoreApplication>

#include "CommonInc.h"
#include "TcpServer.h"

int main(int argc, char *argv[])
{
    QCoreApplication a(argc, argv);

    // 啓動服務核心
    TcpServer *server = new TcpServer();

    if (!server->listen(QHostAddress::Any, Constant::LISTEN_PORT))
    {
        QString strErr(QObject::tr("critical: Satrting server listen fail, error string:: "));
        strErr += server->errorString();
        qDebug("%s", strErr.toStdString().c_str());
        exit(EXIT_FAILURE);
    }


    int ret = a.exec();
    delete server;

    return ret;
}

CommonInc.h

/*********************************************
File Name： CommonInc.h
Description： 封裝公用頭文件
********************************************/
#ifndef COMMONINC_H
#define COMMONINC_H

#include <QObject>
#include <QTcpServer>
#include <QTcpSocket>
#include <QMap>
#include <QTextCodec>
#include <QString>
#include <QFont>
#include <QColor>
#include <QDateTime>
#include <QDataStream>
#include <QSqlDatabase>
#include <QSqlQuery>
#include <QSqlError>
#include <QDebug>
#include <QVector>
#include <QVariant>

#define TEST_SWITCH //測試開關

#endif // COMMONINC_H

structure.h

/*********************************************
File Name： Structure.h
Description： 封裝結構體和常量
********************************************/
#ifndef STRUCTURE_H
#define STRUCTURE_H

#include "CommonInc.h"

class ServerSocket;

//如果沒有合適的翻譯字符串，則返回QString::fromUtf8(sourceText)
static const QString GROUP_NAME_DEFAULT = QString(QObject::tr("New Friend"));

namespace Constant //常量
{
    const unsigned int LOGIN_TIMEOUT = 15;        //(seconds)登錄超時時間

    const unsigned int LISTEN_PORT = 1234;        //TCP監聽端口

    const QString DB_CLI_DRIVER = "QMYSQL";       //客戶端數據庫驅動名稱
    const QString DB_CLI_USERNAME = "root";       //客戶端數據庫用戶名
    const QString DB_CLI_PWD = "";                //客戶端數據庫密碼
    const QString DB_CLI_HOSTNAME = "localhost";  //客戶端數據庫主機名稱(IP)
    const QString DB_CLI_DATABASE = "clientDB";   //客戶端數據庫名稱

    const QString DB_SER_DRIVER = "QMYSQL";       //服務讀數據庫驅動名稱
    const QString DB_SER_USERNAME = "root";       //服務端數據庫用戶名
    const QString DB_SER_PWD = "";                //服務端數據庫密碼
    const QString DB_SER_HOSTNAME = "localhost";  //服務端數據庫主機名稱(IP)
    const QString DB_SER_DATABASE = "serverDB";   //服務端數據庫名稱
}

//服務器節點
struct ServerNode{
    QString serIPaddress; //服務器地址
    int serPort; //服務器端口
    ServerNode(){serPort = 0;}
};

// 登錄狀態
enum LoginStatus
{
    LS_ONLINE = 0,                      // 在線
    LS_INVISIBLE = 1,                   // 隱身
    LS_BUSY = 2,                        // 忙碌
    LS_LEAVE = 3,                       // 離開
    LS_NOTDISTURB = 4,                  // 請勿打擾
    LS_OFFLINE = 5
};

//羣成員身份(未實現，暫保留)
enum FlockStatus
{
    FS_FLOCK_HOST = 0,                  // 羣主
    FS_FLOCK_ADMAIN = 1,                // 羣管理員
    FS_FLOCK_MEMBER = 2                 // 羣成員
};

//討論組成員身份(未實現，暫保留)
enum DiscussionStatus
{
    DS_DISCUSSION_HOST = 0,             // 組長
    DS_DISCUSSION_MEMBER = 1            // 組員
};

//email讀/未讀標記位(未實現，暫保留)
enum EmailStatus
{
    ES_MAIL_UNREAD = 0,
    ES_MAIL_READ = 1
};

// 信息(命令)類型
enum MessageType
{
    /*********************** CToS ************************/
    MT_CHECK_CONNECTION,                // 檢查能否與服務器連接
    MT_LOGIN,                           // 登錄
    MT_REGISTER,                        // 註冊
    MT_GET_QUESTION_ANSWER,             // 獲取密保問題及答案
    MT_GET_QUESTION_ANSWER_SUCCESS,
    MT_GET_QUESTION_ANSWER_FAIL,

    MT_CHECK_MESSAGE,                   // 檢查有無給自己的信息（登錄時檢查）

    MT_GET_ALL_FRIENDS,                 // 獲取所有好友信息

    MT_TALK,                            // 發送對話
    MT_ADD_FRIEND,                      // 添加好友

    MT_GET_FRIEND_INFORMATION,          // 獲取好友信息
    MT_DELETE_FRIEND,                   // 刪除好友
    MT_GET_MYSELF_INFORMATION,          // 獲取自己信息
    MT_CHANGE_INFORMATION,              // 改變自己的信息
    MT_CHANGE_REMARK,                   // 改變（好友）備註
    MT_CHANGE_PASSWORD,                 // 修改密碼
    MT_CHANGE_STATUE,                   // 改變自身狀態
    // 還有分組好友管理
    MT_QUIT,                            // 退出

    /*********************** SToC ************************/

    MT_LOGIN_SUCCESS ,                  // 登錄成功
    MT_LOGIN_FAIL,                      // 登錄失敗
    MT_HAVE_LOGINED,                    // 已經登錄
    MT_REGISTER_SUCCESS,                // 註冊成功
    MT_REGISTER_FAIL,                   // 註冊失敗

    MT_REMOVE_BOX,                      // 刪除分組
    MT_REMOVE_BOX_SUCCESS,
    MT_REMOVE_BOX_FAIL,
    MT_RENAME_BOX,                      // 重命名分組
    MT_RENAME_BOX_SUCCESS,
    MT_RENAME_BOX_FAIL,
    MT_CREATE_BOX,                      // 創建分組
    MT_CREATE_BOX_SUCEESS,
    MT_CREATE_BOX_FAIL,
    MT_MOVE_FRIEND_BOX,                 // 移動好友至其他分組  36
    MT_MOVE_FRIEND_BOX_SUCCESS,
    MT_MOVE_FRIEND_BOX_FAIL,


    MT_GET_ALL_FRIENDS_SUCCESS,         // 獲取所有好友信息成功
    MT_GET_ALL_FRIENDS_FAIL,            // 獲取所有好友信息失敗
    MT_NO_FRIEND,                       // 沒有好友

    MT_NO_MESSAGE,                      // 沒有信息
    MT_HAVE_MESSAGE,                    // 有信息
    MT_HAVE_TALK_MESSAGE,               // 有聊天信息

    MT_FRIEND_REQUESTED,                // 好友請求
    MT_FRIEDN_NO_ACCOUNT,               // 沒有這個用戶名
    MT_ALREAD_FRIENDS,                  // 迴應好友請求

    MT_GET_FRIEND_FAIL,
    MT_GET_FRIEND_SUCCESS,              // 獲得好友資料成功
    MT_DELETE_FRIEND_FAIL,
    MT_DELETE_FRIEND_SUCCESS,           // 刪除好友成功
    MT_GET_MYSELF_FAIL,
    MT_GET_MYSELF_SUCCESS,              // 獲得自己詳細資料成功
    MT_CHANGE_INFORMATION_FAIL,
    MT_CHANGE_INFORMATION_SUCCESS,      // 改變自身資料成功

    MT_CHANGE_REMARK_FAIL,              // 修改備註失敗
    MT_CHANGE_REMARK_SUCCESS,           // 修改備註成功

    MT_OLD_PWD_IS_WRONG,                // 舊密碼是錯的
    MT_CHANGE_PWD_FAIL,                 // 修改密碼失敗
    MT_CHANGE_PWD_SUCCESS,              // 修改密碼成功

    /*********************** CToC ************************/

    MT_REQUEST_FRIEND,                  // 請求添加好友
    MT_AGREE_FRIEND,                    // 同意好友請求
    MT_REFUSE_FRIEND,                   // 拒絕好友請求
    MT_ADD_FRIEND_FAIL,                 // 添加好友失敗
    MT_ADD_FRIEND_SUCCESS,              // 添加好友成功
    MT_END_ADD_FRIEND,                  // 好友添加結束
    MT_BE_DELETED,                      // 自己被好友刪除了

    MT_MESSAGE,				            // 普通信息
    MT_TALK_MESSAGE,                    // 聊天信息

    MT_GET_HISTORY_MESSAGE,             // 歷史消息
    MT_GET_HISTORY_MESSAGE_SUCCESS,
    MT_GET_HISTORY_MESSAGE_FAIL
};

// 存儲聊天的字體和顏色
struct SaveFile
{
    QFont m_Font;
    QColor m_Color;
    // 重載輸入輸出操作符
    friend QDataStream & operator<< (QDataStream &out, const SaveFile &sf)
    {
       out << sf.m_Font << sf.m_Color;
       return out;
    }
    friend QDataStream & operator>> (QDataStream &in, SaveFile &sf)
    {
       in >> sf.m_Font >> sf.m_Color;
       return in;
    }
};

//郵件信息(未實現，暫保留)
struct MailInformation
{
    int m_MailID;
    QString m_SenderAddress;
    QString m_ReceiverAddress;
    QString m_Theme;
    QString m_Content;
    QDateTime m_dateTime;
    int m_Status;

    MailInformation()
    {
        m_Status = ES_MAIL_READ;
    }

    // 重載輸入輸出操作符
    friend QDataStream &operator<<(QDataStream &out, const MailInformation &mail)
    {
        out << mail.m_MailID << mail.m_SenderAddress << mail.m_ReceiverAddress
            << mail.m_Theme << mail.m_Content << mail.m_dateTime << mail.m_Status;
        return out;
    }

    friend QDataStream &operator>>(QDataStream &in, MailInformation &mail)
    {
        in >> mail.m_MailID >> mail.m_SenderAddress >> mail.m_ReceiverAddress
           >> mail.m_Theme >> mail.m_Content >> mail.m_dateTime >> mail.m_Status;
        return in;
    }
};

// 用戶詳細信息
struct UserInformation
{
    int m_HeadPortrait;
    QString m_UserID;
    QString m_NickName;
    QString m_Password;
    QString m_Sex;
    QDate m_Birthday;
    QString m_Question;
    QString m_Answer;
    QString m_Name;
    QString m_Phone;
    QString m_Address;
    QDateTime m_RegDateTime;
    //QString m_Mark;
    int m_Status;

    UserInformation()
    {
       m_Status = LS_OFFLINE;
       m_HeadPortrait = 1;
    }
    // 重載輸入輸出操作符
    friend QDataStream &operator<<(QDataStream &out, const UserInformation &user)
    {
       out << user.m_HeadPortrait << user.m_UserID << user.m_NickName
           << user.m_Password << user.m_Sex << user.m_Birthday
           << user.m_Question << user.m_Answer << user.m_Name
           << user.m_Phone << user.m_Address << user.m_RegDateTime
           << user.m_Status;
       return out;
    }

    friend QDataStream &operator>>(QDataStream &in, UserInformation &user)
    {
       in >> user.m_HeadPortrait >> user.m_UserID >> user.m_NickName
          >> user.m_Password >> user.m_Sex >> user.m_Birthday
          >> user.m_Question >> user.m_Answer >> user.m_Name
          >>  user.m_Phone >> user.m_Address >> user.m_RegDateTime
          >> user.m_Status;
       return in;
    }
};

// 好友信息
struct FriendInformation
{
    int m_HeadPortrait;
    QString m_UserID;
    QString m_NickName;
    QString m_GroupName;
    QString m_RemarkName;
    int m_Status;

    FriendInformation()
    {
       m_Status = LS_OFFLINE;
       m_HeadPortrait = 1;
       m_GroupName = GROUP_NAME_DEFAULT;
    }
    //重載輸入輸出操作符
    friend QDataStream & operator<< (QDataStream &out, const FriendInformation &fri)
    {
       out << fri.m_HeadPortrait << fri.m_UserID << fri.m_NickName
           << fri.m_GroupName << fri.m_RemarkName << fri.m_Status;
       return out;
    }
    friend QDataStream & operator>> (QDataStream &in, FriendInformation &fri)
    {
       in >> fri.m_HeadPortrait >> fri.m_UserID >> fri.m_NickName
          >> fri.m_GroupName >> fri.m_RemarkName >> fri.m_Status;
       return in;
    }
};

// 儲存在各種小窗口中填寫的信息
struct TempStrings
{
    QString m_One;
    QString m_Two;
    QString m_Three;
    //重載輸入輸出操作符
    friend QDataStream & operator<< (QDataStream &out, const TempStrings &str)
    {
       out << str.m_One << str.m_Two << str.m_Three;
       return out;
    }
    friend QDataStream & operator>> (QDataStream &in, TempStrings &str)
    {
       in >> str.m_One >> str.m_Two >> str.m_Three;
       return in;
    }
};

// 聊天框信息
struct ChatInformation
{
    QString m_MyID;
   int m_HeadPortrait;
   QString m_FriendID;          // 好友id
   QString m_FriendNickname;    // 好友暱稱
   QString m_FriendRemark;      // 好友備註
   QString m_FriendMark;        // 好友簽名
   QString m_FriendIP;          // 好友ip地址
   int m_FriendStatus;

    ChatInformation()
    {
       m_FriendStatus = LS_OFFLINE;
       m_HeadPortrait= 1;
       m_FriendMark = QString(QObject::tr("好友簽名"));
    }
};

// 對話信息
struct TalkMessage
{
    int m_Type;               // 消息類型
    QString m_SenderID;       // 發送者ID
    QString m_SenderIP;       // 發送者IP
    QString m_ReceiverID;     // 接收者ID
    QString m_ReceiverIP;     // 接收者IP
    QString m_Text;           // 消息內容
    QDateTime m_dateTime;

    TalkMessage() {	m_Type = 0; }
    //重載輸入輸出操作符
    friend QDataStream & operator<< (QDataStream &out, const TalkMessage &mes)
    {
       out << mes.m_Type << mes.m_SenderID << mes.m_SenderIP << mes.m_ReceiverID
           << mes.m_ReceiverIP << mes.m_Text << mes.m_dateTime;
       return out;
    }

    friend QDataStream & operator>> (QDataStream &in, TalkMessage &mes)
    {
       in >> mes.m_Type >> mes.m_SenderID >> mes.m_SenderIP >> mes.m_ReceiverID
          >> mes.m_ReceiverIP >> mes.m_Text >>  mes.m_dateTime;
       return in;
    }
};

// 用戶登錄信息
struct LoginInformation
{
    QString m_UserID;
    QString m_Password;
    int m_Status;

    LoginInformation()
    {	m_Status = LS_OFFLINE;	}
    //重載輸入輸出操作符
    friend QDataStream & operator<< (QDataStream &out, const LoginInformation &li)
    {
       out << li.m_UserID << li.m_Password << li.m_Status;
       return out;
    }
    friend QDataStream & operator>> (QDataStream &in, LoginInformation &li)
    {
       in >> li.m_UserID >> li.m_Password >> li.m_Status;
       return in;
    }
};

// 羣成員信息(未實現，暫保留)
struct FlockMember
{
    QString m_FlockID;     // 羣ID
    QString m_UserID;      // 用戶ID
    int m_HeadPortrait;    // 用戶頭像編號
    QString m_Nickname;    // 用戶暱稱
    int m_LoginStatus;     // 用戶登錄狀態
    QString m_Remark;      // 用戶羣名片
    int m_FlockStatus;     // 用戶身份
    QDateTime m_dateTimejoin;  // 加入時間
    //重載輸入輸出操作符
    friend QDataStream & operator<< (QDataStream &out, const FlockMember &flockMember)
    {
       out << flockMember.m_FlockID << flockMember.m_UserID
           << flockMember.m_HeadPortrait << flockMember.m_Nickname
           << flockMember.m_LoginStatus << flockMember.m_Remark
           << flockMember.m_FlockStatus << flockMember.m_dateTimejoin;
       return out;
    }
    friend QDataStream & operator>> (QDataStream &in, FlockMember &flockMember)
    {
       in >> flockMember.m_FlockID >> flockMember.m_UserID
          >> flockMember.m_HeadPortrait >> flockMember.m_Nickname
          >> flockMember.m_LoginStatus >> flockMember.m_Remark
          >> flockMember.m_FlockStatus >> flockMember.m_dateTimejoin;
       return in;
    }
};

// 討論組成員信息(未實現，暫保留)
struct DiscussionMember
{
    QString m_DiscussionID;    // 討論組ID
    QString m_UserID;          // 用戶ID
    int m_HeadPortrait;        // 用戶頭像編號
    QString m_NickName;        // 用戶暱稱
    int m_LoginStatus;         // 用戶登錄狀態
    int m_DiscussionStatus;    // 用戶身份
    QDateTime m_dateTimejoin;  // 加入時間
    //重載輸入輸出操作符
    friend QDataStream & operator<< (QDataStream &out, const DiscussionMember &disMember)
    {
       out << disMember.m_DiscussionID << disMember.m_UserID
           << disMember.m_HeadPortrait << disMember.m_NickName
           << disMember.m_LoginStatus << disMember.m_DiscussionStatus
           << disMember.m_dateTimejoin;
       return out;
    }
    friend QDataStream & operator>> (QDataStream &in, DiscussionMember &disMember)
    {
       in >> disMember.m_DiscussionID >> disMember.m_UserID
          >> disMember.m_HeadPortrait >> disMember.m_NickName
          >> disMember.m_LoginStatus  >> disMember.m_DiscussionStatus
          >> disMember.m_dateTimejoin;
       return in;
    }
};

// 羣信息(未實現，暫保留)
struct FlockInformation
{
    QString m_FlockID;
    QString m_CreatorID;
    QDate m_CreateDate;
    QString m_FlockName;
    QString m_Theme;
    //重載輸入輸出操作符
    friend QDataStream & operator<< (QDataStream &out, const FlockInformation &flock)
    {
       out << flock.m_FlockID << flock.m_CreatorID << flock.m_CreateDate
           << flock.m_FlockName << flock.m_Theme;
       return out;
    }
    friend QDataStream & operator>> (QDataStream &in, FlockInformation &flock)
    {
       in >> flock.m_FlockID >> flock.m_CreatorID >> flock.m_CreateDate
          >> flock.m_FlockName >> flock.m_Theme;
       return in;
   }
};

// 討論組信息(未實現，暫保留)
struct DiscussionInformation
{
    QString m_DiscussionID;
    QString m_CreatorID;
    QDate m_CreateDate;
    QString m_DiscussionName;
    QString m_Theme;
    //重載輸入輸出操作符
    friend QDataStream & operator<< (QDataStream &out, const DiscussionInformation &dis)
    {
       out << dis.m_DiscussionID << dis.m_CreatorID << dis.m_CreateDate
           << dis.m_DiscussionName << dis.m_Theme;
       return out;
    }
    friend QDataStream & operator>> (QDataStream &in, DiscussionInformation &dis)
    {
       in >> dis.m_DiscussionID >> dis.m_CreatorID >> dis.m_CreateDate
          >> dis.m_DiscussionName >> dis.m_Theme;
       return in;
    }
};

// 存儲臨時信息的結構體，用於信號/糟
struct SaveTmpInformation
{
    int m_RequestKind;
    int m_ReplyKind;
    ServerSocket *m_ServerSocket;

    QString m_MyID;
    QString m_PeerID;
    QDate m_Date;
    QString m_GroupName;
    QString m_GroupNameF;
    UserInformation m_UserInf;
    LoginInformation m_LoginInf;
    FriendInformation m_FriendInf;
    TalkMessage m_Message;
    int m_Status;
    TempStrings m_TempStr;
    QVector<TalkMessage> m_MessageVec;
    QVector<FriendInformation> m_FriendsVec;
    QVector<QString> m_IDVec;

    QString m_FlockID;
    FlockMember m_FlockMember;
    FlockInformation m_FlockInf;
    QVector<FlockInformation> m_FlocksVec;
    QVector<FlockMember> m_FlockMembersVec;
    QString m_DiscussionID;
    DiscussionMember m_DiscussionMember;
    DiscussionInformation m_DiscussionInf;
    QVector<DiscussionInformation> m_DiscussionsVec;
    QVector<DiscussionMember> m_DiscussionMembersVec;

    int m_MailID;
    MailInformation m_Mail;
    QVector<MailInformation>  m_InMailsVec;
    QVector<MailInformation>  m_OutMailsVec;

    QString m_Title;
    QString m_NewTitle;

    QString m_SocketID;

    QVector<TalkMessage> m_HistoryMessageVec;
};

//頭像尺寸
const int HEAD_BIG_SIZE = 40;
const int HEAD_SMALL_SIZE = 35;
const int HEAD_MINI_SIZE = 20;

//聊天室 工具欄尺寸
const int CHAT_WIDGET_TOOL_SIZE = 20;
const int CHAT_WIDGET_TOOL_SMALL_SIZE = 15;
//字體大小
const int FONT_SIZE = 2;
//郵件地址格式
static const QString MAIL_ADDRESS_FORMAT = "@hackt.com";
//本地/網絡歷史消息記錄存放地址
static const QString LOCAL_HISTORY_MESSAGE_PATH = "resource/history/local/";
static const QString NETWORK_HISTORY_MESSAGE_PATH = "resource/history/network/";

#endif // STRUCTURE_H

Database.h

/*********************************************
File Name： Database.h
Description： 數據庫操作類
********************************************/
#ifndef DATABASE_H
#define DATABASE_H

#include "CommonInc.h"
#include "Structure.h"

class QSqlQuery;
class QDateTime;
class QSqlDatabase;

class Database : public QObject
{
    Q_OBJECT
public:
    Database(QObject *parent = nullptr);
    ~Database();
    //創建連接
    bool createConnection();
    //初始化數據庫，並將所有用戶的狀態status設置爲OFFLINE
    bool initDatabase();
    //關閉數據庫(庫函數)
    void close();
    //打開數據庫(庫函數)
    bool isOpen();
    //修改、更新密碼
    int updatePassword(TempStrings &tmp);
    // 查詢密保問題以及答案
    int searchQuestionAndAnswer(const QString &id, TempStrings & tmp);
    // 查詢用戶
    int searchUserInUserInformation(const LoginInformation &logInf, UserInformation &userInf);
    // 通過id查詢好友信息
    int searchFriendInformationByID(const QString &id, UserInformation &userInf);
    // 通過id查詢自己信息
    int searchMyselfInformationByID(const QString &id, UserInformation &userInf);
    // 修改用戶信息
    int updateMyselfInformation(const UserInformation &user);
    // 添加用戶
    int addUserInUserInformation(const UserInformation &userInf);
    // 獲取一段時間註冊數量
    int countRegisteredInPeriod(const QDateTime &datePre, const QDateTime &dateLater);
    // 獲取所有好友信息
    int searchFriendsInformation(const QString &id, QVector<FriendInformation> &friendsVec);
    // 獲取所有好友ID
    bool searchFriendsID(const QString &id, QVector<QString> &IDVec);
    // 更新好友狀態
    bool updateStatus(int status, const QString & id);
    // 更新好友備註
    int updateRemark(const TempStrings & tmpStr);
    // 刪除好友
    int deleteFriend(const QString & myID, const QString & friendID, QString & groupName, QString & groupNameF);
    // 添加好友
    bool addFriend(const TalkMessage & talMes);
    //插入DateCounterb表某日的數據項
    bool insertValuesInDateCounter(const QDate &date);
    // 獲取某日用戶註冊數量+1
    int getUserTimes(const QDate &date);
    // 增加某日用戶註冊數量
    bool increaseUserTimes(const QDate &date);
    // 移動好友至其他分組
    int moveFriendToGroup(const QString & myID, const QString & friendID, const QString & title, const QString & newTitle);
    // 重命名分組
    int renameGroup(const QString & id, const QString & title,  const QString & newTitle);
    // 存儲歷史記錄
    bool addHistoryMsg(const TalkMessage &talMes);
    // 存儲離線記錄
    bool addOfflineMsg(const TalkMessage &talMes);
    // 獲取用戶的某日與某位好友的聊天記錄
    int searchHistoryMsgOfFriendInDate(const QString & myID, const QString &friendID, const QDate & date, QVector<TalkMessage> & mesVec);


private:
    //檢查SQL語句的正確性
    void errorSQLOrder(QSqlQuery query, QString mark);
    //創建數據庫表格
    void createTable();

    QSqlDatabase *m_SqlDatabase;
};

#endif // DATABASE_H

Database.cpp

/*********************************************
File Name： Database.cpp
Description： 操作數據庫
********************************************/

#include <QSqlDatabase>
#include <QSqlQuery>
#include <QVariant>
#include <qdebug.h>
#include <QSqlError>
#include <QTextDocument>
#include "Database.h"

//更新用戶在線狀態
static const QString UPDATE_STATUS_IN_USERINFORMATION_SQL =
        "update UserInformation set status = 5";

//更新用戶密碼
static const QString UPDATE_PASSWORD_IN_USERINFORMATION_SQL =
        "update UserInformation set password = ?"
        "where user = ? and password = ?";

// 查詢用戶密碼和密保問題
static const QString SEARCH_QUESTION_ANSWER_IN_USERINFORMATION_SQL =
        "select password, question, answer from UserInformation where userID = ?;";

// 查詢用戶信息
static const QString SEARCH_USER_IN_USERINFORMATION_SQL =
        "select * from UserInformation where (userID = ? and password = ?)";

// 查詢用戶信息
static const QString SEARCH_USER_IN_USERINFORMATION_ID_SQL =
        "select * from UserInformation where userID = ?";

// 更新用戶信息
static const QString UPDATE_USERINFORMATION_SQL =
        "update UserInformation set "
        "head = ?, nickname = ?, sex = ?, birthday = ?, "
        "name = ?, phone = ?, address = ?  where userID = ?;";

// 更新用戶在線狀態
static const QString  UPDATE_STATUS_IN_USERINFORMATION_BY_ID_SQL=
        "update UserInformation set status = ? where userID = ?;";

// 更新用戶的個性簽名
static const QString  UPDATE_REMARKNAME_IN_USERFRIENDS_SQL=
        "update UserFriends set remarkName = ? where userID = ? and friendID = ?;";

// 插入一條記錄到UserInformation表中
static const QString ADD_USER_IN_USERINFORMATION_SQL =
        "insert into UserInformation values(?,?,?,?,?,?,?,?,?,?,?,?,?)";

// 查詢某段時間內註冊的用戶數量
static const QString COUNT_REGISTERED_USERS_IN_USERINFORMATION_SQL =
        "select count(*) from UserInformation where regDate >= ? and regDate < ?";

// 在好友中搜索好友信息
static const QString SEARCH_FRIENDS_IN_USERFRIENDS_USERINFORMATION_SQL =
                                        "select UserFriends.groupName, UserFriends.friendID, "
                                        "UserInformation.nickName, UserFriends.remarkName,"
                                        "UserInformation.status, UserInformation.head "
                                        "from UserFriends,UserInformation "
                                        "where UserFriends.userID = ? "
                                        "and UserInformation.userID = UserFriends.friendID;";

// 查詢某個用戶的所有好友
static const QString SEARCH_ID_IN_USERFRIENDS_SQL =
                                        "select friendID from UserFriends where userID = ?";

// 查詢某個分組下的好友信息
static const QString SEARCH_GROUPNAME_IN_USERFRIENDS_SQL =
                                        "select groupName from UserFriends where userID = ? and friendID = ?";

// 從好友分組中移除某個好友
static const QString DELETE_FRIEND_IN_USERFRIENDS_SQL =
                                        "delete from UserFriends where "
                                        "(userID = ? and friendID = ?) or "
                                        "(userID = ? and friendID = ?)";

// 插入好友到用戶的分組中
static const QString ADD_FRIEND_IN_USERFRIENDS_SQL =
                                        "insert into UserFriends(userID, friendID, groupName) values (?, ?, ?);";

// 統計DataCounter表中的記錄數
static const QString COUNT_DATARECORD_IN_DATECOUNTER_SQL =
                                        "select count(*) from DateCounter where date = ?;";

// 插入一條記錄到DataCounter表中
static const QString INSERT_VALUES_IN_DATECOUNTER_SQL =
                                        "insert into DateCounter values(?, 0, 0, 0);";

// 查詢當天用戶註冊數量
static const QString SELECT_USERTIMES_IN_DATECOUNTER_SQL =
                                        "select userTimes from DateCounter where date = ?;";

// 更新當天用戶註冊量
static const QString INCREASE_USERTIMES_IN_DATECOUNTER_SQL =
                                        "update DateCounter set userTimes = userTimes + 1 where date = ?;";

/************************ group ***************************/
// 修改/移動好友所屬分組
static const QString RENAME_GROUPNAME_IN_USERFREINS =
                                    "update UserFriends set groupName = ? where userID = ? and groupName = ?;";

// 修改/移動好友所屬分組
static const QString MOVE_FRIEND_IN_USERFREINS =
                                    "update UserFriends set groupName = ? where userID = ? "
                                    "and friendID = ?;";

/************************ message ************************/
// 插入聊天記錄到HistoryMessage表中
static const QString INSERT_MESSAGE_IN_HISTORYMESSAGE =
                                    "insert into HistoryMessage (senderID, receiverID, content, dateTime) "
                                    "values(?, ?, ?, ?); ";

// 插入離線消息到OfflineMessage表中
static const QString INSERT_MESSAGE_IN_OFFLINEMESSAGE =
                                    "insert into OfflineMessage (kind, senderID, receiverID, content) "
                                    "values(?, ?, ?, ?);";

// 查詢歷史消息
static const QString SEARCH_FRIEND_DATE_MESSAGE_IN_HISTORYMESSAGE =
                                    "select senderID, receiverID, content, dateTime from HistoryMessage "
                                    "where ( (senderID = ? and receiverID = ?) "
                                    "or (senderID = ? and receiverID = ?) ) "
                                    "and dateTime >= ? and dateTime < ?;";

/********************create tables********************/
static const QString CREATE_TABLE_USERINFORMATION_SQL = "";
static const QString CREATE_TABLE_USERFRIENDS_SQL = "";
static const QString CREATE_TABLE_TMPMESSAGE_SQL = "";

//html轉QString
QString simplifyRichText(QString richText)
{
    QTextDocument doc;
    doc.setHtml(richText);
    qDebug() << "say:"<< doc.toPlainText();
    return doc.toPlainText();
}

Database::Database(QObject *parent) :
    QObject(parent)
{
    m_SqlDatabase = new QSqlDatabase;
}

Database::~Database()
{
    if (m_SqlDatabase != nullptr)
    {
        if (m_SqlDatabase->isOpen())
            m_SqlDatabase->close();
        delete m_SqlDatabase;
        m_SqlDatabase = nullptr;
    }
}

bool Database::createConnection() //創建數據庫連接
{
    if(m_SqlDatabase == nullptr)
    {
        m_SqlDatabase = new QSqlDatabase;
    }
    if(m_SqlDatabase->isOpen())
    {
        return true;
    }
    //如果已經打開數據庫
    if(QSqlDatabase::contains("serverDB"))
    {
        *m_SqlDatabase = QSqlDatabase::database("serverDB");
    }else
    {
        //如果沒有打開，則添加
        *m_SqlDatabase = QSqlDatabase::addDatabase(Constant::DB_CLI_DRIVER,Constant::DB_SER_DATABASE);
        m_SqlDatabase->setDatabaseName(Constant::DB_CLI_DATABASE);
    }
    m_SqlDatabase->setUserName(Constant::DB_SER_USERNAME);
    m_SqlDatabase->setPassword(Constant::DB_SER_PWD);
    m_SqlDatabase->setHostName(Constant::DB_SER_HOSTNAME);
    if(!m_SqlDatabase->open())
    {
        QSqlError sqlError = m_SqlDatabase->lastError();
        QString errorString = sqlError.text();
        return false;
    }
    return true;
}

bool Database::initDatabase() //初始化數據庫，將所有用戶狀態status置爲OFFLINE
{
    if(!createConnection())
    {
        return false;
    }
    QSqlQuery sqlQuery(*m_SqlDatabase);
    sqlQuery.exec(UPDATE_STATUS_IN_USERINFORMATION_SQL);
    errorSQLOrder(sqlQuery, "initDatabase");
    if(!sqlQuery.isActive())
    {
        m_SqlDatabase->close();
        return false;
    }
    m_SqlDatabase->close();
    return false;
}

void Database::close() //關閉數據庫
{
    if(m_SqlDatabase != nullptr && m_SqlDatabase->isOpen())
    {
        m_SqlDatabase->close();
    }
}

bool Database::isOpen() //數據庫是否打開
{
    return m_SqlDatabase->isOpen();
}

int Database::updatePassword(TempStrings &tmp)
{
    if(!createConnection())
    {
        return MT_CHANGE_PWD_FAIL;
    }
    //sql事務
    QSqlDatabase::database().transaction(); //如果驅動程序支持事務，則在數據庫上開始事務
    QSqlQuery sqlquery(*m_SqlDatabase);
    sqlquery.prepare(UPDATE_PASSWORD_IN_USERINFORMATION_SQL); //如果要執行的SQL查詢準備成功,返回true,否則返回false

    sqlquery.addBindValue(tmp.m_Three); //使用位置值綁定時，將值val添加到值列表中
    sqlquery.addBindValue(tmp.m_One);
    sqlquery.addBindValue(tmp.m_Two);

    sqlquery.exec(); //執行先前準備的SQL查詢,如果查詢成功執行，返回true,否則返回false
    errorSQLOrder(sqlquery,"update pwd");
    QSqlDatabase::database().commit(); //將修改提交併保存到數據庫，完成這個事務

    if(!sqlquery.isActive()) //isActive():如果查詢處於活動狀態(成功執行exec(),但尚未完成的Qsqlquery)，則返回true
    {
        m_SqlDatabase->close();
        return MT_CHANGE_PWD_FAIL;
    }
    qDebug() << "size:" << sqlquery.numRowsAffected(); //返回受結果SQL語句影響的行數,如果無法確定，則返回-1
    if(sqlquery.numRowsAffected() <= 0)
    {
        m_SqlDatabase->close();
        return MT_CHANGE_PWD_FAIL;
    }
    qDebug() << "size:" << sqlquery.numRowsAffected();
    return MT_CHANGE_PWD_SUCCESS;
}

int Database::searchQuestionAndAnswer(const QString &id, TempStrings &tmp) //查詢密保問題及答案
{
    if(!createConnection())
    {
        return MT_GET_QUESTION_ANSWER_FAIL;
    }
    //Sql事務
    QSqlDatabase::database().transaction();
    QSqlQuery sqlquery(*m_SqlDatabase);
    sqlquery.prepare(SEARCH_QUESTION_ANSWER_IN_USERINFORMATION_SQL);

    sqlquery.addBindValue(id);
    sqlquery.exec();
    errorSQLOrder(sqlquery,"forget pwd");
    QSqlDatabase::database().commit();

    if(!sqlquery.isActive())
    {
        m_SqlDatabase->close();
        return MT_GET_QUESTION_ANSWER_FAIL;
    }
    if(sqlquery.size() <= 0)//返回結果的大小(返回的行數)；如果無法確定大小，或者數據庫不支持有關查詢大小的報告信息，則返回-1
    {
        m_SqlDatabase->close();
        return MT_GET_QUESTION_ANSWER_FAIL;
    }
    /*檢索結果中的下一個記錄(如果有),將查詢放在檢索到的記錄上。
    如果無法檢索記錄,則將結果放置在最後一條記錄之後並返回false。如果成功檢索到記錄，則返回true*/
    if(sqlquery.next())
    {
        tmp.m_One = sqlquery.value(0).toString();
        tmp.m_Two = sqlquery.value(1).toString();
        tmp.m_Three = sqlquery.value(2).toString();

        return MT_GET_QUESTION_ANSWER_SUCCESS;
    }
    return MT_GET_QUESTION_ANSWER_FAIL;
}

int Database::searchUserInUserInformation(const LoginInformation &logInf, UserInformation &userInf) //查詢用戶(登錄信息)
{
    if(!createConnection())
    {
        return MT_LOGIN_FAIL;
    }
    //Sql事務
    QSqlDatabase::database().transaction();
    QSqlQuery sqlquery(*m_SqlDatabase);
    sqlquery.prepare(SEARCH_USER_IN_USERINFORMATION_SQL);

    sqlquery.addBindValue(logInf.m_UserID);
    sqlquery.addBindValue(logInf.m_Password);
    sqlquery.exec();
    errorSQLOrder(sqlquery,"login search");
    QSqlDatabase::database().commit();

    if(!sqlquery.isActive())
    {
        m_SqlDatabase->close();
        return MT_LOGIN_FAIL;
    }
    if(sqlquery.size() <= 0)
    {
        m_SqlDatabase->close();
        return MT_LOGIN_FAIL;
    }
    if(sqlquery.next())
    {
        userInf.m_UserID = sqlquery.value(0).toString();
        userInf.m_NickName = sqlquery.value(1).toString();
        userInf.m_Sex = sqlquery.value(2).toString();
        userInf.m_Birthday = sqlquery.value(3).toDate();
        userInf.m_Address = sqlquery.value(4).toString();
        userInf.m_HeadPortrait = sqlquery.value(5).toInt();
        userInf.m_Phone = sqlquery.value(6).toString();
        userInf.m_RegDateTime = sqlquery.value(7).toDateTime();
        userInf.m_Name = sqlquery.value(8).toString();
        userInf.m_Status = logInf.m_Status;

        qDebug() << "searchUserInUserInformation:" << userInf.m_Status;
        if(updateStatus(logInf.m_Status, logInf.m_UserID))
        {
            return MT_LOGIN_SUCCESS;
        }
    }
    return MT_LOGIN_FAIL;
}

int Database::searchFriendInformationByID(const QString &id, UserInformation &userInf) //通過id查詢好友信息
{
    if(!createConnection())
    {
        return MT_GET_FRIEND_FAIL;
    }
    //sql事務
    QSqlDatabase::database().transaction();
    QSqlQuery sqlquery(*m_SqlDatabase);

    sqlquery.addBindValue(id);
    sqlquery.exec();
    errorSQLOrder(sqlquery, "friend inf search");
    QSqlDatabase::database().commit();

    if(!sqlquery.isActive())
    {
        m_SqlDatabase->close();
        return MT_GET_FRIEND_FAIL;
    }
    if(sqlquery.size() <= 0)
    {
        m_SqlDatabase->close();
        return MT_GET_FRIEND_FAIL;
    }
    if(sqlquery.next())
    {
        userInf.m_UserID = sqlquery.value(0).toString();
        userInf.m_NickName = sqlquery.value(1).toString();
        userInf.m_Sex = sqlquery.value(2).toString();
        userInf.m_Birthday = sqlquery.value(3).toDate();
        userInf.m_Address = sqlquery.value(4).toString();
        userInf.m_HeadPortrait = sqlquery.value(5).toInt();
        userInf.m_Phone = sqlquery.value(6).toString();
        userInf.m_RegDateTime = sqlquery.value(7).toDateTime();
        userInf.m_Name = sqlquery.value(8).toString();
        userInf.m_Status = sqlquery.value(9).toInt();
    }
    return MT_GET_FRIEND_SUCCESS;
}

int Database::searchMyselfInformationByID(const QString &id, UserInformation &userInf) //通過id查詢自己的信息
{
    if(!createConnection())
    {
        return MT_GET_MYSELF_FAIL;
    }
    //sql事務
    QSqlDatabase::database().transaction();
    QSqlQuery sqlquery(*m_SqlDatabase);
    sqlquery.prepare(SEARCH_USER_IN_USERINFORMATION_ID_SQL);

    sqlquery.addBindValue(userInf.m_NickName);
    sqlquery.addBindValue(userInf.m_Sex);
    sqlquery.addBindValue(userInf.m_Birthday);
    sqlquery.addBindValue(userInf.m_Name);
    sqlquery.addBindValue(userInf.m_Phone);
    sqlquery.addBindValue(userInf.m_Address);
    sqlquery.addBindValue(userInf.m_UserID);
    sqlquery.addBindValue(id);
    sqlquery.exec();
    errorSQLOrder(sqlquery, "myself inf search");
    QSqlDatabase::database().commit();

    if(!sqlquery.isActive())
    {
        m_SqlDatabase->close();
        return MT_GET_MYSELF_FAIL;
    }
    if(sqlquery.size() <= 0)
    {
        m_SqlDatabase->close();
        return MT_GET_MYSELF_FAIL;
    }
    if(sqlquery.next())
    {
        userInf.m_UserID = sqlquery.value(0).toString();
        userInf.m_NickName = sqlquery.value(1).toString();
        userInf.m_Sex = sqlquery.value(2).toString();
        userInf.m_Birthday = sqlquery.value(3).toDate();
        userInf.m_Address = sqlquery.value(4).toString();
        userInf.m_HeadPortrait = sqlquery.value(5).toInt();
        userInf.m_Phone = sqlquery.value(6).toString();
        userInf.m_RegDateTime = sqlquery.value(7).toDateTime();
        userInf.m_Name = sqlquery.value(8).toString();
        userInf.m_Status = sqlquery.value(9).toInt();
    }
    return MT_GET_MYSELF_SUCCESS;
}

int Database::updateMyselfInformation(const UserInformation &userInf) //修改用戶信息
{
    if(!createConnection())
    {
        return MT_CHANGE_INFORMATION_FAIL;
    }
    //sql事務
    QSqlDatabase::database().transaction();
    QSqlQuery sqlquery(*m_SqlDatabase);
    sqlquery.prepare(UPDATE_USERINFORMATION_SQL);
    sqlquery.addBindValue(userInf.m_HeadPortrait);
    sqlquery.addBindValue(userInf.m_NickName);
    sqlquery.addBindValue(userInf.m_Sex);
    sqlquery.addBindValue(userInf.m_Birthday);
    sqlquery.addBindValue(userInf.m_Name);
    sqlquery.addBindValue(userInf.m_Phone);
    sqlquery.addBindValue(userInf.m_Address);
    sqlquery.addBindValue(userInf.m_UserID);
    sqlquery.exec();
    errorSQLOrder(sqlquery, "change update userinf");
    QSqlDatabase::database().commit();

    if(!sqlquery.isActive())
    {
        //lastError():返回有關此查詢發生的最後一個錯誤(如果有)的錯誤信息
        qDebug() << "QUERY_FAIL::" << sqlquery.lastError().text();
        m_SqlDatabase->close();
        return MT_CHANGE_INFORMATION_FAIL;
    }
    return MT_CHANGE_INFORMATION_SUCCESS;
}

int Database::addUserInUserInformation(const UserInformation &userInf) //添加用戶
{
    if(!createConnection())
    {
        return MT_REGISTER_FAIL;
    }
    //sql事務
    QSqlDatabase::database().transaction();
    QSqlQuery sqlquery(*m_SqlDatabase);
    sqlquery.prepare(ADD_USER_IN_USERINFORMATION_SQL);

    sqlquery.addBindValue(userInf.m_UserID);
    sqlquery.addBindValue(userInf.m_HeadPortrait);
    sqlquery.addBindValue(userInf.m_NickName);
    sqlquery.addBindValue(userInf.m_Password);
    sqlquery.addBindValue(userInf.m_Sex);
    sqlquery.addBindValue(userInf.m_Birthday.toString("yyyy-MM-dd"));
    sqlquery.addBindValue(userInf.m_Question);
    sqlquery.addBindValue(userInf.m_Answer);
    sqlquery.addBindValue(userInf.m_Name);
    sqlquery.addBindValue(userInf.m_Phone);
    sqlquery.addBindValue(userInf.m_Address);
    sqlquery.addBindValue(userInf.m_RegDateTime);//.toString("yyyy-MM-dd hh:mm:ss"));
    sqlquery.addBindValue(5);
    sqlquery.exec();
    errorSQLOrder(sqlquery, "register add");
    QSqlDatabase::database().commit();

    if(!sqlquery.isActive())
    {
        qDebug() << "QUERY FAIL::" << sqlquery.lastError().text();//返回有關此查詢發生的最後一個錯誤(如果有)的錯誤信息
        m_SqlDatabase->close();
        return MT_REGISTER_FAIL;
    }
    if(!increaseUserTimes(QDate::currentDate()))
    {
        return MT_REGISTER_FAIL;
    }
    return MT_REGISTER_SUCCESS;
}

int Database::countRegisteredInPeriod(const QDateTime &datePre, const QDateTime &dateLater) //獲取一段時間內的用戶註冊數量
{
    int count = 0;
    qDebug() << "A period of date:" << datePre.toString("yyyy-MM-dd hh:mm:ss")
             << dateLater.toString("yyyy-MM-dd hh:mm:ss");
    if(!createConnection())
    {
        return 0;
    }
    //sql事務
    QSqlDatabase::database().transaction();
    QSqlQuery sqlquery(*m_SqlDatabase);
    sqlquery.prepare(COUNT_REGISTERED_USERS_IN_USERINFORMATION_SQL);

    sqlquery.addBindValue(datePre);
    sqlquery.addBindValue(dateLater);
    sqlquery.exec();
    errorSQLOrder(sqlquery, "register count");
    QSqlDatabase::database().commit();

    if(!sqlquery.isActive())
    {
        m_SqlDatabase->close();
        return 0;
    }
    while(sqlquery.next())
    {
        count = sqlquery.value(0).toInt() + 1;
    }
    return count;
}

int Database::searchFriendsInformation(const QString &id, QVector<FriendInformation> &friendsVec) //獲取所有好友信息
{
    if(!createConnection())
    {
        return MT_GET_ALL_FRIENDS_FAIL;
    }
    //sql事務
    QSqlDatabase::database().transaction();
    QSqlQuery sqlquery(*m_SqlDatabase);
    sqlquery.prepare(SEARCH_FRIENDS_IN_USERFRIENDS_USERINFORMATION_SQL);
    sqlquery.addBindValue(id);
    sqlquery.exec();
    errorSQLOrder(sqlquery, "get all friend search");
    QSqlDatabase::database().commit();

    if(!sqlquery.isActive())
    {
        m_SqlDatabase->close();
        return MT_GET_ALL_FRIENDS_FAIL;
    }
    friendsVec.clear(); //從向量中刪除所有元素
    while(sqlquery.next())
    {
        FriendInformation friInf;
        friInf.m_GroupName = sqlquery.value(0).toString();
        friInf.m_UserID = sqlquery.value(1).toString();
        friInf.m_NickName = sqlquery.value(2).toString();
        friInf.m_RemarkName = sqlquery.value(3).toString();
        friInf.m_Status = sqlquery.value(4).toInt();
        friInf.m_HeadPortrait = sqlquery.value(5).toInt();
        friendsVec.push_back(friInf); //提供此功能是爲了實現STL兼容性,等效於append(value)
    }
    if(friendsVec.isEmpty())
    {
        return MT_NO_FRIEND;
    }
    return MT_GET_ALL_FRIENDS_SUCCESS;
}

bool Database::searchFriendsID(const QString &id, QVector<QString> &friVec) //獲取所有好友ID
{
    friVec.clear();
    if(!createConnection())
    {
        return false;
    }
    //sql事務
    QSqlDatabase::database().transaction();
    QSqlQuery sqlquery(*m_SqlDatabase);
    sqlquery.prepare(SEARCH_ID_IN_USERFRIENDS_SQL);

    sqlquery.addBindValue(id);
    sqlquery.exec();
    errorSQLOrder(sqlquery, "get all friend id search");
    QSqlDatabase::database().commit();

    if(!sqlquery.isActive())
    {
        m_SqlDatabase->close();
        return false;
    }
    while(sqlquery.next())
    {
        friVec.push_back(sqlquery.value(0).toString());
    }
    return true;
}

void Database::errorSQLOrder(QSqlQuery sqlquery, QString flog) //檢查SQL語句的正確性
{
    //如果SQL語句錯誤，就彈出警告信息
    if(!sqlquery.isActive())
    {
        QString str = sqlquery.lastError().text() + '\n' + flog;
        qDebug() << "Mysql query error:" << str;
        qDebug() << "sqlquery text:" << sqlquery.lastQuery(); //返回正在使用的當前查詢的文本，如果沒有當前查詢文本，則返回一個空字符串
    }
    else
    {
        qDebug() << Constant::DB_SER_DRIVER << "excute success:";
        qDebug() << "sql text:" << sqlquery.lastQuery();
    }
}

void Database::createTable() //創建數據庫表格
{
    //預留接口函數
}

bool Database::updateStatus(int status, const QString &id) //更新好友狀態
{
    qDebug() << "status is:" << status;
    if(!createConnection())
    {
        return false;
    }
    //sql事務
    QSqlDatabase::database().transaction();
    QSqlQuery sqlquery(*m_SqlDatabase);
    sqlquery.prepare(UPDATE_STATUS_IN_USERINFORMATION_BY_ID_SQL);

    sqlquery.addBindValue(id);
    sqlquery.addBindValue(status);
    sqlquery.exec();
    errorSQLOrder(sqlquery, "quit update status");
    QSqlDatabase::database().commit();

    if(!sqlquery.isActive())
    {
        qDebug() << "QUERY_FAIL::" << sqlquery.lastError().text();
        m_SqlDatabase->close();
        return false;
    }
    return true;
}

int Database::updateRemark(const TempStrings & tmpStr)//更新好友備註
{
    if (!createConnection())
      {
          return MT_CHANGE_REMARK_FAIL;
      }
      //sql 事務
      QSqlDatabase::database().transaction();
      QSqlQuery query(*m_SqlDatabase);
      query.prepare(UPDATE_REMARKNAME_IN_USERFRIENDS_SQL);

      query.addBindValue(tmpStr.m_Three);
      query.addBindValue(tmpStr.m_One);
      query.addBindValue(tmpStr.m_Two);
      query.exec();
      errorSQLOrder(query, "update-remark");
      QSqlDatabase::database().commit();

      if (!query.isActive())
      {
          qDebug() << "QUERY_FAIL::" << query.lastError().text();
          m_SqlDatabase->close();
          return MT_CHANGE_REMARK_FAIL;
      }

      return MT_CHANGE_REMARK_SUCCESS;
}

int Database::deleteFriend(const QString & myID, const QString & friendID, QString & groupName, QString & groupNameF)//刪除好友
{
    if(!createConnection())
    {
        return MT_DELETE_FRIEND_FAIL;
    }
    //sql事務
    QSqlDatabase::database().transaction();
    QSqlQuery sqlquery(*m_SqlDatabase);
    //在groupName下查詢分組信息
    sqlquery.prepare(SEARCH_GROUPNAME_IN_USERFRIENDS_SQL);
    sqlquery.addBindValue(myID);
    sqlquery.addBindValue(friendID);
    sqlquery.exec();
    errorSQLOrder(sqlquery, "search friend through groupName");

    if(!sqlquery.isActive())
    {
        qDebug() << "QUERY FAIL:" << sqlquery.lastError().text();
        m_SqlDatabase->close();
        return MT_DELETE_FRIEND_FAIL;
    }
    if(sqlquery.next())
    {
        groupName = sqlquery.value(0).toString();
    }
    sqlquery.clear();
    //在分組下查詢好友groupNameF
    sqlquery.prepare(SEARCH_GROUPNAME_IN_USERFRIENDS_SQL);
    sqlquery.addBindValue(friendID);
    sqlquery.addBindValue(myID);
    sqlquery.exec();
    errorSQLOrder(sqlquery, "search friend through groupNameF");

    if(!sqlquery.isActive())
    {
        qDebug() << "QUERY FAIL:" << sqlquery.lastError().text();
        m_SqlDatabase->close();
        return MT_DELETE_FRIEND_FAIL;
    }
    if(sqlquery.next())
    {
        groupNameF = sqlquery.value(0).toString();
    }
    sqlquery.clear();
    //刪除好友
    sqlquery.prepare(DELETE_FRIEND_IN_USERFRIENDS_SQL);
    sqlquery.addBindValue(myID);
    sqlquery.addBindValue(friendID);
    sqlquery.addBindValue(friendID);
    sqlquery.addBindValue(myID);
    sqlquery.exec();
    errorSQLOrder(sqlquery, "delete friends");

    QSqlDatabase::database().commit();

    if(!sqlquery.isActive())
    {
        qDebug() << "QUERY FAIL:" << sqlquery.lastError().text();
        m_SqlDatabase->close();
        return MT_DELETE_FRIEND_FAIL;
    }
    return MT_DELETE_FRIEND_SUCCESS;
}

bool Database::addFriend(const TalkMessage &talMes)//添加好友
{
    if(!createConnection())
    {
        return false;
    }
    //sql事務
    QSqlDatabase::database().transaction();
    QSqlQuery sqlquery(*m_SqlDatabase);
    sqlquery.prepare(ADD_FRIEND_IN_USERFRIENDS_SQL);

    sqlquery.addBindValue(talMes.m_SenderID);
    sqlquery.addBindValue(talMes.m_ReceiverID);
    sqlquery.addBindValue(GROUP_NAME_DEFAULT);
    sqlquery.exec();
    errorSQLOrder(sqlquery, "add friend from sender");

    if(!sqlquery.isActive())
    {
        qDebug() << "QUERY FAIL:" << sqlquery.lastError().text();
        m_SqlDatabase->close();
        return false;
    }
    sqlquery.clear();
    sqlquery.prepare(ADD_FRIEND_IN_USERFRIENDS_SQL);

    sqlquery.addBindValue(talMes.m_ReceiverID);
    sqlquery.addBindValue(talMes.m_SenderID);
    sqlquery.addBindValue(GROUP_NAME_DEFAULT);
    sqlquery.exec();
    errorSQLOrder(sqlquery, "add friend from recevier");

    if(!sqlquery.isActive())
    {
        qDebug() << "QUERY FAIL" << sqlquery.lastError().text();
        m_SqlDatabase->close();
        return false;
    }
    QSqlDatabase::database().commit();
    return true;
}

bool Database::insertValuesInDateCounter(const QDate &date)//插入DateCounterb表某日的數據項
{
    int count = 0;
    if (!createConnection())
    {
        return false;
    }
    //sql事務
    QSqlDatabase::database().transaction();
    QSqlQuery sqlquery(*m_SqlDatabase);
    sqlquery.prepare(COUNT_DATARECORD_IN_DATECOUNTER_SQL);
    sqlquery.addBindValue(date.toString("yyyy-MM-dd"));
    sqlquery.exec();
    errorSQLOrder(sqlquery, "count-datecounter");

    if(!sqlquery.isActive())
    {
        qDebug() << "QUERY_FAIL::" << sqlquery.lastError().text();
        m_SqlDatabase->close();
        return false;
    }
    if (sqlquery.next())
    {
        count = sqlquery.value(0).toInt();
    }
    if (count == 0)
    {
        sqlquery.prepare(INSERT_VALUES_IN_DATECOUNTER_SQL);
        sqlquery.addBindValue(date.toString("yyyy-MM-dd"));
        sqlquery.exec();

        errorSQLOrder(sqlquery, "insert datecounter");
        if (!sqlquery.isActive())
        {
            qDebug() << "QUERY_FAIL::" << sqlquery.lastError().text();
            m_SqlDatabase->close();
            return false;
        }

        if (sqlquery.numRowsAffected() <= 0) //返回受結果SQL語句影響的行數,如果無法確定則返回-1,如果查詢未處於活動狀態則返回-1
        {
            return false;
        }
    }
    QSqlDatabase::database().commit();
    return true;
}

int Database::getUserTimes(const QDate &date)//獲取某日用戶註冊數量+1=用戶ID
{
    int count = 0;
    if(!insertValuesInDateCounter(QDate::currentDate()))
    {
        return 0;
    }
    if(!createConnection())
    {
        return 0;
    }
    //sql事務
    QSqlDatabase::database().transaction();
    QSqlQuery sqlquery(*m_SqlDatabase);
    sqlquery.prepare(SELECT_USERTIMES_IN_DATECOUNTER_SQL);

    sqlquery.addBindValue(date.toString("yyyy-MM-dd"));
    sqlquery.exec();
    errorSQLOrder(sqlquery, "select usertimes");

    if(!sqlquery.isActive())
    {
        qDebug() << "QUERY FAIL:" << sqlquery.lastError().text();
        m_SqlDatabase->close();
        return false;
    }
    if(sqlquery.next())
    {
        count = sqlquery.value(0).toInt();
    }
    QSqlDatabase::database().commit();
    return count + 1;
}

bool Database::increaseUserTimes(const QDate &date)//增加某日用戶註冊數量
{
    if(!createConnection())
    {
        return false;
    }
    //sql事務
    QSqlDatabase::database().transaction();
    QSqlQuery sqlquery(*m_SqlDatabase);
    sqlquery.prepare(INCREASE_USERTIMES_IN_DATECOUNTER_SQL);

    sqlquery.addBindValue(date.toString("yyyy-MM-dd"));
    sqlquery.exec();
    errorSQLOrder(sqlquery, "increase usertimes");

    if(!sqlquery.isActive())
    {
        qDebug() << "QUERY FAIL:" << sqlquery.lastError().text();
        m_SqlDatabase->close();
        return false;
    }
    if(sqlquery.numRowsAffected() <= 0)
    {
        return false;
    }
    QSqlDatabase::database().commit();
    return true;
}

int Database::moveFriendToGroup(const QString & myID, const QString & friendID, const QString & title, const QString & newTitle)//移動好友至其他分組
{
    if(!createConnection())
    {
        return MT_MOVE_FRIEND_BOX_FAIL;
    }
    //sql事務
    QSqlDatabase::database().transaction();
    QSqlQuery sqlquery(*m_SqlDatabase);
    sqlquery.prepare(MOVE_FRIEND_IN_USERFREINS);

    sqlquery.addBindValue(title);
    sqlquery.addBindValue(newTitle);
    sqlquery.addBindValue(myID);
    sqlquery.addBindValue(friendID);
    sqlquery.exec();
    errorSQLOrder(sqlquery, "move friend to group");

    if(!sqlquery.isActive())
    {
        qDebug() << "QUERY FAIL:" << sqlquery.lastError().text();
        m_SqlDatabase->close();
        return MT_MOVE_FRIEND_BOX_FAIL;
    }
    if(sqlquery.numRowsAffected() <= 0)
    {
        return MT_MOVE_FRIEND_BOX_FAIL;
    }
    QSqlDatabase::database().commit();
    return MT_MOVE_FRIEND_BOX_SUCCESS;
}

int Database::renameGroup(const QString & id, const QString & title,  const QString & newTitle)//重命名分組
{
    if (!createConnection())
    {
        return MT_RENAME_BOX_FAIL;
    }
    //sql事務
    QSqlDatabase::database().transaction();
    QSqlQuery sqlquery(*m_SqlDatabase);
    sqlquery.prepare(RENAME_GROUPNAME_IN_USERFREINS);

    sqlquery.addBindValue(newTitle);
    sqlquery.addBindValue(id);
    sqlquery.addBindValue(title);
    sqlquery.exec();
    errorSQLOrder(sqlquery, "rename-group");

    if (!sqlquery.isActive())
    {
        qDebug() << "QUERY FAIL:" << sqlquery.lastError().text();
        m_SqlDatabase->close();
        return MT_RENAME_BOX_FAIL;
    }
    if (sqlquery.numRowsAffected() <= 0)
    {
        return MT_RENAME_BOX_FAIL;
    }
    QSqlDatabase::database().commit();

    return MT_RENAME_BOX_SUCCESS;
}

bool Database::addHistoryMsg(const TalkMessage &talMes)// 存儲歷史記錄
{
    if (!createConnection())
    {
       return false;
    }
    //sql 事務
    QSqlDatabase::database().transaction();
    QSqlQuery sqlquery(*m_SqlDatabase);
    sqlquery.prepare(INSERT_MESSAGE_IN_HISTORYMESSAGE);

    sqlquery.addBindValue(talMes.m_SenderID);
    sqlquery.addBindValue(talMes.m_ReceiverID);
    sqlquery.addBindValue(simplifyRichText(talMes.m_Text));
    sqlquery.addBindValue(talMes.m_dateTime);
    sqlquery.exec();
    errorSQLOrder(sqlquery, "inser-mes-to-HistoryMmesage");

    if (!sqlquery.isActive())
    {
       qDebug() << "QUERY_FAIL::" << sqlquery.lastError().text();
       m_SqlDatabase->close();
       return false;
    }
    if(sqlquery.numRowsAffected() <= 0)
    {
       return false;
    }
    QSqlDatabase::database().commit();

    return true;
}

bool Database::addOfflineMsg(const TalkMessage &talMes)// 存儲離線記錄
{
    if(!createConnection())
    {
        return false;
    }
    //sql事務
    QSqlDatabase::database().transaction();
    QSqlQuery sqlquery(*m_SqlDatabase);
    sqlquery.prepare(INSERT_MESSAGE_IN_OFFLINEMESSAGE);

    sqlquery.addBindValue(talMes.m_Type);
    sqlquery.addBindValue(talMes.m_SenderID);
    sqlquery.addBindValue(talMes.m_ReceiverID);
    sqlquery.addBindValue(talMes.m_Text);
    sqlquery.exec();
    errorSQLOrder(sqlquery, "insert talmes to offline message");

    if(!sqlquery.isActive())
    {
        qDebug() << "QUERY FAIL:" << sqlquery.lastError().text();
        m_SqlDatabase->close();
        return false;
    }
    if(sqlquery.numRowsAffected() <= 0)
    {
        return false;
    }
    QSqlDatabase::database().commit();
    return true;
}

// 獲取用戶的某日與某位好友的聊天記錄
int Database::searchHistoryMsgOfFriendInDate(const QString & myID, const QString &friendID,
                                             const QDate & date, QVector<TalkMessage> & mesVec)
{
    if(!createConnection())
    {
        return MT_GET_HISTORY_MESSAGE_FAIL;
    }
    //sql事務
    QSqlDatabase::database().transaction();
    QSqlQuery sqlquery(*m_SqlDatabase);
    sqlquery.prepare(SEARCH_FRIEND_DATE_MESSAGE_IN_HISTORYMESSAGE);
    sqlquery.addBindValue(myID);
    sqlquery.addBindValue(friendID);
    sqlquery.addBindValue(date);
    sqlquery.exec();
    errorSQLOrder(sqlquery, "get history msg of friend in oneDay");
    QSqlDatabase::database().commit();

    if(!sqlquery.isActive())
    {
        m_SqlDatabase->close();
        return MT_GET_HISTORY_MESSAGE_FAIL;
    }
    mesVec.clear(); //從向量中刪除所有元素
    while(sqlquery.next())
    {
        TalkMessage talMes;
        talMes.m_SenderID = sqlquery.value(0).toString();
        talMes.m_SenderIP = sqlquery.value(1).toString();
        talMes.m_ReceiverID = sqlquery.value(2).toString();
        talMes.m_ReceiverIP = sqlquery.value(3).toString();
        talMes.m_dateTime = sqlquery.value(4).toDateTime();
        talMes.m_Type = sqlquery.value(5).toInt();
        talMes.m_Text = sqlquery.value(6).toString();

        mesVec.push_back(talMes); //提供此功能是爲了實現STL兼容性,等效於append(value)
    }
    if(mesVec.isEmpty())
    {
        return MT_NO_MESSAGE;
    }
    return MT_GET_HISTORY_MESSAGE_SUCCESS;
}

ServerSocket.h

/*********************************************
File Name： ServerSocket.h
Description： 來自客戶端的socket，用於處理各種請求
********************************************/
#ifndef SERVERSOCKET_H
#define SERVERSOCKET_H
#include <QTcpSocket>
#include "Database.h"
#include "Structure.h"

class QDateTime;

class ServerSocket:public QTcpSocket
{
    Q_OBJECT
public:
    ServerSocket(QObject *parent = nullptr);
    ~ServerSocket();

signals:
       // 請求server處理
       void sendSignal(const SaveTmpInformation &tmp);
       // 用戶登錄
       void userLoginSignal(const UserInformation &userInf);
       //用戶斷開連接
       void deleteSignal(const QString &acc);

public slots:
       // 接收信息
       void receiveMessage();
       // 發送信息
       void sendMessage(const SaveTmpInformation &tmp);
       // 刪除socket
       void deleteScoket();

   private:
       quint16 m_BlockSize;

       SaveTmpInformation m_SaveTmpInfo;

       Database m_Database;
};

#endif // SERVERSOCKET_H

ServerSocket.cpp

/*********************************************
File Name： ServerSocket.cpp
Description： 用於控制客戶端socket的類
              與客戶端連接的QTcpSocket類
********************************************/
#include "ServerSocket.h"
#include <QMultiMap>
#include <QDateTime>

ServerSocket::ServerSocket(QObject *parent):
    QTcpSocket(parent)
{
    connect(this, SIGNAL(readyRead()), this, SLOT(receiveMessage()));
    connect(this, SIGNAL(disconnected()), this, SLOT(deleteScoket()));
    m_BlockSize = 0;
    m_SaveTmpInfo.m_ServerSocket = this;
}

ServerSocket::~ServerSocket()
{
    if (m_Database.isOpen())
        m_Database.close();
}

void ServerSocket::receiveMessage() //接收用戶發送的信息
{
    QDataStream in(this);
    in.setVersion(QDataStream::Qt_5_11); //修改QDataStream使用的序列化格式
    if(m_BlockSize == 0)
    {
        if(bytesAvailable() < (int)sizeof (quint16)) //bytesAvailable()返回等待讀取的傳入字節數
        {
            return;
        }
        in >> m_BlockSize;
    }
    if(bytesAvailable() < m_BlockSize)
    {
        return;
    }
    in >> m_SaveTmpInfo.m_RequestKind; //將臨時信息的應答類型輸入數據流
    qDebug() << "receive message:" << m_SaveTmpInfo.m_RequestKind;

    switch (m_SaveTmpInfo.m_RequestKind)
    {
        case MT_CHECK_CONNECTION:
        {
            m_SaveTmpInfo.m_ReplyKind = MT_CHECK_CONNECTION;
            sendMessage(m_SaveTmpInfo);
            m_BlockSize = 0;
            break;
        }
        case MT_LOGIN:
        {
            in >> m_SaveTmpInfo.m_LoginInf;
            m_SaveTmpInfo.m_MyID = m_SaveTmpInfo.m_LoginInf.m_UserID;
            m_SaveTmpInfo.m_SocketID = m_SaveTmpInfo.m_MyID;
            break;
        }
        case MT_CHECK_MESSAGE:
        {
            in >> m_SaveTmpInfo.m_MyID;
            break;
        }
        case MT_REGISTER:
        {
            in >> m_SaveTmpInfo.m_UserInf;
            break;
        }
        case MT_GET_QUESTION_ANSWER:
        {
            in >> m_SaveTmpInfo.m_MyID;
            break;
        }
        case MT_TALK:
        {
            in >> m_SaveTmpInfo.m_Message;
            break;
        }
        case MT_GET_ALL_FRIENDS:
        {
            in >> m_SaveTmpInfo.m_MyID >> m_SaveTmpInfo.m_Status;
            m_SaveTmpInfo.m_SocketID = m_SaveTmpInfo.m_MyID;
            break;
        }
        case MT_GET_FRIEND_INFORMATION:
        {
            in >> m_SaveTmpInfo.m_PeerID;
            break;
        }
        case MT_GET_MYSELF_INFORMATION:
        {
            in >> m_SaveTmpInfo.m_MyID;
            break;
        }
        case MT_CHANGE_INFORMATION:
        {
            in >> m_SaveTmpInfo.m_UserInf;
            break;
        }
        case MT_CHANGE_REMARK:
        {
            in >> m_SaveTmpInfo.m_TempStr;
            break;
        }
        case MT_CHANGE_STATUE:
        {
            in >> m_SaveTmpInfo.m_MyID >> m_SaveTmpInfo.m_Status;
            break;
        }
        case MT_ADD_FRIEND:
        {
            in >> m_SaveTmpInfo.m_Message;
            break;
        }
        case MT_REFUSE_FRIEND:
        {
            break;
        }
        case MT_DELETE_FRIEND:
        {
            in >> m_SaveTmpInfo.m_MyID >> m_SaveTmpInfo.m_PeerID;
            break;
        }
        case MT_REMOVE_BOX:
        case MT_CREATE_BOX:
        {
            in >> m_SaveTmpInfo.m_MyID >> m_SaveTmpInfo.m_Title;
            break;
        }
        case MT_MOVE_FRIEND_BOX:
        {
            in >> m_SaveTmpInfo.m_MyID >> m_SaveTmpInfo.m_PeerID >> m_SaveTmpInfo.m_Title >> m_SaveTmpInfo.m_NewTitle;
            break;
        }
        case MT_RENAME_BOX:
        {
            in >> m_SaveTmpInfo.m_MyID >> m_SaveTmpInfo.m_Title >> m_SaveTmpInfo.m_NewTitle;
            break;
        }
        case MT_GET_HISTORY_MESSAGE:
        {
            in >> m_SaveTmpInfo.m_MyID >> m_SaveTmpInfo.m_PeerID >> m_SaveTmpInfo.m_Date;
            break;
        }
        case MT_CHANGE_PASSWORD:
        {
            in >> m_SaveTmpInfo.m_TempStr;
            break;
        }
        default:
        {
            qDebug() << "Unknow receive msg kind:" << m_SaveTmpInfo.m_RequestKind;
            break;
        }
    }
    QByteArray data = this->readAll(); //從設備讀取所有剩餘數據，並將其作爲字節數組返回
    qDebug() << "leave in socket:" << data.size();

    m_BlockSize = 0;
    emit sendSignal(m_SaveTmpInfo);
}

void ServerSocket::sendMessage(const SaveTmpInformation &tmp) //發送消息
{
    QByteArray block;
    QDataStream out(&block, QIODevice::WriteOnly);
    out.setVersion(QDataStream::Qt_5_11);
    out << quint16(0) << (int)tmp.m_ReplyKind;

    switch(tmp.m_ReplyKind)
    {
        case MT_LOGIN_SUCCESS:
        {
            out << tmp.m_UserInf;
            break;
        }
        case MT_REGISTER_SUCCESS:
        {
            out << tmp.m_UserInf.m_UserID << tmp.m_UserInf.m_HeadPortrait;
            break;
        }
        case MT_GET_QUESTION_ANSWER_SUCCESS:
        {
            out << tmp.m_TempStr;
            break;
        }
        case MT_GET_ALL_FRIENDS_SUCCESS:
        {
            int len = tmp.m_FriendsVec.size();
            out << len;
            for(int i = 0;i < len;++i)
            {
                out << tmp.m_FriendsVec[i];
            }
            break;
        }
        case MT_HAVE_MESSAGE:
        {
            break;
        }
        case MT_CHANGE_STATUE:
        {
            out << tmp.m_PeerID << tmp.m_Status;
            break;
        }
        case MT_GET_MYSELF_SUCCESS:
        {
            out << tmp.m_UserInf;
            break;
        }
        case MT_DELETE_FRIEND_SUCCESS:
        {
            out << tmp.m_MyID << tmp.m_PeerID << tmp.m_GroupName;
            break;
        }
        case MT_CHANGE_INFORMATION_SUCCESS:
        {
            out << tmp.m_UserInf;
            break;
        }
        case MT_CHANGE_INFORMATION:
        {
            out << tmp.m_FriendInf;
            break;
        }
        case MT_ADD_FRIEND_SUCCESS:
        {
            out << tmp.m_Message << tmp.m_UserInf;
            break;
        }
        case MT_GET_FRIEND_SUCCESS:
        {
            out << tmp.m_UserInf;
            break;
        }
        case MT_TALK:
        {
            out << tmp.m_Message;
            break;
        }
        case MT_CHANGE_REMARK_SUCCESS:
        {
            out << tmp.m_TempStr;
            break;
        }
        case MT_BE_DELETED:
        {
            out << tmp.m_MyID << tmp.m_PeerID << tmp.m_GroupNameF;
            break;
        }
        case MT_RENAME_BOX_SUCCESS:
        {
            out << tmp.m_MyID << tmp.m_Title << tmp.m_NewTitle;
            break;
        }
        case MT_MOVE_FRIEND_BOX_SUCCESS:
        {
            out << tmp.m_MyID << tmp.m_PeerID << tmp.m_Title << tmp.m_NewTitle;
            break;
        }
        case MT_GET_HISTORY_MESSAGE_SUCCESS:
        {
            out << tmp.m_MyID << tmp.m_PeerID << tmp.m_Date;
            int len = tmp.m_HistoryMessageVec.size();
            out << len;
            for(int i = 0;i < len;++i)
            {
                out << tmp.m_HistoryMessageVec[i];
            }
            break;
        }
        default:
        {
            qDebug() << "unknown send msg kind:" << m_SaveTmpInfo.m_RequestKind;
            break;
        }
    }
    //QIODevice *QDataStream::device()返回當前設置的I/O設備，如果當前未設置任何設備，則返回0
    out.device()->seek(0);//QIODevice::seek(quint16)將當前位置設爲pos，成功返回true，失敗返回false
    out << quint16(block.size() - sizeof(quint16));
    write(block); //將數據的最大maxSize字節從數據寫入設備,返回實際寫入的字節數；如果發生錯誤，則返回-1
}

void ServerSocket::deleteScoket() //刪除socket
{
    if(!m_SaveTmpInfo.m_SocketID.isEmpty())
    {
        if(MT_LOGIN != m_SaveTmpInfo.m_RequestKind)
        {
            emit deleteSignal(m_SaveTmpInfo.m_SocketID);
        }
        deleteLater();
    }
}

TcpServer.h

/*********************************************
File Name： TcpServer.h
Description： 服務器類
********************************************/
#ifndef TCPSERVER_H
#define TCPSERVER_H

#include <QTcpServer>
#include "ServerSocket.h"
#include "Structure.h"
#include "Database.h"

class TcpServer:public QTcpServer
{
    Q_OBJECT
public:
    TcpServer(QObject *parent = nullptr);
    ~TcpServer();

public slots:
    //客戶端斷開連接
    void clientDisconnected(const QString &id);
    //要求ServerSocket發送消息
    void sendMessage(const SaveTmpInformation &tmp);

signals:
    void haveNewID(const QString &id);
    void haveNewAddress(const QString &address);
    void haveLeftID(const QString &id);
    void haveLeftAddress(const QString &address);

private:
    // 當有新的連接時，會調用此函數。用於創建新的線程去控制socket通信
    void incomingConnection(int socketDescriptor);
    // 處理好友請求
    void friendRequest(const SaveTmpInformation &savTmp);
    // 處理對話請求
    void talkRequest(const SaveTmpInformation &save);
    // 通知好友，改變用戶狀態
    void tellFriendsStatusChanged(const QString &id, int status);
    // 通知好友，改變用戶信息
    void tellFriendsInformationChanged(const QString &id);

    // 臨時信息
    SaveTmpInformation m_SaveTmpInfo;
    // 操作數據庫
    Database m_Database;
    // 在線用戶列表
    QMap<QString, ServerSocket*> m_mapOnlineUser;
    // 在線用戶狀態
    QMap<QString, int> m_mapOnlineUserStatus;
    // 郵箱在線的用戶
    QMap<QString, ServerSocket*> m_mapOnlineUserMail;
};

#endif // TCPSERVER_H

TcpServer.cpp

#include "TcpServer.h"

TcpServer::TcpServer(QObject *parent) :
    QTcpServer(parent)
{
    // 初始化一下數據庫
}

TcpServer::~TcpServer()
{

}

void TcpServer::clientDisconnected(const QString &id) //客戶端斷開連接
{
    //某個客戶端斷開連接時，就更新該用戶的狀態
    if(id.contains(MAIL_ADDRESS_FORMAT))//contains():如果此字符串包含出現的字符串str，則返回true,否則返回false
    {
        m_mapOnlineUserMail.remove(id);//remove():返回移除的數量，通常爲1，如果鍵值對不在map內則爲0
        emit haveLeftAddress(id);
    }
    else
    {
        m_mapOnlineUser.remove(id);
        emit haveLeftID(id);
        if(m_Database.updateStatus(LS_OFFLINE, id))
        {
            tellFriendsStatusChanged(id, LS_OFFLINE);
        }
    }
#ifdef TEST_SWITCH
    qDebug() << "userMap-size:" <<m_mapOnlineUser.size();
    QMapIterator<QString, ServerSocket*> i(m_mapOnlineUser);
    while (i.hasNext())
    {
        i.next();
        qDebug() << i.key();
    }
    qDebug() << "userMailMap-size:" << m_mapOnlineUserMail.size();
    QMapIterator<QString, ServerSocket*> j(m_mapOnlineUser);
    while(j.hasNext())
    {
        j.next();
        qDebug() << j.key();
    }
#endif
}

void TcpServer::sendMessage(const SaveTmpInformation &tmp)//處理請求，要求ServerSocket發送信息
{
    m_SaveTmpInfo.m_RequestKind = tmp.m_RequestKind;
    //根據tmp.requestKind來處理信息
    switch(m_SaveTmpInfo.m_RequestKind)
    {
        case MT_LOGIN:
        {
            m_SaveTmpInfo.m_LoginInf = tmp.m_LoginInf;
            m_SaveTmpInfo.m_ServerSocket = tmp.m_ServerSocket;
            //判斷是否已經登錄
            m_SaveTmpInfo.m_MyID = tmp.m_MyID;
            if(m_mapOnlineUser.contains(m_SaveTmpInfo.m_MyID))
            {
                m_SaveTmpInfo.m_ReplyKind = MT_HAVE_LOGINED;
            }
            else
            {
                //訪問數據庫登錄
                m_SaveTmpInfo.m_ReplyKind = m_Database.searchUserInUserInformation(m_SaveTmpInfo.m_LoginInf, m_SaveTmpInfo.m_UserInf);
            }
            tmp.m_ServerSocket->sendMessage(m_SaveTmpInfo);
            break;
        }
        case MT_REGISTER://註冊
        {
            m_SaveTmpInfo.m_UserInf = tmp.m_UserInf;
            int count = m_Database.getUserTimes(QDate::currentDate());
            if(count == 0)
            {
                m_SaveTmpInfo.m_ReplyKind = MT_REGISTER_FAIL;
            }
            else
            {
                QString id = QDate::currentDate().toString("yyyyMMdd");
                id.append(QString::number(0));
                id.append(QString::number(count));
                m_SaveTmpInfo.m_UserInf.m_UserID = id;
                m_SaveTmpInfo.m_UserInf.m_RegDateTime = QDateTime::currentDateTime();
                //訪問數據庫註冊
                m_SaveTmpInfo.m_ReplyKind = m_Database.addUserInUserInformation(m_SaveTmpInfo.m_UserInf);
            }
            tmp.m_ServerSocket->sendMessage(m_SaveTmpInfo);
            break;
        }
        case MT_GET_QUESTION_ANSWER://獲得密保問題及答案
        {
            m_SaveTmpInfo.m_MyID = tmp.m_MyID;
            m_SaveTmpInfo.m_ReplyKind = m_Database.searchQuestionAndAnswer(m_SaveTmpInfo.m_MyID, m_SaveTmpInfo.m_TempStr);
            tmp.m_ServerSocket->sendMessage(m_SaveTmpInfo);
            break;
        }
        case MT_CHECK_MESSAGE://檢查信息
        {
            break;
        }
        case MT_GET_ALL_FRIENDS://獲取所有好友
        {
            m_SaveTmpInfo.m_MyID = tmp.m_MyID;
            m_SaveTmpInfo.m_Status = tmp.m_Status;
            m_SaveTmpInfo.m_ServerSocket = tmp.m_ServerSocket;
            m_SaveTmpInfo.m_ReplyKind = m_Database.searchFriendsInformation(m_SaveTmpInfo.m_MyID, m_SaveTmpInfo.m_FriendsVec);
            tmp.m_ServerSocket->sendMessage(m_SaveTmpInfo);
            if(MT_GET_ALL_FRIENDS_FAIL != m_SaveTmpInfo.m_ReplyKind)
            {
                if(!m_mapOnlineUser.contains(m_SaveTmpInfo.m_MyID))
                {
                    m_mapOnlineUser.insert(m_SaveTmpInfo.m_MyID, m_SaveTmpInfo.m_ServerSocket);
                    emit haveNewID(m_SaveTmpInfo.m_MyID);
                }
                tellFriendsStatusChanged(m_SaveTmpInfo.m_MyID, m_SaveTmpInfo.m_Status);
            }
            break;
        }
        case MT_TALK://對話請求
        {
            talkRequest(tmp);
            break;
        }
        case MT_CHANGE_INFORMATION://用戶改變個人信息
        {
            m_SaveTmpInfo.m_UserInf = tmp.m_UserInf;
            m_SaveTmpInfo.m_ReplyKind = m_Database.updateMyselfInformation(tmp.m_UserInf);
            tmp.m_ServerSocket->sendMessage(m_SaveTmpInfo);
            if(m_SaveTmpInfo.m_ReplyKind == MT_CHANGE_INFORMATION_SUCCESS)
            {
                tellFriendsInformationChanged(m_SaveTmpInfo.m_UserInf.m_UserID);
            }
            break;
        }
        case MT_GET_MYSELF_INFORMATION://獲取個人信息
        {
            m_SaveTmpInfo.m_MyID = tmp.m_MyID;
            m_SaveTmpInfo.m_ReplyKind = m_Database.searchMyselfInformationByID(m_SaveTmpInfo.m_MyID, m_SaveTmpInfo.m_UserInf);
            tmp.m_ServerSocket->sendMessage(m_SaveTmpInfo);
            break;
        }
        case MT_GET_FRIEND_INFORMATION://獲取好友信息
        {
            m_SaveTmpInfo.m_PeerID = tmp.m_PeerID;
            m_SaveTmpInfo.m_ReplyKind = m_Database.searchFriendInformationByID(m_SaveTmpInfo.m_PeerID, m_SaveTmpInfo.m_UserInf);
            tmp.m_ServerSocket->sendMessage(m_SaveTmpInfo);
            break;
        }
        case MT_DELETE_FRIEND://刪除好友
        {
            m_SaveTmpInfo.m_MyID = tmp.m_MyID;
            m_SaveTmpInfo.m_PeerID = tmp.m_MyID;
            m_SaveTmpInfo.m_ReplyKind = m_Database.deleteFriend(m_SaveTmpInfo.m_MyID, m_SaveTmpInfo.m_PeerID,
                                                                m_SaveTmpInfo.m_GroupName,m_SaveTmpInfo.m_GroupNameF);
            tmp.m_ServerSocket->sendMessage(m_SaveTmpInfo);

            //如果好友在線則發送被刪除信息，如果不在線則存儲
            QMap<QString, ServerSocket*>::iterator iter;
            //find(const Key &key):返回一個迭代器,該迭代器指向具有鍵key的項,如果不包含帶有該鍵的項則函數返回end(),如果包含多個帶有該鍵的項則函數返回指向最近插入值的迭代器
            iter = m_mapOnlineUser.find(m_SaveTmpInfo.m_PeerID);
            if(m_mapOnlineUser.end() == iter) //返回一個STL樣式的迭代器，該迭代器指向地圖中最後一個項目之後的假想項目
            {
                TalkMessage talMes;
                talMes.m_Type = MT_BE_DELETED;
                talMes.m_SenderID = tmp.m_MyID;
                talMes.m_ReceiverID = tmp.m_PeerID;
                talMes.m_Text = tmp.m_GroupNameF;
            }
            else
            {
                m_SaveTmpInfo.m_MyID = tmp.m_PeerID;
                m_SaveTmpInfo.m_PeerID = tmp.m_MyID;
                m_SaveTmpInfo.m_ReplyKind = MT_BE_DELETED;
                m_SaveTmpInfo.m_GroupName = tmp.m_GroupNameF;
                //T &iterator::value() const,返回對當前項目值的可修改引用
                iter.value()->sendMessage(m_SaveTmpInfo);
            }
            break;
        }
        case MT_CHANGE_REMARK://修改個籤
        {
            m_SaveTmpInfo.m_TempStr = tmp.m_TempStr;
            m_SaveTmpInfo.m_ReplyKind = m_Database.updateRemark(tmp.m_TempStr);
            tmp.m_ServerSocket->sendMessage(m_SaveTmpInfo);
            break;
        }
        case MT_ADD_FRIEND://添加好友
        {
            friendRequest(tmp);
            break;
        }
        case MT_CHANGE_STATUE://改變在線狀態
        {
            if(m_Database.updateStatus(tmp.m_Status, tmp.m_MyID))
            {
                tellFriendsStatusChanged(tmp.m_MyID, tmp.m_Status);
            }
            break;
        }
        case MT_REMOVE_BOX://刪除分組
        {
            m_SaveTmpInfo.m_MyID = tmp.m_MyID;
            m_SaveTmpInfo.m_Title = tmp.m_Title;
            break;
        }
        case MT_CREATE_BOX://創建分組
        {
            m_SaveTmpInfo.m_MyID = tmp.m_MyID;
            m_SaveTmpInfo.m_Title = tmp.m_Title;
            break;
        }
        case MT_MOVE_FRIEND_BOX://移動好友至其他分組
        {
            m_SaveTmpInfo.m_MyID = tmp.m_MyID;
            m_SaveTmpInfo.m_Title = tmp.m_Title;
            m_SaveTmpInfo.m_PeerID = tmp.m_PeerID;
            m_SaveTmpInfo.m_NewTitle = tmp.m_NewTitle;
            m_SaveTmpInfo.m_ReplyKind = m_Database.moveFriendToGroup(m_SaveTmpInfo.m_MyID, m_SaveTmpInfo.m_PeerID,
                                                                     m_SaveTmpInfo.m_Title, m_SaveTmpInfo.m_NewTitle);
            tmp.m_ServerSocket->sendMessage(m_SaveTmpInfo);
            break;
        }
        case MT_RENAME_BOX://分組重命名
        {
            m_SaveTmpInfo.m_MyID = tmp.m_MyID;
            m_SaveTmpInfo.m_Title = tmp.m_Title;
            m_SaveTmpInfo.m_NewTitle = tmp.m_NewTitle;
            m_SaveTmpInfo.m_ReplyKind = m_Database.renameGroup(m_SaveTmpInfo.m_MyID,
                                                               m_SaveTmpInfo.m_Title, m_SaveTmpInfo.m_NewTitle);
            tmp.m_ServerSocket->sendMessage(m_SaveTmpInfo);
            break;
        }
        case MT_GET_HISTORY_MESSAGE://讀取某天與某好友的消息歷史記錄
        {
            m_SaveTmpInfo.m_MyID = tmp.m_MyID;
            m_SaveTmpInfo.m_PeerID = tmp.m_PeerID;
            m_SaveTmpInfo.m_Date = tmp.m_Date;
            m_SaveTmpInfo.m_ReplyKind = m_Database.searchHistoryMsgOfFriendInDate(m_SaveTmpInfo.m_MyID, m_SaveTmpInfo.m_PeerID,
                                                                                  m_SaveTmpInfo.m_Date, m_SaveTmpInfo.m_HistoryMessageVec);
            tmp.m_ServerSocket->sendMessage(m_SaveTmpInfo);
            break;
        }
        case MT_CHANGE_PASSWORD://修改密碼
        {
            m_SaveTmpInfo.m_TempStr = tmp.m_TempStr;
            m_SaveTmpInfo.m_ReplyKind = m_Database.updatePassword(m_SaveTmpInfo.m_TempStr);
            tmp.m_ServerSocket->sendMessage(m_SaveTmpInfo);
            break;
        }
        default:
        {
            break;
        }
    }
}

void TcpServer::incomingConnection(int socketDescriptor) //當有新的連接時,會調用此函數,用於創建新的線程去控制socket通信
{
    //建立新連接，並鏈接相關的信號和槽
    ServerSocket *serverSocket = new ServerSocket(this);
    serverSocket->setSocketDescriptor(socketDescriptor);
    connect(serverSocket, SIGNAL(deleteSignal(const QString&)),
            this, SLOT(clientDisconnected(const QString&)));
    connect(serverSocket, SIGNAL(sendSignal(const SaveTmpInformation&)),
            this, SLOT(sendMessage(const SaveTmpInformation&)));

    qDebug() << "new client IP:" << serverSocket->peerAddress();
    qDebug() << "userMap-size:" << m_mapOnlineUser.size();

#ifdef TEST_SWITCH
    QMapIterator<QString, ServerSocket *> i(m_mapOnlineUser);
       while (i.hasNext()) {
           i.next();
           qDebug() << i.key();
       }

       qDebug() << "userMailMap-size:" << m_mapOnlineUserMail.size();
       QMapIterator<QString, ServerSocket *> j(m_mapOnlineUserMail);
       while (j.hasNext()) {
           j.next();
           qDebug() << j.key();
       }
#endif
}

void TcpServer::friendRequest(const SaveTmpInformation &savTmp) //處理好友申請請求
{
    m_SaveTmpInfo.m_ReplyKind = MT_ADD_FRIEND;
    m_SaveTmpInfo.m_Message = savTmp.m_Message;
    switch(savTmp.m_Message.m_Type)
    {
        case MT_REQUEST_FRIEND:
        {
            if(MT_ADD_FRIEND_SUCCESS != m_Database.searchFriendInformationByID(m_SaveTmpInfo.m_Message.m_SenderID,
                                                                               m_SaveTmpInfo.m_UserInf))
            {
                return;
            }
            qDebug() << "request add friend";
            break;
        }
        case MT_REFUSE_FRIEND:
        {
            qDebug() << "refuse add friend";
            break;
        }
        case MT_AGREE_FRIEND:
        {
            if(!m_Database.addFriend(m_SaveTmpInfo.m_Message))
            {
                m_SaveTmpInfo.m_Message.m_Type = MT_ADD_FRIEND_FAIL;
                break;
            }
            m_SaveTmpInfo.m_Message.m_SenderID = savTmp.m_Message.m_ReceiverID;
            m_SaveTmpInfo.m_Message.m_ReceiverID = savTmp.m_Message.m_SenderID;
            if(MT_GET_FRIEND_SUCCESS != m_Database.searchFriendInformationByID(m_SaveTmpInfo.m_Message.m_SenderID,
                                                                               m_SaveTmpInfo.m_UserInf))
            {
                m_SaveTmpInfo.m_Message.m_Type = MT_ADD_FRIEND_FAIL;
                return;
            }
            qDebug() << "agree add friend";
            break;
        }
        default:
        {
        return;
        }
    }
    QMap<QString, ServerSocket*>::iterator iter;
    iter = m_mapOnlineUser.find(m_SaveTmpInfo.m_Message.m_ReceiverID);
    if(m_mapOnlineUser.end() == iter)
    {
        //save offline msg
    }
    else
    {
        iter.value()->sendMessage(m_SaveTmpInfo);
    }
}

void TcpServer::talkRequest(const SaveTmpInformation &savTmp) //處理對話請求
{
    m_SaveTmpInfo.m_Message = savTmp.m_Message;
    switch(m_SaveTmpInfo.m_Message.m_Type)
    {
        case MT_TALK_MESSAGE:
        {
            m_SaveTmpInfo.m_ReplyKind = MT_TALK;
            QMap<QString, ServerSocket*>::iterator iter;
            iter = m_mapOnlineUser.find(m_SaveTmpInfo.m_Message.m_ReceiverID);

            //聊天記錄存儲到數據庫
            m_Database.addHistoryMsg(m_SaveTmpInfo.m_Message);
            if(m_mapOnlineUser.end() == iter)
            {
                //存儲離線消息到數據庫
                m_Database.addOfflineMsg(m_SaveTmpInfo.m_Message);
            }
            else
            {
                iter.value()->sendMessage(m_SaveTmpInfo);
            }
            break;
        }
        default:
        {
            break;
        }
    }
}

void TcpServer::tellFriendsStatusChanged(const QString &id, int status) //通知好友，改變用戶狀態
{
    m_SaveTmpInfo.m_ReplyKind = MT_CHANGE_STATUE;
    m_SaveTmpInfo.m_Status = status;
    m_SaveTmpInfo.m_PeerID = id;
    m_Database.searchFriendsID(id, m_SaveTmpInfo.m_IDVec);

    QMap<QString, ServerSocket*>::iterator iter;
    for(int i = 0;i < m_SaveTmpInfo.m_IDVec.size();i++)
    {
        iter = m_mapOnlineUser.find(m_SaveTmpInfo.m_IDVec[i]);
        if(iter != m_mapOnlineUser.end())
        {
            iter.value()->sendMessage(m_SaveTmpInfo);
        }
    }
}

void TcpServer::tellFriendsInformationChanged(const QString &id) //通知好友，改變用戶信息
{
    m_SaveTmpInfo.m_ReplyKind = MT_CHANGE_INFORMATION;
    if(!m_Database.searchFriendsID(id, m_SaveTmpInfo.m_IDVec))
    {
        return;
    }
    m_SaveTmpInfo.m_FriendInf.m_HeadPortrait = m_SaveTmpInfo.m_UserInf.m_HeadPortrait;
    m_SaveTmpInfo.m_FriendInf.m_NickName = m_SaveTmpInfo.m_UserInf.m_NickName;
    m_SaveTmpInfo.m_FriendInf.m_UserID = m_SaveTmpInfo.m_UserInf.m_UserID;
    m_SaveTmpInfo.m_FriendInf.m_Status = m_SaveTmpInfo.m_UserInf.m_Status;

    QMap<QString, ServerSocket*>::iterator iter;
    for(int i = 0;i < m_SaveTmpInfo.m_IDVec.size();i++)
    {
        iter = m_mapOnlineUser.find(m_SaveTmpInfo.m_IDVec[i]);
        if(iter != m_mapOnlineUser.end())
        {
            iter.value()->sendMessage(m_SaveTmpInfo);
        }
    }
}