26.Qt-MyChat06 新用户进入

新用户进入
    提供处理新用户进入函数
    更新右侧TableWidget
    更新聊天记录
    更新在线人数

main.cpp

#include "widget.h"
#include <QApplication>
#include "dialoglist.h"

int main(int argc, char *argv[])
{
    QApplication a(argc, argv);
    //Widget w;
    //w.show();
    DialogList list;
    list.show();

    return a.exec();
}

widget.h

#ifndef WIDGET_H
#define WIDGET_H

#include <QWidget>
#include <QUdpSocket>

namespace Ui {
class Widget;
}

class Widget : public QWidget
{
    Q_OBJECT

    //分别代表 聊天信息、新用户加入、用户退出
    enum MsgType {Msg,UsrEnter,UsrLeft};

public:
    explicit Widget(QWidget *parent, QString name);
    ~Widget();

private:
    Ui::Widget *ui;

signals:
    //关闭窗口发送关闭信息
    void closeWidget();

public:
    //关闭事件
    void closeEvent(QCloseEvent *);

public:
   void sndMsg(MsgType type); //广播UDP消息
   void usrEnter(QString username);//处理新用户加入
   void usrLeft(QString usrname,QString time); //处理用户离开
   QString getUsr(); //获取用户名
   QString getMsg(); //获取聊天信息
private:
   QUdpSocket * udpSocket; //udp套接字
   qint16 port; //端口
   QString uName; //用户名

    void ReceiveMessage();   //接受UDP消息
};

#endif // WIDGET_H

widget.cpp

#include "widget.h"
#include "ui_widget.h"
#include <QDataStream>
#include <QMessageBox>
#include <QDateTime>

Widget::Widget(QWidget *parent, QString name) :
    QWidget(parent),
    ui(new Ui::Widget)
{
    ui->setupUi(this);

    //初始化操作
    udpSocket = new QUdpSocket(this);
    //用户名获取
    uName = name;
    //端口号
    this->port = 9999;

    //绑定端口号  绑定模式  共享地址，断线重连
    udpSocket->bind(this->port,QUdpSocket::ShareAddress |QUdpSocket::ReuseAddressHint);

    //发送新用户进入
    sndMsg(UsrEnter);
    //点击发送按钮发送消息
    connect(ui->sendBtn,&QPushButton::clicked,[=](){
        sndMsg(Msg);
    });

    //监听别人发送的数据
    connect(udpSocket,&QUdpSocket::readyRead,this,&Widget::ReceiveMessage);

}

//接受UDP的消息
void Widget::ReceiveMessage()
{
    //拿到数据报文
    //获取长度
    qint64 size = udpSocket->pendingDatagramSize();
    QByteArray array = QByteArray(size,0);
    udpSocket->readDatagram(array.data(),size);

    //解析数据
    //第一段  类型  第二段 用户名 第三段 内容
    QDataStream stream (&array,QIODevice::ReadOnly);

    int msgType; //读取到类型
    QString usrName;
    QString msg;

    //获取当前时间
    QString time = QDateTime::currentDateTime().toString("yyyy-MM-dd hh:mm:ss");

    stream >>msgType;

    switch (msgType) {
        case Msg: //普通聊天
            stream >> usrName >> msg;

            //最加聊天记录
            ui->msgBrowser->setTextColor(Qt::blue);
            ui->msgBrowser->append("[" + usrName  + "]" + time);
            ui->msgBrowser->append(msg);

            break;
        case UsrEnter:
            //更新右侧TableWidget
            stream >> usrName;
            usrEnter(usrName);

            break;
        case UsrLeft:
            //stream >> usrName;
           // usrLeft(usrName,time);
            break;
        default:
            break;
    }
}

//处理新用户加入
void Widget::usrEnter(QString username)
{
    //更新右侧TableWidget

    //判断右侧的列表是不是空的，空的就添加内容
    bool isEmpty1 = ui->usrTblWidget->findItems(username,Qt::MatchExactly).isEmpty();
    if(isEmpty1)
    {
        QTableWidgetItem * usr = new QTableWidgetItem(username);

        //插入行
        ui->usrTblWidget->insertRow(0);
        ui->usrTblWidget->setItem(0,0,usr);

        //追加聊天记录
        ui->msgBrowser->setTextColor(Qt::gray);
        ui->msgBrowser->append( QString("%1 上线了").arg(username));
        //在线人数更新
        ui->usrNumLbl->setText(QString("在线用户：%1人").arg(ui->usrTblWidget->rowCount()));

        //把自身信息广播出去
        sndMsg(UsrEnter);
    }

}

