123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174 |
- /**
- * 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/Servant.h"
- #include "servant/BaseF.h"
- #include "servant/Application.h"
- #include "servant/AppCache.h"
- #include "servant/RemoteLogger.h"
- #include <cerrno>
- namespace tars
- {
- thread_local shared_ptr<CallbackThreadData> CallbackThreadData::g_sp;
- Servant::Servant():_handle(NULL)
- {
- }
- Servant::~Servant()
- {
- }
- void Servant::setName(const string &name)
- {
- _name = name;
- }
- string Servant::getName() const
- {
- return _name;
- }
- void Servant::setApplication(Application *application)
- {
- _application = application;
- setNotifyObserver(application->getNotifyObserver());
- }
- Application* Servant::getApplication() const
- {
- return _application;
- }
- void Servant::setHandle(TC_EpollServer::Handle* handle)
- {
- _handle = handle;
- }
- TC_EpollServer::Handle* Servant::getHandle()
- {
- return _handle;
- }
- int Servant::dispatch(CurrentPtr current, vector<char> &buffer)
- {
- int ret = TARSSERVERUNKNOWNERR;
- if (current->getFuncName() == "tars_ping")
- {
- TLOGTARS("[Servant::dispatch] tars_ping ok from [" << current->getIp() << ":" << current->getPort() << "]" << endl);
- ret = TARSSERVERSUCCESS;
- }
- else if (!current->getBindAdapter()->isTarsProtocol())
- {
- TC_LockT<TC_ThreadRecMutex> lock(*this);
- ret = doRequest(current, buffer);
- }
- else
- {
- TC_LockT<TC_ThreadRecMutex> lock(*this);
- ret = onDispatch(current, buffer);
- }
- return ret;
- }
- TC_CasQueue<ReqMessagePtr>& Servant::getResponseQueue()
- {
- return _asyncResponseQueue;
- }
- ///////////////////////////////////////////////////////////////////////////
- ServantCallback::ServantCallback(const string& type, const ServantPtr& servant, const CurrentPtr& current)
- : _servant(servant)
- , _current(current)
- {
- ServantProxyCallback::setType(type);
- }
- int ServantCallback::onDispatch(ReqMessagePtr msg)
- {
- _servant->getResponseQueue().push_back(msg);
- _servant->getHandle()->notifyFilter();
- return 0;
- }
- const ServantPtr& ServantCallback::getServant()
- {
- return _servant;
- }
- const CurrentPtr& ServantCallback::getCurrent()
- {
- return _current;
- }
- ///////////////////////////////////////////////////////////////////////////
- CallbackThreadData::CallbackThreadData():_contextValid(false)
- {
- }
- // void CallbackThreadData::destructor(void* p)
- // {
- // CallbackThreadData * pCbtd = (CallbackThreadData*)p;
- // if(pCbtd)
- // delete pCbtd;
- // }
- CallbackThreadData * CallbackThreadData::getData()
- {
- if(!g_sp)
- {
- g_sp.reset(new CallbackThreadData());
- }
- return g_sp.get();
- }
- void CallbackThreadData::setResponseContext(const map<std::string, std::string> & context)
- {
- _contextValid = true;
- if(context.empty())
- {
- _responseContext.clear();
- }
- else
- {
- _responseContext = context;
- }
- }
- map<std::string, std::string> & CallbackThreadData::getResponseContext()
- {
- return _responseContext;
- }
- void CallbackThreadData::delResponseContext()
- {
- _contextValid = false;
- _responseContext.clear();
- }
- ////////////////////////////////////////////////////////////////////////
- }
|