peakzhang 发表于 2008-9-10 22:37:33

创建一个可正常结束的Proactor服务器

作者:Qinglan
这是APG上关于Proactor使用的例子,去掉了trace信息,做了一点小修改:


#include <ace/Os_main.h>
#include <ace/Asynch_Acceptor.h>
#include <ace/Proactor.h>
#define LISTEN_PORT 5222

class HA_Proactive_Service : public ACE_Service_Handler
{
public:
       ~HA_Proactive_Service ()
       {
            if (this->handle () != ACE_INVALID_HANDLE)
                     ACE_OS::closesocket (this->handle ());
       }

       virtual void open (ACE_HANDLE h, ACE_Message_Block&)
       {
            this->handle (h);
            if (this->reader_.open (*this) != 0 || this->writer_.open (*this) != 0   )
            {
                     delete this;
                     return;
            }

            ACE_Message_Block *mb;
            ACE_NEW_NORETURN (mb, ACE_Message_Block (1024));
            if (this->reader_.read (*mb, mb->space ()) != 0)
            {
                     mb->release ();
                     delete this;
                     return;
            }
       }

       virtual void handle_read_stream (const ACE_Asynch_Read_Stream::Result &result)
       {
            ACE_Message_Block &mb = result.message_block ();
            if (!result.success () || result.bytes_transferred () == 0)
            {
                     mb.release ();
                     delete this;
            }
            else
            {
                     ACE_DEBUG ((LM_DEBUG, ACE_TEXT("Received Data : %c\n"), *mb.rd_ptr()));
                     mb.release();

                     ACE_Message_Block *new_mb;
                     ACE_NEW_NORETURN (new_mb, ACE_Message_Block (1024));
                     this->reader_.read (*new_mb, new_mb->space ());
            }
       }

       virtual void handle_write_stream (const ACE_Asynch_Write_Stream::Result &result)
       { result.message_block ().release (); }

private:
       ACE_Asynch_Read_Stream reader_;
       ACE_Asynch_Write_Stream writer_;
};

class HA_Proactive_Acceptor : public ACE_Asynch_Acceptor<HA_Proactive_Service>
{
};

int ACE_TMAIN (int, ACE_TCHAR *[])
{
       ACE_INET_Addr listen_addr( LISTEN_PORT );
       HA_Proactive_Acceptor aio_acceptor;
       if (0 != aio_acceptor.open (listen_addr, 0, 0, 5, 1, 0, 0))
            ACE_ERROR_RETURN ((LM_ERROR, ACE_TEXT ("%p\n"), ACE_TEXT ("acceptor open")), 1);

       ACE_Proactor::instance ()->proactor_run_event_loop ();

       return 0;
}

程序在5222端口上监听,对于每个已建立的连接,服务器将所有收到的数据打印到控制台上。这就是整个程序所实现的功能。
但是这个程序存在一个很大的问题:没有办法正常结束。当运行这个程序后,只能强制结束这个进程,因为程序没有提供从proactor事件循环中退出来的方法。
让程序能够正常退出是很有必要的,比如在进行内存泄漏的检测时,如果程序没有办法正常退出,那么内存泄漏的检测工作将很难进行。

为了实现能够正常关闭程序的目的,需要稍微修改一下上面的程序实现。
这次程序将会有两个线程在运行,一个是Proactor事件循环线程,另一个则为控制线程。在控制线程中,要能够关闭Proactor线程,然后将程序干净地退出。
得益于ACE强大的封装,所做的改动并不大:


class ProactorThread : public ACE_Task_Base
{
public:
       int open()
       {
            return this->activate();
       }

       int close()
       {
            ACE_Proactor::instance()->proactor_end_event_loop();
            this->wait();
            return 0;
       }

       virtual int svc()
       {
            ACE_INET_Addr listen_addr( LISTEN_PORT );
            HA_Proactive_Acceptor aio_acceptor;
            if (0 != aio_acceptor.open (listen_addr, 0, 0, 5, 1, 0, 0))
                     ACE_ERROR_RETURN ((LM_ERROR, ACE_TEXT ("%p\n"), ACE_TEXT ("acceptor open")), 1);

            ACE_Proactor::instance ()->proactor_run_event_loop ();
            ACE_DEBUG ((LM_DEBUG, ACE_TEXT("Thread ended\n")));

            return 0;
       }
};

int ACE_TMAIN (int, ACE_TCHAR *[])
{
       ProactorThread thread;
       thread.open();

       int x;
       std::cin >> x;

       thread.close();

       return 0;
}

仅仅只是添加了一个新的类,用于封装Proactor事件循环线程,另外修改了程序的主函数。这次程序将首先创建一个Proactor事件循环线程,然后主线程等待用户输入任意数后结束Proactor线程,在Proactor线程正常退出后主线程退出。

注意这里有两个要点:
1.当执行proactor_end_event_loop()之后,Proactor线程是从proactor_run_event_loop()处退出,所以会接着执行下面的代码,打印出” Thread ended”信息。
2.主线程必须使用wait()等待Proactor线程正常结束后再退出,否则如果主线程退出后,Proactor线程仍有一些资源释放工作未完成,那么就会存在资源未正确释放的问题。

peakzhang 发表于 2008-9-10 22:37:51

更深入的处理可以是这样的:

       int x;
       std::cin >> x;
改为注册成信号,用户按ctrl+c,向系统插入一个特殊的ACE_Message_Block的实例,各线程收到后,均正常结束,在ACE_Stream框架中,常用这种办法.
这处结束过程还可以把一些未处理的消息写入磁盘,下次启动后载入后接着处理.

psycheqiqi 发表于 2009-9-29 11:27:50

我把这个程序移植到MFC里面的时候,在ProactorThread中aio_acceptor.open的时候跳转不到HA_Proactive_Service中,为什么?

winston 发表于 2009-9-29 22:37:54

错误是什么?
页: [1]
查看完整版本: 创建一个可正常结束的Proactor服务器