/**
 * 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 "QueryImp.h"
#include "RequestDecoder.h"
#include "QueryItem.h"
#include "servant/Application.h"

using namespace std;

//////////////////////////////////////////////////////
void QueryImp::initialize()
{
    //initialize servant here:
    //...
}

//////////////////////////////////////////////////////
void QueryImp::destroy()
{
    //destroy servant here:
    //...
}


int QueryImp::doRequest(tars::TarsCurrentPtr current, vector<char>& response)
{
    const vector<char>& request = current->getRequestBuffer();
    string buf((const char*)(&request[0]), request.size());
    string sUid = TC_Common::tostr(g_app.genUid()) + "|";

    FDLOG("inout") << "QueryImp::doRequest sUid:" << sUid << "doRequest input:" <<  buf << endl;
    TLOGDEBUG(sUid << "QueryImp::doRequest sUid:" << sUid << "doRequest input:" <<  buf << endl);

    doQuery(sUid, buf, false, current);

    return 0;
}

int QueryImp::doQuery(const string sUid, const string &sIn, bool bTarsProtocol, tars::TarsCurrentPtr current)
{
    try
    {
        size_t pos = sIn.rfind("}"); // find json end

        string s("");

        if (pos != string::npos)
        {
            s = sIn.substr(0, pos+1);
        }
        else
        {
            throw TC_Exception("bad query string");
        }

        RequestDecoder decoder(s);
        TLOGDEBUG("QueryImp::doQuery"<<endl);
        int ret = decoder.decode();

        decoder.addUid(sUid); //传人uid，供打印使用

        TLOGDEBUG("QueryImp::doQuery " << sUid << "decode json ret:" << ret << "|sqlPart: " << TC_Common::tostr(decoder.getSql()) << endl);

        int64_t tStart    = 0;
        int64_t tEnd    = 0;

        if(ret == RequestDecoder::TIMECHECK)
        {
            string sResult("");

            tStart    = TC_TimeProvider::getInstance()->getNowMs();

            string lasttime = _proxy.getLastTime(decoder.getSql());
            sResult += "lasttime:" + lasttime + "\n";
            sResult += "endline\n";

            tEnd = TC_TimeProvider::getInstance()->getNowMs();

            current->sendResponse(sResult.c_str(), sResult.length());

            FDLOG("inout") << "QueryImp::doQuery time_check sUid:" << sUid << "send result succ,size:"<< sResult.length() << "|timecost(ms):" << (tEnd-tStart) << endl;
            TLOGDEBUG("QueryImp::doQuery time_check sUid:" << sUid << "send result succ,size:"<< sResult.length() << "|timecost(ms):" << (tEnd-tStart) << endl);
        }

        else if(ret == RequestDecoder::QUERY)
        {
            current->setResponse(false);
            QueryItem * pItem = new QueryItem();
            pItem->sUid     = sUid;
            pItem->current  = current;
            pItem->mQuery   = decoder.getSql();

            map<string,string> &mSqlPart    = decoder.getSql();

            map<string,string>::iterator it;
            for(it=mSqlPart.begin();it!=mSqlPart.end();it++)
            {
                TLOGDEBUG("QueryImp::mysql "<<it->first<<"|"<<it->second<<endl);
            }
            string sGroupField                = mSqlPart["groupField"];
            vector<string> vGroupField        = TC_Common::sepstr<string>(sGroupField, ", ");
            
            pItem->bFlag = true;
            g_app.getThreadPoolQueryDb()->put(pItem);

            TLOGDEBUG("QueryImp::doQuery all dbcount." << endl);
        }
        else
        {
            TLOGERROR("QueryImp::doQuery " << sUid << "decode request failed\n" <<endl);

            string sResult = "Ret:-1\ndecode request failed\n";
            sResult += "endline\n";

            current->sendResponse(sResult.c_str(), sResult.length());

            FDLOG("inout") << "QueryImp::doQuery failed sUid:" << sUid << endl;
        }
    }
    catch(exception &ex)
    {
        TLOGERROR("QueryImp::doQuery exception:" << ex.what() << endl);
        string sResult = "Ret:-1\n" +  string(ex.what()) + "\nendline\n";
        current->sendResponse(sResult.c_str(), sResult.length());

        FDLOG("inout") << "QueryImp::doQuery exception sUid:" << sUid << endl;
    }
    return 0;
}