add version ability

rancher · Oct 12, 2016 · f7167cb · f7167cb
1 parent a9a60d4
commit f7167cb
Showing 1 changed file with 12 additions and 2 deletions.
diff --git a/main.go b/main.go
@@ -1,13 +1,23 @@
 package main
 
 import (
-	"os"
-
+	"flag"
+	"fmt"
 	"github.com/Sirupsen/logrus"
 	"github.com/rancher/agent/events"
+	"os"
 )
 
+var VERSION = "dev"
+
 func main() {
+	version := flag.Bool("version", false, "go-agent version")
+	flag.Parse()
+	if *version {
+		fmt.Printf("go-agent version %s \n", VERSION)
+		os.Exit(0)
+	}
+
 	logrus.Info("Launching agent")
 
 	url := os.Getenv("CATTLE_URL")