1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
|
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 err
}
err = me.cluster.UnmarshalJSON(pfile)
if err != nil {
log.Info("read json failed", err)
os.Exit(-1)
return err
}
// 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() {
test := time.Now().Format("YYYYMMDD-HH-MM")
if !writeConfigFileTmp("virtigo.json.new." + test) {
log.Println("config file write error")
os.Exit(-1)
}
if !writeConfigFileTmp("virtigo.json") {
log.Println("config file write error")
os.Exit(-1)
}
origname := filepath.Join(os.Getenv("VIRTIGO_HOME"), "virtigo.json")
newname := filepath.Join(os.Getenv("VIRTIGO_HOME"), "virtigo.json.old")
err := os.Rename(origname, newname)
if err != nil {
log.Printf("rename fail: %s", err)
os.Exit(-1)
}
}
func writeConfigFileTmp(filename string) bool {
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 false
}
json := me.cluster.FormatJSON()
fmt.Fprintln(cfgfile, json)
log.Info("Write:", fullname, "OK")
return true
}
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")
}
|
