Merge pull request #29 from debeando/feat-env-getInt

feat - in env submodule is add GetInt, GetUInt8, GetUInt16 and GetBool.

Author: nicola-strappazzon

aws/rds/logs_test.go

@@ -24,4 +24,4 @@ func TestLogsSort(t *testing.T) {
 	}
 
 	assert.True(t, true)
-}
+}

env/main.go

@@ -2,20 +2,60 @@ package env
 
 import (
 	"os"
+	"strconv"
 )
 
-// Verify environment variable.
+// Verify exist environment variable.
 func Exist(key string) bool {
 	if _, ok := os.LookupEnv(key); ok {
 		return true
 	}
 	return false
 }
 
-// Retrieve environment variable.
+// Retrieve environment variable in String type.
 func Get(key, fallback string) string {
 	if value, ok := os.LookupEnv(key); ok {
 		return value
 	}
 	return fallback
 }
+
+// Retrieve environment variable in Int type.
+func GetInt(key string, fallback int) int {
+	if envvalue, ok := os.LookupEnv(key); ok {
+		value, _ := strconv.Atoi(envvalue)
+		return value
+	}
+
+	return fallback
+}
+
+// Retrieve environment variable in Int type.
+func GetUInt8(key string, fallback uint8) uint8 {
+	if envvalue, ok := os.LookupEnv(key); ok {
+		value, _ := strconv.Atoi(envvalue)
+		return uint8(value)
+	}
+
+	return fallback
+}
+
+func GetUInt16(key string, fallback uint16) uint16 {
+	if envvalue, ok := os.LookupEnv(key); ok {
+		value, _ := strconv.Atoi(envvalue)
+		return uint16(value)
+	}
+
+	return fallback
+}
+
+// Retrieve environment variable in Bool type.
+func GetBool(key string, fallback bool) bool {
+	if envvalue, ok := os.LookupEnv(key); ok {
+		value, _ := strconv.ParseBool(envvalue)
+		return value
+	}
+
+	return fallback
+}

env/main_test.go

@@ -14,6 +14,34 @@ func TestGet(t *testing.T) {
 	assert.Equal(t, env.Get("ENV_TMP", "value1"), "value1")
 }
 
