TCP Server示例
TcpServer.h
#pragma once
#include <mutex>
#include <queue>
using namespace std;
class TcpServer
{
public:
void Run(uint16_t nPort);
private:
void Init(uint16_t nPort);
void AddTask(int nSocket);
void Consume();
private:
mutex m_mutex;
queue<int> m_nSocketQueue;
};
TcpServer.cpp
#include "TcpServer.h"
#include <iostream>
#include <thread>
#include <chrono>
#include <stdlib.h>
#include <string.h>
#include <errno.h>
#include <sys/types.h>
#include <netinet/in.h>
#include <unistd.h>
#include <arpa/inet.h>
#include <sys/socket.h>
using namespace std;
#define BUFFER_LENGTH 2000
void TcpServer::Init(uint16_t nPort)
{
int error;
int svrSocket, cliSocket;
struct sockaddr_in svrAddr, cliAddr;
socklen_t nAddrLen = sizeof(cliAddr);
memset(&svrAddr, 0, sizeof(svrAddr));
memset(&cliAddr, 0, sizeof(cliAddr));
//创建套接字
svrSocket = socket(AF_INET, SOCK_STREAM, 0);
if (svrSocket == -1)
{
cout << "Invalid socket" << endl;
exit(1);
}
//设置地址
svrAddr.sin_addr.s_addr = htonl(INADDR_ANY);
svrAddr.sin_family = AF_INET;
svrAddr.sin_port = htons(nPort);
//绑定套接字
error = bind(svrSocket, (struct sockaddr*) & svrAddr, sizeof(svrAddr));
if (error == -1)
{
cout << "Bind faild:" << strerror(errno) << endl;
//关闭套接字
close(svrSocket);
exit(1);
}
//监听套接字
error = listen(svrSocket, 5);
if (error == -1)
{
cout << "Listen faild:" << strerror(errno) << endl;
//关闭套接字
close(svrSocket);
exit(1);
}
cout << "Server is running!" << endl;
while (1)
{
cliSocket = accept(svrSocket, (struct sockaddr*) & cliAddr, &nAddrLen);
if (cliSocket == -1)
{
cout << "Accept faild:" << strerror(errno) << endl;
break;
}
//从客户端接收消息
AddTask(cliSocket);
}
//关闭套接字
close(svrSocket);
}
void TcpServer::AddTask(int nSocket)
{
unique_lock <mutex> lockGuard(m_mutex);
m_nSocketQueue.push(nSocket);
}
void TcpServer::Consume()
{
while (true) {
queue<int> nSocketQueue;
do
{
unique_lock<mutex> lockGuard(m_mutex);
nSocketQueue = m_nSocketQueue;
queue<int> empty;
swap(empty, m_nSocketQueue);
} while (false);
while (!nSocketQueue.empty())
{
int cliSocket = nSocketQueue.front();
char recvBuf[BUFFER_LENGTH];
memset(recvBuf, 0, BUFFER_LENGTH);
socklen_t recvBuflen = BUFFER_LENGTH;
recv(cliSocket, recvBuf, recvBuflen, 0);
cout << "Consume thread:" << this_thread::get_id() << ",Consume " << "i:" << cliSocket << ",Receive from client:" << recvBuf << endl;
close(cliSocket);
nSocketQueue.pop();
}
this_thread::sleep_for(chrono::microseconds(50));
}
}
void TcpServer::Run(uint16_t nPort)
{
for (size_t i = 0; i < 2; i++)
{
thread consume(&TcpServer::Consume, this);
consume.detach();
}
Init(nPort);
}
main.cpp
#include "TcpServer.h"
int main(int argc, char** argv)
{
uint16_t nPort = 6600;
if (argc >= 2)
{
nPort = (uint16_t)atoi(argv[1]);
}
TcpServer server;
server.Run(nPort);
return 0;
}
Makefile示例
CC = g++ -std=c++11
OPT_LEVEL := -O2
RM = rm -rf
BIN_NAME = Server
OBJS = main.o TcpServer.o
INC_DIRS = -I./
OS_DEFS = -DNDEBUG
LIBS = -lpthread
CPP_DEPS = main.d TcpServer.d
all: ${BIN_NAME}
${BIN_NAME}: ${OBJS}
${CC} ${OBJS} $(LIBS) -o $@
%.o:%.cpp
$(CC) $(INC_DIRS) $(OS_DEFS) $(OPT_LEVEL) -Wall -c -fmessage-length=0 -MMD -MP -MF"$(@:%.o=%.d)" -MT"$(@:%.o=%.d)" -o "$@" "$<"
-include $(CPP_DEPS)
clean:
-$(RM) $(OBJS) $(CPP_DEPS) $(BIN_NAME)
编译与运行结果
文件结构:
文件编译:
通过curl模拟一条客户端请求:curl 127.0.0.1:6000
运行结果: