fix: load ucloud env file when it exists

Author: fioncat

cmd/cnivpc/main.go

@@ -35,6 +35,7 @@ import (
 	"github.com/containernetworking/cni/pkg/version"
 	"github.com/containernetworking/plugins/pkg/ip"
 	"github.com/containernetworking/plugins/pkg/ns"
+	"github.com/joho/godotenv"
 	"github.com/vishvananda/netlink"
 )
 
@@ -313,6 +314,8 @@ func tickSuicide(done chan bool) {
 	}
 }
 
+const envFilePath = "/etc/kubernetes/ucloud"
+
 func main() {
 	// Print version
 	if len(os.Args) == 2 && os.Args[1] == "version" {
@@ -322,6 +325,20 @@ func main() {
 
 	ulog.BinaryMode("/var/log/cnivpc.log")
 
+	_, err := os.Stat(envFilePath)
+	if err != nil {
+		if os.IsNotExist(err) {
+			ulog.Warnf("Env file %s not found", envFilePath)
+		} else {
+			ulog.Errorf("Check env file %s error: %v", envFilePath, err)
+		}
+	} else {
+		err = godotenv.Load(envFilePath)
+		if err != nil {
+			ulog.Errorf("Load env file %s error: %v", envFilePath, err)
+		}
+	}
+
 	about := fmt.Sprintf("ucloud-uk8s-cnivpc version %s", vs.CNIVersion)
 
 	done := make(chan bool, 1)

go.mod

@@ -8,6 +8,7 @@ require (
 	github.com/containernetworking/plugins v0.9.1
 	github.com/coreos/go-iptables v0.6.0
 	github.com/j-keck/arping v1.0.2
+	github.com/joho/godotenv v1.5.1
 	github.com/mattn/go-shellwords v1.0.12
 	github.com/sirupsen/logrus v1.6.0
 	github.com/spf13/cobra v1.7.0

go.sum

@@ -88,6 +88,8 @@ github.com/inconshreveable/mousetrap v1.1.0/go.mod h1:vpF70FUmC8bwa3OWnCshd2FqLf
 github.com/j-keck/arping v0.0.0-20160618110441-2cf9dc699c56/go.mod h1:ymszkNOg6tORTn+6F6j+Jc8TOr5osrynvN6ivFWZ2GA=
 github.com/j-keck/arping v1.0.2 h1:hlLhuXgQkzIJTZuhMigvG/CuSkaspeaD9hRDk2zuiMI=
 github.com/j-keck/arping v1.0.2/go.mod h1:aJbELhR92bSk7tp79AWM/ftfc90EfEi2bQJrbBFOsPw=
+github.com/joho/godotenv v1.5.1 h1:7eLL/+HRGLY0ldzfGMeQkb7vMd0as4CfYvUVzLqw0N0=
+github.com/joho/godotenv v1.5.1/go.mod h1:f4LDr5Voq0i2e/R5DDNOoa2zzDfwtkZa6DnEwAbqwq4=
 github.com/josharian/intern v1.0.0 h1:vlS4z54oSdjm0bgjRigI+G1HpF+tI+9rE5LLzOg8HmY=
 github.com/josharian/intern v1.0.0/go.mod h1:5DoeVV0s6jJacbCEi61lwdGj/aVlrQvzHFFd8Hwg//Y=
 github.com/json-iterator/go v1.1.12 h1:PV8peI4a0ysnczrg+LtxykD8LfKY9ML6u2jnxaEnrnM=