採用C++的ACE庫實現的一個通用的C/S架構通信程序(最終版)

轉自:http://blog.csdn.net/chenruichuan/article/details/384476

ACE自適配通信環境(ADAPTIVE Communication Environment)是可以自由使用、開放源碼的面向對象(OO)框架(Framework),在其中實現了許多用於併發通信軟件的核心模式。ACE提供了一組豐富的可複用C++ Wrapper Facade(包裝外觀)和框架組件,可跨越多種平臺完成通用的通信軟件任務,其中包括:事件多路分離和事件處理器分派、信號處理、服務初始化、進程間通信、共享內存管理、消息路由、分佈式服務動態(重)配置、併發執行和同步,等等。

ACE的目標用戶是高性能和實時通信服務和應用的開發者。它簡化了使用進程間通信、事件多路分離、顯式動態鏈接和併發的OO網絡應用和服務的開發。此外,通過服務在運行時與應用的動態鏈接,ACE還使系統的配置和重配置得以自動化。

我最近採用ACE實現了一個通用的C/S架構通信程序,具體實現簡述如下:

1.  服務器端:一個採用領導者/跟隨者模型的線程池不斷地接受從多個客戶端發來的消息,並放入一個消息隊列,然後又有一個採用半同步/半異步模型的線程池不斷地從這個消息隊列中取出消息進行處理。

服務器端代碼如下:(共10個文件)

ACE_Server.cpp

#include "ace/SOCK_Acceptor.h"

#include "ace/Acceptor.h"

#include "ace/Thread_Manager.h"

#include "ace/TP_Reactor.h"

#include "ace/Reactor.h"

#include "ace/INET_Addr.h"

#include "ace/OS.h"

#include "Request_Handler.h"

#include "Server.h"

#include "Constants.h"

using namespace ACE_Server;

int main(int argc, char *argv[])

{

     ACE_TP_Reactor tp_reactor;

     ACE_Reactor reactor(&tp_reactor, 1);

     ACE_Reactor::instance(&reactor, 1);

     ACE_Acceptor<Request_Handler, ACE_SOCK_ACCEPTOR> acceptor;

     ACE_INET_Addr addr(SERVER_PORT_NUM);

     if(acceptor.open(addr) == -1)

         return -1;

     Server server_tp;

     server_tp.activate(THR_NEW_LWP | THR_JOINABLE, SERVER_THREAD_POOL_SIZE);

     ACE_Thread_Manager::instance()->wait();

     return 0;

}

Constants.h

#ifndef __CONSTANTS_H_

#define __CONSTANTS_H_

namespace ACE_Server

{

     static const size_t SERVER_THREAD_POOL_SIZE = 5;   //進行數據接收的線程池大小

     static const size_t TASK_THREAD_POOL_SIZE = 5; //進行數據處理的線程池大小

     static const size_t BUFFER_SIZE = 4096;   //數據緩衝區大小

     static const size_t SERVER_PORT_NUM = 10101;   //服務器的通信端口號

}

#endif

Server.h

#ifndef __SERVER_H_

#define __SERVER_H_

#include "ace/Task.h"

namespace ACE_Server

{

     class Server: public ACE_Task_Base

     {

     public:

         virtual int svc(void);

     };

}

#endif

Server.cpp

#include "ace/Reactor.h"

#include "Server.h"

namespace ACE_Server

{

     int Server::svc(void)

     {

         int result = ACE_Reactor::instance()->run_reactor_event_loop();

         if(result == -1)

              return -1;

         return 0;

     }

}

Request_Handler.h

#ifndef __REQUEST_HANDLER_H_

#define __REQUEST_HANDLER_H_

#include "ace/Svc_Handler.h"

#include "ace/SOCK_Stream.h"

#include "ace/Synch.h"

#include "ace/Thread_Manager.h"

#include "Task_Manager.h"

namespace ACE_Server

{

     class Request_Handler: public ACE_Svc_Handler<ACE_SOCK_STREAM, ACE_MT_SYNCH>

     {

     public:

         Request_Handler(ACE_Thread_Manager *thr_mgr = 0);

     protected:

