Add support for check-id

check.go

@@ -1,6 +1,7 @@
 package gohealthchecks
 
 import (
+	"fmt"
 	"path"
 
 	"github.com/google/uuid"
@@ -11,28 +12,50 @@ type Check interface {
 	params() map[string]string
 }
 
-type uuidCheck uuid.UUID
+type uuidCheck struct {
+	checkID uuid.UUID
+	runID   uuid.UUID
+}
 
 func (c uuidCheck) path() string {
-	return uuid.UUID(c).String()
+	return c.checkID.String()
 }
 
 func (c uuidCheck) params() map[string]string {
+	if c.runID == uuid.Nil {
 		return nil
+	}
+
+	return map[string]string{
+		"rid": c.runID.String(),
+	}
 }
 
-func NewUUIDCheck(rawUUID string) (Check, error) {
-	val, err := uuid.Parse(rawUUID)
+func NewUUIDCheck(checkRawID, runRawID string) (Check, error) {
+	checkID, err := uuid.Parse(checkRawID)
 	if err != nil {
 		return nil, err
 	}
-	return uuidCheck(val), nil
+
+	runID := uuid.Nil
+	if runRawID != "" {
+		runID, err = uuid.Parse(runRawID)
+		if err != nil {
+			return nil, err
+		}
+	}
+
+	return uuidCheck{
+		checkID: checkID,
+		runID:   runID,
+	}, nil
 }
 
 type slugCheck struct {
 	pingKey    string
 	slug       string
 	autoCreate bool
+	runID      uuid.UUID
 }
 
 func (c slugCheck) path() string {
@@ -47,13 +70,27 @@ func (c slugCheck) params() map[string]string {
 	}
 	res["create"] = createVal
 
+	if c.runID != uuid.Nil {
+		res["rid"] = c.runID.String()
+	}
+
 	return res
 }
 
-func NewSlugCheck(pingKey, slug string, autoCreate bool) Check {
+func NewSlugCheck(pingKey, slug string, autoCreate bool, runRawID string) (Check, error) {
+	runID := uuid.Nil
+	if runRawID != "" {
+		var err error
+		runID, err = uuid.Parse(runRawID)
+		if err != nil {
+			return nil, fmt.Errorf("invalid run ID: %w", err)
+		}
+	}
+
 	return slugCheck{
 		pingKey:    pingKey,
 		slug:       slug,
 		autoCreate: autoCreate,
-	}
+		runID:      runID,
+	}, nil
 }

cmd/failure.go

@@ -0,0 +1,30 @@
+package main
+
+import (
+	"fmt"
+	"os"
+
+	gohealthchecks "git.faercol.me/faercol/go-healthchecks"
+	"github.com/spf13/cobra"
+)
+
+var failureCmd = &cobra.Command{
+	Use:   "failure",
+	Short: "Signal the check has failed",
+	Long:  ``,
+	Run: func(cmd *cobra.Command, args []string) {
+		check := parseCheckFlags()
+		clt := gohealthchecks.NewPingClient(pingHost)
+		err := clt.ReportFailure(cmd.Context(), check)
+		if err != nil {
+			failf("Failed to notify failure: %s\n", err)
+		} else {
+			fmt.Println("Notified check failure")
+			os.Exit(0)
+		}
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(failureCmd)
+}

cmd/log.go

@@ -0,0 +1,32 @@
+package main
+
+import (
+	"fmt"
+	"os"
+
+	gohealthchecks "git.faercol.me/faercol/go-healthchecks"
+	"github.com/spf13/cobra"
+)
+
+var logCmd = &cobra.Command{
+	Use:   "log <message>",
+	Short: "Send a log for the check",
+	Long:  ``,
+	Args:  cobra.ExactArgs(1),
+	Run: func(cmd *cobra.Command, args []string) {
+		check := parseCheckFlags()
+		msg := []byte(args[0])
+		clt := gohealthchecks.NewPingClient(pingHost)
+		err := clt.LogMessage(cmd.Context(), check, msg)
+		if err != nil {
+			failf("Failed to send log: %s\n", err)
+		} else {
+			fmt.Println("Log sent")
+			os.Exit(0)
+		}
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(logCmd)
+}

cmd/returncode.go

@@ -0,0 +1,36 @@
+package main
+
+import (
+	"fmt"
+	"os"
+	"strconv"
+
+	gohealthchecks "git.faercol.me/faercol/go-healthchecks"
+	"github.com/spf13/cobra"
+)
+
+var returncodeCmd = &cobra.Command{
+	Use:   "returncode <message>",
+	Short: "Notify specific returncode for the check",
+	Long:  ``,
+	Args:  cobra.ExactArgs(1),
+	Run: func(cmd *cobra.Command, args []string) {
+		check := parseCheckFlags()
+		returncode, err := strconv.Atoi(args[0])
+		if err != nil {
+			failf("Invalid value for returncode: %s\n", err)
+		}
+		clt := gohealthchecks.NewPingClient(pingHost)
+		err = clt.ReportExitCode(cmd.Context(), check, returncode)
+		if err != nil {
+			failf("Failed to send log: %s\n", err)
+		} else {
+			fmt.Println("Log sent")
+			os.Exit(0)
+		}
+	},
+}
+
+func init() {
+	rootCmd.AddCommand(returncodeCmd)
+}

cmd/root.go

@@ -19,6 +19,7 @@ var (
 	pingKey    string
 	checkSlug  string
 	autoCreate bool
+	runID      string
 )
 
 // Execute adds all child commands to the root command and sets flags appropriately.
@@ -33,6 +34,7 @@ func Execute() {
 func init() {
 	rootCmd.PersistentFlags().StringVar(&pingHost, "host", gohealthchecks.HealthchecksIOPingHost, "healthchecks host to use")
 	rootCmd.PersistentFlags().BoolVar(&autoCreate, "auto-create", false, "auto create check if it does not exist (slug only)")
+	rootCmd.PersistentFlags().StringVar(&runID, "run-id", "", "Optional run-id to identify the job")
 	rootCmd.PersistentFlags().StringVar(&checkUUID, "uuid", "", "check UUID")
 	rootCmd.PersistentFlags().StringVar(&pingKey, "ping-key", "", "ping key")
 	rootCmd.PersistentFlags().StringVar(&checkSlug, "slug", "", "check-slug")

cmd/utils.go

@@ -7,8 +7,8 @@ import (
 	gohealthchecks "git.faercol.me/faercol/go-healthchecks"
 )
 
-func mustParseUUIDCheck(rawVal string) gohealthchecks.Check {
-	check, err := gohealthchecks.NewUUIDCheck(rawVal)
+func mustParseUUIDCheck(checkID, runID string) gohealthchecks.Check {
+	check, err := gohealthchecks.NewUUIDCheck(checkID, runID)
 	if err != nil {
 		failf("Invalid UUID: %s\n", err)
 	}
@@ -17,10 +17,14 @@ func mustParseUUIDCheck(rawVal string) gohealthchecks.Check {
 
 func parseCheckFlags() gohealthchecks.Check {
 	var check gohealthchecks.Check
+	var err error
 	if checkUUID != "" {
-		check = mustParseUUIDCheck(checkUUID)
+		check = mustParseUUIDCheck(checkUUID, runID)
 	} else {
-		check = gohealthchecks.NewSlugCheck(pingKey, checkSlug, autoCreate)
+		check, err = gohealthchecks.NewSlugCheck(pingKey, checkSlug, autoCreate, runID)
+		if err != nil {
+			failf("Invalid UUID: %s\n", err)
+		}
 	}
 	return check
 }