UNIX TCP回射服务器/客户端(6):线程池服务器

来源:互联网 发布:prime输入法知乎 编辑:程序博客网 时间:2024/06/05 14:09

    《Unix网络编程》这本书附带了许多短小精美的小程序,我在阅读此书的时候,将书上的代码按照自己的理解重写了一遍(大部分是抄书上的),加深一下自己的理解(纯看书太困了,呵呵)。此例子在Ubuntu10.04上测试通过。

    PS:程序里使用了包裹函数(首字母是大写的函数)和常量(所有字母都是大写的常量)的声明在my_unp.h文件中,定义在unp_base.c和unp_thread.c中,地址:http://blog.csdn.net/aaa20090987/article/details/8096701

    程序简介:这是一个基本的线程池服务器模型。服务器启动时,先启动N个用于处理与客户端通信的子线程。当一个客户端连接上服务器时(accept),各个空闲的子线程都会去accept这个连接,但因为互斥锁的关系,总能保证有且只有一个线程会accept成功。然后这个子线程就会与客户端进行通信。


上代码:

#include "my_unp.h"typedef struct {pthread_t thread_tid;//线程IDlong thread_count; //线程处理的条数} Thread;Thread *tptr;int    listenfd, nthreads;socklen_t  addrlen;pthread_mutex_t lock = PTHREAD_MUTEX_INITIALIZER;//接受来自客户端字符串,然后再原样返回void str_echo(int sockfd){ssize_t n;  char buf[MAXLINE];  again:  //从套接字中读取数据,写到buffer中去  //再将buffer中的数据写到套接字中去  while( (n=read(sockfd, buf, MAXLINE)) > 0 )  Writen(sockfd, buf, n); //由于信号中断,没写或读成功任何数据  if( n<0 && errno==EINTR )  goto again;  else if( n < 0 )  error_quit("str_echo: read error");  }void *thread_main(void *arg){int    connfd;socklen_t  clilen;struct sockaddr *cliaddr;cliaddr = Malloc(addrlen);printf("thread %d starting\n", (int) arg);while(1){//当请求到达时,加锁,接收,解锁,处理数据//这样就保证了每个请求只能被一个随机线程所处理clilen = addrlen;Pthread_mutex_lock(&lock);connfd = Accept(listenfd, cliaddr, &clilen);Pthread_mutex_unlock(&lock);tptr[ (int)arg ].thread_count++;str_echo(connfd);Close(connfd);}}int main(int argc, char **argv){int  i;struct sockaddr_in servaddr; if( argc != 2 )error_quit("Using: server <#thread num>");//创建用于TCP协议的套接字   listenfd = Socket(AF_INET, SOCK_STREAM, 0);  memset(&servaddr, 0, sizeof(servaddr));  servaddr.sin_family = AF_INET;  servaddr.sin_addr.s_addr = htonl(INADDR_ANY);  servaddr.sin_port = htons(SERV_PORT);  //把socket和socket地址结构联系起来    Bind(listenfd, (SA*)&servaddr, sizeof(servaddr));  //开始监听LISTENQ端口  Listen(listenfd, LISTENQ);  nthreads = atoi(argv[1]);tptr = Calloc(nthreads, sizeof(Thread));//创建处理线程,并让主线程挂起for (i = 0; i < nthreads; i++){Pthread_create(&tptr[i].thread_tid, NULL, &thread_main, (void *)i);}pause();return 0;}

与其配套的客户端在这里:http://blog.csdn.net/aaa20090987/article/details/8462262