Check that collectors given on --collectors.enabled exist before trying to construct them

This commit is contained in:
Calle Pettersson
2020-05-24 17:23:13 +02:00
parent 245b2b4d28
commit 2ea20ff628

View File

@@ -1,6 +1,7 @@
package collector package collector
import ( import (
"fmt"
"strconv" "strconv"
"strings" "strings"
@@ -72,7 +73,11 @@ func Available() []string {
return cs return cs
} }
func Build(collector string) (Collector, error) { func Build(collector string) (Collector, error) {
return builders[collector]() builder, exists := builders[collector]
if !exists {
return nil, fmt.Errorf("Unknown collector %q", collector)
}
return builder()
} }
func getPerfQuery(collectors []string) string { func getPerfQuery(collectors []string) string {
parts := make([]string, 0, len(collectors)) parts := make([]string, 0, len(collectors))