ACE通用服务端框架

最新推荐文章于 2022-08-20 19:29:54 发布

ishow99

最新推荐文章于 2022-08-20 19:29:54 发布

阅读量2.6k

点赞数

分类专栏： ACE应用文章标签：框架 server buffer byte system signal

本文链接：https://blog.csdn.net/ishow99/article/details/4265880

版权

ACE应用专栏收录该内容

8 篇文章 0 订阅

订阅专栏

本文详细介绍了如何使用ACE框架构建一个通用的服务端，包括从《ACE程序员指南》中获取灵感，创建ServerAcceptor和Server端的实现。文章提供了一段主程序和Server的代码示例，涉及信号处理、消息队列和多线程编程。同时，强调了阅读《ACE自适配通信环境中文技术文档》的重要性。

摘要由CSDN通过智能技术生成

认真学习了《The ACE Programmers Guide》中7.6 Using the Acceptor-Connector Framework章节，整理了通用服务端和客户端框架，首次在网络上共享学习成果，希望对大家有所帮助。

不爱废话，开始我们的编码旅程。补充一句，学习ACE不得不研读一下《ACE自适配通信环境中文技术文档-中篇：ACE程序员教程》这份文档，作者和马维达的翻译团队严谨的态度值得我们赞赏。

本框架部分代码来自《The ACE Programmers Guide》7.6 Using the Acceptor-Connector Framework章节和《ACE自适配通信环境中文技术文档-中篇：ACE程序员教程》第七章的思路。

1、main.cpp

2、Server.cpp

