2016-08-3116:44:09

server 端

  1. /*******************************************************************
  2. author:xuqing
  3. date:2016-5-20
  4. about the io thread
  5. the init part must be less than 1000,just the real_part_count
  6. THREAD_DEAL_COUNT is the count one thread can deal with
  7. if the real_part_count is 10;
  8. the THREAD_DEAL_COUNT is 5
  9. we just need 2 thread to deal the all io write
  10. success return 0
  11. error return 负数
  12. sucess 有意义 正数
  13. ***********************************************************************/
  14. #include "stdafx.h"
  15. #include "rts_server.h"
  16. #include "memory_part.h"
  17. #pragma comment(lib,"ws2_32.lib")
  18.  
  19. #define THREAD_DEAL_COUNT 5
  20. HANDLE h_complete_point;
  21. extern memory_part *temp_memory[];
  22.  
  23. int socket_init()
  24. {
  25.     RTS_DIARY_RET_ERROR_VAR;
  26.     WSAData wsaData;
  27.     if (WSAStartup(MAKEWORD(, ), &wsaData) != )
  28.     {
  29.         printf("WSAStartup失败!\n");
  30.         RTS_DIARY_ERROR_GOTO;
  31.     }
  32.     if (LOBYTE(wsaData.wVersion) !=  || HIBYTE(wsaData.wVersion) != )
  33.     {
  34.         printf("版本不对！\n");
  35.         RTS_DIARY_ERROR_GOTO;
  36.     }
  37.     return ;
  38. ret_error:
  39.     if (i_ret) return i_ret;
  40.     return ;
  41.  
  42. }
  43.  
  44. int write_memory(char *record)
  45. {
  46.     RTS_DIARY_RET_ERROR_VAR;
  47.     char str_cust_id[];
  48.     get_custid2data(record, str_cust_id);
  49.     int int_cust_id = atoi(str_cust_id);
  50.     int part_index = int_cust_id % MAX_PART;
  51.     memory_part *save_part = temp_memory[part_index];
  52.     if (save_part == NULL) RTS_DIARY_ERROR_GOTO;
  53.     SqQueue *save_queue = save_part->queue_offset;
  54.     if (save_queue == NULL) RTS_DIARY_ERROR_GOTO;
  55.  
  56.     if ((save_queue->rear + ) % MAX_RECORD_OF_PART == save_queue->front)
  57.     {
  58.         //write memory wait!!!
  59.         Sleep();
  60.     }
  61.     i_ret=save_data2memory(part_index, record);
  62.     if (i_ret != ) RTS_DIARY_ERROR_GOTO;
  63.  
  64.     return ;
  65.  
  66. ret_error:
  67.     if (i_ret) return i_ret;
  68.     return ;
  69.  
  70. }
  71.  
  72. DWORD WINAPI ServerWorkerThread(LPVOID CompletionPort)
  73. {
  74.     RTS_DIARY_RET_ERROR_VAR;
  75.     HANDLE ComPort = (HANDLE)CompletionPort;
  76.     DWORD BytesTransferred;
  77.     //LPOVERLAPPED Overlapped;
  78.     LPPER_HANDLE_DATA PerHandleData;
  79.     LPPER_IO_OPERATION_DATA PerIoData;
  80.     DWORD SendBytes, RecvBytes;
  81.     DWORD Flags;
  82.     int recv_int;
  83.     char str[] = "";
  84.     while (TRUE)
  85.     {
  86.         //等待完成端口上SOCKET的完成  阻塞
  87.         printf( "等待完成端口上SOCKET的完成!\n");
  88.         if ((GetQueuedCompletionStatus(ComPort,&BytesTransferred, (PULONG_PTR)&PerHandleData,(LPOVERLAPPED *)&PerIoData, INFINITE)) == NULL)
  89.         {
  90.             continue;
  91.         }
  92.         recv_int = BytesTransferred;
  93.         printf("BytesTransferred work:%d\n", BytesTransferred);
  94.         //检查是否有错误产生
  95.         if (BytesTransferred ==  &&(PerIoData->OperationType == RECV_POSTED ||PerIoData->OperationType == SEND_POSTED))
  96.         {
  97.             //关闭SOCKET
  98.             printf("%d SOCKET关闭\n", PerHandleData->sock);
  99.             closesocket(PerHandleData->sock);
  100.             free(PerHandleData);
  101.             free(PerIoData);
  102.             continue;
  103.         }
  104.         //为请求服务
  105.         if (PerIoData->OperationType == RECV_POSTED)
  106.         {
  107.             printf( "接收处理\n");
  108.             memcpy(str, PerIoData->Buff, recv_int);
  109.             str[recv_int] = '\0';
  110.             printf("%d SOCKET :%s\n", PerHandleData->sock, str);
  111.             //although it is multithread but they don't visit the same memory,we have use the queue structure to avoid it
  112.             write_memory(str);
  113.             ZeroMemory(PerIoData->Buff, );
  114.             strcpy(PerIoData->Buff, str);
  115.             Flags = ;
  116.             ZeroMemory((LPVOID)&(PerIoData->Overlapped), sizeof(OVERLAPPED));
  117.             PerIoData->DataBuff[].len = ;
  118.             PerIoData->DataBuff[].buf = PerIoData->Buff;
  119.             PerIoData->OperationType = SEND_POSTED;
  120.             WSASend(PerHandleData->sock, PerIoData->DataBuff,, &SendBytes, , &(PerIoData->Overlapped), NULL);
  121.             printf("the send byte is %d\n", SendBytes);
  122.         }
  123.         else //if(PerIoData->OperationType == SEND_POSTED)
  124.         {
  125.             printf( "发送处理!\n");
  126.             Flags = ;
  127.             ZeroMemory((LPVOID)&(PerIoData->Overlapped), sizeof(OVERLAPPED));
  128.             ZeroMemory(PerIoData->Buff, );
  129.             PerIoData->DataBuff[].len = ;
  130.             PerIoData->DataBuff[].buf = PerIoData->Buff;
  131.             PerIoData->OperationType = RECV_POSTED;
  132.             WSARecv(PerHandleData->sock, PerIoData->DataBuff,, &RecvBytes, &Flags, &(PerIoData->Overlapped), NULL);
  133.             //recv_int = RecvBytes;
  134.             printf("the recv byte is %d\n", RecvBytes);
  135.         }
  136.     }
  137.     return ;
  138. ret_error:
  139.     if (i_ret) return i_ret;
  140.     return ;
  141.  
  142. }
  143.  
  144. DWORD WINAPI write_disk(LPVOID p)
  145. {
  146.     RTS_DIARY_RET_ERROR_VAR;
  147.     Sleep();
  148.     //循环处理不同的part   一个线程可以负责处理多个不同的part
  149.     memory_part *write_part;
  150.     SqQueue *queue;
  151.     int i = (int)p;
  152.     while (true)
  153.     {
  154.         write_part = temp_memory[i];
  155.         queue = write_part->queue_offset;
  156.         if (write_part == NULL || queue == NULL)  continue;
  157.         printf("rear : %d, front :%d", queue->rear, queue->front);
  158.         while (true)
  159.         {
  160.             if (queue->front != (queue->rear))
  161.             {
  162.                 printf("while front %d\n", queue->front);
  163.                 printf("hello\n");
  164.                 save_data2disk(+i);
  165.             }
  166.             else
  167.             {
  168.                 //if the read thread pointer==write thread pointer
  169.                 //the read thread must sleep 500
  170.                 Sleep();
  171.                 break;
  172.             }
  173.         }
  174.         i = i + ;
  175.         i = i % THREAD_DEAL_COUNT + (int)p;
  176.     }
  177.     return ;
  178. ret_error:
  179.     if (i_ret) return i_ret;
  180.     return ;
  181.  
  182. }
  183.  
  184. int start_rts_server()
  185. {
  186.     RTS_DIARY_RET_ERROR_VAR;
  187.     LPPER_HANDLE_DATA perHandleData;
  188.     LPPER_IO_OPERATION_DATA ioperdata;
  189.     SYSTEM_INFO siSys;
  190.     SOCKET sockListen;
  191.     struct sockaddr_in addrLocal;
  192.     int nRet = ;
  193.     DWORD nThreadID;
  194.     SOCKET sockAccept;
  195.     DWORD dwFlags;
  196.     DWORD dwRecvBytes;
  197.     int nReuseAddr = ;
  198.     int ret_code = ;
  199.     int real_part_count = ;
  200.     printf("初始环境...!\n");
  201.     if (socket_init() != ) RTS_DIARY_ERROR_GOTO;
  202.     init_memory_part(real_part_count);
  203.     //创建一个IO完成端口
  204.     printf("创建一个IO完成端口!\n");
  205.     h_complete_point = CreateIoCompletionPort(INVALID_HANDLE_VALUE, NULL, , );
  206.     if (h_complete_point == INVALID_HANDLE_VALUE)
  207.     {
  208.         printf("创建IO完成端口失败!\n");
  209.         RTS_DIARY_ERROR_GOTO;
  210.     }
  211.     //获取CPU数目
  212.     GetSystemInfo(&siSys);
  213.     for (int i = ; i<(int)siSys.dwNumberOfProcessors * ; i++)//NumberOfProcessors
  214.     {
  215.         HANDLE hThread;
  216.         hThread = CreateThread(NULL, , ServerWorkerThread, (LPVOID)h_complete_point, , &nThreadID);
  217.         printf("创建工作者线程 ---->%d!\n", i);
  218.         CloseHandle(hThread);
  219.     }
  220.     //创建监听SOCKET
  221.     printf("创建监听SOCKET!\n");
  222.     sockListen = WSASocket(AF_INET, SOCK_STREAM, , NULL, , WSA_FLAG_OVERLAPPED);
  223.     if (sockListen == SOCKET_ERROR)
  224.     {
  225.         printf("WSASocket错误!\n");
  226.         RTS_DIARY_ERROR_GOTO;
  227.     }
  228.  
  229.     if (setsockopt(sockListen, SOL_SOCKET, SO_REUSEADDR, (const char *)&nReuseAddr, sizeof(int)) != )
  230.     {
  231.         printf("setsockopt错误!\n");
  232.         RTS_DIARY_ERROR_GOTO;
  233.     }
  234.     addrLocal.sin_family = AF_INET;
  235.     addrLocal.sin_addr.s_addr = htonl(INADDR_ANY);
  236.     addrLocal.sin_port = htons();
  237.     if (bind(sockListen, (struct sockaddr *)&addrLocal, sizeof(sockaddr_in)) != )
  238.     {
  239.         printf("bind错误\n");
  240.         i_ret = WSAGetLastError();
  241.         RTS_DIARY_ERROR_GOTO;
  242.     }
  243.     //准备监听
  244.     printf("准备监听\n");
  245.     if (listen(sockListen, ) != )
  246.     {
  247.         printf("listen错误\n");
  248.         RTS_DIARY_ERROR_GOTO;
  249.     }
  250.     int client_addr = sizeof(SOCKADDR_IN);
  251.     int index_part = ;
  252.     // the thread  to save data into the disk
  253.     for (int i = ; i < real_part_count / THREAD_DEAL_COUNT;i++)
  254.     {
  255.         index_part = THREAD_DEAL_COUNT*i;
  256.         HANDLE write_handle = CreateThread(NULL, , write_disk,(LPVOID)index_part, , NULL);
  257.     }
  258.  
  259.     //main thread to get client accept
  260.     while (true)
  261.     {
  262.         //阻塞
  263.         sockAccept = WSAAccept(sockListen, NULL, NULL, NULL, );
  264.         perHandleData = (LPPER_HANDLE_DATA)malloc(sizeof(PER_HANDLE_DATA));
  265.         if (perHandleData == NULL) continue;
  266.         printf("socket number --->   %d   接入！\n", sockAccept);
  267.         perHandleData->sock = sockAccept;
  268.         if (CreateIoCompletionPort((HANDLE)sockAccept, h_complete_point, (ULONG_PTR)perHandleData, ) == NULL)
  269.         {
  270.             free(perHandleData);
  271.             printf("iocp client!\n");
  272.             continue;
  273.         }
  274.         ioperdata = (LPPER_IO_OPERATION_DATA)malloc(sizeof(PER_IO_OPERATION_DATA));
  275.         memset(&(ioperdata->Overlapped), , sizeof(OVERLAPPED));
  276.         (ioperdata->DataBuff[]).len = ;
  277.         (ioperdata->DataBuff[]).buf = ioperdata->Buff;
  278.         ioperdata->OperationType = RECV_POSTED;
  279.         if (ioperdata == NULL)
  280.         {
  281.             free(perHandleData);
  282.             continue;
  283.         }
  284.         printf("投递接收操作!\n");
  285.         dwFlags = ;
  286.         int b=WSARecv(perHandleData->sock, &ioperdata->DataBuff[], , &dwRecvBytes, &dwFlags, &(ioperdata->Overlapped), NULL);
  287.         printf("the recv byte main():%d \n", dwRecvBytes);
  288.         if (b == SOCKET_ERROR)
  289.         {
  290.             ret_code = WSAGetLastError();
  291.             if (ret_code= WSA_IO_PENDING)
  292.             {
  293.                 printf("was recv sucess %d\n", WSAGetLastError());
  294.  
  295.             }
  296.             else
  297.             {
  298.                 printf("was recv error!\n");
  299.                 RTS_DIARY_ERROR_GOTO;
  300.             }
  301.         }
  302.     }
  303. ret_error:
  304.     if (i_ret) return i_ret;
  305.     closesocket(sockListen);
  306.     return ;
  307. }

