15ffd83dbSDimitry Andric //===-- CommandObjectStats.cpp --------------------------------------------===//
2*0b57cec5SDimitry Andric //
3*0b57cec5SDimitry Andric // Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions.
4*0b57cec5SDimitry Andric // See https://llvm.org/LICENSE.txt for license information.
5*0b57cec5SDimitry Andric // SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception
6*0b57cec5SDimitry Andric //
7*0b57cec5SDimitry Andric //===----------------------------------------------------------------------===//
8*0b57cec5SDimitry Andric 
9*0b57cec5SDimitry Andric #include "CommandObjectStats.h"
10*0b57cec5SDimitry Andric #include "lldb/Interpreter/CommandReturnObject.h"
11*0b57cec5SDimitry Andric #include "lldb/Target/Target.h"
12*0b57cec5SDimitry Andric 
13*0b57cec5SDimitry Andric using namespace lldb;
14*0b57cec5SDimitry Andric using namespace lldb_private;
15*0b57cec5SDimitry Andric 
16*0b57cec5SDimitry Andric class CommandObjectStatsEnable : public CommandObjectParsed {
17*0b57cec5SDimitry Andric public:
CommandObjectStatsEnable(CommandInterpreter & interpreter)18*0b57cec5SDimitry Andric   CommandObjectStatsEnable(CommandInterpreter &interpreter)
19*0b57cec5SDimitry Andric       : CommandObjectParsed(interpreter, "enable",
20*0b57cec5SDimitry Andric                             "Enable statistics collection", nullptr,
21*0b57cec5SDimitry Andric                             eCommandProcessMustBePaused) {}
22*0b57cec5SDimitry Andric 
23*0b57cec5SDimitry Andric   ~CommandObjectStatsEnable() override = default;
24*0b57cec5SDimitry Andric 
25*0b57cec5SDimitry Andric protected:
DoExecute(Args & command,CommandReturnObject & result)26*0b57cec5SDimitry Andric   bool DoExecute(Args &command, CommandReturnObject &result) override {
279dba64beSDimitry Andric     Target &target = GetSelectedOrDummyTarget();
28*0b57cec5SDimitry Andric 
299dba64beSDimitry Andric     if (target.GetCollectingStats()) {
30*0b57cec5SDimitry Andric       result.AppendError("statistics already enabled");
31*0b57cec5SDimitry Andric       return false;
32*0b57cec5SDimitry Andric     }
33*0b57cec5SDimitry Andric 
349dba64beSDimitry Andric     target.SetCollectingStats(true);
35*0b57cec5SDimitry Andric     result.SetStatus(eReturnStatusSuccessFinishResult);
36*0b57cec5SDimitry Andric     return true;
37*0b57cec5SDimitry Andric   }
38*0b57cec5SDimitry Andric };
39*0b57cec5SDimitry Andric 
40*0b57cec5SDimitry Andric class CommandObjectStatsDisable : public CommandObjectParsed {
41*0b57cec5SDimitry Andric public:
CommandObjectStatsDisable(CommandInterpreter & interpreter)42*0b57cec5SDimitry Andric   CommandObjectStatsDisable(CommandInterpreter &interpreter)
43*0b57cec5SDimitry Andric       : CommandObjectParsed(interpreter, "disable",
44*0b57cec5SDimitry Andric                             "Disable statistics collection", nullptr,
45*0b57cec5SDimitry Andric                             eCommandProcessMustBePaused) {}
46*0b57cec5SDimitry Andric 
47*0b57cec5SDimitry Andric   ~CommandObjectStatsDisable() override = default;
48*0b57cec5SDimitry Andric 
49*0b57cec5SDimitry Andric protected:
DoExecute(Args & command,CommandReturnObject & result)50*0b57cec5SDimitry Andric   bool DoExecute(Args &command, CommandReturnObject &result) override {
519dba64beSDimitry Andric     Target &target = GetSelectedOrDummyTarget();
52*0b57cec5SDimitry Andric 
539dba64beSDimitry Andric     if (!target.GetCollectingStats()) {
54*0b57cec5SDimitry Andric       result.AppendError("need to enable statistics before disabling them");
55*0b57cec5SDimitry Andric       return false;
56*0b57cec5SDimitry Andric     }
57*0b57cec5SDimitry Andric 
589dba64beSDimitry Andric     target.SetCollectingStats(false);
59*0b57cec5SDimitry Andric     result.SetStatus(eReturnStatusSuccessFinishResult);
60*0b57cec5SDimitry Andric     return true;
61*0b57cec5SDimitry Andric   }
62*0b57cec5SDimitry Andric };
63*0b57cec5SDimitry Andric 
64*0b57cec5SDimitry Andric class CommandObjectStatsDump : public CommandObjectParsed {
65*0b57cec5SDimitry Andric public:
CommandObjectStatsDump(CommandInterpreter & interpreter)66*0b57cec5SDimitry Andric   CommandObjectStatsDump(CommandInterpreter &interpreter)
67*0b57cec5SDimitry Andric       : CommandObjectParsed(interpreter, "dump", "Dump statistics results",
68*0b57cec5SDimitry Andric                             nullptr, eCommandProcessMustBePaused) {}
69*0b57cec5SDimitry Andric 
70*0b57cec5SDimitry Andric   ~CommandObjectStatsDump() override = default;
71*0b57cec5SDimitry Andric 
72*0b57cec5SDimitry Andric protected:
DoExecute(Args & command,CommandReturnObject & result)73*0b57cec5SDimitry Andric   bool DoExecute(Args &command, CommandReturnObject &result) override {
749dba64beSDimitry Andric     Target &target = GetSelectedOrDummyTarget();
75*0b57cec5SDimitry Andric 
76*0b57cec5SDimitry Andric     uint32_t i = 0;
779dba64beSDimitry Andric     for (auto &stat : target.GetStatistics()) {
78*0b57cec5SDimitry Andric       result.AppendMessageWithFormat(
79*0b57cec5SDimitry Andric           "%s : %u\n",
80480093f4SDimitry Andric           lldb_private::GetStatDescription(
81480093f4SDimitry Andric               static_cast<lldb_private::StatisticKind>(i))
82*0b57cec5SDimitry Andric               .c_str(),
83*0b57cec5SDimitry Andric           stat);
84*0b57cec5SDimitry Andric       i += 1;
85*0b57cec5SDimitry Andric     }
86*0b57cec5SDimitry Andric     result.SetStatus(eReturnStatusSuccessFinishResult);
87*0b57cec5SDimitry Andric     return true;
88*0b57cec5SDimitry Andric   }
89*0b57cec5SDimitry Andric };
90*0b57cec5SDimitry Andric 
CommandObjectStats(CommandInterpreter & interpreter)91*0b57cec5SDimitry Andric CommandObjectStats::CommandObjectStats(CommandInterpreter &interpreter)
92*0b57cec5SDimitry Andric     : CommandObjectMultiword(interpreter, "statistics",
93*0b57cec5SDimitry Andric                              "Print statistics about a debugging session",
94*0b57cec5SDimitry Andric                              "statistics <subcommand> [<subcommand-options>]") {
95*0b57cec5SDimitry Andric   LoadSubCommand("enable",
96*0b57cec5SDimitry Andric                  CommandObjectSP(new CommandObjectStatsEnable(interpreter)));
97*0b57cec5SDimitry Andric   LoadSubCommand("disable",
98*0b57cec5SDimitry Andric                  CommandObjectSP(new CommandObjectStatsDisable(interpreter)));
99*0b57cec5SDimitry Andric   LoadSubCommand("dump",
100*0b57cec5SDimitry Andric                  CommandObjectSP(new CommandObjectStatsDump(interpreter)));
101*0b57cec5SDimitry Andric }
102*0b57cec5SDimitry Andric 
103*0b57cec5SDimitry Andric CommandObjectStats::~CommandObjectStats() = default;
104