From be2463b95899d258c5ee0f911a8feef0693a4049 Mon Sep 17 00:00:00 2001
From: Shivam Sharma <shivam691999@gmail.com>
Date: Mon, 31 Jan 2022 13:59:03 +0000
Subject: [PATCH] read/write IOPS added

---
 internal/cli/bor_fingerprint.go | 67 +++++++++++++++++++++++++++++++++
 1 file changed, 67 insertions(+)

diff --git a/internal/cli/bor_fingerprint.go b/internal/cli/bor_fingerprint.go
index bb7e3de7f0..8e2fd8b0dd 100644
--- a/internal/cli/bor_fingerprint.go
+++ b/internal/cli/bor_fingerprint.go
@@ -3,6 +3,7 @@ package cli
 import (
 	"fmt"
 	"math"
+	"os/exec"
 
 	"github.com/ethereum/go-ethereum/params"
 	"github.com/mitchellh/cli"
@@ -72,6 +73,63 @@ func convertBytesToGB(bytesValue uint64) float64 {
 	return math.Floor(float64(bytesValue)/(1024*1024*1024)*100) / 100
 }
 
+// Install fio on the node if it does not exist
+func (c *FingerprintCommand) installFio() error {
+
+	cmd := exec.Command("/bin/sh", "-c", "fio -v")
+
+	_, err := cmd.CombinedOutput()
+	if err != nil {
+		c.UI.Output("fio not installed\nInstalling Fio...")
+
+		cmd := exec.Command("/bin/sh", "-c", "sudo apt-get update && sudo apt-get install fio -y")
+		_, err := cmd.CombinedOutput()
+		if err == nil {
+			return nil
+		}
+
+		cmd = exec.Command("/bin/sh", "-c", "sudo yum install fio -y")
+		_, err = cmd.CombinedOutput()
+		if err == nil {
+			return nil
+		}
+
+		cmd = exec.Command("/bin/sh", "-c", "sudo dnf install fio -y")
+		_, err = cmd.CombinedOutput()
+		if err == nil {
+			return nil
+		}
+
+		c.UI.Output("Unable to install fio")
+		return err
+	}
+
+	return nil
+}
+
+// Run the IOPS benchmark for the node
+func (c *FingerprintCommand) benchmark() error {
+	var b []byte
+	err := c.installFio()
+	if err != nil {
+		return err
+	}
+
+	c.UI.Output("\nRunning a 10 second test...\n")
+
+	cmd := exec.Command("/bin/sh", "-c", "sudo fio --filename=/file --size=2GB --direct=1 --rw=randrw --bs=64k --ioengine=libaio --iodepth=64 --runtime=10 --numjobs=4 --time_based --group_reporting --name=throughput-test-job --eta-newline=1 | grep -e 'read:' -e 'write:' | awk '{print $1,$2}' ")
+
+	b, err = cmd.CombinedOutput()
+	if err != nil {
+		return err
+	}
+
+	out := string(b)
+	c.UI.Output(out)
+
+	return nil
+}
+
 // Run implements the cli.Command interface
 func (c *FingerprintCommand) Run(args []string) int {
 
@@ -120,5 +178,14 @@ func (c *FingerprintCommand) Run(args []string) int {
 	}
 
 	c.UI.Output(formatFingerprint(borFingerprint))
+
+	if borFingerprint.OsName == "linux" {
+		err = c.benchmark()
+		if err != nil {
+			c.UI.Error(err.Error())
+			return 1
+		}
+	}
+
 	return 0
 }