warthreadforpool.cpp

ftpserver very good sample

#include "StdAfx.h"
#include "WarThreadForPool.h"   // class implemented

#ifndef WAR_THREAD_POOL_H
#   include "WarThreadPool.h"
#endif

#ifndef WAR_THREAD_POOL_TASK_H
#   include "WarThreadPoolTask.h"
#endif

#ifndef WAR_LOG_H
#   include "WarLog.h"
#endif

/////////////////////////////// PUBLIC ///////////////////////////////////////

//============================= LIFECYCLE ====================================

WarThreadForPool::WarThreadForPool(WarThreadPool * pPool)
    : mpPool(pPool),
    mAutoCnt(pPool->mCntThreadsExist, pPool->mLock)
{
}

//============================= OPERATORS ====================================
//============================= OPERATIONS ===================================
//============================= ACCESS     ===================================
//============================= INQUIRY    ===================================
/////////////////////////////// PROTECTED  ///////////////////////////////////

void WarThreadForPool::ProcessRequest(WarThreadPoolTask * pTask)
{
    // The default processing is to let the task object
    // handle its own buisness
    pTask->OnProcessing(this);
}

WarThreadForPool::~WarThreadForPool()
{
}

/////////////////////////////// PRIVATE    ///////////////////////////////////



void
WarThreadForPool::Run()
{
    SetState(STATE_HOUSEKEEPING);

    mpPool->Attach(this);

    while (!IsExiting())
    {
        ResetCancelFlag();
        WarThreadPoolTask * pTask = mpPool->PopTask();

        if (pTask)
        {
            WarPtrWrapper<WarThreadPoolTask> MySmartie(pTask);

            pTask->mpThread = this;
            
            SetState(STATE_BUSY);
            pTask->SetState(WarThreadPoolTask::STATE_PROCESSING);

            try
            {
                ProcessRequest(pTask);
            }
            catch(WarException& ex)
            {
                WarLogError("WarThreadForPool::Run()",
                    "Caught unhandled WarException",
                    &ex);
            }

#if WAR_CATCH_ALL            
            catch(...)
            {
                WarLogError("WarThreadForPool::Run()",
                    "Caught unknown exception", NULL);
                break;
            }
#endif

            pTask->mpThread = NULL;

#if WAR_CATCH_ALL
            try
#endif
            {
                mpPool->TaskDone(MySmartie, GetCancelFlag());
            }

#if WAR_CATCH_ALL            
            catch(...)
            {
                WarLogError("WarThreadForPool::Run()",
                    "Caught unknown exception", NULL);
                break;
            }
#endif
        }

        if (IsExiting())	// New test
            break;

        SetState(STATE_IDLE);
        mpPool->mSignal.WaitForEvent();
        SetState(STATE_HOUSEKEEPING);
    }

    mpPool->Detach(this);
}