/**
 * 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.
 */

#include "LoadDbThread.h"
#include "NotifyServer.h"

LoadDbThread::LoadDbThread()
: _interval(30)
, _terminate(false)
{

}

LoadDbThread::~LoadDbThread()
{
    if(isAlive())
    {
        terminate();
        getThreadControl().join();
    }
}

void LoadDbThread::terminate()
{
    _terminate = true;

    TC_ThreadLock::Lock lock(*this);

    notifyAll();
}

void LoadDbThread::init()
{
    try
    {
        map<string, string> mDbConf = g_pconf->getDomainMap("/tars/db");
        assert(!mDbConf.empty());

        TC_DBConf tcDBConf;
        tcDBConf.loadFromMap(mDbConf);

        _mysql.init(tcDBConf);

        TLOGDEBUG("LoadDbThread::init init mysql conf succ." << endl);
    }
    catch(exception &ex)
    {
        TLOGERROR("LoadDbThread::init ex:" << ex.what() << endl);
        FDLOG("EX") << "LoadDbThread::init ex:" << ex.what() << endl;
    }
}

void LoadDbThread::run()
{
    size_t iLastTime = 0;
    size_t iNow = TNOW;

    while (!_terminate)
    {
        if(iNow - iLastTime >= _interval)
        {
            loadData();

            iLastTime = iNow;
        }

        {
            TC_ThreadLock::Lock lock(*this);
            timedWait(1000);
        }
    }
}

void LoadDbThread::loadData()
{
    try
    {
        TC_Mysql::MysqlData mysqlData;
        map<string, string> &mTmep = _data.getWriter();
        mTmep.clear();
        size_t iOffset(0);

        do
        {
            string sSql("select application, set_name, set_area, set_group, server_name, node_name from t_server_conf limit 1000 offset ");
            sSql = sSql + TC_Common::tostr<size_t>(iOffset) +";";

            mysqlData = _mysql.queryRecord(sSql);

            for (size_t i = 0; i < mysqlData.size(); i++)
            {
                string sValue = mysqlData[i]["set_name"] +"."+ mysqlData[i]["set_area"]+ "." + mysqlData[i]["set_group"];
                if (mysqlData[i]["set_name"].empty())
                {
                    continue;
                }

                string sKey = mysqlData[i]["application"] + "." +  mysqlData[i]["server_name"] + mysqlData[i]["node_name"];
                mTmep.insert(map<string, string>::value_type(sKey, sValue));
            }

            iOffset += mysqlData.size();

        } while (iOffset % 1000 == 0);

        _data.swap();

        TLOGDEBUG("LoadDbThread::loadData load data finish, _mSetApp size:" << mTmep.size() << endl);

    }
    catch (exception &ex)
    {
        TLOGERROR("LoadDbThread::loadData exception:" << ex.what() << endl);
        FDLOG("EX") << "LoadDbThread::loadData exception:" << ex.what() << endl;
    }
}