refactor the profile add

Signed-off-by: jason yang <jasonyangshadow@gmail.com>
This commit is contained in:
jason yang
2023-04-12 10:27:44 +00:00
parent 0ad69ca3ce
commit 0faada6aae
8 changed files with 180 additions and 72 deletions

View File

@@ -14,8 +14,10 @@ import (
"gopkg.in/yaml.v2"
)
var ConfigFile string
var DefaultConfig string
var (
ConfigFile string
DefaultConfig string
)
var cachedDB NodeYaml
@@ -91,7 +93,8 @@ func New() (NodeYaml, error) {
wwlog.Debug("Returning node object")
cachedDB = ret
cachedDB.current = true
return ret, nil
cachedDB.persist = true
return cachedDB, nil
}
/*

View File

@@ -38,7 +38,6 @@ func (config *NodeYaml) AddNode(nodeID string) (NodeInfo, error) {
}
func (config *NodeYaml) DelNode(nodeID string) error {
if _, ok := config.Nodes[nodeID]; !ok {
return errors.New("Nodename does not exist: " + nodeID)
}
@@ -76,14 +75,14 @@ func (config *NodeYaml) AddProfile(profileID string) (NodeInfo, error) {
}
config.NodeProfiles[profileID] = &node
config.NodeProfiles[profileID].NetDevs = make(map[string]*NetDevs)
n.Id.Set(profileID)
n.NetDevs = make(map[string]*NetDevEntry)
return n, nil
}
func (config *NodeYaml) DelProfile(profileID string) error {
if _, ok := config.NodeProfiles[profileID]; !ok {
return errors.New("Profile does not exist: " + profileID)
}
@@ -131,7 +130,7 @@ func (config *NodeYaml) Persist() error {
if !config.persist {
return err
}
file, err := os.OpenFile(ConfigFile, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0644)
file, err := os.OpenFile(ConfigFile, os.O_RDWR|os.O_CREATE|os.O_TRUNC, 0o644)
if err != nil {
wwlog.Error("%s", err)
os.Exit(1)