Compare commits
No commits in common. "24c3d5c5b25a7fb4f4f80dd5aaf5ea75a7cd2a72" and "bc34b2d813f257589251303839521634a47ae4fb" have entirely different histories.
24c3d5c5b2
...
bc34b2d813
6 changed files with 12 additions and 153 deletions
53
check.go
53
check.go
|
@ -1,7 +1,6 @@
|
||||||
package gohealthchecks
|
package gohealthchecks
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"fmt"
|
|
||||||
"path"
|
"path"
|
||||||
|
|
||||||
"github.com/google/uuid"
|
"github.com/google/uuid"
|
||||||
|
@ -12,50 +11,28 @@ type Check interface {
|
||||||
params() map[string]string
|
params() map[string]string
|
||||||
}
|
}
|
||||||
|
|
||||||
type uuidCheck struct {
|
type uuidCheck uuid.UUID
|
||||||
checkID uuid.UUID
|
|
||||||
runID uuid.UUID
|
|
||||||
}
|
|
||||||
|
|
||||||
func (c uuidCheck) path() string {
|
func (c uuidCheck) path() string {
|
||||||
return c.checkID.String()
|
return uuid.UUID(c).String()
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c uuidCheck) params() map[string]string {
|
func (c uuidCheck) params() map[string]string {
|
||||||
if c.runID == uuid.Nil {
|
return nil
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
return map[string]string{
|
|
||||||
"rid": c.runID.String(),
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewUUIDCheck(checkRawID, runRawID string) (Check, error) {
|
func NewUUIDCheck(rawUUID string) (Check, error) {
|
||||||
checkID, err := uuid.Parse(checkRawID)
|
val, err := uuid.Parse(rawUUID)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return nil, err
|
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 {
|
type slugCheck struct {
|
||||||
pingKey string
|
pingKey string
|
||||||
slug string
|
slug string
|
||||||
autoCreate bool
|
autoCreate bool
|
||||||
runID uuid.UUID
|
|
||||||
}
|
}
|
||||||
|
|
||||||
func (c slugCheck) path() string {
|
func (c slugCheck) path() string {
|
||||||
|
@ -70,27 +47,13 @@ func (c slugCheck) params() map[string]string {
|
||||||
}
|
}
|
||||||
res["create"] = createVal
|
res["create"] = createVal
|
||||||
|
|
||||||
if c.runID != uuid.Nil {
|
|
||||||
res["rid"] = c.runID.String()
|
|
||||||
}
|
|
||||||
|
|
||||||
return res
|
return res
|
||||||
}
|
}
|
||||||
|
|
||||||
func NewSlugCheck(pingKey, slug string, autoCreate bool, runRawID string) (Check, error) {
|
func NewSlugCheck(pingKey, slug string, autoCreate bool) Check {
|
||||||
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{
|
return slugCheck{
|
||||||
pingKey: pingKey,
|
pingKey: pingKey,
|
||||||
slug: slug,
|
slug: slug,
|
||||||
autoCreate: autoCreate,
|
autoCreate: autoCreate,
|
||||||
runID: runID,
|
}
|
||||||
}, nil
|
|
||||||
}
|
}
|
||||||
|
|
|
@ -1,30 +0,0 @@
|
||||||
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)
|
|
||||||
}
|
|
32
cmd/log.go
32
cmd/log.go
|
@ -1,32 +0,0 @@
|
||||||
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)
|
|
||||||
}
|
|
|
@ -1,36 +0,0 @@
|
||||||
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)
|
|
||||||
}
|
|
|
@ -19,7 +19,6 @@ var (
|
||||||
pingKey string
|
pingKey string
|
||||||
checkSlug string
|
checkSlug string
|
||||||
autoCreate bool
|
autoCreate bool
|
||||||
runID string
|
|
||||||
)
|
)
|
||||||
|
|
||||||
// Execute adds all child commands to the root command and sets flags appropriately.
|
// Execute adds all child commands to the root command and sets flags appropriately.
|
||||||
|
@ -34,7 +33,6 @@ func Execute() {
|
||||||
func init() {
|
func init() {
|
||||||
rootCmd.PersistentFlags().StringVar(&pingHost, "host", gohealthchecks.HealthchecksIOPingHost, "healthchecks host to use")
|
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().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(&checkUUID, "uuid", "", "check UUID")
|
||||||
rootCmd.PersistentFlags().StringVar(&pingKey, "ping-key", "", "ping key")
|
rootCmd.PersistentFlags().StringVar(&pingKey, "ping-key", "", "ping key")
|
||||||
rootCmd.PersistentFlags().StringVar(&checkSlug, "slug", "", "check-slug")
|
rootCmd.PersistentFlags().StringVar(&checkSlug, "slug", "", "check-slug")
|
||||||
|
|
12
cmd/utils.go
12
cmd/utils.go
|
@ -7,8 +7,8 @@ import (
|
||||||
gohealthchecks "git.faercol.me/faercol/go-healthchecks"
|
gohealthchecks "git.faercol.me/faercol/go-healthchecks"
|
||||||
)
|
)
|
||||||
|
|
||||||
func mustParseUUIDCheck(checkID, runID string) gohealthchecks.Check {
|
func mustParseUUIDCheck(rawVal string) gohealthchecks.Check {
|
||||||
check, err := gohealthchecks.NewUUIDCheck(checkID, runID)
|
check, err := gohealthchecks.NewUUIDCheck(rawVal)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
failf("Invalid UUID: %s\n", err)
|
failf("Invalid UUID: %s\n", err)
|
||||||
}
|
}
|
||||||
|
@ -17,14 +17,10 @@ func mustParseUUIDCheck(checkID, runID string) gohealthchecks.Check {
|
||||||
|
|
||||||
func parseCheckFlags() gohealthchecks.Check {
|
func parseCheckFlags() gohealthchecks.Check {
|
||||||
var check gohealthchecks.Check
|
var check gohealthchecks.Check
|
||||||
var err error
|
|
||||||
if checkUUID != "" {
|
if checkUUID != "" {
|
||||||
check = mustParseUUIDCheck(checkUUID, runID)
|
check = mustParseUUIDCheck(checkUUID)
|
||||||
} else {
|
} else {
|
||||||
check, err = gohealthchecks.NewSlugCheck(pingKey, checkSlug, autoCreate, runID)
|
check = gohealthchecks.NewSlugCheck(pingKey, checkSlug, autoCreate)
|
||||||
if err != nil {
|
|
||||||
failf("Invalid UUID: %s\n", err)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
return check
|
return check
|
||||||
}
|
}
|
||||||
|
|
Loading…
Reference in a new issue