/******************************************************************************************** * huangjf 2009年4月于宁夏银川 ********************************************************************************************/ #include "Global_Define.h" #include <ace/OS.h> #include <ace/Message_Block.h> #include <ace/Message_Queue.h> #include <ace/Time_Value.h> #include <ace/INET_Addr.h> #include <ace/Log_Msg.h> /*记录最近一次接入的客户端ACE_SOCK_STREAM，用于控制一次只能允许一个客户端接入的情况*/ //ACE_SOCK_STREAM * last_peer = NULL; Server21::Server21(void) : active_(false) /*方式2*/ /*, notifier_(this->reactor(), this, ACE_Event_Handler::WRITE_MASK)*/ { this->active_ = false; this->msg_queue()->high_water_mark(SEND_HIGHT_MARK); /*方式2*/ //this->notifier_.reactor(this->reactor()); //this->msg_queue()->notification_strategy(&this->notifier_); ACE_DEBUG((LM_DEBUG, ACE_TEXT("(%P|%t) Server21 start/n"))); return; } Server21::~Server21(void) { this->active_ = false; this->msg_queue()->close(); ACE_DEBUG((LM_DEBUG, ACE_TEXT("(%P|%t) Server21 stop/n"))); return; } // 在连接建立后被自动回调 int Server21::open(void *p) { if (super::open(p) == -1) return -1; ACE_TCHAR peer_name[MAXHOSTNAMELEN]; ACE_INET_Addr peer_addr; if (this->peer().get_remote_addr(peer_addr) == 0 && peer_addr.addr_to_string (peer_name, MAXHOSTNAMELEN) == 0) ACE_DEBUG((LM_INFO, ACE_TEXT("(%P|%t) Connection from %s/n"), peer_name)); this->active_ = true; task_recv.set_active(true); task_send.set_active(true); task_recv.set_task_send(&task_send); if (task_recv.activate(THR_NEW_LWP | THR_JOINABLE, TASK_RECV_POOL_SIZE) == -1) { ACE_DEBUG((LM_ERROR, ACE_TEXT("(%P|%t) TaskRecv activate failed/n"))); } task_send.set_task_peer((ACE_Task<ACE_MT_SYNCH> *)this); if (task_send.activate(THR_NEW_LWP | THR_JOINABLE, TASK_SEND_POOL_SIZE) == -1) { ACE_DEBUG((LM_ERROR, ACE_TEXT("(%P|%t) TaskSend activate failed/n"))); } /*回调*/ this->start_0(); /*开启定时处理线程*/ ACE_Time_Value interval(1); // One seconds return this->reactor()->schedule_timer(this, 0, ACE_Time_Value::zero, interval); //return 0; } // 服务端接收到客户端发送的数据后，回调此函数 int Server21::handle_input(ACE_HANDLE) { char buffer[INPUT_SIZE] = {0}; ssize_t recv_cnt = 0; #if (MSG_HEAD_MODE==1) /*接收消息头模式*/ /**************************************************************** * TODO 根据实际消息头格式修改解析方法 *****************************************************************/ ssize_t msg_len = 0; recv_cnt = this->peer().recv(buffer, MSG_HEAD_SIZE); if (recv_cnt > 0 && recv_cnt == MSG_HEAD_SIZE) { /************************* * TODO 注意这里的4个字节 **************************/ msg_len = SET_BYTE4(buffer[0]&0xFF) + SET_BYTE3(buffer[1]&0xFF) + SET_BYTE2(buffer[2]&0xFF) + SET_BYTE1(buffer[3]&0xFF); ACE_DEBUG((LM_DEBUG, ACE_TEXT("(%P|%t) Message length (%d) %s/n"), msg_len, msg_len<INPUT_SIZE-MSG_HEAD_SIZE?ACE_TEXT(""):ACE_TEXT(": warning!!! out of range"))); if (msg_len > 0 && msg_len < INPUT_SIZE-MSG_HEAD_SIZE) { if (msg_len == MSG_HEAD_SIZE) { ACE_DEBUG((LM_DEBUG, ACE_TEXT("(%P|%t) Recv(%d+%d):[%s]/n"), MSG_HEAD_SIZE, msg_len-MSG_HEAD_SIZE, buffer)); ACE_Message_Block *mb = NULL; ACE_NEW_RETURN(mb, ACE_Message_Block(msg_len+1), -1); ACE_OS::memcpy(mb->wr_ptr(), buffer, msg_len); mb->wr_ptr(msg_len); if (task_recv.putq(mb) == -1) { ACE_DEBUG((LM_ERROR, ACE_TEXT ("(%P|%t) %p; discarding data/n"), ACE_TEXT ("Server21 enqueue to TaskRecv failed"))); mb->release(); return 0; } return 0; } else { recv_cnt = this->peer().recv(buffer+MSG_HEAD_SIZE, (msg_len-MSG_HEAD_SIZE)); if (recv_cnt > 0 && recv_cnt == (msg_len-MSG_HEAD_SIZE)) { ACE_DEBUG((LM_DEBUG, ACE_TEXT("(%P|%t) Recv(%d+%d):[%s]/n"), MSG_HEAD_SIZE, msg_len-MSG_HEAD_SIZE, buffer)); ACE_Message_Block *mb = NULL; ACE_NEW_RETURN(mb, ACE_Message_Block(msg_len+1), -1); ACE_OS::memcpy(mb->wr_ptr(), buffer, msg_len); mb->wr_ptr(msg_len); if (task_recv.putq(mb) == -1) { ACE_DEBUG((LM_ERROR, ACE_TEXT ("(%P|%t) %p; discarding data/n"), ACE_TEXT ("Server21 enqueue to TaskRecv failed"))); mb->release(); return 0; } return 0; } if (recv_cnt <= 0 || ACE_OS::last_error() != EWOULDBLOCK) { this->active_ = false; ACE_DEBUG((LM_ERROR, ACE_TEXT("(%P|%t) Connection closed1/n"))); /*表示客户端已断开*/ return -1; } } } return 0; } if (recv_cnt <= 0 || ACE_OS::last_error() != EWOULDBLOCK) { this->active_ = false; ACE_DEBUG((LM_ERROR, ACE_TEXT("(%P|%t) Connection closed2/n"))); /*表示客户端已断开*/ return -1; } #else /*不接收消息头模式*/ recv_cnt = this->peer().recv(buffer, sizeof(buffer)); if (recv_cnt > 0) { ACE_DEBUG((LM_DEBUG, ACE_TEXT("(%P|%t) Recv(%d):[%s]/n"), recv_cnt, buffer)); ACE_Message_Block *mb = NULL; ACE_NEW_RETURN(mb, ACE_Message_Block(recv_cnt+1), -1); ACE_OS::memcpy(mb->wr_ptr(), buffer, recv_cnt); mb->wr_ptr(recv_cnt); if (task_recv.putq(mb) == -1) { ACE_DEBUG((LM_ERROR, ACE_TEXT ("(%P|%t) %p; discarding data/n"), ACE_TEXT ("Server21 enqueue to TaskRecv failed"))); mb->release(); return 0; } return 0; } if (recv_cnt <= 0 || ACE_OS::last_error() != EWOULDBLOCK) { this->active_ = false; ACE_DEBUG((LM_ERROR, ACE_TEXT("(%P|%t) Connection closed/n"))); /*表示客户端已断开*/ return -1; } #endif return 0; } // 服务端向客户端发送数据时，回调此函数 int Server21::handle_output (ACE_HANDLE) { ACE_Message_Block *mb; ACE_Time_Value nowait (ACE_OS::gettimeofday()); while (!this->msg_queue()->is_empty()) if (-1 != this->getq(mb, &nowait)) { ssize_t send_cnt = this->peer().send(mb->rd_ptr(), mb->length()); if (send_cnt == -1) { ACE_ERROR ((LM_ERROR, ACE_TEXT ("(%P|%t) %p/n"), ACE_TEXT ("Server21 send"))); } else { ACE_DEBUG((LM_DEBUG, ACE_TEXT("(%P|%t) Send(%d):[%s]/n"), send_cnt, mb->rd_ptr())); mb->rd_ptr(ACE_static_cast(size_t, send_cnt)); } mb->release (); } /*方式2*/ /* if (this->msg_queue()->is_empty()) this->reactor()->cancel_wakeup(this, ACE_Event_Handler::WRITE_MASK); else this->reactor()->schedule_wakeup(this, ACE_Event_Handler::WRITE_MASK); */ /*(方法1)*/ this->reactor()->cancel_wakeup(this, ACE_Event_Handler::WRITE_MASK); return 0; } // 定时处理函数 int Server21::handle_timeout(const ACE_Time_Value ¤t_time, const void *act) { if (this->active_ != true) return 0; this->timeout_0(); return 0; } // 客户端断开时，回调此函数 int Server21::handle_close (ACE_HANDLE h, ACE_Reactor_Mask mask) { if (mask == ACE_Event_Handler::WRITE_MASK) return 0; else { ACE_TCHAR peer_name[MAXHOSTNAMELEN]; ACE_INET_Addr peer_addr; if (this->peer().get_remote_addr(peer_addr) == 0 && peer_addr.addr_to_string (peer_name, MAXHOSTNAMELEN) == 0) ACE_DEBUG((LM_INFO, ACE_TEXT("(%P|%t) Disconnected %s/n"), peer_name)); /*释放队列资源*/ task_send.set_active(false); task_recv.set_active(false); this->set_active(false); task_send.msg_queue()->close(); task_recv.msg_queue()->close(); this->msg_queue()->close(); task_send.wait(); task_recv.wait(); this->wait(); /*回调*/ this->stop_0(); /*关闭链路*/ this->peer().close_reader(); this->peer().close_writer(); this->peer().close(); return super::handle_close (h, mask); } } /**********************************************************************************************************************************/ void Server21::set_active(bool b) { this->active_ = b; } bool Server21::get_active() { return this->active_; } /**********************************************************************************************************************************/ void Server21::start_0() { /* if (last_peer != NULL) { last_peer->close_reader(); last_peer->close_writer(); last_peer->close(); } last_peer = &this->peer(); */ /**************************************************************** * TODO 可以在此实现客户端接入后需要的初始化工作 - （已经实际接入） *****************************************************************/ } void Server21::stop_0() { //if (last_peer == &this->peer()) last_peer = NULL; /**************************************************************** * TODO 可以在此实现客户端断开后需要执行的工作 *****************************************************************/ } void Server21::timeout_0() { /**************************************************************** * TODO 实现服务端定时任务，如发送心跳等，(注：多客户端时，只对本连接有效) *****************************************************************/ ACE_DEBUG((LM_DEBUG, ACE_TEXT("(%P|%t) handle_timeout/n"))); }

