一步一步完善我的设计（5）：各个模块编码码

#include "TsQueue.h"



TsQueue GVideoTsQueue;

TsQueue GDocOpenofficeTsQueue;

TsQueue GDocWpsTsQueue;

TsQueue GDocOfficeTsQueue;

TsQueue GDpptTsQueue;



bool TsQueue::PushTs(TransServer * ts)

{

        m_tsList.push_back(ts);



        return true;

}



TransServer* TsQueue::GetTsInSpecId(string tsId)

{

        list<TransServer*>::iterator it;



        for(it = m_tsList.begin();it != m_tsList.end();it ++){

                if((*it)->GetId() == tsId){

                        return (TransServer *)(*it);

                }

        }



        return NULL;

}



bool TsQueue::CheckSameTs(string tsId)

{

        list<TransServer*>::iterator it;

        bool ret = false;



        for(it = m_tsList.begin();it != m_tsList.end();it ++){

                if((*it)->GetId() == tsId){

                        ret = true;

                }

        }



        return ret;

}



bool TsQueue::DelTs(string tsId)

{

        list<TransServer*>::iterator it;

        bool ret = false;



        for(it = m_tsList.begin();it != m_tsList.end();it ++){

                if((*it)->GetId() == tsId){

                        m_tsList.erase(it);

                        ret = true;

                }

        }



        return ret;

}



bool TsQueue::DelTs(TransServer *ts)

{

        list<TransServer*>::iterator it;

        bool ret = false;



        for(it = m_tsList.begin();it != m_tsList.end();it ++){

                if((*it) == ts){

                        m_tsList.erase(it);

                        ret = true;

                }

        }



        return ret;

}



TransServer* TsQueue::GetTsIdle()

{

        list<TransServer*>::iterator it;



        for(it = m_tsList.begin();it != m_tsList.end();it ++){

                if((*it)->GetState() == TsIdle){

                        return (TransServer *)(*it);

                }

        }



        return NULL;

}



TransServer* TsQueue::GetTransServer(TsSession* session)

{

        list<TransServer*>::iterator it;



        for(it = m_tsList.begin();it != m_tsList.end();it ++){

                if((*it)->GetSession() == session)

                        return *it;

        }



        return NULL;

}

#ifndef __TS_UPDATE_H__

#define __TS_UPDATE_H__



#include "stdio.h"

#include "TsInterface.h"

#include "TaskQueue.h"



class TsUpdate : public ITsUpdateInfo

{

        TsUpdate* next;

protected:

        string m_taskType;

        TaskRunQueue* m_taskRunQueue;

public:

        bool isThis(string taskType)

        {

                if(m_taskType == taskType)

                        return true;

                else

                        return false;

        }

        void setNext(TsUpdate *n)

        {

                next = n;

        }

        void add(TsUpdate *n)

        {

                if (next)

                        next->add(n);

                else

                        next = n;

        }

        

        virtual void OnUpdateInfo(Json::Value& value,TsSession *session)

        {

                if(next != NULL)

                        next->OnUpdateInfo(value,session);

                else {  

                        printf("error");

                        // TODO: log

                }

        }

};



class TsVideoTransProgress : public TsUpdate

{

        TsVideoTransProgress();

public:

        static TsVideoTransProgress& GetInstance();

        virtual void OnUpdateInfo(Json::Value& value,TsSession *session);

};









#endif

#include "TsUpdate.h"



////////////////////////////////////////



extern TaskRunQueue GVideoTaskRunQueue;



///////////////// TsVideoTransProgress //////////////////

TsVideoTransProgress::TsVideoTransProgress()

{

        m_taskType = "Transfer";

        m_taskRunQueue = &GVideoTaskRunQueue;

}



TsVideoTransProgress& TsVideoTransProgress::GetInstance()

{

        static TsVideoTransProgress instance_;

        return instance_;



}

void TsVideoTransProgress::OnUpdateInfo(Json::Value& value,TsSession *session)

{

        Task* task;



        

        if(value["TaskType"].isNull() || value["id"].isNull() || value["progress"].isNull()){



                // TODO:  log

                return;

        }

                

        if(isThis(value["TaskType"].asString())){

                task = m_taskRunQueue->GetTaskInSpecId(value["id"].asString());

                if(!task){

                        // TODO: log

                        return;

                }



                if(!value["progress"].isInt()){

                        // TODO: log

                        return;

                }



                task->SetProgress(value["progress"].asInt());

                

        } else {

                TsUpdate::OnUpdateInfo(value,session);

        }

}