Remove init funciton usage from env helper

tdunlap607 · Feb 24, 2023 · 3303711 · 3303711
1 parent 9817de0
commit 3303711
Show file tree

Hide file tree

Showing 2 changed files with 8 additions and 6 deletions.
diff --git a/cmd/exporter/main.go b/cmd/exporter/main.go
@@ -5,11 +5,17 @@ import (
 
 	"github.com/kuskoman/logstash-exporter/collectors"
 	"github.com/kuskoman/logstash-exporter/config"
+	"github.com/kuskoman/logstash-exporter/helpers"
 	"github.com/kuskoman/logstash-exporter/server"
 	"github.com/prometheus/client_golang/prometheus"
 )
 
 func main() {
+	warn := helpers.InitializeEnv()
+	if warn != nil {
+		log.Println(warn)
+	}
+
 	port := config.Port
 	host := config.Host
 	logstashUrl := config.LogstashUrl

diff --git a/helpers/env_helper.go b/helpers/env_helper.go
@@ -2,17 +2,13 @@ package helpers
 
 import (
 	"errors"
-	"log"
 	"os"
 
 	"github.com/joho/godotenv"
 )
 
-func init() {
-	err := godotenv.Load()
-	if err != nil {
-		log.Println("Error loading .env file")
-	}
+func InitializeEnv() error {
+	return godotenv.Load()
 }
 
 func GetEnv(key string) string {