fix: code scanning alert no. 2: Incorrect conversion between integer …

…types Co-authored-by: Copilot Autofix powered by AI <62310815+github-advanced-security[bot]@users.noreply.github.com>
TBXark · Dec 2, 2024 · 1e8745c · 1e8745c
1 parent cd0c0c2
commit 1e8745c
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/pkg/utils/idgenerator/idgenerator.go b/pkg/utils/idgenerator/idgenerator.go
@@ -9,8 +9,9 @@ import (
 
 func init() {
 	workerIDRaw := os.Getenv("WORKER_ID")
-	workerID, _ := strconv.Atoi(workerIDRaw)
-	if workerID == 0 {
+	workerIDRaw := os.Getenv("WORKER_ID")
+	workerID, err := strconv.ParseUint(workerIDRaw, 10, 16)
+	if err != nil || workerID == 0 {
 		workerID = 1
 	}
 	options := idgen.NewIdGeneratorOptions(uint16(workerID))