From 0786db016d81be88c18eb60fb9d3ba6875e94829 Mon Sep 17 00:00:00 2001
From: Carlos Salguero <carlos.salguero@percona.com>
Date: Wed, 13 Dec 2017 13:51:08 -0300
Subject: [PATCH] PT-218 Added RocksDB status section to pt-stalk

---
 bin/pt-stalk        | 16 ++++++++++++++++
 lib/bash/collect.sh | 16 ++++++++++++++++
 2 files changed, 32 insertions(+)

diff --git a/bin/pt-stalk b/bin/pt-stalk
index 099177b0..f91a8a24 100755
--- a/bin/pt-stalk
+++ b/bin/pt-stalk
@@ -840,6 +840,8 @@ collect() {
    fi
    innodb_status 1
    tokudb_status 1
+   rocksdb_status 1
+
    $CMD_MYSQL $EXT_ARGV -e "$mutex" >> "$d/$p-mutex-status1" &
    open_tables                      >> "$d/$p-opentables1"   &
 
@@ -1006,6 +1008,8 @@ collect() {
 
    innodb_status 2
    tokudb_status 2
+   rocksdb_status 2
+
    $CMD_MYSQL $EXT_ARGV -e "$mutex" >> "$d/$p-mutex-status2" &
    open_tables                      >> "$d/$p-opentables2"   &
 
@@ -1101,6 +1105,18 @@ innodb_status() {
    }
 }
 
+rocksdb_status() {
+    local n=$1
+
+    has_rocksdb=`$CMD_MYSQL $EXT_ARGV -e "SHOW ENGINES" | grep -i 'rocksdb'`
+    exit_code=$?
+
+    if [ $exit_code -eq 0 ]; then
+        $CMD_MYSQL $EXT_ARGV -e "SHOW ENGINE ROCKSDB STATUS\G" \
+                   >> "$d/$p-rocksdbstatus$n" || rm -f "$d/$p-rocksdbstatus$n"
+    fi
+}
+
 ps_locks_transactions() {
    local outfile=$1 
    
diff --git a/lib/bash/collect.sh b/lib/bash/collect.sh
index c3f8f221..ba0ca6a3 100644
--- a/lib/bash/collect.sh
+++ b/lib/bash/collect.sh
@@ -110,6 +110,8 @@ collect() {
    fi
    innodb_status 1
    tokudb_status 1
+   rocksdb_status 1
+
    $CMD_MYSQL $EXT_ARGV -e "$mutex" >> "$d/$p-mutex-status1" &
    open_tables                      >> "$d/$p-opentables1"   &
 
@@ -301,6 +303,8 @@ collect() {
 
    innodb_status 2
    tokudb_status 2
+   rocksdb_status 2
+
    $CMD_MYSQL $EXT_ARGV -e "$mutex" >> "$d/$p-mutex-status2" &
    open_tables                      >> "$d/$p-opentables2"   &
 
@@ -405,6 +409,18 @@ innodb_status() {
    }
 }
 
+rocksdb_status() {
+    local n=$1
+
+    has_rocksdb=`$CMD_MYSQL $EXT_ARGV -e "SHOW ENGINES" | grep -i 'rocksdb'`
+    exit_code=$?
+
+    if [ $exit_code -eq 0 ]; then
+        $CMD_MYSQL $EXT_ARGV -e "SHOW ENGINE ROCKSDB STATUS\G" \
+                   >> "$d/$p-rocksdbstatus$n" || rm -f "$d/$p-rocksdbstatus$n"
+    fi
+}
+
 ps_locks_transactions() {
    local outfile=$1