wip v1.0.0

internal/health/script.go

@@ -0,0 +1,73 @@
+package health
+
+import (
+	"context"
+	"fmt"
+	"os/exec"
+	"time"
+)
+
+type ScriptChecker struct {
+	name    string
+	script  string
+	args    []string
+	timeout time.Duration
+}
+
+func NewScriptChecker(name string, config map[string]interface{}) (*ScriptChecker, error) {
+	script, ok := config["script"].(string)
+	if !ok {
+		return nil, fmt.Errorf("script checker: missing or invalid 'script' field")
+	}
+
+	var args []string
+	if argsInterface, ok := config["args"].([]interface{}); ok {
+		args = make([]string, len(argsInterface))
+		for i, arg := range argsInterface {
+			if argStr, ok := arg.(string); ok {
+				args[i] = argStr
+			}
+		}
+	}
+
+	timeout := 10 * time.Second
+	if t, ok := config["timeout"].(string); ok {
+		if d, err := time.ParseDuration(t); err == nil {
+			timeout = d
+		}
+	}
+
+	return &ScriptChecker{
+		name:    name,
+		script:  script,
+		args:    args,
+		timeout: timeout,
+	}, nil
+}
+
+func (c *ScriptChecker) Name() string {
+	return c.name
+}
+
+func (c *ScriptChecker) Type() string {
+	return "script"
+}
+
+func (c *ScriptChecker) Check(ctx context.Context) CheckResult {
+	cmdCtx, cancel := context.WithTimeout(ctx, c.timeout)
+	defer cancel()
+
+	cmd := exec.CommandContext(cmdCtx, c.script, c.args...)
+
+	err := cmd.Run()
+	if err != nil {
+		if exitErr, ok := err.(*exec.ExitError); ok {
+			if exitErr.ExitCode() != 0 {
+				return CheckResultFailure
+			}
+		}
+		return CheckResultFailure
+	}
+
+	return CheckResultSuccess
+}