git init

Author: Philipp Dorschner

.gitignore

@@ -0,0 +1 @@
+.idea

LICENSE

@@ -0,0 +1,34 @@
+package cmd
+
+import (
+	"check_elasticsearch/internal/client"
+	"net/url"
+	"strconv"
+)
+
+type Config struct {
+	Hostname string
+	Port     int
+	TLS      bool
+	Username string
+	Password string
+	Insecure bool
+}
+
+var cliConfig Config
+
+func (c *Config) Client() *client.Client {
+	u := url.URL{
+		Scheme: "http",
+		Host:   c.Hostname + ":" + strconv.Itoa(c.Port),
+	}
+
+	if c.TLS {
+		u.Scheme = "https"
+	}
+
+	cl := client.NewClient(u.String(), c.Username, c.Password)
+	cl.Insecure = c.Insecure
+
+	return cl
+}

cmd/config.go

@@ -0,0 +1,61 @@
+package cmd
+
+import (
+	"github.com/NETWAYS/go-check"
+	"github.com/NETWAYS/go-check/perfdata"
+	"github.com/spf13/cobra"
+)
+
+// healthCmd represents the health command
+var healthCmd = &cobra.Command{
+	Use:   "health",
+	Short: "A brief description of your command",
+	Long: `A longer description that spans multiple lines and likely contains examples
+and usage of using your command. For example:
+
+Cobra is a CLI library for Go that empowers applications.
+This application is a tool to generate the needed files
+to quickly create a Cobra application.`,
+	Run: func(cmd *cobra.Command, args []string) {
+		client := cliConfig.Client()
+		err := client.Connect()
+		if err != nil {
+			check.ExitError(err)
+		}
+
+		health, err := client.Health()
+		if err != nil {
+			check.ExitError(err)
+		}
+
+		rc := 3
+		output := "Cluster " + health.ClusterName + " is " + health.Status
+
+		switch health.Status {
+		case "green":
+			rc = 0
+		case "yellow":
+			rc = 1
+		default:
+			rc = 2
+		}
+
+		// green = 0
+		// yellow = 1
+		// red = 2
+		p := perfdata.PerfdataList{
+			{Label: "status", Value: rc},
+			{Label: "nodes", Value: health.NumberOfNodes},
+			{Label: "data_nodes", Value: health.NumberOfDataNodes},
+			{Label: "active_primary_shards", Value: health.ActivePrimaryShards},
+			{Label: "active_shards", Value: health.ActiveShards},
+		}
+
+		check.ExitRaw(rc, output, "|", p.String())
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(healthCmd)
+	healthCmd.DisableFlagsInUseLine = true
+}

cmd/health.go

@@ -0,0 +1,85 @@
+package cmd
+
+import (
+	"fmt"
+	"github.com/NETWAYS/go-check"
+	"github.com/NETWAYS/go-check/perfdata"
+	"github.com/spf13/cobra"
+)
+
+type QueryConfig struct {
+	Index      string
+	Query      string
+	MessageKey string
+	MessageLen int
+	Critical   uint
+	Warning    uint
+}
+
+var cliQueryConfig QueryConfig
+
+// queryCmd represents the query command
+var queryCmd = &cobra.Command{
+	Use:   "query",
+	Short: "A brief description of your command",
+	Long: `A longer description that spans multiple lines and likely contains examples
+and usage of using your command. For example:
+
+Cobra is a CLI library for Go that empowers applications.
+This application is a tool to generate the needed files
+to quickly create a Cobra application.`,
+	Run: func(cmd *cobra.Command, args []string) {
+		client := cliConfig.Client()
+		err := client.Connect()
+		if err != nil {
+			check.ExitError(err)
+		}
+
+		total, messages, err := client.SearchMessages(
+			cliQueryConfig.Index,
+			cliQueryConfig.Query,
+			cliQueryConfig.MessageKey)
+
+		rc := 3
+		output := fmt.Sprintf("Total hits: %d", total)
+
+		if len(messages) > 0 {
+			output += "\n"
+			for _, msg := range messages {
+				if len(msg) > cliQueryConfig.MessageLen {
+					msg = msg[0:cliQueryConfig.MessageLen]
+				}
+				output += msg + "\n"
+			}
+		}
+
+		if total >= cliQueryConfig.Critical {
+			rc = check.Critical
+		} else if total >= cliQueryConfig.Warning {
+			rc = check.Warning
+		} else if total == 0 {
+			rc = check.OK
+		}
+
+		p := perfdata.PerfdataList{
+			{Label: "total", Value: total,
+				Warn: &check.Threshold{Upper: float64(cliQueryConfig.Warning)},
+				Crit: &check.Threshold{Upper: float64(cliQueryConfig.Critical)},
+			},
+		}
+
+		check.ExitRaw(rc, output, "|", p.String())
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(queryCmd)
+
+	fs := queryCmd.Flags()
+	fs.StringVarP(&cliQueryConfig.Query, "query", "q", "", "")
+	fs.StringVarP(&cliQueryConfig.Index, "index", "I", "", "")
+	fs.StringVarP(&cliQueryConfig.MessageKey, "msgkey", "k", "", "")
+	fs.IntVarP(&cliQueryConfig.MessageLen, "msglen", "m", 80, "")
+	fs.UintVarP(&cliQueryConfig.Warning, "warning", "w", 20, "")
+	fs.UintVarP(&cliQueryConfig.Critical, "critical", "w", 50, "")
+}

cmd/query.go

@@ -0,0 +1,58 @@
+package cmd
+
+import (
+	"github.com/NETWAYS/go-check"
+	"github.com/spf13/cobra"
+	"os"
+)
+
+var rootCmd = &cobra.Command{
+	Use:   "check_elasticsearch",
+	Short: "A brief description of your application",
+	Long: `A longer description that spans multiple lines and likely contains
+examples and usage of using your application. For example:
+
+Cobra is a CLI library for Go that empowers applications.
+This application is a tool to generate the needed files
+to quickly create a Cobra application.`,
+	// Uncomment the following line if your bare application
+	// has an action associated with it:
+	Run: Help,
+}
+
+func Execute(version string) {
+	defer check.CatchPanic()
+
+	rootCmd.Version = version
+	rootCmd.VersionTemplate()
+
+	if err := rootCmd.Execute(); err != nil {
+		check.ExitError(err)
+	}
+}
+
+func init() {
+	rootCmd.CompletionOptions.DisableDefaultCmd = true
+	rootCmd.DisableAutoGenTag = true
+
+	rootCmd.SetHelpCommand(&cobra.Command{
+		Use:    "no-help",
+		Hidden: true,
+	})
+
+	pfs := rootCmd.PersistentFlags()
+	pfs.StringVarP(&cliConfig.Hostname, "hostname", "H", "localhost", "")
+	pfs.StringVarP(&cliConfig.Username, "username", "U", "", "")
+	pfs.StringVarP(&cliConfig.Password, "password", "P", "", "")
+	pfs.IntVarP(&cliConfig.Port, "port", "p", 9200, "")
+	pfs.BoolVarP(&cliConfig.TLS, "tls", "S", false, "")
+	pfs.BoolVar(&cliConfig.Insecure, "insecure", false, "")
+
+	pfs.SortFlags = false
+}
+
+func Help(cmd *cobra.Command, strings []string) {
+	_ = cmd.Usage()
+
+	os.Exit(3)
+}

cmd/root.go

@@ -0,0 +1,10 @@
+module check_elasticsearch
+
+go 1.16
+
+require (
+	github.com/NETWAYS/go-check v0.2.0
+	github.com/elastic/go-elasticsearch/v7 v7.15.1
+	github.com/spf13/cobra v1.2.1
+	golang.org/x/sys v0.0.0-20211124211545-fe61309f8881 // indirect
+)