/**
 * Tencent is pleased to support the open source community by making Tars available.
 *
 * Copyright (C) 2016THL A29 Limited, a Tencent company. All rights reserved.
 *
 * Licensed under the BSD 3-Clause License (the "License"); you may not use this file except 
 * in compliance with the License. You may obtain a copy of the License at
 *
 * https://opensource.org/licenses/BSD-3-Clause
 *
 * Unless required by applicable law or agreed to in writing, software distributed 
 * under the License is distributed on an "AS IS" BASIS, WITHOUT WARRANTIES OR 
 * CONDITIONS OF ANY KIND, either express or implied. See the License for the 
 * specific language governing permissions and limitations under the License.
 */

#ifndef __BATCH_PATCH_THREAD_H_
#define __BATCH_PATCH_THREAD_H_
#include "servant/Application.h"
#include "Node.h"
#include "ServerObject.h"
#include "util/tc_thread_queue.h"

class BatchPatchThread;

class BatchPatch //: public tars::TC_ThreadLock
{
public:
    /**
    * 构造函数
    */
    BatchPatch();

    /**
    * 析构函数
    */
    ~BatchPatch();

    /**
     * 结束线程
     */
    void terminate();

    /**
     * 线程运行
     */
    void start(int iThreadNum);

    void timedWait(int millsecond);
public:
    /**
     * 插入发布请求
     */
    void push_back(const tars::PatchRequest & request, ServerObjectPtr servePtr);

    /**
     * 从发布队列中获取发布请求
     */
    bool pop_front(pair<tars::PatchRequest,ServerObjectPtr>& item);

    /**
     * 设置下载暂存目录
     */
    void setPath(const std::string & sDownloadPath)
    {
        _downloadPath = sDownloadPath;
    }
private:
    tars::TC_ThreadLock                  _queueMutex;

    TC_ThreadQueue<pair<tars::PatchRequest,ServerObjectPtr> > _patchQueue;

    std::set<std::string>               _patchIng;

    std::vector<BatchPatchThread *>     _runners;

    std::string                         _downloadPath;       
};


class BatchPatchThread : public TC_Thread
{
public:
    BatchPatchThread(BatchPatch * patch);

    ~BatchPatchThread();

    virtual void run();

    void terminate();

public:
    /**
     * 设置下载暂存目录
     */
    void setPath(const std::string & sDownloadPath)
    {
        _downloadPath = sDownloadPath;
    }

    /**
     * 执行发布单个请求
     */
    void doPatchRequest(const tars::PatchRequest & request, ServerObjectPtr server);

protected:
    BatchPatch *    _batchPatch;

    std::string     _downloadPath; //文件下载目录

protected:
    bool            _shutDown; 
};

#endif