2017-06-19 16:41:42 +08:00
/**
2017-01-18 16:19:06 +08:00
* 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 \n decode 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 ( ) ) + " \n endline \n " ;
current - > sendResponse ( sResult . c_str ( ) , sResult . length ( ) ) ;
FDLOG ( " inout " ) < < " QueryImp::doQuery exception sUid: " < < sUid < < endl ;
}
return 0 ;
}
