QT网络编程Tcp下C/S架构的即时通信实例

#ifndef TCPCLIENT_H#define TCPCLIENT_H #include <QDialog>#include <QListWidget>#include <QLineEdit>#include <QPushButton>#include <QLabel>#include <QGridLayout>#include <QtNetWork/QHostAddress>#include <QtNetWork/QTcpSocket> class TcpClient : public QDialog{ Q_OBJECT public: TcpClient(QWidget *parent = 0,Qt::WindowFlags f=0); ~TcpClient();private: QListWidget *contentListWidget; QLineEdit *sendLineEdit; QPushButton *sendBtn; QLabel *userNameLabel; QLineEdit *userNameLineEdit; QLabel *serverIPLabel; QLineEdit *serverIPLineEdit; QLabel *portLabel; QLineEdit *portLineEdit; QPushButton *enterBtn; QGridLayout *mainLayout; bool status; int port; QHostAddress *serverIP; QString userName; QTcpSocket *tcpSocket;public slots: void slotEnter(); void slotConnected(); void slotDisconnected(); void dataReceived(); void slotSend();}; #endif // TCPCLIENT_H

#include "tcpclient.h"#include <QMessageBox>#include <QHostInfo> TcpClient::TcpClient(QWidget *parent,Qt::WindowFlags f) : QDialog(parent,f){ setWindowTitle(tr("TCP Client"));  contentListWidget = new QListWidget;  sendLineEdit = new QLineEdit; sendBtn = new QPushButton(tr("send"));  userNameLabel = new QLabel(tr("name")); userNameLineEdit = new QLineEdit;  serverIPLabel = new QLabel(tr("server")); serverIPLineEdit = new QLineEdit;  portLabel = new QLabel(tr("port")); portLineEdit = new QLineEdit;  enterBtn= new QPushButton(tr("join chat"));  mainLayout = new QGridLayout(this); mainLayout->addWidget(contentListWidget,0,0,1,2); mainLayout->addWidget(sendLineEdit,1,0); mainLayout->addWidget(sendBtn,1,1); mainLayout->addWidget(userNameLabel,2,0); mainLayout->addWidget(userNameLineEdit,2,1); mainLayout->addWidget(serverIPLabel,3,0); mainLayout->addWidget(serverIPLineEdit,3,1); mainLayout->addWidget(portLabel,4,0); mainLayout->addWidget(portLineEdit,4,1); mainLayout->addWidget(enterBtn,5,0,1,2);  status = false;  port = 8010; portLineEdit->setText(QString::number(port));  serverIP =new QHostAddress();  connect(enterBtn,SIGNAL(clicked()),this,SLOT(slotEnter())); connect(sendBtn,SIGNAL(clicked()),this,SLOT(slotSend()));  sendBtn->setEnabled(false);} TcpClient::~TcpClient(){ } void TcpClient::slotEnter(){ if(!status) {  QString ip = serverIPLineEdit->text();  if(!serverIP->setAddress(ip))  {   QMessageBox::information(this,tr("error"),tr("server ip address error!"));   return;  }   if(userNameLineEdit->text()=="")  {   QMessageBox::information(this,tr("error"),tr("User name error!"));   return;  }   userName=userNameLineEdit->text();   tcpSocket = new QTcpSocket(this);  connect(tcpSocket,SIGNAL(connected()),this,SLOT(slotConnected()));  connect(tcpSocket,SIGNAL(disconnected()),this,SLOT(slotDisconnected()));  connect(tcpSocket,SIGNAL(readyRead()),this,SLOT(dataReceived()));   tcpSocket->connectToHost(*serverIP,port);   status=true; } else {  int length=0;  QString msg=userName+tr(":Leave Chat Room");  if((length=tcpSocket->write(msg.toLatin1(),msg.length()))!=msg. length())  {   return;  }   tcpSocket->disconnectFromHost();   status=false; }} void TcpClient::slotConnected(){ sendBtn->setEnabled(true); enterBtn->setText(tr("离开"));  int length=0; QString msg=userName+tr(":Enter Chat Room"); if((length=tcpSocket->write(msg.toLatin1(),msg.length()))!=msg.length()) {  return; }} void TcpClient::slotSend(){ if(sendLineEdit->text()=="") {  return ; }  QString msg=userName+":"+sendLineEdit->text();  tcpSocket->write(msg.toLatin1(),msg.length()); sendLineEdit->clear();} void TcpClient::slotDisconnected(){ sendBtn->setEnabled(false); enterBtn->setText(tr("join chat"));} void TcpClient::dataReceived(){ while(tcpSocket->bytesAvailable()>0) {  QByteArray datagram;  datagram.resize(tcpSocket->bytesAvailable());   tcpSocket->read(datagram.data(),datagram.size());   QString msg=datagram.data();  contentListWidget->addItem(msg.left(datagram.size())); }}