+func TestGetInt(t *testing.T) {
+	t.Setenv("ENV_VAR", "10")
+
+	assert.Equal(t, env.GetInt("ENV_VAR", 20), 10)
+	assert.Equal(t, env.GetInt("ENV_TMP", 0), 0)
+}
+
+func TestGetUInt8(t *testing.T) {
+	t.Setenv("ENV_VAR", "255")
+
+	assert.Equal(t, env.GetUInt8("ENV_VAR", uint8(10)), uint8(255))
+	assert.Equal(t, env.GetUInt8("ENV_TMP", uint8(0)), uint8(0))
+}
+
+func TestGetUInt16(t *testing.T) {
+	t.Setenv("ENV_VAR", "3306")
+
+	assert.Equal(t, env.GetUInt16("ENV_VAR", uint16(10)), uint16(3306))
+	assert.Equal(t, env.GetUInt16("ENV_TMP", uint16(0)), uint16(0))
+}
+
+func TestGetBool(t *testing.T) {
+	t.Setenv("ENV_VAR", "true")
+
+	assert.Equal(t, env.GetBool("ENV_VAR", false), true)
+	assert.Equal(t, env.GetBool("ENV_TMP", false), false)
+}
+
 func TestExist(t *testing.T) {
 	t.Setenv("ENV_VAR", "value1")
 

go.mod

@@ -1,48 +1,51 @@
 module github.com/debeando/go-common
 
-go 1.19
+go 1.22
+
+toolchain go1.23.2
 
 require (
-	github.com/ClickHouse/clickhouse-go/v2 v2.14.2
-	github.com/aws/aws-sdk-go v1.44.319
-	github.com/fatih/color v1.15.0
-	github.com/go-sql-driver/mysql v1.7.1
-	github.com/google/uuid v1.3.1
+	github.com/ClickHouse/clickhouse-go/v2 v2.30.0
+	github.com/aws/aws-sdk-go v1.55.5
+	github.com/fatih/color v1.18.0
+	github.com/go-sql-driver/mysql v1.8.1
+	github.com/google/uuid v1.6.0
 	github.com/hpcloud/tail v1.0.0
 	github.com/sirupsen/logrus v1.9.3
-	github.com/stretchr/testify v1.8.4
-	go.mongodb.org/mongo-driver v1.12.1
-	golang.org/x/sys v0.14.0
+	github.com/stretchr/testify v1.9.0
+	go.mongodb.org/mongo-driver v1.17.1
+	golang.org/x/sys v0.26.0
 )
 
 require (
-	github.com/ClickHouse/ch-go v0.58.2 // indirect
-	github.com/andybalholm/brotli v1.0.5 // indirect
+	filippo.io/edwards25519 v1.1.0 // indirect
+	github.com/ClickHouse/ch-go v0.63.1 // indirect
+	github.com/andybalholm/brotli v1.1.1 // indirect
 	github.com/davecgh/go-spew v1.1.1 // indirect
 	github.com/fsnotify/fsnotify v1.7.0 // indirect
 	github.com/go-faster/city v1.0.1 // indirect
-	github.com/go-faster/errors v0.6.1 // indirect
-	github.com/golang/snappy v0.0.1 // indirect
+	github.com/go-faster/errors v0.7.1 // indirect
+	github.com/golang/snappy v0.0.4 // indirect
 	github.com/jmespath/go-jmespath v0.4.0 // indirect
-	github.com/klauspost/compress v1.16.7 // indirect
+	github.com/klauspost/compress v1.17.11 // indirect
 	github.com/mattn/go-colorable v0.1.13 // indirect
-	github.com/mattn/go-isatty v0.0.17 // indirect
-	github.com/montanaflynn/stats v0.0.0-20171201202039-1bf9dbcd8cbe // indirect
-	github.com/paulmach/orb v0.10.0 // indirect
-	github.com/pierrec/lz4/v4 v4.1.18 // indirect
+	github.com/mattn/go-isatty v0.0.20 // indirect
+	github.com/montanaflynn/stats v0.7.1 // indirect
+	github.com/paulmach/orb v0.11.1 // indirect
+	github.com/pierrec/lz4/v4 v4.1.21 // indirect
 	github.com/pkg/errors v0.9.1 // indirect
 	github.com/pmezard/go-difflib v1.0.0 // indirect
 	github.com/segmentio/asm v1.2.0 // indirect
-	github.com/shopspring/decimal v1.3.1 // indirect
+	github.com/shopspring/decimal v1.4.0 // indirect
 	github.com/xdg-go/pbkdf2 v1.0.0 // indirect
 	github.com/xdg-go/scram v1.1.2 // indirect
 	github.com/xdg-go/stringprep v1.0.4 // indirect
-	github.com/youmark/pkcs8 v0.0.0-20181117223130-1be2e3e5546d // indirect
-	go.opentelemetry.io/otel v1.19.0 // indirect
-	go.opentelemetry.io/otel/trace v1.19.0 // indirect
-	golang.org/x/crypto v0.0.0-20220622213112-05595931fe9d // indirect
-	golang.org/x/sync v0.3.0 // indirect
-	golang.org/x/text v0.7.0 // indirect
+	github.com/youmark/pkcs8 v0.0.0-20240726163527-a2c0da244d78 // indirect
+	go.opentelemetry.io/otel v1.31.0 // indirect
+	go.opentelemetry.io/otel/trace v1.31.0 // indirect
+	golang.org/x/crypto v0.28.0 // indirect
+	golang.org/x/sync v0.8.0 // indirect
+	golang.org/x/text v0.19.0 // indirect
 	gopkg.in/fsnotify.v1 v1.4.7 // indirect
 	gopkg.in/tomb.v1 v1.0.0-20141024135613-dd632973f1e7 // indirect
 	gopkg.in/yaml.v3 v3.0.1 // indirect