issue #174

rbkit-lib/subscriber.cpp

@@ -74,6 +74,10 @@ void Subscriber::takeSnapshot()
    commandSocket->sendCommand(triggerSnapshot);
 }
 
+void Subscriber::startCPUSnapshot() {
+    qDebug() <<"Start CPU Snapshot command received";
+}
+
 void Subscriber::startSubscriber()
 {
     qDebug() << "** (start subscriber) Thread id is : " << QThread::currentThreadId();

rbkit-lib/subscriber.h

@@ -83,6 +83,7 @@ public slots:
     void startCPUProfiling();
     void stopCPUProfiling();
     void takeSnapshot();
+    void startCPUSnapshot();
     void startSubscriber();
 
 private:

rbkit-lib/ui/actiontoolbar.cpp

@@ -70,6 +70,7 @@ void ActionToolbar::setupSubscriber()
     connect(this, SIGNAL(takeSnapshot()), subscriber, SLOT(takeSnapshot()));
     connect(this, SIGNAL(startCPUProfiling()), subscriber, SLOT(startCPUProfiling()));
     connect(this, SIGNAL(stopCPUProfiling()), subscriber, SLOT(stopCPUProfiling()));
+    connect(this, SIGNAL(startCPUSnapshot()), subscriber, SLOT(startCPUSnapshot()));
 
     connect(subscriber, &Subscriber::errored, centralWidget, &CentralWidget::onError);
     connect(subscriber, &Subscriber::connected, this, &ActionToolbar::connectedToSocket);
@@ -156,8 +157,15 @@ void ActionToolbar::setupToolBar()
                              "stop_cpu_profiling",
                              QIcon(":/icons/Compare-32.png"),
                              "cpu_profiling");
+
     connect(stopCPUButton, &QToolButton::clicked, this, &ActionToolbar::performStopCPUAction);
 
+    cpusnapshotButton = toolBar->addRibbonAction("Take Snapshot",
+                                              "start_snapshot",
+                                              QIcon(":/icons/snapshot-32.png"),
+                                                 "cpu_profiling");
+    connect(cpusnapshotButton, &QToolButton::clicked, this, &ActionToolbar::performCPUSnapshot);
+
     toolBar->loadStyleSheet(":/icons/style.css");
 }
 
@@ -174,6 +182,10 @@ void ActionToolbar::performStopCPUAction() {
     emit stopCPUProfiling();
 }
 
+void ActionToolbar::performCPUSnapshot() {
+    emit startCPUSnapshot();
+}
+
 void ActionToolbar::takeSnapshotAction()
 {
     if (centralWidget->attemptMemorySnapshot()) {

rbkit-lib/ui/actiontoolbar.h

@@ -53,6 +53,7 @@ class ActionToolbar : public QObject
     QToolButton *stopCPUButton;
     QToolButton *connectButton;
     QToolButton *snapshotButton;
+    QToolButton *cpusnapshotButton;
     QToolButton *compareSnapshotButton;
 signals:
     void connectToSocket(QString commandSocket, QString eventSocket);
@@ -61,11 +62,13 @@ class ActionToolbar : public QObject
     void stopCPUProfiling();
     void disconnectSubscriber();
     void takeSnapshot();
+    void startCPUSnapshot();
 public slots:
     void performGCAction();
     void performStartCPUAction();
     void performStopCPUAction();
     void takeSnapshotAction();
+    void performCPUSnapshot();
     void attemptConnection();
     void compareSnapshots();
     void useSelectedHost(QString commandSocket, QString eventSocket);