#ifndef TCPSERVER_H#define TCPSERVER_H #include <QDialog>#include <QListWidget>#include <QLabel>#include <QLineEdit>#include <QPushButton>#include <QGridLayout>#include "server.h" class TcpServer : public QDialog{ Q_OBJECT public: TcpServer(QWidget *parent = 0,Qt::WindowFlags f=0); ~TcpServer();private: QListWidget *ContentListWidget; QLabel *PortLabel; QLineEdit *PortLineEdit; QPushButton *CreateBtn; QGridLayout *mainLayout; int port; Server *server;public slots: void slotCreateServer(); void updateServer(QString,int);}; #endif // TCPSERVER_H

#include "tcpserver.h" TcpServer::TcpServer(QWidget *parent,Qt::WindowFlags f) : QDialog(parent,f){ setWindowTitle(tr("TCP Server"));  ContentListWidget = new QListWidget;  PortLabel = new QLabel(tr(" port")); PortLineEdit = new QLineEdit;  CreateBtn = new QPushButton(tr("create chat")); mainLayout = new QGridLayout(this); mainLayout->addWidget(ContentListWidget,0,0,1,2); mainLayout->addWidget(PortLabel,1,0); mainLayout->addWidget(PortLineEdit,1,1); mainLayout->addWidget(CreateBtn,2,0,1,2);  port=8010; PortLineEdit->setText(QString::number(port));  connect(CreateBtn,SIGNAL(clicked()),this,SLOT(slotCreateServer()));} TcpServer::~TcpServer(){ } void TcpServer::slotCreateServer(){ server = new Server(this,port); connect(server,SIGNAL(updateServer(QString,int)),this,SLOT(updateServer(QString,int)));  CreateBtn->setEnabled(false);} void TcpServer::updateServer(QString msg,int length){ ContentListWidget->addItem(msg.left(length));}

#ifndef TCPCLIENTSOCKET_H#define TCPCLIENTSOCKET_H #include <QtNetWork/QTcpSocket>#include <QObject> class TcpClientSocket : public QTcpSocket{ Q_OBJECTpublic: TcpClientSocket();signals: void updateClients(QString,int); void disconnected(int);protected slots: void dataReceived(); void slotDisconnected();}; #endif // TCPCLIENTSOCKET_H

#include "tcpclientsocket.h" TcpClientSocket::TcpClientSocket(){ connect(this,SIGNAL(readyRead()),this,SLOT(dataReceived())); connect(this,SIGNAL(disconnected()),this,SLOT(slotDisconnected()));} void TcpClientSocket::dataReceived(){ while(bytesAvailable()>0) {  int length = bytesAvailable();  char buf[1024];  read(buf,length);   QString msg=buf;  emit updateClients(msg,length); }} void TcpClientSocket::slotDisconnected(){ emit disconnected(this->socketDescriptor());}

#ifndef SERVER_H#define SERVER_H #include <QtNetWork/QTcpServer>#include <QObject>#include "tcpclientsocket.h" class Server : public QTcpServer{ Q_OBJECTpublic: Server(QObject *parent=0,int port=0); QList<TcpClientSocket*> tcpClientSocketList;signals: void updateServer(QString,int);public slots: void updateClients(QString,int); void slotDisconnected(int);protected: void incomingConnection(int socketDescriptor);}; #endif // SERVER_H

#include "server.h" Server::Server(QObject *parent,int port) :QTcpServer(parent){ listen(QHostAddress::Any,port);} void Server::incomingConnection(int socketDescriptor){ TcpClientSocket *tcpClientSocket = new TcpClientSocket; connect(tcpClientSocket,SIGNAL(updateClients(QString,int)),this,SLOT(updateClients(QString,int))); connect(tcpClientSocket,SIGNAL(disconnected(int)),this,SLOT(slotDisconnected(int)));  tcpClientSocket->setSocketDescriptor(socketDescriptor);  tcpClientSocketList.append(tcpClientSocket);} void Server::updateClients(QString msg,int length){ emit updateServer(msg,length); for(int i=0;i<tcpClientSocketList.count();i++) {  QTcpSocket *item = tcpClientSocketList.at(i);  if(item->write(msg.toLatin1(),length)!=length)  {   continue;  } }} void Server::slotDisconnected(int descriptor){ for(int i=0;i<tcpClientSocketList.count();i++) {  QTcpSocket *item = tcpClientSocketList.at(i);  if(item->socketDescriptor()==descriptor)  {   tcpClientSocketList.removeAt(i);   return;  } } return;}