还是之前udp部分的socket函数,这里只是简单说明一下与udp的差异
int socket(int domain, int type, int protocol);
只需将第二个参数type换成:
SOCK_STREAM: 基于TCP的网络通信,流式套接字,提供的是流式服务(对应TCP的特点:面向字节流)
还是和之前一样的接口
UDP服务器的初始化操作只有2步,第一步:创建套接字,第二步:是绑定。但是TCP服务器是面向连接的,客户端在正式向TCP服务器发送数据之前,需要先与TCP服务器建立连接,然后才能与服务器进行通信。
因此TCP服务器需要时刻注意是否有客户端发来连接请求,此时就需要将TCP服务器创建的套接字设置为监听状态
listen for connections on a socket: 监听套接字上的连接
头文件:
#include
#include
函数原型:
int listen(int sockfd, int backlog);
参数说明:
第一个参数sockfd: 需要设置为监听状态的套接字对应的文件描述符
第二个参数backlog: 这里当成一个整数,后续详细解释
返回值:
监听成功: 返回0
监听失败: 失败返回-1,并设置错误码
客户端有新链接到来,服务端可以获取到新链接,这一步需要死循环获取客户端新链接。
accept a connection on a socket: 接收套接字上的连接
头文件:
#include
#include
函数原型:
int accept(int sockfd, struct sockaddr *addr, socklen_t *addrlen);
参数说明:
第一个参数sockfd: 监听套接字
第二个参数addr: 获取对方一端网络相关的属性信息
第三个参数addrlen: addr的长度
返回值:
连接成功: 返回接收到的套接字的文件描述符
连接失败: 失败返回-1,并设置错误码
关于accept的返回值: 也是一个文件描述符
为什么又返回一个新的文件描述符??返回的这个新的文件描述符跟旧的文件描述符_sockfd有什么关系?
感性理解:
对比listen监听套接字与accept函数返回的套接字
因为TC 提供的是流式服务,所以这里利用read和write来实现读取与写入
4步:创建套接字,客户端向服务器发起连接请求,bind(不需要自己绑定,由OS自动分配),处理数据读取与写入
initiate a connection on a socket: 在套接字上发起连接
头文件:
#include
#include
函数原型:
int connect(int sockfd, const struct sockaddr *addr, socklen_t addrlen);
参数说明:
第一个参数sockfd: 表示通过该套接字发起连接请求
第二个参数addr: 对方一端网络相关的属性信息
第三个参数addrlen: addr的长度
返回值:
连接成功: 返回0
连接失败: 失败返回-1,并设置错误码
这里一共提供4个版本的tcp代码
err.hpp
:这个代码是公用的后续不在给出
#pragma once
enum
{
USAGE_ERR=1,
SOCKET_ERR,
BIND_ERR,
LISTEN_ERR,
CONNECT_ERR,
};
客户端向服务端发送消息,服务端收到后再把消息发回给客户端
tcpServer.hpp
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include"err.hpp"
using namespace std;
// 问题: 目前的服务器, 无法处理多个client的问题, 为什么?
// 单进程服务, 当服务端向客户端提供业务处理服务时, 没有办法accet, 不能处理连接
namespace ns_server
{
static const uint16_t defaultport=8081;
static int backlog=32;
using func_t=function<string(const string&)>; // 回调函数,一种处理逻辑
class TcpServer
{
public:
TcpServer(func_t func, uint16_t port=defaultport)
:func_(func)
,port_(port)
,quit_(true)
{
}
void InitServer()
{
// 1. 创建socket文件
listensock_=socket(AF_INET,SOCK_STREAM,0);
if(listensock_<0)
{
cerr<<"create socket error"<<endl;
exit(SOCKET_ERR);
}
// 2. bind
struct sockaddr_in local;
memset(&local,0,sizeof(local));
local.sin_port=htons(port_);
local.sin_family=AF_INET;
local.sin_addr.s_addr=INADDR_ANY;
int n=bind(listensock_,(struct sockaddr*)&local,sizeof(local));
if(n<0)
{
cerr<<"bind socket error"<<endl;
exit(BIND_ERR);
}
// 3. 监听
int m=listen(listensock_,backlog);
if(m<0)
{
cerr<<"listen socket error"<<endl;
exit(LISTEN_ERR);
}
}
void Start()
{
quit_=false;
while(!quit_)
{
struct sockaddr_in client;
socklen_t len=sizeof(client);
// 4. 获取连接, accept
int sock=accept(listensock_,(struct sockaddr*)&client,&len);
if (sock < 0) // accept失败并不会终止进程, 只要获取下一个连接
{
cerr << "accept error" << endl;
continue;
}
// 提取client信息 --- debug
string clientip=inet_ntoa(client.sin_addr); // 把4字节对应的IP转化成字符串风格
uint16_t clientport=ntohs(client.sin_port); // 网络序列转主机序列
// 5. 获取新连接成功后, 开始进行业务处理
cout<<"获取新连接成功: "<<sock<<" from "<<listensock_<<", "<< clientip << "- " <<clientport<<endl;
// v1
service(sock,clientip,clientport);
}
}
// 流式 - 利用read和write
void service(int sock, const string&clientip,const uint16_t clientport)
{
string who=clientip + "-" + to_string(clientport);
char buffer[1024];
while(true)
{
ssize_t s=read(sock,buffer,sizeof(buffer)-1);
if(s>0)
{
buffer[s]=0;
string res=func_(buffer); // 进行回调
cout<<who<< ">>> " <<res<<endl;
// 把收到的消息返回(写给客户端)
write(sock,res.c_str(),res.size());
}
else if(s==0)
{
// 对方将连接关闭了
close(sock);
cout<< who <<" quit, me too"<<endl;
break;
}
else
{
close(sock);
cerr<<"read error: "<<strerror(errno)<<endl;
break;
}
}
}
~TcpServer()
{
}
private:
uint16_t port_;
int listensock_;
bool quit_; // 标志服务器是否运行字段
func_t func_;
};
}
tcpServer.cc
#include"tcpServer.hpp"
using namespace ns_server;
// ./tcp_server port
// 使用手册
static void usage(string proc)
{
cout<<"usage:\n\t"<<proc<<" port\n"<<endl;
}
string echo(const string&message)
{
return message;
}
int main(int argc,char*argv[])
{
if(argc!=2)
{
usage(argv[0]);
exit(USAGE_ERR);
}
uint16_t port=atoi(argv[1]);
unique_ptr<TcpServer> tsvr(new TcpServer(echo,port));
tsvr->InitServer();
tsvr->Start();
return 0;
}
tcpClient.cc
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include"err.hpp"
using namespace std;
static void usage(string proc)
{
cout<<"usage:\n\t"<<proc<<" serverip serverport\n" <<endl;
}
// ./tcp_client serverip serverport
int main(int argc,char*argv[])
{
// 准备工作
if(argc!=3)
{
usage(argv[0]);
exit(USAGE_ERR);
}
string serverip=argv[1];
uint16_t serverport=atoi(argv[2]);
// 1.创建套接字
int sock=socket(AF_INET,SOCK_STREAM,0);
if (sock < 0)
{
cerr << "create socket error: " << strerror(errno) << endl;
exit(SOCKET_ERR);
}
// (2) 客户端要不要bind呢? 要
// 要不要自己bind呢? 不要, 因为client要让OS自动给用户进行bind
// (3) 要不要listen?不要, 客户端连别人, 永远都是别人listen; 要不要accept?不要, 服务器来连接
// 2. connect 客户端向服务器发起连接请求
struct sockaddr_in server;
memset(&server,0,sizeof(server));
server.sin_port=htons(serverport);
server.sin_family=AF_INET;
// server.sin_addr.s_addr=INADDR_ANY; 绝对不是
inet_aton(serverip.c_str(),&(server.sin_addr)); // 字符串风格ip转成点分十进制
int cnt=5;
while(connect(sock,(struct sockaddr*)&server,sizeof(server))!=0) // 连接失败
{
sleep(1);
cout<<"正在给你重连, 重连次数还有: "<<cnt--<<endl;
if(cnt<=0)
break;
}
if(cnt<=0)
{
cerr<<"连接失败"<<endl;
exit(CONNECT_ERR);
}
char buffer[1024];
// 3. 连接成功
while(true)
{
string line;
cout<<"Enter>> ";
getline(cin,line);
write(sock,line.c_str(),line.size());
ssize_t s = read(sock, buffer, sizeof(buffer)-1);
if (s > 0)
{
buffer[s] = 0;
cout<<"server echo >>>"<<buffer<<endl;
}
else if (s == 0)
{
cerr << "server quit" << endl;
break;
}
else
{
cerr << "read error: " << strerror(errno) << endl;
break;
}
}
close(sock);
return 0;
}
运行结果:
v2版本是把单执行流服务器改成多进程版的服务器
在accept获取新连接成功后,fork创建创建子进程,此时子进程对外提供服务, 父进程只进行accept
父进程的文件描述符会被子进程继承,但并不是父子共用同一张文件描述符表,因为子进程会拷贝继承父进程的文件描述符表
对于套接字文件也是相同的,父进程创建的子进程也会继承父进程的套接字文件,此时子进程就能够对特定的套接字文件进行读写操作,进而完成对对应客户端的服务
关于阻塞等待与非阻塞等待
常见的方式有两种:
下面是创建孙进程的方案:
tcpServer.hpp
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include"err.hpp"
using namespace std;
namespace ns_server
{
static const uint16_t defaultport=8081;
static int backlog=32;
using func_t=function<string(const string&)>; // 回调函数,一种处理逻辑
class TcpServer
{
public:
TcpServer(func_t func, uint16_t port=defaultport)
:func_(func)
,port_(port)
,quit_(true)
{
}
void InitServer()
{
// 1. 创建socket文件
listensock_=socket(AF_INET,SOCK_STREAM,0);
if(listensock_<0)
{
cerr<<"create socket error"<<endl;
exit(SOCKET_ERR);
}
// 2. bind
struct sockaddr_in local;
memset(&local,0,sizeof(local));
local.sin_port=htons(port_);
local.sin_family=AF_INET;
local.sin_addr.s_addr=INADDR_ANY;
int n=bind(listensock_,(struct sockaddr*)&local,sizeof(local));
if(n<0)
{
cerr<<"bind socket error"<<endl;
exit(BIND_ERR);
}
// 3. 监听
int m=listen(listensock_,backlog);
if(m<0)
{
cerr<<"listen socket error"<<endl;
exit(LISTEN_ERR);
}
}
void Start()
{
// signal(SIGCHLD,SIG_IGN); // ok, 最推荐
// signal(SIGCHLD,handler); // 回收子进程, 不太推荐
quit_=false;
while(!quit_)
{
struct sockaddr_in client;
socklen_t len=sizeof(client);
// 4. 获取连接, accept
int sock=accept(listensock_,(struct sockaddr*)&client,&len);
if (sock < 0) // accept失败并不会终止进程, 只要获取下一个连接
{
cerr << "accept error" << endl;
continue;
}
// 提取client信息 --- debug
string clientip=inet_ntoa(client.sin_addr); // 把4字节对应的IP转化成字符串风格
uint16_t clientport=ntohs(client.sin_port); // 网络序列转主机序列
// 5. 获取新连接成功后, 开始进行业务处理
cout<<"获取新连接成功: "<<sock<<" from "<<listensock_<<", "<< clientip << "- " <<clientport<<endl;
// v2: 多进程版本
// 子进程对外提供服务, 父进程只进行accept
pid_t id=fork();
if(id<0)
{
close(sock);
continue;
}
else if(id==0) // child, 父进程的fd会被子进程继承吗? 会; 父子会用同一张文件描述符表吗?不会, 子进程会拷贝继承父进程的fd table
{
// 建议关闭掉不需要的fd
close(listensock_);
if(fork()>0) exit(0); // 就这一行代码
// 子进程已经退了(则下面的wait立马返回, 回收子进程资源), 孙子进程在运行(无父进程, 变成孤儿进程, 被系统领养),提供服务
// 孙子进程的回收工作由系统来承担
service(sock,clientip,clientport);
exit(0);
}
// 父进程, 一定要关闭不需要的fd(否则会导致父进程的文件描述符变少, 即父进程文件描述符资源的浪费[文件描述符泄露])
close(sock);
// 不等待子进程, 会导致子进程僵尸之后无法回收, 近而导致内存泄漏
pid_t ret=waitpid(id,nullptr,0); // 父进程默认是阻塞的, waitpid(id,nullptr,WNOHANG);不推荐
if(ret==id)
cout<< "wait child "<<id<< " success" <<endl;
}
}
// 流式 - 利用read和write
void service(int sock, const string&clientip,const uint16_t clientport)
{
string who=clientip + "-" + to_string(clientport);
char buffer[1024];
while(true)
{
ssize_t s=read(sock,buffer,sizeof(buffer)-1);
if(s>0)
{
buffer[s]=0;
string res=func_(buffer); // 进行回调
cout<<who<< ">>> " <<res<<endl;
// 把收到的消息返回(写给客户端)
write(sock,res.c_str(),res.size());
}
else if(s==0)
{
// 对方将连接关闭了
close(sock);
cout<< who <<" quit, me too"<<endl;
break;
}
else
{
close(sock);
cerr<<"read error: "<<strerror(errno)<<endl;
break;
}
}
}
~TcpServer()
{
}
private:
uint16_t port_;
int listensock_;
bool quit_; // 标志服务器是否运行字段
func_t func_;
};
}
tcpServer.cc
#include"tcpServer.hpp"
using namespace ns_server;
// ./tcp_server port
// 使用手册
static void usage(string proc)
{
cout<<"usage:\n\t"<<proc<<" port\n"<<endl;
}
string echo(const string&message)
{
return message;
}
int main(int argc,char*argv[])
{
if(argc!=2)
{
usage(argv[0]);
exit(USAGE_ERR);
}
uint16_t port=atoi(argv[1]);
unique_ptr<TcpServer> tsvr(new TcpServer(echo,port));
tsvr->InitServer();
tsvr->Start();
return 0;
}
tcpClient.cc
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include"err.hpp"
using namespace std;
static void usage(string proc)
{
cout<<"usage:\n\t"<<proc<<" serverip serverport\n" <<endl;
}
// ./tcp_client serverip serverport
int main(int argc,char*argv[])
{
// 准备工作
if(argc!=3)
{
usage(argv[0]);
exit(USAGE_ERR);
}
string serverip=argv[1];
uint16_t serverport=atoi(argv[2]);
// 1.创建套接字
int sock=socket(AF_INET,SOCK_STREAM,0);
if (sock < 0)
{
cerr << "create socket error: " << strerror(errno) << endl;
exit(SOCKET_ERR);
}
// (2) 客户端要不要bind呢? 要
// 要不要自己bind呢? 不要, 因为client要让OS自动给用户进行bind
// (3) 要不要listen?不要, 客户端连别人, 永远都是别人listen; 要不要accept?不要, 服务器来连接
// 2. connect 客户端向服务器发起连接请求
struct sockaddr_in server;
memset(&server,0,sizeof(server));
server.sin_port=htons(serverport);
server.sin_family=AF_INET;
// server.sin_addr.s_addr=INADDR_ANY; 绝对不是
inet_aton(serverip.c_str(),&(server.sin_addr)); // 字符串风格ip转成点分十进制
int cnt=5;
while(connect(sock,(struct sockaddr*)&server,sizeof(server))!=0) // 连接失败
{
sleep(1);
cout<<"正在给你重连, 重连次数还有: "<<cnt--<<endl;
if(cnt<=0)
break;
}
if(cnt<=0)
{
cerr<<"连接失败"<<endl;
exit(CONNECT_ERR);
}
char buffer[1024];
// 3. 连接成功
while(true)
{
string line;
cout<<"Enter>> ";
getline(cin,line);
write(sock,line.c_str(),line.size());
ssize_t s = read(sock, buffer, sizeof(buffer)-1);
if (s > 0)
{
buffer[s] = 0;
cout<<"server echo >>>"<<buffer<<endl;
}
else if (s == 0)
{
cerr << "server quit" << endl;
break;
}
else
{
cerr << "read error: " << strerror(errno) << endl;
break;
}
}
close(sock);
return 0;
}
运行结果:
频繁的创建进程会给OS带来巨大的负担,并且创建线程的成本比创建线程高得多。因此在实现多执行流的服务器时最好采用多线程进行实现。
主线程创建出新线程后,也是需要等待新线程退出的,否则也会造成类似于僵尸进程这样的问题。但对于线程来说,如果不想让主线程等待新线程退出,直接线程分离即可,当这个线程退出时系统会自动回收该线程所对应的资源。
各个线程共享是同一张文件描述符表,也就是说服务进程(主线程)调用accept函数获取到一个文件描述符后,其他创建的新线程是能够直接访问这个文件描述符的。
所以不能关闭不要的套接字文件描述符,该文件描述符的关闭操作应该又新线程来执行。因为是新线程为客户端提供服务的,只有当新线程为客户端提供的服务结束后才能将该文件描述符关闭。
tcpServer.hpp
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include"err.hpp"
using namespace std;
namespace ns_server
{
static const uint16_t defaultport=8081;
static int backlog=32;
using func_t=function<string(const string&)>; // 回调函数,一种处理逻辑
class TcpServer;
class ThreadData
{
public:
ThreadData(int fd, const string&ip,const uint16_t&port,TcpServer*ts)
:sock(fd)
,clientip(ip)
,clientport(port)
,current(ts)
{
}
public:
int sock;
string clientip;
uint16_t clientport;
TcpServer*current;
};
class TcpServer
{
public:
TcpServer(func_t func, uint16_t port=defaultport)
:func_(func)
,port_(port)
,quit_(true)
{
}
void InitServer()
{
// 1. 创建socket文件
listensock_=socket(AF_INET,SOCK_STREAM,0);
if(listensock_<0)
{
cerr<<"create socket error"<<endl;
exit(SOCKET_ERR);
}
// 2. bind
struct sockaddr_in local;
memset(&local,0,sizeof(local));
local.sin_port=htons(port_);
local.sin_family=AF_INET;
local.sin_addr.s_addr=INADDR_ANY;
int n=bind(listensock_,(struct sockaddr*)&local,sizeof(local));
if(n<0)
{
cerr<<"bind socket error"<<endl;
exit(BIND_ERR);
}
// 3. 监听
int m=listen(listensock_,backlog);
if(m<0)
{
cerr<<"listen socket error"<<endl;
exit(LISTEN_ERR);
}
}
void Start()
{
// signal(SIGCHLD,SIG_IGN); // ok, 最推荐
// signal(SIGCHLD,handler); // 回收子进程, 不太推荐
quit_=false;
while(!quit_)
{
struct sockaddr_in client;
socklen_t len=sizeof(client);
// 4. 获取连接, accept
int sock=accept(listensock_,(struct sockaddr*)&client,&len);
if (sock < 0) // accept失败并不会终止进程, 只要获取下一个连接
{
cerr << "accept error" << endl;
continue;
}
// 提取client信息 --- debug
string clientip=inet_ntoa(client.sin_addr); // 把4字节对应的IP转化成字符串风格
uint16_t clientport=ntohs(client.sin_port); // 网络序列转主机序列
// 5. 获取新连接成功后, 开始进行业务处理
cout<<"获取新连接成功: "<<sock<<" from "<<listensock_<<", "<< clientip << "- " <<clientport<<endl;
// v3: 多线程版本 --- 原生多线程
// 1. 要不要关闭不要的socket? 绝对不能,一个进程的文件描述符表共享, 关了影响其他线程
// 2. 要不要回收线程?要;如何回收?会不会阻塞
pthread_t tid;
ThreadData*td=new ThreadData(sock,clientip,clientport,this); // 要开出一块独立的空间
pthread_create(&tid,nullptr,threadRoutine,td);
}
}
static void*threadRoutine(void*args)
{
pthread_detach(pthread_self());
ThreadData*td=static_cast<ThreadData*>(args);
td->current->service(td->sock,td->clientip,td->clientport);
delete td;
}
// 流式 - 利用read和write
void service(int sock, const string&clientip,const uint16_t clientport)
{
string who=clientip + "-" + to_string(clientport);
char buffer[1024];
while(true)
{
ssize_t s=read(sock,buffer,sizeof(buffer)-1);
if(s>0)
{
buffer[s]=0;
string res=func_(buffer); // 进行回调
cout<<who<< ">>> " <<res<<endl;
// 把收到的消息返回(写给客户端)
write(sock,res.c_str(),res.size());
}
else if(s==0)
{
// 对方将连接关闭了
close(sock);
cout<< who <<" quit, me too"<<endl;
break;
}
else
{
close(sock);
cerr<<"read error: "<<strerror(errno)<<endl;
break;
}
}
}
~TcpServer()
{
}
private:
uint16_t port_;
int listensock_;
bool quit_; // 标志服务器是否运行字段
func_t func_;
};
}
tcpServer.cc
#include"tcpServer.hpp"
using namespace ns_server;
// ./tcp_server port
// 使用手册
static void usage(string proc)
{
cout<<"usage:\n\t"<<proc<<" port\n"<<endl;
}
string echo(const string&message)
{
return message;
}
int main(int argc,char*argv[])
{
if(argc!=2)
{
usage(argv[0]);
exit(USAGE_ERR);
}
uint16_t port=atoi(argv[1]);
unique_ptr<TcpServer> tsvr(new TcpServer(echo,port));
tsvr->InitServer();
tsvr->Start();
return 0;
}
tcpClient.cc
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include"err.hpp"
using namespace std;
static void usage(string proc)
{
cout<<"usage:\n\t"<<proc<<" serverip serverport\n" <<endl;
}
// ./tcp_client serverip serverport
int main(int argc,char*argv[])
{
// 准备工作
if(argc!=3)
{
usage(argv[0]);
exit(USAGE_ERR);
}
string serverip=argv[1];
uint16_t serverport=atoi(argv[2]);
// 1.创建套接字
int sock=socket(AF_INET,SOCK_STREAM,0);
if (sock < 0)
{
cerr << "create socket error: " << strerror(errno) << endl;
exit(SOCKET_ERR);
}
// (2) 客户端要不要bind呢? 要
// 要不要自己bind呢? 不要, 因为client要让OS自动给用户进行bind
// (3) 要不要listen?不要, 客户端连别人, 永远都是别人listen; 要不要accept?不要, 服务器来连接
// 2. connect 客户端向服务器发起连接请求
struct sockaddr_in server;
memset(&server,0,sizeof(server));
server.sin_port=htons(serverport);
server.sin_family=AF_INET;
// server.sin_addr.s_addr=INADDR_ANY; 绝对不是
inet_aton(serverip.c_str(),&(server.sin_addr)); // 字符串风格ip转成点分十进制
int cnt=5;
while(connect(sock,(struct sockaddr*)&server,sizeof(server))!=0) // 连接失败
{
sleep(1);
cout<<"正在给你重连, 重连次数还有: "<<cnt--<<endl;
if(cnt<=0)
break;
}
if(cnt<=0)
{
cerr<<"连接失败"<<endl;
exit(CONNECT_ERR);
}
char buffer[1024];
// 3. 连接成功
while(true)
{
string line;
cout<<"Enter>> ";
getline(cin,line);
write(sock,line.c_str(),line.size());
ssize_t s = read(sock, buffer, sizeof(buffer)-1);
if (s > 0)
{
buffer[s] = 0;
cout<<"server echo >>>"<<buffer<<endl;
}
else if (s == 0)
{
cerr << "server quit" << endl;
break;
}
else
{
cerr << "read error: " << strerror(errno) << endl;
break;
}
}
close(sock);
return 0;
}
运行结果:
多线程版的问题:
线程池
task.hpp
#pragma once
#include
#include
#include
#include
#include
using namespace std;
using cb_t=function<void(int sock, const string&,const uint16_t&)>;
class Task
{
public:
Task()
{
}
Task(int sock, const string& ip,const uint16_t&port,cb_t cb)
:_sock(sock)
,_ip(ip)
,_port(port)
,_cb(cb)
{
}
void operator()()
{
_cb(_sock,_ip,_port);
}
~Task()
{
}
private:
int _sock;
string _ip;
uint16_t _port;
cb_t _cb;
};
LockGuard.hpp
#include
#include
using namespace std;
class Mutex //自己不维护锁,由外部传入
{
public:
Mutex(pthread_mutex_t* mutex)
:_pmutex(mutex)
{
}
void lock()
{
pthread_mutex_lock(_pmutex);
}
void unlock()
{
pthread_mutex_unlock(_pmutex);
}
~Mutex()
{}
private:
pthread_mutex_t* _pmutex; //锁的指针
};
class LockGuard //自己不维护锁,由外部传入
{
public:
LockGuard(pthread_mutex_t* mutex)
:_mutex(mutex)
{
_mutex.lock();
}
~LockGuard()
{
_mutex.unlock();
}
private:
Mutex _mutex; //锁的指针
};
thread.hpp
#include
#include
using namespace std;
class Thread
{
public:
typedef enum
{
NEW=0,
RUNNING,
EXITED
}ThreadStatus;
typedef void (*func_t)(void*); //函数指针, 参数是void*
Thread(int num, func_t func, void*args)
:_tid(0)
,_status(NEW)
,_func(func)
,_args(args)
{
char name[128];
snprintf(name,sizeof(name),"thread-%d",num);
_name=name;
}
int status() {return _status;}
string threadname() {return _name;}
pthread_t thread_id()
{
if(_status==RUNNING)
return _tid;
else
return 0;
}
// runHelper是不是类的成员函数, 而类的成员函数, 具有默认参数this, 需要static
// void*runHelper(Thread*this, void*args) , 而pthread_create要求传的参数必须是: void*的, 即参数不匹配
// 但是static会有新的问题: static成员函数, 无法直接访问类属性和其他成员函数
static void*runHelper(void*args)
{
Thread*ts=(Thread*)args; //就拿到了当前对象
// _func(_args);
(*ts)();
}
//仿函数
void operator()()
{
_func(_args);
}
void run()
{
int n=pthread_create(&_tid,nullptr,runHelper,this); //this: 是当前线程对象Thread
if(n!=0) exit(-1);
_status=RUNNING;
}
void join()
{
int n=pthread_join(_tid,nullptr);
if(n!=0)
{
cerr<<" main thread join thread "<< _name << " error "<<endl;
}
_status=EXITED;
}
~Thread()
{}
private:
pthread_t _tid;
string _name;
func_t _func; //线程未来要执行的回调
void*_args; //调用回调函数时的参数
ThreadStatus _status;
};
threadPool_v4.hpp
#include
#include
#include
#include
#include
#include"thread.hpp"
#include"lockGuard.hpp"
using namespace std;
const static int N=5;
template<class T>
class threadPool
{
public:
pthread_mutex_t* getlock()
{
return &_lock;
}
void threadWait()
{
pthread_cond_wait(&_cond,&_lock);
}
void threadWakeup()
{
pthread_cond_signal(&_cond); // 唤醒在条件变量下等待的线程
}
bool isEmpty()
{
return _tasks.empty();
}
T popTask()
{
T t=_tasks.front();
_tasks.pop();
return t;
}
static void threadRoutine(void*args)
{
threadPool<T>*tp=static_cast<threadPool<T>*>(args);
while(true)
{
// 1. 检测有没有任务 --- 本质是看队列是否为空
// --- 本质就是在访问共享资源 --- 必定加锁
// 2. 有: 处理
// 3. 无: 等待
// 细节: 必定加锁
T t;
{
LockGuard lockguard(tp->getlock());
while (tp->isEmpty())
{
// 等待, 在条件变量下等待
tp->threadWait();
}
t = tp->popTask(); // 把任务从公共区域拿到私有区域
}
// for test
// 处理任务应不应该在临界区中处理, 不应该, 这是线程自己私有的事情
t();
}
}
static threadPool<T> * getinstance()
{
if (instance == nullptr) // 为什么要这样? 提高效率, 减少加锁的次数
{
LockGuard lockguard(&instance_lock);
if (instance == nullptr)
{
cout<<"线程池单例形成"<<endl;
instance = new threadPool<T>();
instance->init();
instance->start();
}
}
return instance;
}
void init()
{
for(int i=0;i<_num;++i)
{
_threads.push_back(Thread(i,threadRoutine,this));
cout<<i<<" thread running"<<endl;
}
}
void check()
{
for(auto&t:_threads)
{
cout<<t.threadname()<<" running..."<<endl;
}
}
void start()
{
for(auto&t:_threads)
{
t.run();
}
}
void pushTask(const T&t)
{
LockGuard lockguard(&_lock);
_tasks.push(t);
threadWakeup();
}
~threadPool()
{
for(auto&t:_threads)
{
t.join();
}
pthread_mutex_destroy(&_lock);
pthread_cond_destroy(&_cond);
}
private:
threadPool(int num=N)
:_num(num)
{
pthread_mutex_init(&_lock,nullptr);
pthread_cond_init(&_cond,nullptr);
}
threadPool(const threadPool<T>&tp)=delete;
void operator=(const threadPool<T>&tp)=delete;
private:
vector<Thread> _threads;
int _num;
queue<T> _tasks;
pthread_mutex_t _lock;
pthread_cond_t _cond;
static threadPool<T>*instance;
static pthread_mutex_t instance_lock;
};
template<class T>
threadPool<T> * threadPool<T>::instance=nullptr;
template<class T>
pthread_mutex_t threadPool<T>::instance_lock=PTHREAD_MUTEX_INITIALIZER;
tcpServer.hpp
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include"err.hpp"
#include"threadPool_v4.hpp"
#include"task.hpp"
using namespace std;
namespace ns_server
{
static const uint16_t defaultport=8081;
static int backlog=32;
using func_t=function<string(const string&)>; // 回调函数,一种处理逻辑
class TcpServer;
class ThreadData
{
public:
ThreadData(int fd, const string&ip,const uint16_t&port,TcpServer*ts)
:sock(fd)
,clientip(ip)
,clientport(port)
,current(ts)
{
}
public:
int sock;
string clientip;
uint16_t clientport;
TcpServer*current;
};
class TcpServer
{
public:
TcpServer(func_t func, uint16_t port=defaultport)
:func_(func)
,port_(port)
,quit_(true)
{
}
void InitServer()
{
// 1. 创建socket文件
listensock_=socket(AF_INET,SOCK_STREAM,0);
if(listensock_<0)
{
cerr<<"create socket error"<<endl;
exit(SOCKET_ERR);
}
// 2. bind
struct sockaddr_in local;
memset(&local,0,sizeof(local));
local.sin_port=htons(port_);
local.sin_family=AF_INET;
local.sin_addr.s_addr=INADDR_ANY;
int n=bind(listensock_,(struct sockaddr*)&local,sizeof(local));
if(n<0)
{
cerr<<"bind socket error"<<endl;
exit(BIND_ERR);
}
// 3. 监听
int m=listen(listensock_,backlog);
if(m<0)
{
cerr<<"listen socket error"<<endl;
exit(LISTEN_ERR);
}
}
void Start()
{
quit_=false;
while(!quit_)
{
struct sockaddr_in client;
socklen_t len=sizeof(client);
// 4. 获取连接, accept
int sock=accept(listensock_,(struct sockaddr*)&client,&len);
if (sock < 0) // accept失败并不会终止进程, 只要获取下一个连接
{
cerr << "accept error" << endl;
continue;
}
// 提取client信息 --- debug
string clientip=inet_ntoa(client.sin_addr); // 把4字节对应的IP转化成字符串风格
uint16_t clientport=ntohs(client.sin_port); // 网络序列转主机序列
// 5. 获取新连接成功后, 开始进行业务处理
cout<<"获取新连接成功: "<<sock<<" from "<<listensock_<<", "<< clientip << "- " <<clientport<<endl;
// v4: 线程池版本
// 一旦用户来了,你才创建线程, 线程池吗
// 使用线程池的时候, 一定是有限的线程个数, 一定要处理短任务
Task t(sock,clientip,clientport, bind(&TcpServer::service, this,placeholders::_1,placeholders::_2,placeholders::_3));
threadPool<Task>::getinstance()->pushTask(t);
}
}
static void*threadRoutine(void*args)
{
pthread_detach(pthread_self());
ThreadData*td=static_cast<ThreadData*>(args);
td->current->service(td->sock,td->clientip,td->clientport);
delete td;
}
// 流式 - 利用read和write
void service(int sock, const string&clientip,const uint16_t clientport)
{
string who=clientip + "-" + to_string(clientport);
char buffer[1024];
ssize_t s = read(sock, buffer, sizeof(buffer) - 1);
if (s > 0)
{
buffer[s] = 0;
string res = func_(buffer); // 进行回调
cout << who << ">>> " << res << endl;
// 把收到的消息返回(写给客户端)
write(sock, res.c_str(), res.size());
}
else if (s == 0)
{
cout << who << " quit, me too" << endl;
}
else
{
close(sock);
cerr << "read error: " << strerror(errno) << endl;
}
close(sock);
}
~TcpServer()
{
}
private:
uint16_t port_;
int listensock_;
bool quit_; // 标志服务器是否运行字段
func_t func_;
};
}
tcpServer.cc
#include"tcpServer.hpp"
using namespace ns_server;
// ./tcp_server port
// 使用手册
static void usage(string proc)
{
cout<<"usage:\n\t"<<proc<<" port\n"<<endl;
}
string echo(const string&message)
{
return message;
}
int main(int argc,char*argv[])
{
if(argc!=2)
{
usage(argv[0]);
exit(USAGE_ERR);
}
uint16_t port=atoi(argv[1]);
unique_ptr<TcpServer> tsvr(new TcpServer(echo,port));
tsvr->InitServer();
tsvr->Start();
return 0;
}
tcpClient.cc
#include
#include
#include
#include
#include
#include
#include
#include
#include
#include"err.hpp"
using namespace std;
static void usage(string proc)
{
cout<<"usage:\n\t"<<proc<<" serverip serverport\n" <<endl;
}
// ./tcp_client serverip serverport
int main(int argc,char*argv[])
{
// 准备工作
if(argc!=3)
{
usage(argv[0]);
exit(USAGE_ERR);
}
string serverip=argv[1];
uint16_t serverport=atoi(argv[2]);
// 1.创建套接字
int sock=socket(AF_INET,SOCK_STREAM,0);
if (sock < 0)
{
cerr << "create socket error: " << strerror(errno) << endl;
exit(SOCKET_ERR);
}
// (2) 客户端要不要bind呢? 要
// 要不要自己bind呢? 不要, 因为client要让OS自动给用户进行bind
// (3) 要不要listen?不要, 客户端连别人, 永远都是别人listen; 要不要accept?不要, 服务器来连接
// 2. connect 客户端向服务器发起连接请求
struct sockaddr_in server;
memset(&server,0,sizeof(server));
server.sin_port=htons(serverport);
server.sin_family=AF_INET;
// server.sin_addr.s_addr=INADDR_ANY; 绝对不是
inet_aton(serverip.c_str(),&(server.sin_addr)); // 字符串风格ip转成点分十进制
int cnt=5;
while(connect(sock,(struct sockaddr*)&server,sizeof(server))!=0) // 连接失败
{
sleep(1);
cout<<"正在给你重连, 重连次数还有: "<<cnt--<<endl;
if(cnt<=0)
break;
}
if(cnt<=0)
{
cerr<<"连接失败"<<endl;
exit(CONNECT_ERR);
}
char buffer[1024];
// 3. 连接成功
while(true)
{
string line;
cout<<"Enter>> ";
getline(cin,line);
write(sock,line.c_str(),line.size());
ssize_t s = read(sock, buffer, sizeof(buffer)-1);
if (s > 0)
{
buffer[s] = 0;
cout<<"server echo >>>"<<buffer<<endl;
}
else if (s == 0)
{
cerr << "server quit" << endl;
break;
}
else
{
cerr << "read error: " << strerror(errno) << endl;
break;
}
}
close(sock);
return 0;
}
运行结果:
ip serverport\n" <
// ./tcp_client serverip serverport
int main(int argc,char*argv[])
{
// 准备工作
if(argc!=3)
{
usage(argv[0]);
exit(USAGE_ERR);
}
string serverip=argv[1];
uint16_t serverport=atoi(argv[2]);
// 1.创建套接字
int sock=socket(AF_INET,SOCK_STREAM,0);
if (sock < 0)
{
cerr << "create socket error: " << strerror(errno) << endl;
exit(SOCKET_ERR);
}
// (2) 客户端要不要bind呢? 要
// 要不要自己bind呢? 不要, 因为client要让OS自动给用户进行bind
// (3) 要不要listen?不要, 客户端连别人, 永远都是别人listen; 要不要accept?不要, 服务器来连接
// 2. connect 客户端向服务器发起连接请求
struct sockaddr_in server;
memset(&server,0,sizeof(server));
server.sin_port=htons(serverport);
server.sin_family=AF_INET;
// server.sin_addr.s_addr=INADDR_ANY; 绝对不是
inet_aton(serverip.c_str(),&(server.sin_addr)); // 字符串风格ip转成点分十进制
int cnt=5;
while(connect(sock,(struct sockaddr*)&server,sizeof(server))!=0) // 连接失败
{
sleep(1);
cout<<"正在给你重连, 重连次数还有: "<> ";
getline(cin,line);
write(sock,line.c_str(),line.size());
ssize_t s = read(sock, buffer, sizeof(buffer)-1);
if (s > 0)
{
buffer[s] = 0;
cout<<"server echo >>>"<
}
运行结果:
![在这里插入图片描述](https://img-blog.csdnimg.cn/eadac05d3a6c43dbb8c5e44f9ccebca6.png)