mirror of
https://gitee.com/TarsCloud/TarsCpp.git
synced 2024-12-22 22:16:38 +08:00
133 lines
2.9 KiB
C++
133 lines
2.9 KiB
C++
/**
|
|
* 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 "servant/AsyncProcThread.h"
|
|
#include "servant/Communicator.h"
|
|
#include "servant/StatReport.h"
|
|
#include "servant/TarsLogger.h"
|
|
|
|
namespace tars
|
|
{
|
|
|
|
AsyncProcThread::AsyncProcThread(size_t iQueueCap, bool merge)
|
|
: _terminate(false), _iQueueCap(iQueueCap), _merge(merge)
|
|
{
|
|
_msgQueue = new TC_CasQueue<ReqMessage*>();
|
|
|
|
if(!_merge)
|
|
{
|
|
start();
|
|
}
|
|
}
|
|
|
|
AsyncProcThread::~AsyncProcThread()
|
|
{
|
|
terminate();
|
|
|
|
if(_msgQueue)
|
|
{
|
|
delete _msgQueue;
|
|
_msgQueue = NULL;
|
|
}
|
|
}
|
|
|
|
void AsyncProcThread::terminate()
|
|
{
|
|
if(!_merge) {
|
|
TC_ThreadLock::Lock lock(*this);
|
|
|
|
_terminate = true;
|
|
|
|
notifyAll();
|
|
}
|
|
}
|
|
|
|
void AsyncProcThread::push_back(ReqMessage * msg)
|
|
{
|
|
if(_merge) {
|
|
//合并了, 直接回调
|
|
callback(msg);
|
|
}
|
|
else {
|
|
if(_msgQueue->size() >= _iQueueCap)
|
|
{
|
|
TLOGERROR("[TARS][AsyncProcThread::push_back] async_queue full:" << _msgQueue->size() << ">=" << _iQueueCap << endl);
|
|
delete msg;
|
|
}
|
|
else
|
|
{
|
|
_msgQueue->push_back(msg);
|
|
|
|
TC_ThreadLock::Lock lock(*this);
|
|
notify();
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
void AsyncProcThread::run()
|
|
{
|
|
while (!_terminate)
|
|
{
|
|
ReqMessage * msg;
|
|
|
|
if (_msgQueue->pop_front(msg))
|
|
{
|
|
callback(msg);
|
|
}
|
|
else
|
|
{
|
|
TC_ThreadLock::Lock lock(*this);
|
|
timedWait(1000);
|
|
}
|
|
}
|
|
}
|
|
|
|
|
|
void AsyncProcThread::callback(ReqMessage * msg)
|
|
{
|
|
TLOGTARS("[TARS][AsyncProcThread::run] get one msg." << endl);
|
|
|
|
//从回调对象把线程私有数据传递到回调线程中
|
|
ServantProxyThreadData * pServantProxyThreadData = ServantProxyThreadData::getData();
|
|
assert(pServantProxyThreadData != NULL);
|
|
|
|
//把染色的消息设置在线程私有数据里面
|
|
pServantProxyThreadData->_dyeing = msg->bDyeing;
|
|
pServantProxyThreadData->_dyeingKey = msg->sDyeingKey;
|
|
|
|
if(msg->adapter)
|
|
{
|
|
pServantProxyThreadData->_szHost = msg->adapter->endpoint().desc();
|
|
}
|
|
|
|
try
|
|
{
|
|
ReqMessagePtr msgPtr = msg;
|
|
msg->callback->onDispatch(msgPtr);
|
|
}
|
|
catch (exception& e)
|
|
{
|
|
TLOGERROR("[TARS][AsyncProcThread exception]:" << e.what() << endl);
|
|
}
|
|
catch (...)
|
|
{
|
|
TLOGERROR("[TARS][AsyncProcThread exception.]" << endl);
|
|
}
|
|
}
|
|
/////////////////////////////////////////////////////////////////////////
|
|
}
|