[update_probers] Add flags instead of hard-coding paths
Bug: skia:12544
Change-Id: Icf814a66c503958cb31cf1a32a1ef6d3fa8d5054
Reviewed-on: https://skia-review.googlesource.com/c/buildbot/+/611219
Auto-Submit: Eric Boren <borenet@google.com>
Reviewed-by: Joe Gregorio <jcgregorio@google.com>
Commit-Queue: Joe Gregorio <jcgregorio@google.com>
diff --git a/skfe/go/update_probers/main.go b/skfe/go/update_probers/main.go
index 56e1365..9859448 100644
--- a/skfe/go/update_probers/main.go
+++ b/skfe/go/update_probers/main.go
@@ -5,6 +5,7 @@
package main
import (
+ "flag"
"fmt"
"io/ioutil"
"sort"
@@ -15,12 +16,6 @@
"go.skia.org/infra/go/util"
)
-const (
- probersFilename = "probersk.json"
-
- envoyFilename = "envoy-starter.json"
-)
-
var (
proberRedirectURLsPath = []string{"envoy-redirects", "urls"}
@@ -30,8 +25,12 @@
)
func main() {
+ probersFilename := flag.String("probers-file", "probersk.json", "Path to the probers file.")
+ envoyFilename := flag.String("envoy-file", "envoy-starter.json", "Path to the envoy starter file.")
+ flag.Parse()
+
// Load the existing probers file.
- probers, err := gabs.ParseJSONFile(probersFilename)
+ probers, err := gabs.ParseJSONFile(*probersFilename)
if err != nil {
sklog.Fatal(err)
}
@@ -47,7 +46,7 @@
}
// Load the envoy config file.
- redirects, err := gabs.ParseJSONFile(envoyFilename)
+ redirects, err := gabs.ParseJSONFile(*envoyFilename)
if err != nil {
sklog.Fatal(err)
}
@@ -85,7 +84,7 @@
}
// Rewrite the probers file.
- if err := ioutil.WriteFile(probersFilename, []byte(probers.StringIndent("", " ")), 0644); err != nil {
+ if err := ioutil.WriteFile(*probersFilename, []byte(probers.StringIndent("", " ")), 0644); err != nil {
sklog.Fatal(err)
}
}