//广播UDP消息
void Widget::sndMsg(MsgType type)
{
    //发送的消息分为三种数据类型
    //发送的数据 做分段处理 第一段：类型  第二段：用户名 第三段 具体内容
    QByteArray array;
    QDataStream stream(&array, QIODevice::WriteOnly);

    stream << type << getUsr() ; //第一段内容 添加到流中 第二段 用户名
    switch(type){
    case Msg://发送普通消息
        if(ui->msgTxtEdit->toPlainText() == "") //判断如果用户没有输入内容，不发任何消息
                {
                    QMessageBox::warning(this,"警告","发送内容不能为空");
                    return ;
                }
        //第三段数据，具体说的话
        stream <<getMsg();
        break;
    case UsrEnter://发送新用户进入的消息
        break;
    case UsrLeft://发送用户离开的消息
        break;
    default:
        break;
    }

    //书写报文
    udpSocket->writeDatagram(array, QHostAddress::Broadcast, port);
}

//获取到用户名
QString Widget::getUsr()
{
    return this->uName;
}

//获取聊天信息
QString Widget::getMsg()
{
    QString str = ui->msgTxtEdit->toHtml();
    //清空输入框
    ui->msgTxtEdit->clear();
    ui->msgTxtEdit->setFocus();

    return str;
}

//重写关闭窗口的事件
void Widget::closeEvent(QCloseEvent * e)
{

    emit this->closeWidget();
    //sndMsg(UsrLeft);
    //断开套接字
    //udpSocket->close();
    //udpSocket->destroyed();

    QWidget::closeEvent(e);
}

Widget::~Widget()
{
    delete ui;
}

dialoglist.h

#ifndef DIALOGLIST_H
#define DIALOGLIST_H

#include <QWidget>

namespace Ui {
class DialogList;
}

class DialogList : public QWidget
{
    Q_OBJECT

public:
    explicit DialogList(QWidget *parent = 0);
    ~DialogList();

private:
    Ui::DialogList *ui;
    QVector< bool> isShow;
};

#endif // DIALOGLIST_H

dialoglist.cpp

#include "dialoglist.h"
#include "ui_dialoglist.h"
#include <QToolButton>
#include "widget.h"
#include <QMessageBox>

DialogList::DialogList(QWidget *parent) :
    QWidget(parent),
    ui(new Ui::DialogList)
{
    ui->setupUi(this);
    //设置标题
    setWindowTitle("MyChat");
    //设置图标
    setWindowIcon(QPixmap(":/images/qq.png"));

    //准备图标
    QList<QString>nameList;
    nameList << "aaaaaa1111" << "aaaaaa2222" <<"aaaaaa3333"<<"aaaaaa4444"<<"aaaaaa5555"
             <<"aaaaaa6666"<<"aaaaaa7777"<<"aaaaaa8888"<<"aaaaaa9999";
    QStringList iconNameList; //图标资源列表
        iconNameList << "ftbz"<< "ymrl" <<"qq" <<"Cherry"<< "dr"
                     <<"jj"<<"lswh"<<"qmnn"<<"wy";

    QVector <QToolButton *> vToolBtn;

    for(int i = 0; i < 9; i++)
    {
        //设置头像
        QToolButton * btn = new QToolButton;
        //设置文字
        //btn->setText("aaaaaaa");
        btn->setText(nameList[i]);
        //设置头像
        //btn->setIcon(QPixmap(":/images/ftbz.png"));
        //设置头像
        QString str = QString(":/images/%1.png").arg(iconNameList.at(i));
        btn->setIcon(QPixmap(str));
        //设置头像大小
        btn->setIconSize(QPixmap(":/images/ftbz.png").size());
        //设置按钮的风格  透明
        btn->setAutoRaise(true);
        //设置文字和图片一起显示
        btn->setToolButtonStyle(Qt::ToolButtonTextBesideIcon);
        //加到垂直布局中
        ui->vLayout->addWidget(btn);
        //将创建的button加入到容器中,保存储存的按钮
        vToolBtn.push_back(btn);
        //9个标示默认初始化
        isShow.push_back(false);
    }

    //对9个按钮 进行添加信号槽
    for(int i = 0 ; i < vToolBtn.size();i++)
    {
        connect(vToolBtn[i],&QToolButton::clicked,[=](){
            //如果被打开了，就不要再次打开
            if(isShow[i])
            {
                QString str = QString("%1窗口已经被打开了").arg(vToolBtn[i]->text());
                QMessageBox::warning(this,"警告",str);
                return;
            }

            isShow[i] = true;

            //弹出聊天对话框
            //构造聊天窗口时候 告诉这个窗口他的名字 参数1  顶层方式弹出 参数2  窗口名字
            //注意！ Widget构造函数 并没有这两个参数，需要重新写构造函数，增加一个新的参数QString name
            Widget * widget = new Widget(0,vToolBtn[i]->text());
            //设置窗口标题
            widget->setWindowTitle(vToolBtn[i]->text());
            widget->setWindowIcon(vToolBtn[i]->icon());
            widget->show();

            //监听窗口关闭的信息，发送者是widget发送的
            connect(widget,&Widget::closeWidget,[=](){

                    isShow[i] = false;
            });

        });

    }
}



DialogList::~DialogList()
{
    delete ui;
}