mirror of
https://github.com/k8snetworkplumbingwg/multus-cni.git
synced 2025-09-14 21:47:25 +00:00
delete multus conf file generated by thin_entrypoint
- if cleanup-config-on-exit is set delete generated multus config file on exit. - add an option to skip watch for master cni config and kubeconfig as cleanup-config-on-exit with multus-conf-file=auto also triggered the watch for cases when deletion of multus config is desired but watch isnt - setup signal handling to allow config file cleanup on exit Signed-off-by: adrianc <adrianc@nvidia.com>
This commit is contained in:
@@ -31,6 +31,7 @@ import (
|
|||||||
"github.com/spf13/pflag"
|
"github.com/spf13/pflag"
|
||||||
|
|
||||||
"gopkg.in/k8snetworkplumbingwg/multus-cni.v4/pkg/cmdutils"
|
"gopkg.in/k8snetworkplumbingwg/multus-cni.v4/pkg/cmdutils"
|
||||||
|
"gopkg.in/k8snetworkplumbingwg/multus-cni.v4/pkg/signals"
|
||||||
)
|
)
|
||||||
|
|
||||||
// Options stores command line options
|
// Options stores command line options
|
||||||
@@ -57,6 +58,7 @@ type Options struct {
|
|||||||
AdditionalBinDir string
|
AdditionalBinDir string
|
||||||
ForceCNIVersion bool
|
ForceCNIVersion bool
|
||||||
SkipTLSVerify bool
|
SkipTLSVerify bool
|
||||||
|
SkipMultusConfWatch bool
|
||||||
}
|
}
|
||||||
|
|
||||||
const (
|
const (
|
||||||
@@ -84,7 +86,8 @@ func (o *Options) addFlags() {
|
|||||||
fs.StringVar(&o.MultusLogLevel, "multus-log-level", "", "multus log level")
|
fs.StringVar(&o.MultusLogLevel, "multus-log-level", "", "multus log level")
|
||||||
fs.StringVar(&o.MultusLogFile, "multus-log-file", "", "multus log file")
|
fs.StringVar(&o.MultusLogFile, "multus-log-file", "", "multus log file")
|
||||||
fs.BoolVar(&o.OverrideNetworkName, "override-network-name", false, "override network name from master cni file (used only with --multus-conf-file=auto)")
|
fs.BoolVar(&o.OverrideNetworkName, "override-network-name", false, "override network name from master cni file (used only with --multus-conf-file=auto)")
|
||||||
fs.BoolVar(&o.CleanupConfigOnExit, "cleanup-config-on-exit", false, "cleanup config file on exit (used only with --multus-conf-file=auto)")
|
fs.BoolVar(&o.CleanupConfigOnExit, "cleanup-config-on-exit", false, "cleanup config file on exit")
|
||||||
|
fs.BoolVar(&o.SkipMultusConfWatch, "skip-config-watch", false, "dont watch for config (master cni and kubeconfig) changes (used only with --multus-conf-file=auto)")
|
||||||
fs.BoolVar(&o.RenameConfFile, "rename-conf-file", false, "rename master config file to invalidate (used only with --multus-conf-file=auto)")
|
fs.BoolVar(&o.RenameConfFile, "rename-conf-file", false, "rename master config file to invalidate (used only with --multus-conf-file=auto)")
|
||||||
fs.StringVar(&o.ReadinessIndicatorFile, "readiness-indicator-file", "", "readiness indicator file (used only with --multus-conf-file=auto)")
|
fs.StringVar(&o.ReadinessIndicatorFile, "readiness-indicator-file", "", "readiness indicator file (used only with --multus-conf-file=auto)")
|
||||||
fs.StringVar(&o.AdditionalBinDir, "additional-bin-dir", "", "adds binDir option to configuration (used only with --multus-conf-file=auto)")
|
fs.StringVar(&o.AdditionalBinDir, "additional-bin-dir", "", "adds binDir option to configuration (used only with --multus-conf-file=auto)")
|
||||||
@@ -604,9 +607,24 @@ func main() {
|
|||||||
fmt.Printf("multus config file is created.\n")
|
fmt.Printf("multus config file is created.\n")
|
||||||
}
|
}
|
||||||
|
|
||||||
if opt.CleanupConfigOnExit && opt.MultusConfFile == "auto" {
|
ctx := signals.SetupSignalHandler()
|
||||||
|
|
||||||
|
if opt.CleanupConfigOnExit {
|
||||||
|
defer cleanupMultusConf(&opt)
|
||||||
|
}
|
||||||
|
|
||||||
|
watchChanges := opt.CleanupConfigOnExit && opt.MultusConfFile == "auto" && !opt.SkipMultusConfWatch
|
||||||
|
if watchChanges {
|
||||||
fmt.Printf("Entering watch loop...\n")
|
fmt.Printf("Entering watch loop...\n")
|
||||||
for {
|
masterConfigExists := true
|
||||||
|
|
||||||
|
outer:
|
||||||
|
for range time.Tick(1 * time.Second) {
|
||||||
|
select {
|
||||||
|
case <-ctx.Done():
|
||||||
|
// signal received break from loop
|
||||||
|
break outer
|
||||||
|
default:
|
||||||
// Check kubeconfig and update if different (i.e. service account updated)
|
// Check kubeconfig and update if different (i.e. service account updated)
|
||||||
caHash, saTokenHash, err = opt.createKubeConfig(caHash, saTokenHash)
|
caHash, saTokenHash, err = opt.createKubeConfig(caHash, saTokenHash)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
@@ -619,28 +637,42 @@ func main() {
|
|||||||
|
|
||||||
// if masterConfigFilePath is no longer exists
|
// if masterConfigFilePath is no longer exists
|
||||||
if os.IsNotExist(err) {
|
if os.IsNotExist(err) {
|
||||||
fmt.Printf("Master plugin @ %q has been deleted. Allowing 45 seconds for its restoration...\n", masterConfigFilePath)
|
if masterConfigExists {
|
||||||
time.Sleep(10 * time.Second)
|
fmt.Printf("Master plugin @ %q has been deleted. waiting for its restoration...\n", masterConfigFilePath)
|
||||||
|
}
|
||||||
|
masterConfigExists = false
|
||||||
|
continue
|
||||||
|
}
|
||||||
|
|
||||||
for range time.Tick(1 * time.Second) {
|
if !masterConfigExists {
|
||||||
_, err = os.Stat(masterConfigFilePath)
|
|
||||||
if !os.IsNotExist(err) {
|
|
||||||
fmt.Printf("Master plugin @ %q was restored. Regenerating given configuration.\n", masterConfigFilePath)
|
fmt.Printf("Master plugin @ %q was restored. Regenerating given configuration.\n", masterConfigFilePath)
|
||||||
break
|
masterConfigExists = true
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
masterConfigFilePath, masterConfigHash, err = opt.createMultusConfig(masterConfigHash)
|
masterConfigFilePath, masterConfigHash, err = opt.createMultusConfig(masterConfigHash)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
fmt.Fprintf(os.Stderr, "failed to create multus config: %v\n", err)
|
fmt.Fprintf(os.Stderr, "failed to create multus config: %v\n", err)
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
time.Sleep(1 * time.Second)
|
}
|
||||||
}
|
}
|
||||||
} else {
|
} else {
|
||||||
// sleep infinitely
|
// wait until signal received
|
||||||
for {
|
<-ctx.Done()
|
||||||
time.Sleep(time.Duration(1<<63 - 1))
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func cleanupMultusConf(opt *Options) {
|
||||||
|
// try remove multus conf
|
||||||
|
if opt.MultusConfFile == "auto" {
|
||||||
|
multusConfFilePath := fmt.Sprintf("%s/00-multus.conf", opt.CNIConfDir)
|
||||||
|
_ = os.Remove(multusConfFilePath)
|
||||||
|
|
||||||
|
multusConfFilePath = fmt.Sprintf("%s/00-multus.conflist", opt.CNIConfDir)
|
||||||
|
_ = os.Remove(multusConfFilePath)
|
||||||
|
} else {
|
||||||
|
confFileName := filepath.Base(opt.MultusConfFile)
|
||||||
|
_ = os.Remove(filepath.Join(opt.CNIConfDir, confFileName))
|
||||||
|
}
|
||||||
|
|
||||||
|
}
|
||||||
|
Reference in New Issue
Block a user