summaryrefslogtreecommitdiff
path: root/src/mongo/db/commands/cpuprofile.cpp
diff options
context:
space:
mode:
Diffstat (limited to 'src/mongo/db/commands/cpuprofile.cpp')
-rw-r--r--src/mongo/db/commands/cpuprofile.cpp205
1 files changed, 106 insertions, 99 deletions
diff --git a/src/mongo/db/commands/cpuprofile.cpp b/src/mongo/db/commands/cpuprofile.cpp
index b50dd0233a5..4273666a70f 100644
--- a/src/mongo/db/commands/cpuprofile.cpp
+++ b/src/mongo/db/commands/cpuprofile.cpp
@@ -42,9 +42,9 @@
* The commands defined here, and profiling, are only available when enabled at
* build-time with the "--use-cpu-profiler" argument to scons.
*
- * Example SCons command line:
+ * Example SCons command line:
*
- * scons --release --use-cpu-profiler
+ * scons --release --use-cpu-profiler
*/
#include "gperftools/profiler.h"
@@ -63,102 +63,109 @@
namespace mongo {
- namespace {
-
- /**
- * Common code for the implementation of cpu profiler commands.
- */
- class CpuProfilerCommand : public Command {
- public:
- CpuProfilerCommand( char const *name ) : Command( name ) {}
- virtual bool slaveOk() const { return true; }
- virtual bool adminOnly() const { return true; }
- virtual bool localHostOnlyIfNoAuth( const BSONObj& cmdObj ) { return true; }
- virtual void addRequiredPrivileges(const std::string& dbname,
- const BSONObj& cmdObj,
- std::vector<Privilege>* out) {
- ActionSet actions;
- actions.addAction(ActionType::cpuProfiler);
- out->push_back(Privilege(ResourcePattern::forClusterResource(), actions));
- }
-
- // This is an abuse of the global dbmutex. We only really need to
- // ensure that only one cpuprofiler command runs at once; it would
- // be fine for it to run concurrently with other operations.
- virtual bool isWriteCommandForConfigServer() const { return true; }
- };
-
- /**
- * Class providing implementation of the _cpuProfilerStart command.
- */
- class CpuProfilerStartCommand : public CpuProfilerCommand {
- public:
- CpuProfilerStartCommand() : CpuProfilerCommand( commandName ) {}
-
- virtual bool run( OperationContext* txn,
- std::string const &db,
- BSONObj &cmdObj,
- int options,
- std::string &errmsg,
- BSONObjBuilder &result);
-
- static char const *const commandName;
- } cpuProfilerStartCommandInstance;
-
- /**
- * Class providing implementation of the _cpuProfilerStop command.
- */
- class CpuProfilerStopCommand : public CpuProfilerCommand {
- public:
- CpuProfilerStopCommand() : CpuProfilerCommand( commandName ) {}
-
- virtual bool run( OperationContext* txn,
- std::string const &db,
- BSONObj &cmdObj,
- int options,
- std::string &errmsg,
- BSONObjBuilder &result);
-
- static char const *const commandName;
- } cpuProfilerStopCommandInstance;
-
- char const *const CpuProfilerStartCommand::commandName = "_cpuProfilerStart";
- char const *const CpuProfilerStopCommand::commandName = "_cpuProfilerStop";
-
- bool CpuProfilerStartCommand::run( OperationContext* txn,
- std::string const &db,
- BSONObj &cmdObj,
- int options,
- std::string &errmsg,
- BSONObjBuilder &result) {
- ScopedTransaction transaction(txn, MODE_IX);
- Lock::DBLock dbXLock(txn->lockState(), db, MODE_X);
- // The lock here is just to prevent concurrency, nothing will write.
- OldClientContext ctx(txn, db);
-
- std::string profileFilename = cmdObj[commandName]["profileFilename"].String();
- if ( ! ::ProfilerStart( profileFilename.c_str() ) ) {
- errmsg = "Failed to start profiler";
- return false;
- }
- return true;
- }
-
- bool CpuProfilerStopCommand::run( OperationContext* txn,
- std::string const &db,
- BSONObj &cmdObj,
- int options,
- std::string &errmsg,
- BSONObjBuilder &result) {
- ScopedTransaction transaction(txn, MODE_IX);
- Lock::DBLock dbXLock(txn->lockState(), db, MODE_X);
- OldClientContext ctx(txn, db);
-
- ::ProfilerStop();
- return true;
- }
-
- } // namespace
+namespace {
-} // namespace mongo
+/**
+ * Common code for the implementation of cpu profiler commands.
+ */
+class CpuProfilerCommand : public Command {
+public:
+ CpuProfilerCommand(char const* name) : Command(name) {}
+ virtual bool slaveOk() const {
+ return true;
+ }
+ virtual bool adminOnly() const {
+ return true;
+ }
+ virtual bool localHostOnlyIfNoAuth(const BSONObj& cmdObj) {
+ return true;
+ }
+ virtual void addRequiredPrivileges(const std::string& dbname,
+ const BSONObj& cmdObj,
+ std::vector<Privilege>* out) {
+ ActionSet actions;
+ actions.addAction(ActionType::cpuProfiler);
+ out->push_back(Privilege(ResourcePattern::forClusterResource(), actions));
+ }
+
+ // This is an abuse of the global dbmutex. We only really need to
+ // ensure that only one cpuprofiler command runs at once; it would
+ // be fine for it to run concurrently with other operations.
+ virtual bool isWriteCommandForConfigServer() const {
+ return true;
+ }
+};
+
+/**
+ * Class providing implementation of the _cpuProfilerStart command.
+ */
+class CpuProfilerStartCommand : public CpuProfilerCommand {
+public:
+ CpuProfilerStartCommand() : CpuProfilerCommand(commandName) {}
+ virtual bool run(OperationContext* txn,
+ std::string const& db,
+ BSONObj& cmdObj,
+ int options,
+ std::string& errmsg,
+ BSONObjBuilder& result);
+
+ static char const* const commandName;
+} cpuProfilerStartCommandInstance;
+
+/**
+ * Class providing implementation of the _cpuProfilerStop command.
+ */
+class CpuProfilerStopCommand : public CpuProfilerCommand {
+public:
+ CpuProfilerStopCommand() : CpuProfilerCommand(commandName) {}
+
+ virtual bool run(OperationContext* txn,
+ std::string const& db,
+ BSONObj& cmdObj,
+ int options,
+ std::string& errmsg,
+ BSONObjBuilder& result);
+
+ static char const* const commandName;
+} cpuProfilerStopCommandInstance;
+
+char const* const CpuProfilerStartCommand::commandName = "_cpuProfilerStart";
+char const* const CpuProfilerStopCommand::commandName = "_cpuProfilerStop";
+
+bool CpuProfilerStartCommand::run(OperationContext* txn,
+ std::string const& db,
+ BSONObj& cmdObj,
+ int options,
+ std::string& errmsg,
+ BSONObjBuilder& result) {
+ ScopedTransaction transaction(txn, MODE_IX);
+ Lock::DBLock dbXLock(txn->lockState(), db, MODE_X);
+ // The lock here is just to prevent concurrency, nothing will write.
+ OldClientContext ctx(txn, db);
+
+ std::string profileFilename = cmdObj[commandName]["profileFilename"].String();
+ if (!::ProfilerStart(profileFilename.c_str())) {
+ errmsg = "Failed to start profiler";
+ return false;
+ }
+ return true;
+}
+
+bool CpuProfilerStopCommand::run(OperationContext* txn,
+ std::string const& db,
+ BSONObj& cmdObj,
+ int options,
+ std::string& errmsg,
+ BSONObjBuilder& result) {
+ ScopedTransaction transaction(txn, MODE_IX);
+ Lock::DBLock dbXLock(txn->lockState(), db, MODE_X);
+ OldClientContext ctx(txn, db);
+
+ ::ProfilerStop();
+ return true;
+}
+
+} // namespace
+
+} // namespace mongo
View Lorry Controller Status