client 端

  1. #include "stdafx.h"
  2. #include "rts_client.h"
  3. #pragma comment(lib,"ws2_32.lib")
  4. SOCKET sockClient;
  5. struct sockaddr_in addrServer;
  6. char buf[1024];
  7. int n = 0;
  8. int socket_init()
  9. {
  10. 	WSAData wsaData;
  11. 	if (WSAStartup(MAKEWORD(2, 2), &wsaData) != 0)
  12. 	{
  13. 		printf("WSAStartup失败!\n");
  14. 		return -1;
  15. 	}
  16.  
  17. 	if (LOBYTE(wsaData.wVersion) != 2 || HIBYTE(wsaData.wVersion) != 2)
  18. 	{
  19. 		printf("版本不对！\n");
  20. 		WSACleanup();
  21. 		return -1;
  22. 	}
  23. 	return 0;
  24. }
  25.  
  26. DWORD WINAPI recv_thread(LPVOID par)
  27. {
  28. 	while (true)
  29. 	{
  30. 		if (recv(sockClient, buf, 1024, 0) <= 0)
  31. 		{
  32. 		printf("recv失败,可能连接断开!\n");
  33. 		//break;
  34. 		goto theend;
  35. 		}
  36. 		printf("服务器应答：%s", buf);
  37. 		memset(buf, 0, 1024);
  38.  
  39. 		Sleep(200);
  40. 	}
  41.  
  42. theend:
  43. 	WSACleanup();
  44. 	getchar();
  45. 	return 0;
  46. }
  47.  
  48. int client_start()
  49. {
  50. 	int i_ret_error = 0;
  51. 	ff2028_gset_ctx_t gset_st;
  52. 	ff2028_gset_ctx_t *p_gset_st=&gset_st;
  53. 	unsigned char tag[4];
  54. 	i_ret_error = ff2028_gset_ctx_init(p_gset_st);
  55. 	if (i_ret_error < 0)
  56. 	{
  57. 		printf("error-> %s,%s,%d\n", __FILE__, __FUNCTION__, __LINE__);
  58. 		return i_ret_error - __LINE__;
  59. 	}
  60. 	if (socket_init() != 0)
  61. 		goto theend;
  62.  
  63. 	sockClient = socket(AF_INET, SOCK_STREAM, 0);
  64. 	if (sockClient == INVALID_SOCKET)
  65. 	{
  66. 		printf("socket 失败!\n");
  67. 		WSACleanup();
  68. 		goto theend;
  69. 	}
  70. 	memset(&addrServer, 0, sizeof(sockaddr_in));
  71. 	addrServer.sin_family = AF_INET;
  72. 	addrServer.sin_addr.s_addr = inet_addr("127.0.0.1");
  73. 	addrServer.sin_port = htons(9090);
  74. 	printf("连接服务器...\n");
  75. 	if (connect(sockClient, (const struct sockaddr *)&addrServer, sizeof(sockaddr)) != 0)
  76. 	{
  77. 		printf("connect 失败\n");
  78. 		WSACleanup();
  79. 		goto theend;
  80. 	}
  81. 	memset(buf, 0, 1024);
  82. 	sprintf(buf, "1005,5443543546546,,6566,4564645646546,t435435345353453534535  --->%d\r\n", n);
  83. 	tag[0] = 100;
  84. 	i_ret_error = ff2028_gset_add_app_data(p_gset_st, tag, 1, (unsigned char *)buf, strlen(buf));
  85. 	if (i_ret_error < 0)
  86. 	{
  87. 		printf("error-> %s,%s,%d\n", __FILE__, __FUNCTION__, __LINE__);
  88. 		return i_ret_error - __LINE__;
  89. 	}
  90. 	ff2028_gset_add_app_data_end(p_gset_st);
  91. 	//HANDLE receive = CreateThread(NULL, 0, recv_thread, NULL, 0, NULL);
  92. 	while (true)
  93. 	{
  94. 		sprintf(buf, "1005,5443543546546,,6566,4564645646546,t435435345353453534535  --->%d\r\n", n);
  95. 		if (send(sockClient, buf, strlen(buf), 0) <= 0)
  96. 		{
  97. 			printf( "send失败,可能连接断开!\n");
  98. 			//break;
  99. 			goto theend;
  100. 		}
  101. 		memset(buf, 0, 1024);
  102.  
  103. 		//接收服务端应答
  104. 		if (recv(sockClient, buf, 1024, 0) <= 0)
  105. 		{
  106. 			printf("recv失败,可能连接断开!\n");
  107. 			//break;
  108. 			goto theend;
  109. 		}
  110. 		printf("服务器应答：%s", buf);
  111. 		memset(buf, 0, 1024);
  112. 		Sleep(200);
  113. 		n++;
  114. 		if (n == 20)
  115. 			break;
  116. 	}
  117. theend:
  118. 	WSACleanup();
  119. 	return 0;
  120.  
  121. }

