如何定制一款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页面讲解其上详细的过程。

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

扫码关注云+社区

领取腾讯云代金券