         virtual int handle_input(ACE_HANDLE fd = ACE_INVALID_HANDLE);

     private:

         static Task_Manager task_mgr;

     };

}

#endif

Request_Handler.cpp

#include "ace/OS.h"

#include "ace/Message_Block.h"

#include "ace/Thread_Manager.h"

#include "ace/Svc_Handler.h"

#include "ace/SOCK_Stream.h"

#include "ace/Synch.h"

#include "ace/Reactor.h"

#include "Request_Handler.h"

#include "Task_Manager.h"

#include "Constants.h"

namespace ACE_Server

{

     Task_Manager Request_Handler::task_mgr;

     Request_Handler::Request_Handler(ACE_Thread_Manager *thr_mgr): ACE_Svc_Handler<ACE_SOCK_STREAM, ACE_MT_SYNCH> (thr_mgr)

     {

         this->reactor(ACE_Reactor::instance());

         task_mgr.activate();

     }

     int Request_Handler::handle_input(ACE_HANDLE fd)

     {

         char length[4] = {0};

         if(this->peer().recv_n(length, 4) == 4)

         {

              size_t msg_len = 0;

              for(int i = 0; i < 4; i++)

              {

                   msg_len |= (size_t)length[i] << (8 * i);

              }

              char msg[BUFFER_SIZE] = {0};

              if(this->peer().recv_n(msg, msg_len) == msg_len)

              {

                   ACE_Message_Block *mb = NULL;

                   ACE_NEW_RETURN(mb, ACE_Message_Block(msg_len, ACE_Message_Block::MB_DATA, 0, msg), -1);

                   mb->wr_ptr(msg_len);

                   task_mgr.putq(mb);

                   return 0;

              }

         }

         return -1;

     }

}

Task_Manager.h

#ifndef __TASK_MANAGER_H_

#define __TASK_MANAGER_H_

#include "ace/Task.h"

#include "ace/Synch.h"

namespace ACE_Server

{

     class Task_Manager: public ACE_Task<ACE_MT_SYNCH>

     {

     public:

         virtual int svc(void);

     };

}

#endif

Task_Manager.cpp

#include "ace/Message_Block.h"

#include "Task_Manager.h"

#include "Task_Worker.h"

#include "Constants.h"

namespace ACE_Server

{

     int Task_Manager::svc(void)

     {

         Task_Worker task_tp;

         task_tp.activate(THR_NEW_LWP | THR_JOINABLE, TASK_THREAD_POOL_SIZE);

         while(1)

         {

              ACE_Message_Block *mb = NULL;

              if(this->getq(mb) < 0)

              {

                   task_tp.msg_queue()->deactivate();

                   task_tp.wait();

              }

              task_tp.putq(mb);

         }

         return 0;

     }

}

Task_Worker.h

#ifndef __TASK_WORKER_H_

#define __TASK_WORKER_H_

#include "ace/Task.h"

#include "ace/Synch.h"

#include "ace/Message_Block.h"

namespace ACE_Server

{

     class Task_Worker: public ACE_Task<ACE_MT_SYNCH>

     {

     public:

         virtual int svc(void);

     private:

         void process_task(ACE_Message_Block *mb);

     };

}

#endif

Task_Worker.cpp

#include "ace/OS.h"

#include "ace/Message_Block.h"

#include "Task_Worker.h"

namespace ACE_Server

{

     int Task_Worker::svc(void)

     {

         while(1)

         {

              ACE_Message_Block *mb = NULL;

              if(this->getq(mb) == -1)

              {

                   continue;

              }

              process_task(mb);

         }

         return 0;

     }

     void Task_Worker::process_task(ACE_Message_Block *mb)

     {

         //進行數據處理,數據的起始地址爲mb->rd_ptr(),長度爲mb->length()

         ACE_DEBUG((LM_DEBUG, ACE_TEXT("(%t) Processing task: %s length %d/n"), mb->rd_ptr(), mb->length()));

         ACE_OS::sleep(3);  //模擬數據處理過程

         mb->release();

     }

}

2.  客戶端:應用程序將需要發送的若干消息放入一個消息隊列,然後激活一個線程來發送所有消息到服務器端。

客戶端代碼如下:(共4個文件)

