/**
 * 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 _QueryServer_H_
#define _QueryServer_H_

#include <iostream>
#include <set>
#include "servant/Application.h"
#include "util/tc_thread_mutex.h"
#include "util/tc_lock.h"
#include "util/tc_thread_pool.h"
#include "util/tc_atomic.h"
#include "DbThread.h"
#include "QueryDbThread.h"

using namespace std;
using namespace tars;

/////////////////////////////////////////////////////////////////////
class QueryParam
{
public:
 int _run_times;
 int _run_result;
 TC_Atomic _atomic;
 TC_ThreadLock _monitor;

 QueryParam()
 : _run_times(0)
 , _run_result(0)
 , _atomic(0)
 {}
};
/////////////////////////////////////////////////////////////////////
class QueryServer : public Application , public TC_ThreadMutex
{
public:
 /**
 *
 **/
 virtual ~QueryServer() {};

 /**
 *
 **/
 virtual void initialize();

 /**
 *
 **/
 virtual void destroyApp();

 int getInsertInterval() const { return _insertInterval; };

 size_t getDbNumber() const { return _dbStatInfo.size(); }

 size_t getActiveDbSize() const { return _activeDbInfo.size(); }

 void setActiveDb( const vector<TC_DBConf> &vDbInfo);

 vector<TC_DBConf> getAllActiveDbInfo() const;

 vector<TC_DBConf> getDbInfo() const;

 vector<TC_DBConf> getActiveDbInfo() const;

 TC_DBConf getDbInfo(int iIndex) { return _dbStatInfo[iIndex]; }

 string dumpDbInfo(const vector<TC_DBConf>& vDbInfo) const;

 uint32_t genUid();

 TC_ThreadPool & getThreadPoolTimeCheck() { return _timeCheck; }

 TC_ThreadPool & getThreadPoolDb() { return _poolDb; }

 QueryDbThread * getThreadPoolQueryDb() { return _tpoolQueryDb; }

 bool searchQueryFlag(const string &sKey);

 //匹配非tars被调服务名
 set<string>& getNotTarsSlaveName();

private:
 uint32_t _uniqId; //唯一id

 int _insertInterval; //查询最后入库时间使用,入库耗时不会超过入库间隔,单位为分钟

 DBThread * _dBThread; //定时检查数据库实例是否存活

 QueryDbThread *_tpoolQueryDb; //用于处理数据库的查询操作

 vector<TC_DBConf> _dbStatInfo; //数据库信息

 vector<TC_DBConf> _activeDbInfo; //存活的数据库信息

 set<string> _queryFlag; //数据库字段

 TC_ThreadPool _timeCheck; //处理timecheck操作的线程池

 TC_ThreadPool _poolDb; //具体查询压缩维度后的数据库实例数据的线程池

 
 set<string> _notTarsSlaveName; //匹配非tars被调服务名
};

extern QueryServer g_app;
extern TC_Config * g_pconf;
////////////////////////////////////////////
#endif