iocp 小例子的更多相关文章

  1. springmvc入门的第一个小例子

    今天我们探讨一下springmvc,由于是初学,所以简单的了解一下 springmvc的流程,后续会持续更新... 由一个小例子来简单的了解一下 springmvc springmvc是spring框 ...

  2. java即时通信小例子

    学习java一段时间了,今天写来一个即时通信的小例子练手在其过程中也学到了一些知识拿出来和大家分享,请路过的各位大神多多赐教... 好了下面讲一下基本的思路: 首先,编写服务器端的程序,简单点说吧就是 ...

  3. Runtime的几个小例子（含Demo）

    一.什么是runtime(也就是所谓的“运行时”,因为是在运行时实现的.)           1.runtime是一套底层的c语言API(包括很多强大实用的c语言类型,c语言函数);  [runti ...

  4. bootstrap 模态 modal 小例子

    bootstrap 模态 modal  小例子 <html> <head> <meta charset="utf-8" /> <title ...

  5. INI配置文件分析小例子

    随手写个解析INI配置字符串的小例子 带测试 #include <iostream> #include <map> #include <string> #inclu ...

  6. JavaScript小例子：复选框全选

    JavaScript小例子:复选框全选 这只是一个小例子,很简单,但是这个功能还是很常用的: 实现后效果如图: JavaScript代码: <script type="text/jav ...

  7. 【zTree】 zTree使用的 小例子

    使用zTree树不是第一次了  但是 还是翻阅着之前做的 对照着 使用起来比较方便  这里就把小例子列出来   总结一下使用步骤 这样方便下次使用起来方便一点 使用zTree树的步骤: 1.首先  在 ...

  8. js小例子（标签页）

    运用js写的一个小例子,实现点击不同的标签出现不同的内容: <!DOCTYPE html> <html> <head> <meta chaset=" ...

  9. sbrk与brk的使用小例子

    sbrk() 和 brk() - Unix的系统函数   sbrk()和brk() 系统的底层会维护一个位置,通过位置的移动完成内存的分配和回收.映射内存时 以一个内存页作为基本单位.   void* ...

随机推荐

  1. eclipse 自动排版

    自动排版:  ctrl + i 同名变量高亮: shift+alt+ o 导包:  shift+ctrl +o

  2. RTTI: dynamic_cast typeid

    dynamic_cast:将基类类型的指针向派生类指针安全转换.多用于下行转换.上行转换时,和static_cast是一样的.C++类型转换看这里.而const_cast用来修改类型的const或vo ...

  3. tomcat源码分析(三)一次http请求的旅行-从Socket说起

    p { margin-bottom: 0.25cm; line-height: 120% } tomcat源码分析(三)一次http请求的旅行 在http请求旅行之前,我们先来准备下我们所需要的工具. ...

  4. word域1

    Description: 域是WORD中的一种特殊命令,它由花括号.域名(域代码)及选项开关构成.域代码类似于公式,域选项并关是特殊指令,在域中可触发特定的操作.在用WORD处理文档时若能巧妙应用域, ...

  5. html5悬浮球效果

    自己想做一个自己的网站,觉得自适应的效果会好一点,但是放到手机端的话,菜单显示是个问题.所以自己试着写了一个悬浮球菜单的效果. 好了,上代码. 这里有四个文件要用: jqurey.js//因为基于jq ...

  6. SLES 10安装Oracle10gR2笔记

    SLES 10安装Oracle10gR2笔记 一. 数据库安装 . 安装C/C++ Compiler gcc --version验证是否安装 . 验证Service Pack版本 SPident –v ...

  7. UVA 11464 偶数矩阵

    https://uva.onlinejudge.org/index.php?option=com_onlinejudge&Itemid=8&page=show_problem& ...

  8. C#的四种Timer介绍

    一.Timer的几个类别 1.System.Threading.Timer 2.System.Timers.Timer 3.System.Windows.Forms.Timer 4.System.Wi ...

  9. 关于ES3、ES5、ES6以及ES7所有数组的方法(api)的总结

    起因:工作用经常用到操作数组的方法,这里进行一下总结,我尽量以简洁的语言概括每个方法(api)的作用.如果您想快速定位,可以Control+F 然后搜相应的方法即可定位 :) ES3的数组方法 joi ...

  10. Date Picker和UITool Bar的使用

    p.p1 { margin: 0.0px 0.0px 0.0px 0.0px; font: 13.0px Verdana } span.s1 { } span.s2 { background-colo ...