feat: updating template for supporting int64 values (#161)

* updating template for supporting int64 values * review changes --------- Co-authored-by: Varkeychan Jacob <[email protected]>
hypertrace · Feb 21, 2024 · 0c1dbb4 · 0c1dbb4
1 parent f05da39
commit 0c1dbb4
Showing 1 changed file with 14 additions and 0 deletions.
diff --git a/tools/go-generator/cmd/generator/_templates/loader.go.tmpl b/tools/go-generator/cmd/generator/_templates/loader.go.tmpl
@@ -58,6 +58,20 @@ func getInt32Env(name string) (int32, bool) {
 	return 0, false
 }
 
+// getInt64Env returns the int64 value for an env var and a confirmation
+// if the var exists
+func getInt64Env(name string) (int64, bool) {
+	if val := os.Getenv(name); val != "" {
+		intVal, err := strconv.Atoi(val)
+		if err != nil {
+		    return 0, false
+		}
+		return int64(intVal), true
+	}
+
+	return 0, false
+}
+
 // loadFromFile loads the agent config from a file
 func loadFromFile(c *{{ .MainType}}, filename string) error {
 	unmarshaler := protojson.UnmarshalOptions{DiscardUnknown: true}