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

#include "RegistryImp.h"
#include "RegistryProcThread.h"
#include "RegistryServer.h"

//初始化配置db连接
extern TC_Config * g_pconf;
extern RegistryServer g_app;

void RegistryImp::initialize()
{
    TLOGDEBUG("begin RegistryImp init"<<endl);

    _db.init(g_pconf);

    TLOGDEBUG("RegistryImp init ok."<<endl);

}

int RegistryImp::registerNode(const string & name, const NodeInfo & ni, const LoadInfo & li, tars::TarsCurrentPtr current)
{
    return _db.registerNode(name, ni, li);
}

int RegistryImp::keepAlive(const string & name, const LoadInfo & ni, tars::TarsCurrentPtr current)
{
    RegistryProcInfo procInfo;
    procInfo.nodeName = name;
    procInfo.loadinfo = ni;
    procInfo.cmd      = EM_NODE_KEEPALIVE;

    //放入异步处理线程中
    g_app.getRegProcThread()->put(procInfo);

    return 0;
}

vector<tars::ServerDescriptor> RegistryImp::getServers(const std::string & app,const std::string & serverName,const std::string & nodeName,tars::TarsCurrentPtr current)
{
    return  _db.getServers(app, serverName, nodeName);
}

int RegistryImp::updateServer(const string & nodeName, const string & app, const string & serverName, const tars::ServerStateInfo & stateInfo, tars::TarsCurrentPtr current)
{
    return _db.updateServerState(app, serverName, nodeName, "present_state", stateInfo.serverState, stateInfo.processId);
}

int RegistryImp::updateServerBatch(const std::vector<tars::ServerStateInfo> & vecStateInfo, tars::TarsCurrentPtr current)
{
    return _db.updateServerStateBatch(vecStateInfo);
}

int RegistryImp::destroy(const string & name, tars::TarsCurrentPtr current)
{
    return _db.destroyNode(name);
}

int RegistryImp::reportVersion(const string & app, const string & serverName, const string & nodeName, const string & version, tars::TarsCurrentPtr current)
{
    RegistryProcInfo procInfo;
    procInfo.appName    = app;
    procInfo.serverName = serverName;
    procInfo.nodeName   = nodeName;
    procInfo.tarsVersion = version;
    procInfo.cmd        = EM_REPORTVERSION;

    //放入异步处理线程中
    g_app.getRegProcThread()->put(procInfo);

    return 0;
}

int RegistryImp::getNodeTemplate(const std::string & nodeName,std::string &profileTemplate,tars::TarsCurrentPtr current)
{
    string sTemplateName;
    int iRet = _db.getNodeTemplateName(nodeName, sTemplateName);
    if(iRet != 0 || sTemplateName == "")
    {
        //默认模板配置
        sTemplateName = (*g_pconf)["/tars/nodeinfo<defaultTemplate>"];
    }

    string sDesc;
    profileTemplate = _db.getProfileTemplate(sTemplateName, sDesc);

    TLOGDEBUG(nodeName << " get sTemplateName:" << sTemplateName << " result:" << sDesc << endl);

    return 0;
}

int RegistryImp::getClientIp(std::string &sClientIp,tars::TarsCurrentPtr current)
{
    sClientIp = current->getIp();
    TLOGDEBUG("RegistryImp::getClientIp ip: " << sClientIp <<  endl);

    return 0;
}

int RegistryImp::updatePatchResult(const PatchResult & result, tars::TarsCurrentPtr current)
{
    TLOGDEBUG( "RegistryImp::updatePatchResult " << result.sApplication + "." + result.sServerName + "_" + result.sNodeName << "|V:" << result.sVersion << "|U:" <<  result.sUserName << endl);

    return _db.setPatchInfo(result.sApplication, result.sServerName, result.sNodeName, result.sVersion, result.sUserName);
}