通用的Makefile编写与简单的TCP Server

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
运行结果：