- [+] add int flag env var support to commandlineFlag

Author: suntong

test/commandlineFlag.ref

@@ -24,6 +24,7 @@ type Options struct {
 	ExtTmpl		string	// `extension` of template file
 	StrFrom		string	// replace from, the from string used for the replace function
 	StrTo		string	// replace to, the to string used for the replace function
+	IntV		int	// int var
 	Batch		time.Duration	// batch interval
 	Debug		int	// debugging `level`
 }
@@ -56,6 +57,8 @@ func initVars() {
 		"replace from, the from string used for the replace function")
 	flag.StringVar(&Opts.StrTo, "rt", "",
 		"replace to, the to string used for the replace function")
+	flag.IntVar(&Opts.IntV, "intv", 0,
+		"int var")
 	flag.DurationVar(&Opts.Batch, "batch", 120*time.Second,
 		"batch interval")
 	flag.IntVar(&Opts.Debug, "d", 0,
@@ -98,6 +101,24 @@ func initVals() {
 		len(os.Getenv("EASYGEN_RT")) != 0 {
 		Opts.StrTo = os.Getenv("EASYGEN_RT")
 	}
+	if Opts.IntV == 0 ||
+		len(os.Getenv("EASYGEN_INTV")) != 0 {
+		if i, err := strconv.Atoi(os.Getenv("EASYGEN_INTV")); err == nil {
+			Opts.IntV = i
+		}
+	}
+	if Opts.Debug == 0 ||
+		len(os.Getenv("EASYGEN_D")) != 0 {
+		if i, err := strconv.Atoi(os.Getenv("EASYGEN_D")); err == nil {
+			Opts.Debug = i
+		}
+	}
+	if Opts.Debug == 0 ||
+		len(os.Getenv("EASYGEN_DEBUG")) != 0 {
+		if i, err := strconv.Atoi(os.Getenv("EASYGEN_DEBUG")); err == nil {
+			Opts.Debug = i
+		}
+	}
 
 }
 

test/commandlineFlag.tmpl

@@ -55,7 +55,19 @@ func initVals() {
 	}{{end}}{{else}}{{if eq .Type "bool" }}
 	if _, exists = os.LookupEnv("{{$envVar}}"); {{$.StructVar}}.{{.Name}}|| exists {
 		{{$.StructVar}}.{{.Name}} = true
-	}{{end}}{{end}}{{end}}{{end}}
+	}{{else}}{{if eq .Type "int" }}{{if (stringsContains .Flag ",")}}{{$cf := .}}{{range (stringsSplit .Flag ",")}}{{$envVar := printf "%s_%s" (clk2ss $.ProgramName) (clk2ss .)}}
+	if {{$.StructVar}}.{{$cf.Name}} == 0 ||
+		len(os.Getenv("{{$envVar}}")) != 0 {
+		if i, err := strconv.Atoi(os.Getenv("{{$envVar}}")); err == nil {
+			{{$.StructVar}}.{{$cf.Name}} = i
+		}
+	}{{end}}{{else}}
+	if {{$.StructVar}}.{{.Name}} == 0 ||
+		len(os.Getenv("{{$envVar}}")) != 0 {
+		if i, err := strconv.Atoi(os.Getenv("{{$envVar}}")); err == nil {
+			{{$.StructVar}}.{{.Name}} = i
+		}
+	}{{end}}{{end}}{{end}}{{end}}{{end}}{{end}}
 
 }
 

test/commandlineFlag.yaml

@@ -62,6 +62,12 @@ Options:
     Value: '""'
     Usage: "replace to, the to string used for the replace function"
 
+  - Name: IntV
+    Type: int
+    Flag: intv
+    Value: 0
+    Usage: "int var"
+
   - Name: Batch
     Type: duration
     Flag: batch