如何定制一款12306抢票浏览器——启动“人”线程

启动“人”线程

在《如何定制一款12306抢票浏览器——构架》一文中，我们提到“人”线程。对于熟悉Window编程的同学来说，线程间通信和信息传递不是难题。但是由于浏览器和我们“人”线程之间传递的是COM对象，这个过程就没有那么简单了。（转载请指明出于breaksoftware的csdn博客）

首先我们要明确下传递的到底是什么COM对象。一般来说，如果我们要操控浏览器中的页面，都是从IWebBrowser接口对象开始的。我们这儿也是要传递这个接口对象

VOID SetWebBrowser(CComPtr<IWebBrowser2> & spWeb);

其次我们要明确下什么时候要传递IWebBrowser接口对象。我选择是的NavigateComplete2消息触发时告诉“人”线程该接口对象。

STDMETHODIMP_(void) CBrowserHost::NavigateComplete2( IDispatch *pDisp, VARIANT *URL )
{
    CComPtr<IWebBrowser2> spWeb;
    m_webBrowser.QueryControl( IID_IWebBrowser2, (LPVOID*)&spWeb);
    m_AutoMan.SetWebBrowser( spWeb );
}

假如12306一个页面加载完，只会触发一次NavigateComplete2事件，那我们可能就没必要在此特别独立出一篇文章来说“人”线程的启动了。观察过12306页面的同学应该发现，它的页面中嵌入了多个Iframe。而每一个Iframe完成，都会触发一次NavigateComplete2事件。这样就导致我们产生在多线程编程中的经典问题：“消费者”和“生产者”问题。此处的生产者是浏览器，它会不时制造个产品（IWebBrowser对象）出来。而“消费者”就是我们的“人”线程，面对这么多的产品，它将如何做出选择？

我们先看下生产者的行为

VOID CAutoMan::SetWebBrowser( CComPtr<IWebBrowser2> & spWeb )
{
    CoInitializeEx(NULL, COINIT_APARTMENTTHREADED);
    do {
        EnterCriticalSection(&m_cs);
        CComPtr<IUnknown> spIUnknown;
        IStream* spIStream = NULL;
        HRESULT hr = spWeb->QueryInterface(IID_IUnknown, (LPVOID*)&spIUnknown);
        CHECKHRPOINTER(hr, spIUnknown);

        try {
            hr = ::CoMarshalInterThreadInterfaceInStream( __uuidof(IWebBrowser2), spIUnknown, &spIStream );
        }
        catch (...) {
        	hr = E_FAIL;
        }  
        CHECKHRPOINTER(hr, spIStream);
        m_ListIStream.push_back(spIStream);
        LeaveCriticalSection(&m_cs);
    } while (0);
    CoUninitialize(); 
}

因为是多线程，我们使用了临界区m_cs保证了对产品仓库——m_ListIStream的有序化管理。我们对于浏览器制造出来的初级产品进行包装——CoMarshalInterThreadInterfaceInStream，产生一个流对象。再将这个流对象放到产品库最后一个位置。此处要特别注意一下流对象，像我比较喜欢用ATL管理COM的人，此时对流对象IStream* spIStream没有使用CComPtr进行管理。因为这个流对象在这个函数内部还不能释放掉，我们要在“人”线程中读取它。“人”线程中的“解开包装”的函数会负责对它的释放。

对于“人”线程，它可能在处理完一个IWebBrowser接口对象后，要接着处理产品库中其他接口对象。那么它该如何选择呢？我们可以把它想成一个人，其实我们在浏览网页的过程中，浏览器发出了很多个事件，而我们却不会关心这些事件，我们只是关心最后的状态——是的，我们的“人”线程也是如此，它只关心最后一个产品——因为它是最新的，有最新的干嘛要用过时的东西呢？

