// client.cpp /** * Copyright (C) 2009 10gen Inc. * * This program is free software: you can redistribute it and/or modify * it under the terms of the GNU Affero General Public License, version 3, * as published by the Free Software Foundation. * * This program is distributed in the hope that it will be useful, * but WITHOUT ANY WARRANTY; without even the implied warranty of * MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the * GNU Affero General Public License for more details. * * You should have received a copy of the GNU Affero General Public License * along with this program. If not, see . * * As a special exception, the copyright holders give permission to link the * code of portions of this program with the OpenSSL library under certain * conditions as described in each individual source file and distribute * linked combinations including the program with the OpenSSL library. You * must comply with the GNU Affero General Public License in all respects for * all of the code used other than as permitted herein. If you modify file(s) * with this exception, you may extend this exception to your version of the * file(s), but you are not obligated to do so. If you do not wish to do so, * delete this exception statement from your version. If you delete this * exception statement from all source files in the program, then also delete * it in the license file. */ /* Client represents a connection to the database (the server-side) and corresponds to an open socket (or logical connection if pooling on sockets) from a client. */ #include "mongo/platform/basic.h" #include "mongo/db/client.h" #include #include #include #include "mongo/base/status.h" #include "mongo/db/lasterror.h" #include "mongo/db/service_context.h" #include "mongo/stdx/thread.h" #include "mongo/util/concurrency/thread_name.h" #include "mongo/util/exit.h" #include "mongo/util/mongoutils/str.h" namespace mongo { TSP_DECLARE(ServiceContext::UniqueClient, currentClient) TSP_DEFINE(ServiceContext::UniqueClient, currentClient) void Client::initThreadIfNotAlready(const char* desc) { if (currentClient.getMake()->get()) return; initThread(desc); } void Client::initThreadIfNotAlready() { initThreadIfNotAlready(getThreadName().c_str()); } void Client::initThread(const char* desc, AbstractMessagingPort* mp) { initThread(desc, getGlobalServiceContext(), mp); } void Client::initThread(const char* desc, ServiceContext* service, AbstractMessagingPort* mp) { invariant(currentClient.getMake()->get() == nullptr); std::string fullDesc; if (mp != NULL) { fullDesc = str::stream() << desc << mp->connectionId(); } else { fullDesc = desc; } setThreadName(fullDesc.c_str()); // Create the client obj, attach to thread *currentClient.get() = service->makeClient(fullDesc, mp); } void Client::destroy() { invariant(currentClient.get()); invariant(currentClient.get()->get()); currentClient.reset(nullptr); } namespace { int64_t generateSeed(const std::string& desc) { size_t seed = 0; boost::hash_combine(seed, Date_t::now().asInt64()); boost::hash_combine(seed, desc); return seed; } } // namespace Client::Client(std::string desc, ServiceContext* serviceContext, AbstractMessagingPort* p) : ClientBasic(serviceContext, p), _desc(std::move(desc)), _threadId(stdx::this_thread::get_id()), _connectionId(p ? p->connectionId() : 0), _prng(generateSeed(_desc)) {} void Client::reportState(BSONObjBuilder& builder) { builder.append("desc", desc()); std::stringstream ss; ss << _threadId; builder.append("threadId", ss.str()); if (_connectionId) { builder.appendNumber("connectionId", _connectionId); } if (hasRemote()) { builder.append("client", getRemote().toString()); } } ServiceContext::UniqueOperationContext Client::makeOperationContext() { return getServiceContext()->makeOperationContext(this); } void Client::setOperationContext(OperationContext* txn) { // We can only set the OperationContext once before resetting it. invariant(txn != NULL && _txn == NULL); _txn = txn; } void Client::resetOperationContext() { invariant(_txn != NULL); _txn = NULL; } std::string Client::clientAddress(bool includePort) const { if (!hasRemote()) { return ""; } if (includePort) { return getRemote().toString(); } return getRemote().host(); } ClientBasic* ClientBasic::getCurrent() { return currentClient.getMake()->get(); } Client& cc() { Client* c = currentClient.getMake()->get(); verify(c); return *c; } bool haveClient() { return currentClient.getMake()->get(); } } // namespace mongo