day5 QT

作业

客户端

#include "widget.h"
#include "ui_widget.h"

Widget::Widget(QWidget *parent)
    : QWidget(parent)
    , ui(new Ui::Widget)
    , socket(new QTcpSocket(this))//给客户端指针实例化空间
{
    ui->setupUi(this);

    //初始化界面
    ui->send_btn->setEnabled(false);
    ui->msg_edit->setEnabled(false);
    ui->disconnect_btn->setEnabled(false);



    //如果客户端成功连接服务器，那么客户端自动发射一个connected（）信号
    //我们就可以将该信号连接到自定义的槽函数处理逻辑代码，由于只需要连接一次，所以连接函数写在构造函数中
    connect(socket,&QTcpSocket::connected,this,&Widget::connected_slot);

    //如果服务器向客户端发送数据，那么客户端就会自动发射一个readyRead()信号
    //我们就可以将该信号链接到自定义的槽函数，读取发送来的数据
    connect(socket,&QTcpSocket::readyRead,this,&Widget::readyRead_slot);



    //如果成功与服务器断开连接，那么客户端就会自动发射disconnected信号
    //我们就可以将该函数连接到自定义的槽函数中，处理逻辑代码，由于字需要连接一次，所以我们将连接函数写在构造函数中
    connect(socket,&QTcpSocket::disconnected,this,&Widget::disconnected_slot);
}

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



//连接服务器按钮对应的槽函数
void Widget::on_connect_btn_clicked()
{
    //获取ui界面上的ip和端口号
    QString ip = ui->ip_edit->text();

    quint16 port = ui->port_edit->text().toUInt();

    //让客户端连接服务器
    //connectToHost(const QString &hostName, quint16 port, OpenMode mode = ReadWrite, NetworkLayerProtocol protocol = AnyIPProtocol);
    //参数1：连接服务器的IP地址
    //参数2：连接服务器的端口号
    socket->connectToHost(ip,port);
}

void Widget::connected_slot()
{
    username = ui->username_edit->text();

    //告诉服务器，客户端已连接
    QString msg = username + ":进入聊天室";

    //将信息发送给服务器
    socket->write(msg.toLocal8Bit());

    ui->send_btn->setEnabled(true);
    ui->msg_edit->setEnabled(true);
    ui->disconnect_btn->setEnabled(true);

    ui->username_edit->setEnabled(false);
    ui->ip_edit->setEnabled(false);
    ui->port_edit->setEnabled(false);
    ui->connect_btn->setEnabled(false);

}

//发送按钮对应的槽函数
void Widget::on_send_btn_clicked()
{
    //获取ui界面输入框中的文本
    QString msg = username + ":" + ui->msg_edit->text();

    //将信息发送给服务器
    socket->write(msg.toLocal8Bit());

    //申请单独列表项用于显示自己发的信息
    QListWidgetItem *item = new QListWidgetItem();

    //申请标签存放自己的信息，并设置标签对齐方式为右对齐
    QLabel *label = new QLabel(ui->msg_edit->text(),this);
    label->setAlignment(Qt::AlignRight);

    //添加列表项
    ui->listWidget->addItem(item);

    //设置label作为新添加列表项的显示内容
    ui->listWidget->setItemWidget(item,label);


    //清空输入框
    ui->msg_edit->clear();
}

void Widget::readyRead_slot()
{
    //读取服务器发来的数据
    QByteArray msg = socket->readAll();

    //将读取到的信息放入ui界面上
    ui->listWidget->addItem(QString::fromLocal8Bit(msg));
}

//断开连接按钮对应的槽函数
void Widget::on_disconnect_btn_clicked()
{
    //告诉服务器，我走了
    QString msg = username + ":" + "离开聊天室";

    socket->write(msg.toLocal8Bit());

    //断开与服务器的连接
    socket->disconnectFromHost();

}

//成功与服务器断开连接后执行的槽函数
void Widget::disconnected_slot()
{
    ui->send_btn->setEnabled(false);
    ui->msg_edit->setEnabled(false);
    ui->disconnect_btn->setEnabled(false);

    ui->username_edit->setEnabled(true);
    ui->ip_edit->setEnabled(true);
    ui->port_edit->setEnabled(true);
    ui->connect_btn->setEnabled(true);
}

服务器端

#include "widget.h"
#include "ui_widget.h"



Widget::Widget(QWidget *parent)
    : QWidget(parent)
    , ui(new Ui::Widget)
    , server(new QTcpServer(this))//给服务器指针实例化空间
{
    ui->setupUi(this);
}

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

//启动按钮槽函数
void Widget::on_start_btn_clicked()
{
    //获取ui界面的端口号
    quint16 port = ui->port_edit->text().toUInt();//将字符串转换成整形


    //服务器设置监听
    //bool server->listen(const QHostAddress &address = QHostAddress::Any, quint16 port = 0)
    //监听成功返回true，失败返回false

    if(server->listen(QHostAddress::Any,port))
    {
        //监听成功
        QMessageBox::information(this,"提示","启动服务器成功");
        //成功后按钮不可按
        ui->start_btn->setEnabled(false);
    }
    else
    {
        //监听失败
        QMessageBox::information(this,"","启动服务器失败");
        return;
    }
    //程序至此，说明监听成功，如果客户端发来连接请求，那服务器就会自动发射一个newConnection()信号
    //我们就可以将该信号连接到自定义的槽函数中，处理相关逻辑代码

    connect(server,&QTcpServer::newConnection,this,&Widget::newConnection_slot);

}


//槽函数的实现
void Widget::newConnection_slot()
{
    qDebug() << "有新的客户端连接...";

    //获取最新连接的客户端的套接字
    QTcpSocket *s = server->nextPendingConnection();


    //将客户端放入客户端容器中
    socketList.push_back(s);

    //此时说明客户端和服务器已经建立了连接，如果客户端向服务器端发来数据，那么客户端就会自动发射一个readyRead（）信号
    //我们就可以将该信号连接到自定义的槽函数中，读取客户端的数据
    connect(s,&QTcpSocket::readyRead,this,&Widget::readyRead_slot);

}


//readyRead对应的槽函数
void Widget::readyRead_slot()
{
    //读取客户端的数据


    //遍历客户端容器，移除无效客户端
                    //容器里的元素个数
    for(int i=0; i<socketList.count(); i++)
    {
        //判断客户端和服务器的连接状态
        //为连接状态的枚举值是 0
        if(socketList.at(i)->state() == 0)
        {
            //移除无效客户端
            socketList.removeAt(i);
        }

        //遍历客户端容器，判断哪个客户端有数据待读
        for(int i=0; i< socketList.count(); i++)
        {
            //判断哪个客户端有数据带读
            //qint64 bytesAvailable() const override;
            //接收数据的字节大小
            if(socketList.at(i)->bytesAvailable() != 0)
            {
                //读取数据

                QByteArray msg = socketList.at(i)->readAll();

                //将读取到的数据放到ui界面上
                ui->listWidget->addItem(QString::fromLocal8Bit(msg));

                //将数据广播给所有客户端
                for(int j=0; j<socketList.count(); j++)
                {
                    if(j != i)
                    {
                        socketList.at(j)->write(msg);
                    }
                }
            }
        }
    }
}