HRESULT CAutoMan::ConvertInterface()
{
    HRESULT hr = E_FAIL;
    CComPtr<IWebBrowser2> spTempWebB = NULL;
    EnterCriticalSection(&m_cs);
    do {
        // 获取最后一个IStream，以它作为标准
        ListIStreamRIter iterLast = m_ListIStream.rbegin();
        if ( iterLast == m_ListIStream.rend() || NULL == *iterLast ) {
            break;
        }

        // 释放其他的IStream
        for ( ListIStreamIter iter = m_ListIStream.begin(); iter != m_ListIStream.end(); iter++ ) {
             if ( *iter == *iterLast || NULL == *iter ) {
                 continue;
             }
            (*iter)->Release();
            *iter = NULL;
        }

        spTempWebB = NULL;
        CHECKPOINT(*iterLast);
        try {
            hr = CoGetInterfaceAndReleaseStream((*iterLast), __uuidof(IWebBrowser2), (LPVOID*)&spTempWebB );
        }
        catch(...) {
            hr = E_FAIL;
        }
        CHECKHRPOINTER(hr, spTempWebB);
        *iterLast = NULL; 
    } while (0);
    m_ListIStream.clear();
    LeaveCriticalSection(&m_cs);
    if ( NULL != spTempWebB ) {
        m_spWindow = NULL;
        m_spWindow = spTempWebB;
    }
    return hr;
}

以上代码注释写的很清楚了，“人”线程拿到最后一个最新的IStream，并对它进行了解包装，把结果保存在临时变量spTempWebB中。同时它释放了仓库中其他的过时的IStream接口对象。此处有个地方要注意，我没有直接将IStream转换成m_spWindow，因为在转之前要将m_spWindow置为NULL。而恰恰是这个置为NULL的过程，可能会和之前SetWebBrowser的过程发生死锁。所以此处我用一个临时变量去接收转换结果，最后再将m_spWindow设置为该结果。
线程函数的代码是

VOID CAutoMan::ThreadFun()
{
    m_dwQueryTime = QUERYTIMESLOW;
    while ( WAIT_TIMEOUT == WaitForSingleObject(m_hStopEvent, m_dwQueryTime )) {
        ConvertInterface();
        if ( NULL == m_spWindow ) {
            continue;
        }
        CComBSTR bstrUrl;
        HRESULT hr = m_spWindow->get_LocationURL(&bstrUrl);
        CComPtr<IHTMLDocument2> spDoc;

        CComPtr<IDispatch> spDispatch;
        hr = m_spWindow->get_Document(&spDispatch);
        if ( FAILED(hr) || NULL == spDispatch ){
            continue;
        }
        hr = spDispatch->QueryInterface( IID_IHTMLDocument2, (LPVOID*)&spDoc);
        try {
            if ( m_DealWebPage.IsQueryPage(spDoc, bstrUrl)) {
                hr = m_DealWebPage.InsertButtonInQueryPage(spDoc);
                if ( m_bStartQuery ) {
                    hr = m_DealWebPage.QueryTicketsInfo(spDoc);
                    if (FAILED(hr)) {
                        hr = m_DealWebPage.StartQueryInQueryPage(spDoc);
                    }
                }
                else {
                }
            }
            else if ( m_DealWebPage.IsBookingPage(spDoc, bstrUrl) ) {
                hr = m_DealWebPage.BookTickets(spDoc);
                if (SUCCEEDED(hr)) {
                    // 待处理，退出线程
                }
                else {
                }
            }
        }
        catch(...) {
        }
    }
}

“人”线程会每隔m_dwQueryTime毫秒进行一次轮询操作。操作的内容就是：

1 查询当前URL

2 如果当前URL是订票查询页面（m_DealWebPage.IsQueryPage(spDoc, bstrUrl)），则进行

A 插入控制按钮（hr = m_DealWebPage.InsertButtonInQueryPage(spDoc);）

B 检查是否有票（hr = m_DealWebPage.QueryTicketsInfo(spDoc);）

C 在不存在其票的情况下点击刷新按钮的操作（hr = m_DealWebPage.StartQueryInQueryPage(spDoc);）

3 如果当前是确认订单页面（m_DealWebPage.IsBookingPage(spDoc, bstrUrl)），则进行订票操作（hr = m_DealWebPage.BookTickets(spDoc);）

由于为时已晚，我无法图文并茂的讲解之后的订票过程。今天就到此休息了，今天晚上我会结合12306页面讲解其上详细的过程。