Switch to kingpin flags (prometheus#639)

* Switch to kingpin flags

* Fix logrus vendoring

* Fix flags in main tests

* Fix vendoring versions

Author: carlpett

README.md

@@ -157,9 +157,9 @@ docker run -d -p 9100:9100 \
   -v "/:/rootfs:ro" \
   --net="host" \
   quay.io/prometheus/node-exporter \
-    -collector.procfs /host/proc \
-    -collector.sysfs /host/sys \
-    -collector.filesystem.ignored-mount-points "^/(sys|proc|dev|host|etc)($|/)"
+    --collector.procfs /host/proc \
+    --collector.sysfs /host/sys \
+    --collector.filesystem.ignored-mount-points "^/(sys|proc|dev|host|etc)($|/)"
 ```
 
 Be aware though that the mountpoint label in various metrics will now have

collector/diskstats_linux.go

@@ -17,7 +17,6 @@ package collector
 
 import (
 	"bufio"
-	"flag"
 	"fmt"
 	"io"
 	"os"
@@ -27,6 +26,7 @@ import (
 
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/prometheus/common/log"
+	"gopkg.in/alecthomas/kingpin.v2"
 )
 
 const (
@@ -35,7 +35,7 @@ const (
 )
 
 var (
-	ignoredDevices = flag.String("collector.diskstats.ignored-devices", "^(ram|loop|fd|(h|s|v|xv)d[a-z]|nvme\\d+n\\d+p)\\d+$", "Regexp of devices to ignore for diskstats.")
+	ignoredDevices = kingpin.Flag("collector.diskstats.ignored-devices", "Regexp of devices to ignore for diskstats.").Default("^(ram|loop|fd|(h|s|v|xv)d[a-z]|nvme\\d+n\\d+p)\\d+$").String()
 )
 
 type diskstatsCollector struct {

collector/filesystem_common.go

@@ -17,10 +17,10 @@
 package collector
 
 import (
-	"flag"
 	"regexp"
 
 	"github.com/prometheus/client_golang/prometheus"
+	"gopkg.in/alecthomas/kingpin.v2"
 )
 
 // Arch-dependent implementation must define:
@@ -30,15 +30,14 @@ import (
 // * filesystemCollector.GetStats
 
 var (
-	ignoredMountPoints = flag.String(
+	ignoredMountPoints = kingpin.Flag(
 		"collector.filesystem.ignored-mount-points",
-		defIgnoredMountPoints,
-		"Regexp of mount points to ignore for filesystem collector.")
-
-	ignoredFSTypes = flag.String(
+		"Regexp of mount points to ignore for filesystem collector.",
+	).Default(defIgnoredMountPoints).String()
+	ignoredFSTypes = kingpin.Flag(
 		"collector.filesystem.ignored-fs-types",
-		defIgnoredFSTypes,
-		"Regexp of filesystem types to ignore for filesystem collector.")
+		"Regexp of filesystem types to ignore for filesystem collector.",
+	).Default(defIgnoredFSTypes).String()
 
 	filesystemLabelNames = []string{"device", "mountpoint", "fstype"}
 )

collector/ipvs_linux_test.go

@@ -14,7 +14,6 @@
 package collector
 
 import (
-	"flag"
 	"fmt"
 	"io/ioutil"
 	"net/http"
@@ -24,10 +23,11 @@ import (
 
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/prometheus/client_golang/prometheus/promhttp"
+	"gopkg.in/alecthomas/kingpin.v2"
 )
 
 func TestIPVSCollector(t *testing.T) {
-	if err := flag.Set("collector.procfs", "fixtures/proc"); err != nil {
+	if _, err := kingpin.CommandLine.Parse([]string{"--collector.procfs", "fixtures/proc"}); err != nil {
 		t.Fatal(err)
 	}
 	collector, err := newIPVSCollector()
@@ -76,7 +76,7 @@ func (c miniCollector) Describe(ch chan<- *prometheus.Desc) {
 }
 
 func TestIPVSCollectorResponse(t *testing.T) {
-	if err := flag.Set("collector.procfs", "fixtures/proc"); err != nil {
+	if _, err := kingpin.CommandLine.Parse([]string{"--collector.procfs", "fixtures/proc"}); err != nil {
 		t.Fatal(err)
 	}
 	collector, err := NewIPVSCollector()

collector/megacli.go

@@ -17,13 +17,13 @@ package collector
 
 import (
 	"bufio"
-	"flag"
 	"io"
 	"os/exec"
 	"strconv"
 	"strings"
 
 	"github.com/prometheus/client_golang/prometheus"
+	"gopkg.in/alecthomas/kingpin.v2"
 )
 
 const (
@@ -32,7 +32,7 @@ const (
 )
 
 var (
-	megacliCommand = flag.String("collector.megacli.command", defaultMegaCli, "Command to run megacli.")
+	megacliCommand = kingpin.Flag("collector.megacli.command", "Command to run megacli.").Default(defaultMegaCli).String()
 )
 
 type megaCliCollector struct {

collector/megacli_test.go

@@ -16,11 +16,11 @@
 package collector
 
 import (
-	"flag"
 	"os"
 	"testing"
 
 	"github.com/prometheus/client_golang/prometheus"
+	"gopkg.in/alecthomas/kingpin.v2"
 )
 
 const (
@@ -74,7 +74,7 @@ func TestMegaCliDisks(t *testing.T) {
 }
 
 func TestMegaCliCollectorDoesntCrash(t *testing.T) {
-	if err := flag.Set("collector.megacli.command", "./fixtures/megacli"); err != nil {
+	if _, err := kingpin.CommandLine.Parse([]string{"--collector.megacli.command", "./fixtures/megacli"}); err != nil {
 		t.Fatal(err)
 	}
 	collector, err := NewMegaCliCollector()

collector/netdev_common.go

@@ -17,18 +17,16 @@
 package collector
 
 import (
-	"flag"
 	"fmt"
 	"regexp"
 	"strconv"
 
 	"github.com/prometheus/client_golang/prometheus"
+	"gopkg.in/alecthomas/kingpin.v2"
 )
 
 var (
-	netdevIgnoredDevices = flag.String(
-		"collector.netdev.ignored-devices", "^$",
-		"Regexp of net devices to ignore for netdev collector.")
+	netdevIgnoredDevices = kingpin.Flag("collector.netdev.ignored-devices", "Regexp of net devices to ignore for netdev collector.").Default("^$").String()
 )
 
 type netDevCollector struct {

collector/ntp.go

@@ -16,17 +16,17 @@
 package collector
 
 import (
-	"flag"
 	"fmt"
 
 	"github.com/beevik/ntp"
 	"github.com/prometheus/client_golang/prometheus"
 	"github.com/prometheus/common/log"
+	"gopkg.in/alecthomas/kingpin.v2"
 )
 
 var (
-	ntpServer          = flag.String("collector.ntp.server", "", "NTP server to use for ntp collector.")
-	ntpProtocolVersion = flag.Int("collector.ntp.protocol-version", 4, "NTP protocol version")
+	ntpServer          = kingpin.Flag("collector.ntp.server", "NTP server to use for ntp collector.").Default("").String()
+	ntpProtocolVersion = kingpin.Flag("collector.ntp.protocol-version", "NTP protocol version").Default("4").Int()
 )
 
 type ntpCollector struct {

collector/paths.go

@@ -14,16 +14,16 @@
 package collector
 
 import (
-	"flag"
 	"path"
 
 	"github.com/prometheus/procfs"
+	"gopkg.in/alecthomas/kingpin.v2"
 )
 
 var (
 	// The path of the proc filesystem.
-	procPath = flag.String("collector.procfs", procfs.DefaultMountPoint, "procfs mountpoint.")
-	sysPath  = flag.String("collector.sysfs", "/sys", "sysfs mountpoint.")
+	procPath = kingpin.Flag("collector.procfs", "procfs mountpoint.").Default(procfs.DefaultMountPoint).String()
+	sysPath  = kingpin.Flag("collector.sysfs", "sysfs mountpoint.").Default("/sys").String()
 )
 
 func procFilePath(name string) string {

collector/paths_test.go

@@ -14,14 +14,14 @@
 package collector
 
 import (
-	"flag"
 	"testing"
 
 	"github.com/prometheus/procfs"
+	"gopkg.in/alecthomas/kingpin.v2"
 )
 
 func TestDefaultProcPath(t *testing.T) {
-	if err := flag.Set("collector.procfs", procfs.DefaultMountPoint); err != nil {
+	if _, err := kingpin.CommandLine.Parse([]string{"--collector.procfs", procfs.DefaultMountPoint}); err != nil {
 		t.Fatal(err)
 	}
 
@@ -35,7 +35,7 @@ func TestDefaultProcPath(t *testing.T) {
 }
 
 func TestCustomProcPath(t *testing.T) {
-	if err := flag.Set("collector.procfs", "./../some/./place/"); err != nil {
+	if _, err := kingpin.CommandLine.Parse([]string{"--collector.procfs", "./../some/./place/"}); err != nil {
 		t.Fatal(err)
 	}
 
@@ -49,7 +49,7 @@ func TestCustomProcPath(t *testing.T) {
 }
 
 func TestDefaultSysPath(t *testing.T) {
-	if err := flag.Set("collector.sysfs", "/sys"); err != nil {
+	if _, err := kingpin.CommandLine.Parse([]string{"--collector.sysfs", "/sys"}); err != nil {
 		t.Fatal(err)
 	}
 
@@ -63,7 +63,7 @@ func TestDefaultSysPath(t *testing.T) {
 }
 
 func TestCustomSysPath(t *testing.T) {
-	if err := flag.Set("collector.sysfs", "./../some/./place/"); err != nil {
+	if _, err := kingpin.CommandLine.Parse([]string{"--collector.sysfs", "./../some/./place/"}); err != nil {
 		t.Fatal(err)
 	}