4

server-exp3/togglest_pool.c

@@ -0,0 +1,113 @@
+#include "common.h"
+#include "thpool.h"
+
+#define NTHREADS   4
+#define SBUFSIZE   16
+
+typedef struct _param {
+   int hit;
+   int conn_sock;
+} param;
+
+void toggle(int conn_sock,int hit);
+void handle_request( void *arg) ;
+
+void toggle(int conn_sock,int hit) 
+{
+    size_t n; int i,no=0;
+    char buf[MAXLINE]; 
+
+    
+    while((n =recv(conn_sock, buf, MAXLINE,0))> 0) {
+	printf("toggle服务器收到第%d个客户第%d个消息,长度为%d字节\n", hit,++no,(int)n);
+
+        for(i=0; i<n; i++)
+            if(isupper(buf[i])) 
+                buf[i]=tolower(buf[i]);
+            else if(islower(buf[i]))
+                buf[i]=toupper(buf[i]);
+        //sleep(1);      
+	send (conn_sock, buf, n, 0);
+    }
+    
+}
+
+//示例任务 
+void handle_request(void *arg) 
+{
+    param* datap=(param *) arg;
+    int conn_sock = datap->conn_sock;
+    int hit = datap->hit;
+
+    printf("第%d个客户通信开始\n",hit);
+    //usleep(1);
+    toggle(conn_sock,hit);
+    printf("第%d个客户通信结束\n",hit);
+    
+    free(arg);
+    close(conn_sock);
+}
+
+//主函数
+int main(int argc, char **argv) {
+
+    int listen_sock, conn_sock, port,i; 
+    struct sockaddr_in clientaddr; 
+    struct hostent *hp;
+    char *haddrp;
+    int hit;
+ 
+    int nth[NTHREADS];
+    
+    socklen_t clientlen=sizeof(struct sockaddr_in);
+
+    if (argc != 2) {
+	    fprintf(stderr, "usage: %s <port>\n", argv[0]);
+	    exit(1);
+    }
+    
+    listen_sock = open_listen_sock(port);
+    if( listen_sock==-1) {
+    	printf("端口号%d繁忙\n",port);
+    	exit(1);
+    }    
+    
+    port = atoi(argv[1]);
+    
+    listen_sock = open_listen_sock(port);
+    if( listen_sock==-1) {
+    	printf("端口号%d繁忙\n",port);
+    	exit(1);
+    }
+
+    printf("start main\n");
+    // 初始化线程池
+    thpool_* pool = thpool_init(NTHREADS);
+
+    // 创建示例任务并将其添加到线程池
+     for (hit=1; ; hit++) { 
+	    conn_sock = accept(listen_sock, (SA *) &clientaddr, &clientlen);
+	    
+           //determine the domain name and IP address of the client 
+	    hp = gethostbyaddr((const char *)&clientaddr.sin_addr.s_addr, 
+			   sizeof(clientaddr.sin_addr.s_addr), AF_INET);
+	    haddrp = inet_ntoa(clientaddr.sin_addr);
+	    //printf("server connected to %s (%s)\n", hp->h_name, haddrp);
+	               
+            param *new_param = (param*) malloc(sizeof(param));
+            
+            new_param->hit = hit;   
+            new_param->conn_sock=conn_sock;
+
+	    thpool_add_work(pool,handle_request, (void*)new_param); 
+    }
+    
+    // 等待线程池中的所有任务执行完毕
+    thpool_wait(pool);
+
+    // 销毁线程池
+    thpool_destroy(pool);
+
+    printf("stop main\n");
+    return 0;
+}