3、TaskRecv.cpp

4、TaskSend.cpp

5、ScheduleHandler.cpp

6、Config.cpp

/******************************************************************************************** * huangjf 2009年5月于福州 ********************************************************************************************/ #include "Global_Define.h" #include <ace/OS.h> #include <ace/Configuration.h> #include <ace/Configuration_Import_Export.h> // 载入配置文件 int GlobalValue::load_config(const char* config_filename_) { ACE_TString str; long num = 0; ACE_Configuration_Heap config; if (config.open() == -1) { ACE_ERROR_RETURN((LM_ERROR, ACE_TEXT("(%P|%t) %p/n"), ACE_TEXT("config.open()")), -1); } ACE_Ini_ImpExp config_importer(config); if (config_importer.import_config(config_filename_) == -1) { ACE_ERROR_RETURN((LM_ERROR, ACE_TEXT("(%P|%t) %p/n"), ACE_TEXT(config_filename_)), -1); } ACE_Configuration_Section_Key status_section; if (config.open_section (config.root_section(), ACE_TEXT("SYSTEM"), 0, status_section) == -1) { ACE_ERROR_RETURN ((LM_ERROR, ACE_TEXT("(%P|%t) %p/n"), ACE_TEXT ("Can't open [SYSTEM] section")), -1); } // 服务地址 if (config.get_string_value(status_section, ACE_TEXT("HOST"), str) != -1) { if (str.compare(this->host) != 0) { ACE_DEBUG((LM_DEBUG, ACE_TEXT("(%P|%t) [SYSTEM]->'HOST' is updated (%s)->(%s)/n"), this->host.c_str(), str.c_str())); this->host = str; } str.clear(); } else { this->host.clear(); this->host = ACE_TEXT("127.0.0.1"); ACE_DEBUG((LM_ERROR, ACE_TEXT("(%P|%t) [SYSTEM]->'HOST' does not exist, default value is (%s)/n"), this->host.c_str())); } // 服务端口 if (config.get_string_value(status_section, ACE_TEXT("PORT"), str) != -1) { try { num = ACE_OS::strtol(str.c_str(), NULL, 10); if (num != this->port) { ACE_DEBUG((LM_DEBUG, ACE_TEXT("(%P|%t) [SYSTEM]->'PORT' is updated (%d)->(%d)/n"), this->port, num)); this->port = static_cast<u_short> (num); } } catch (...) { this->port = 50001; ACE_DEBUG((LM_ERROR, ACE_TEXT("(%P|%t) [SYSTEM]->'PORT' is catch exception, default value is (%d)/n"), this->port)); } num = 0; } else { this->port = 50001; ACE_DEBUG((LM_ERROR, ACE_TEXT("(%P|%t) [SYSTEM]->'PORT' does not exist, default value is (%d)/n"), this->port)); } /**************************************************************** * TODO 在此增加读取配置文件的参数 *****************************************************************/ return 0; }

