feat: init
This commit is contained in:
@@ -0,0 +1,76 @@
|
||||
package config
|
||||
|
||||
import (
|
||||
"os"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
const (
|
||||
envCSIServerURL = "KKS_CSI_SERVER_URL"
|
||||
envCSIAccessToken = "KKS_CSI_ACCESS_TOKEN"
|
||||
envCSIDriverMode = "KKS_CSI_DRIVER_MODE"
|
||||
envCSINodeID = "KKS_CSI_NODE_ID"
|
||||
envCSIDriverName = "KKS_CSI_DRIVER_NAME"
|
||||
envCSIDriverEndpoint = "KKS_CSI_DRIVER_ENDPOINT"
|
||||
)
|
||||
|
||||
// ApplyEnvOverrides applies environment variables used by the Helm chart.
|
||||
func ApplyEnvOverrides(cfg *Config) {
|
||||
if cfg == nil {
|
||||
return
|
||||
}
|
||||
if cfg.Driver == nil {
|
||||
cfg.Driver = &DriverConf{}
|
||||
}
|
||||
if cfg.Client == nil {
|
||||
cfg.Client = &ClientConf{}
|
||||
}
|
||||
|
||||
if v := strings.TrimSpace(os.Getenv(envCSIServerURL)); v != "" {
|
||||
cfg.Client.ServerURL = strings.TrimRight(v, "/")
|
||||
}
|
||||
if v := strings.TrimSpace(os.Getenv(envCSIAccessToken)); v != "" {
|
||||
cfg.Client.AccessToken = v
|
||||
}
|
||||
if v := strings.TrimSpace(os.Getenv(envCSIDriverMode)); v != "" {
|
||||
cfg.Driver.Mode = strings.ToLower(v)
|
||||
}
|
||||
if v := strings.TrimSpace(os.Getenv(envCSINodeID)); v != "" {
|
||||
cfg.Driver.NodeID = v
|
||||
}
|
||||
if v := strings.TrimSpace(os.Getenv(envCSIDriverName)); v != "" {
|
||||
cfg.Driver.Name = v
|
||||
}
|
||||
if v := strings.TrimSpace(os.Getenv(envCSIDriverEndpoint)); v != "" {
|
||||
cfg.Driver.Endpoint = v
|
||||
}
|
||||
|
||||
if cfg.Client.TimeoutSeconds <= 0 {
|
||||
cfg.Client.TimeoutSeconds = 30
|
||||
}
|
||||
if cfg.Driver.Mode == "" {
|
||||
cfg.Driver.Mode = CSIModeNode
|
||||
}
|
||||
if cfg.Driver.Name == "" {
|
||||
cfg.Driver.Name = "storage.csi.kloud.team"
|
||||
}
|
||||
if cfg.Driver.Endpoint == "" {
|
||||
cfg.Driver.Endpoint = "unix:///var/lib/kubelet/plugins/storage.csi.kloud.team/csi.sock"
|
||||
}
|
||||
}
|
||||
|
||||
func ApplyClientTimeoutFromEnv(cfg *Config) {
|
||||
if cfg == nil || cfg.Client == nil {
|
||||
return
|
||||
}
|
||||
raw := strings.TrimSpace(os.Getenv("KKS_CSI_CLIENT_TIMEOUT_SECONDS"))
|
||||
if raw == "" {
|
||||
return
|
||||
}
|
||||
seconds, err := strconv.Atoi(raw)
|
||||
if err != nil || seconds <= 0 {
|
||||
return
|
||||
}
|
||||
cfg.Client.TimeoutSeconds = seconds
|
||||
}
|
||||
Reference in New Issue
Block a user