From e63e6e3db777c56ad35e9fd26185de4c471deb6d Mon Sep 17 00:00:00 2001 From: Andrew Stuart Date: Fri, 8 Jul 2016 14:05:10 -0700 Subject: [PATCH] Extend MustRegister function to accept variadic args --- prometheus/registry.go | 12 +++++++----- 1 file changed, 7 insertions(+), 5 deletions(-) diff --git a/prometheus/registry.go b/prometheus/registry.go index 7e3560a..f6ae51b 100644 --- a/prometheus/registry.go +++ b/prometheus/registry.go @@ -115,11 +115,13 @@ func Register(m Collector) error { } // MustRegister works like Register but panics where Register would have -// returned an error. -func MustRegister(m Collector) { - err := Register(m) - if err != nil { - panic(err) +// returned an error. MustRegister is also Variadic, where Register only +// accepts a single Collector to register. +func MustRegister(m ...Collector) { + for i := range m { + if err := Register(m[i]); err != nil { + panic(err) + } } }