healthcheck: implement GetOrRegisterHealthcheck and NewRegisteredHealthcheck

healthcheck.go

@@ -8,6 +8,15 @@ type Healthcheck interface {
 	Unhealthy(error)
 }
 
+// GetOrRegisterHealthcheck returns an existing Healthcheck or
+// constructs and registers a new StandardHealthcheck.
+func GetOrRegisterHealthcheck(name string, r Registry, f func(Healthcheck)) Healthcheck {
+	if nil == r {
+		r = DefaultRegistry
+	}
+	return r.GetOrRegister(name, func() Healthcheck { return NewHealthcheck(f) }).(Healthcheck)
+}
+
 // NewHealthcheck constructs a new Healthcheck which will use the given
 // function to update its status.
 func NewHealthcheck(f func(Healthcheck)) Healthcheck {
@@ -17,6 +26,16 @@ func NewHealthcheck(f func(Healthcheck)) Healthcheck {
 	return &StandardHealthcheck{nil, f}
 }
 
+// NewRegisteredHealthcheck constructs and registers a new StandardHealthcheck.
+func NewRegisteredHealthcheck(name string, r Registry, f func(Healthcheck)) Healthcheck {
+	c := NewHealthcheck(f)
+	if nil == r {
+		r = DefaultRegistry
+	}
+	r.Register(name, c)
+	return c
+}
+
 // NilHealthcheck is a no-op.
 type NilHealthcheck struct{}
 

healthcheck_test.go

@@ -0,0 +1,17 @@
+package metrics
+
+import (
+	"errors"
+	"testing"
+)
+
+func TestGetOrRegisterHealthcheck(t *testing.T) {
+	r := NewRegistry()
+	check := func(h Healthcheck) {
+		h.Unhealthy(errors.New("foo"))
+	}
+	NewRegisteredHealthcheck("foo", r, check).Check()
+	if h := GetOrRegisterHealthcheck("foo", r, check); h.Error().Error() != "foo" {
+		t.Fatal(h)
+	}
+}