Allow running multiple custom commands

bisohns · Nov 10, 2022 · 9aac418 · 9aac418
1 parent 6ee59c9
commit 9aac418
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 3 deletions.
diff --git a/config.example.yaml b/config.example.yaml
@@ -41,5 +41,6 @@ metrics:
  memory:
  tcp:
  # custom command to show uptime
- custom: "cat /proc/uptime"
+ custom-uptime: "cat /proc/uptime"
+ custom-dir: 'dir C:\'
 poll-interval: 30
diff --git a/inspector/inspector.go b/inspector/inspector.go
@@ -3,6 +3,7 @@ package inspector
 import (
  "errors"
  "fmt"
+ "strings"
 
  "github.com/bisohns/saido/driver"
 )
@@ -32,8 +33,8 @@ var inspectorMap = map[string]NewInspector{
 
 // Valid : checks if inspector is a valid inspector
 func Valid(name string) bool {
- for key, _ := range inspectorMap {
- if name == key {
+ for key := range inspectorMap {
+ if name == key || strings.HasPrefix(name, "custom") {
  return true
  }
  }
@@ -42,6 +43,9 @@ func Valid(name string) bool {
 
 // Init : initializes the specified inspector using name and driver
 func Init(name string, driver *driver.Driver, custom ...string) (Inspector, error) {
+ if strings.HasPrefix(name, "custom") {
+ name = "custom"
+ }
  val, ok := inspectorMap[name]
  if ok {
  inspector, err := val(driver, custom...)