7、Global_Define.h

/******************************************************************************************** * huangjf 2009年4月于宁夏银川 ********************************************************************************************/ #ifndef _Global_Define_h #define _Global_Define_h #include <ace/OS.h> #include <ace/SString.h> #include <ace/Task.h> #include <ace/Synch.h> #include <ace/Message_Block.h> #include <ace/Log_Msg.h> #include <ace/Svc_Handler.h> #include <ace/Acceptor.h> #include <ace/SOCK_Stream.h> #include <ace/SOCK_Acceptor.h> #define RECV_HIGHT_MARK 8*1024 /*接收队列高水位标*/ #define SEND_HIGHT_MARK 8*1024 /*发送队列高水位标*/ #define INPUT_SIZE 4*1024 /*接收数据缓冲区大小*/ #define TASK_RECV_POOL_SIZE 10 //处理服务器接收到的客户端数据队列的线程池大小 #define TASK_SEND_POOL_SIZE 1 //处理服务器向客户端发送的数据队列的线程池大小 #define MSG_HEAD_MODE 0 /*表示是否采用消息头模式，0:表示不使用，1:表示使用*/ #define MSG_HEAD_SIZE 12 #define CONFIG_FILE "../config/config.cfg" #define GET_BYTE1(a) ( (a)%0x100 ) /*低1字节*/ #define GET_BYTE2(a) ( ((a)/0x100)%0x100 ) /*低2字节*/ #define GET_BYTE3(a) ( ((a)/0x10000)%0x100 ) /*低3字节*/ #define GET_BYTE4(a) ( ((a)/0x1000000)%0x100 ) /*低4字节*/ #define SET_BYTE1(a) ( (a) ) /*低1字节*/ #define SET_BYTE2(a) ( (a)*0x100 ) /*低2字节*/ #define SET_BYTE3(a) ( (a)*0x10000 ) /*低3字节*/ #define SET_BYTE4(a) ( (a)*0x1000000 ) /*低4字节*/ /**********************************************************************************************************************************/ /******************************************************************************************** * TODO 全局参数类，在此增加全局参数 ********************************************************************************************/ class GlobalValue { public: ACE_TString host; /*服务器主机地址*/ u_short port; /*服务器监听端口*/ int load_config(const char* config_filename_); }; /**********************************************************************************************************************************/ /******************************************************************************************** * 采用ACE_Task任务或主动对象处理模式，处理服务器向客户端发送的数据队列 ********************************************************************************************/ class TaskSend: public ACE_Task<ACE_MT_SYNCH> { public: TaskSend() : active_(false), task_peer_(NULL) { this->active_ = false; this->msg_queue()->high_water_mark(SEND_HIGHT_MARK); ACE_DEBUG((LM_DEBUG, ACE_TEXT("(%P|%t) TaskSend start/n"))); } virtual ~TaskSend() { this->active_ = false; this->msg_queue()->close(); ACE_DEBUG((LM_DEBUG, ACE_TEXT("(%P|%t) TaskSend stop/n"))); } virtual int svc(void); void set_active(bool b); bool get_active(); void set_task_peer(ACE_Task<ACE_MT_SYNCH> * _task_peer_); ACE_Task<ACE_MT_SYNCH> * get_task_peer(); int send_msg(char * _buf, size_t _size); private: bool active_; ACE_Task<ACE_MT_SYNCH> * task_peer_; void process_send(ACE_Message_Block *mb = NULL); }; /******************************************************************************************** * 采用ACE_Task任务或主动对象处理模式，处理服务器接收到的客户端数据队列 ********************************************************************************************/ class TaskRecv: public ACE_Task<ACE_MT_SYNCH> { public: TaskRecv() : active_(false), task_send_(NULL) { this->active_ = false; this->msg_queue()->high_water_mark(RECV_HIGHT_MARK); ACE_DEBUG((LM_DEBUG, ACE_TEXT("(%P|%t) TaskRecv start/n"))); } virtual ~TaskRecv() { this->active_ = false; this->msg_queue()->close(); ACE_DEBUG((LM_DEBUG, ACE_TEXT("(%P|%t) TaskRecv stop/n"))); } virtual int svc(void); void set_active(bool b); bool get_active(); void set_task_send(TaskSend *_task_send); TaskSend * get_task_send(); private: bool active_; TaskSend *task_send_; void process_recv(ACE_Message_Block *mb = NULL); int send_msg(char * _buf, size_t _size); }; /******************************************************************************************** * 采用ACE接受器（Acceptor）连接建立模式，实现双工单路服务端 ********************************************************************************************/ class Server21 : public ACE_Svc_Handler<ACE_SOCK_STREAM, ACE_MT_SYNCH> { typedef ACE_Svc_Handler<ACE_SOCK_STREAM, ACE_MT_SYNCH> super; public: Server21(void); virtual ~Server21(void); // 在连接建立后被自动回调 virtual int open (void * = 0); // Called when input is available from the client. virtual int handle_input(ACE_HANDLE fd = ACE_INVALID_HANDLE); // Called when output is possible. virtual int handle_output(ACE_HANDLE fd = ACE_INVALID_HANDLE); // Called when a timer expires. virtual int handle_timeout (const ACE_Time_Value ¤t_time, const void *act = 0); // Called when this handler is removed from the ACE_Reactor. virtual int handle_close(ACE_HANDLE handle, ACE_Reactor_Mask close_mask); void set_active(bool b); bool get_active(); private: bool active_; TaskRecv task_recv; TaskSend task_send; /******************************************************************************************** * 策略类，实现了Strategy模式。如果ACE_Message_Queue拥有一个策略对象，无论何时有ACE_Message_Block对象 * 进入队列，ACE_Message_Queue都会调用该策略对象的notify()方法，把一个通知放入队列，通知的目标是对象的handle_output()方法 ********************************************************************************************/ /*方式2*/ //ACE_Reactor_Notification_Strategy notifier_; /*回调函数，使用者在此实现对应功能*/ void start_0(); /*对等端接入后回调*/ void stop_0(); /*对等端关闭后回调*/ void timeout_0(); /*定时任务*/ }; /**********************************************************************************************************************************/ /******************************************************************************************** * 应用程序定时任务，及信号处理 ********************************************************************************************/ class ScheduleHandler : public ACE_Event_Handler { public: virtual int handle_timeout(const ACE_Time_Value ¤t_time, const void *act = 0); virtual int handle_signal(int signum, siginfo_t * = 0, ucontext_t * = 0); }; /**********************************************************************************************************************************/ typedef ACE_Acceptor<Server21, ACE_SOCK_ACCEPTOR> ServerAcceptor; #endif