ACE_Client.cpp

#include "Client.h"

using namespace ACE_Client;

int main(int argc, char *argv[])

{

     Client client("localhost"); //服務器的IP地址或者服務器名稱

     for(int i = 0; i < 5; i++)

     {

         char *task1 = "Is it a good day?";   //第1個task的數據

         size_t task1_len = 18; //第1個task的數據長度

         char *task1_t;     //無需修改

         ACE_NEW_RETURN(task1_t, char[task1_len + 4], -1);  //無需修改

         client.put_task(task1_t, task1, task1_len);    //無需修改

         char *task2 = "Yeah, it really is."; //第2個task的數據

         size_t task2_len = 20; //第2個task的數據長度

         char *task2_t;     //無需修改

         ACE_NEW_RETURN(task2_t, char[task2_len + 4], -1);  //無需修改

         client.put_task(task2_t, task2, task2_len);    //無需修改

         client.send_tasks();   //將上面的task全部發到服務器

         delete [] task1_t; //釋放task1的內存

         delete [] task2_t; //釋放task2的內存

     }

     return 0;

}

Constants.h

#ifndef __CONSTANTS_H_

#define __CONSTANTS_H_

#include "ace/Time_Value.h"

namespace ACE_Client

{

     static const size_t BUFFER_SIZE = 4096;   //數據緩衝區大小

     static const size_t SERVER_PORT_NUM = 10101;   //服務器的通信端口號

     static const ACE_Time_Value TIME_INTERVAL(0, 1000000);  //兩次數據發送之間的時間間隔(0 s + 1000000 us = 1 s)

}

#endif

Client.h

#ifndef __CLIENT_H_

#define __CLIENT_H_

#include "ace/Task.h"

#include "ace/INET_Addr.h"

#include "ace/Synch.h"

namespace ACE_Client

{

     class Client: public ACE_Task<ACE_NULL_SYNCH>

     {

     public:

         Client(char *server);

         virtual int svc(void);

         char *put_task(char *msg_t, char *msg_s, size_t msg_len);

         void send_tasks(void);

     private:

         ACE_INET_Addr addr;

     };

}

#endif

Client.cpp

#include "ace/OS.h"

#include "ace/SOCK_Stream.h"

#include "ace/SOCK_Connector.h"

#include "ace/Message_Block.h"

#include "ace/Thread_Manager.h"

#include "ace/INET_Addr.h"

#include "Constants.h"

#include "Client.h"

namespace ACE_Client

{

     Client::Client(char *server)

     {

         addr = ACE_INET_Addr(SERVER_PORT_NUM, server);

     }

     int Client::svc(void)

     {

         ACE_SOCK_Stream stream;

         ACE_SOCK_Connector connector;

         if(connector.connect(stream, addr) < 0)

         {

              return -1;

         }

         else

         {

              while(1)

              {

                   ACE_Message_Block *mb = NULL;

                   if(this->getq(mb) == -1)

                   {

                       break;

                   }

                   ACE_DEBUG((LM_DEBUG, ACE_TEXT("(%t) Sending %s/n"), &(mb->rd_ptr()[4])));

                   stream.send_n(mb->rd_ptr(), mb->length());

                   mb->release();

                   ACE_OS::sleep(TIME_INTERVAL);

              }

         }

         stream.close();

          return 0;

     }

     char *Client::put_task(char *msg_t, char *msg_s, size_t msg_len)

     {

         for(int i = 0; i < 4; i++)

         {

              msg_t[i] = (char)((msg_len >> (8 * i)) & 0xff);

         }

         ACE_OS::memcpy(&msg_t[4], msg_s, msg_len);

         ACE_Message_Block *mb = NULL;

         ACE_NEW_RETURN(mb, ACE_Message_Block(msg_len + 4, ACE_Message_Block::MB_DATA, 0, msg_t), 0);

         mb->wr_ptr(msg_len + 4);

         this->putq(mb);

         return msg_t;

     }

     void Client::send_tasks(void)

     {

         this->activate();

         ACE_Thread_Manager::instance()->wait();

     }

}

全部完畢,希望大家能夠喜歡:)

 

發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章