summaryrefslogtreecommitdiffstats
path: root/main.go
diff options
context:
space:
mode:
Diffstat (limited to 'main.go')
-rw-r--r--main.go13
1 files changed, 10 insertions, 3 deletions
diff --git a/main.go b/main.go
index 1c75e50..3a8671e 100644
--- a/main.go
+++ b/main.go
@@ -143,7 +143,7 @@ func main() {
Value: defaultLogLevel,
},
cli.StringFlag{
- Name: "url",
+ Name: "url, u",
EnvVar: "XDS_SERVER_URL",
Value: "localhost:8000",
Usage: "remote XDS server url",
@@ -247,8 +247,15 @@ func XdsConnInit(ctx *cli.Context) error {
// Define HTTP and WS url
baseURL := ctx.String("url")
- if !strings.HasPrefix(ctx.String("url"), "http://") {
- baseURL = "http://" + ctx.String("url")
+
+ // Allow to only set port number
+ if match, _ := regexp.MatchString("^([0-9]+)$", baseURL); match {
+ baseURL = "http://localhost:" + ctx.String("url")
+ }
+
+ // Add http prefix if missing
+ if !strings.HasPrefix(baseURL, "http://") {
+ baseURL = "http://" + baseURL
}
// Create HTTP client