refactor: environment variable loading mechanism (#85)

- Remove unused import for `github.com/joho/godotenv/autoload` - Add conditional loading of environment variables from `/run/drone/env` if the file exists Signed-off-by: appleboy <[email protected]>
appleboy · Jun 29, 2024 · 890cfe5 · 890cfe5
1 parent 8535838
commit 890cfe5
Showing 1 changed file with 4 additions and 1 deletion.
diff --git a/main.go b/main.go
@@ -7,7 +7,6 @@ import (
 	"time"
 
 	"github.com/joho/godotenv"
-	_ "github.com/joho/godotenv/autoload"
 	"github.com/urfave/cli/v2"
 )
 
@@ -22,6 +21,10 @@ func main() {
 		_ = godotenv.Load(filename)
 	}
 
+	if _, err := os.Stat("/run/drone/env"); err == nil {
+		_ = godotenv.Overload("/run/drone/env")
+	}
+
 	app := cli.NewApp()
 	app.Name = "Drone Lambda"
 	app.Usage = "Deploying Lambda code with drone CI to an existing function"