package main import ( "errors" "fmt" "os" "path/filepath" "time" pb "go.wit.com/lib/protobuf/virtbuf" "go.wit.com/log" ) var ErrorNoFile error = errors.New("missing file") var ErrorParseJSON error = errors.New("invalid json") var ErrorParseXML error = errors.New("invalid xml") // something is wrong somewhere and sometimes the // protobuf json files get written out with garbage func cfgfile() { err := readConfigFile("virtigo.json") if err == nil { return } if err == ErrorParseJSON { os.Exit(-1) } err = readConfigFile("virtigo.json.last") if err == nil { log.Info("read json failed", err) os.Exit(-1) } if err == ErrorNoFile { log.Info("no config file created yet", err) os.Exit(-1) } } func readConfigFile(filename string) error { me.cluster = new(pb.Cluster) fullname := filepath.Join(os.Getenv("VIRTIGO_HOME"), filename) pfile, err := os.ReadFile(fullname) if err != nil { log.Info("open config file :", err) return ErrorNoFile } err = me.cluster.UnmarshalJSON(pfile) if err != nil { log.Info("read json failed", err) os.Exit(-1) return ErrorParseJSON } // initialize each hypervisor for _, pbh := range me.cluster.Hypervisors { h := findHypervisor(pbh.Hostname) if h != nil { continue } // this is a new unknown droplet (not in the config file) h = new(HyperT) h.pb = pbh h.lastpoll = time.Now() me.hypers = append(me.hypers, h) log.Log(EVENT, "config new hypervisors", h.pb.Hostname) } // initialize values for each droplet for _, pbd := range me.cluster.Droplets { d := findDroplet(pbd.Hostname) if d != nil { continue } // this is a new unknown droplet (not in the config file) d = new(DropletT) d.pb = pbd me.droplets = append(me.droplets, d) log.Log(EVENT, "config new droplet", d.pb.Hostname, d.pb.StartState, d.pb.PreferredHypervisor) } return nil } func writeConfigFile() { fullname := filepath.Join(os.Getenv("VIRTIGO_HOME"), "virtigo.json") cfgfile, err := os.OpenFile(fullname, os.O_RDWR|os.O_CREATE, 0666) defer cfgfile.Close() if err != nil { log.Info("open config file :", err) return } json := me.cluster.FormatJSON() fmt.Fprintln(cfgfile, json) log.Info("Write:", fullname, "OK") } func writeConfigFileDroplets() { fullname := filepath.Join(os.Getenv("VIRTIGO_HOME"), "droplets.text") cfgfile, err := os.OpenFile(fullname, os.O_RDWR|os.O_CREATE, 0666) defer cfgfile.Close() if err != nil { log.Info("open config file :", err) return } // text := me.cluster.Droplets.FormatTEXT() text := me.cluster.FormatTEXT() fmt.Fprintln(cfgfile, text) log.Info("Write:", fullname, "OK") }