validate code shouldn't be here
Signed-off-by: Jeff Carr <jcarr@wit.com>
This commit is contained in:
parent
255ffc9488
commit
e23c99d771
|
|
@ -308,11 +308,13 @@ func updateNetwork(cluster *pb.Cluster, d *pb.Droplet, domcfg *libvirtxml.Domain
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
if !found {
|
if !found {
|
||||||
if CheckUniqueMac(cluster, mac) {
|
/*
|
||||||
} else {
|
if CheckUniqueMac(cluster, mac) {
|
||||||
log.Info("droplet", d.Hostname, "duplicate mac address", mac)
|
} else {
|
||||||
return nil, errors.New("duplicate mac address")
|
log.Info("droplet", d.Hostname, "duplicate mac address", mac)
|
||||||
}
|
return nil, errors.New("duplicate mac address")
|
||||||
|
}
|
||||||
|
*/
|
||||||
var eth *pb.Network
|
var eth *pb.Network
|
||||||
eth = new(pb.Network)
|
eth = new(pb.Network)
|
||||||
eth.Mac = mac
|
eth.Mac = mac
|
||||||
|
|
@ -360,7 +362,7 @@ func updateDisk(cluster *pb.Cluster, d *pb.Droplet, domcfg *libvirtxml.Domain) (
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
|
|
||||||
e, err := InsertFilename(cluster, d, filename)
|
e, err := InsertFilename(d, filename)
|
||||||
if err != nil {
|
if err != nil {
|
||||||
return alle, err
|
return alle, err
|
||||||
}
|
}
|
||||||
|
|
@ -368,23 +370,6 @@ func updateDisk(cluster *pb.Cluster, d *pb.Droplet, domcfg *libvirtxml.Domain) (
|
||||||
continue
|
continue
|
||||||
}
|
}
|
||||||
alle = append(alle, e)
|
alle = append(alle, e)
|
||||||
|
|
||||||
/*
|
|
||||||
var found bool = false
|
|
||||||
for _, disk := range d.Disks {
|
|
||||||
if disk.Filename == filename {
|
|
||||||
log.Verbose("OKAY. FOUND filename", filename)
|
|
||||||
found = true
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if !found {
|
|
||||||
var disk *pb.Disk
|
|
||||||
disk = new(pb.Disk)
|
|
||||||
disk.Filename = filename
|
|
||||||
d.Disks = append(d.Disks, disk)
|
|
||||||
log.Info("New filename", filename)
|
|
||||||
}
|
|
||||||
*/
|
|
||||||
}
|
}
|
||||||
return alle, nil
|
return alle, nil
|
||||||
}
|
}
|
||||||
|
|
|
||||||
|
|
@ -0,0 +1,37 @@
|
||||||
|
package virtigoxml
|
||||||
|
|
||||||
|
/*
|
||||||
|
makes a droplet hard disk record
|
||||||
|
The full path doesn't matter.
|
||||||
|
We really just care about the qcow2 filename
|
||||||
|
it probably should be forced to be <hostname>.qcow2
|
||||||
|
*/
|
||||||
|
|
||||||
|
import (
|
||||||
|
"path/filepath"
|
||||||
|
|
||||||
|
pb "go.wit.com/lib/protobuf/virtbuf"
|
||||||
|
"go.wit.com/log"
|
||||||
|
)
|
||||||
|
|
||||||
|
func InsertFilename(d *pb.Droplet, filename string) (*pb.Event, error) {
|
||||||
|
filebase := filepath.Base(filename)
|
||||||
|
dir := filepath.Dir(filename)
|
||||||
|
for _, disk := range d.Disks {
|
||||||
|
if disk.Filename == filebase {
|
||||||
|
log.Info("droplet", d.Hostname, "already has this disk", filename)
|
||||||
|
return nil, nil
|
||||||
|
}
|
||||||
|
}
|
||||||
|
// make a new Add Event
|
||||||
|
e := d.NewChangeEvent("Add Disk", "", filename)
|
||||||
|
|
||||||
|
// add the disk protobuf entry
|
||||||
|
var disk *pb.Disk
|
||||||
|
disk = new(pb.Disk)
|
||||||
|
disk.Filename = filebase
|
||||||
|
disk.Filepath = dir
|
||||||
|
d.Disks = append(d.Disks, disk)
|
||||||
|
log.Info("New filename", filebase, dir)
|
||||||
|
return e, nil
|
||||||
|
}
|
||||||
221
validate.go
221
validate.go
|
|
@ -1,221 +0,0 @@
|
||||||
package virtigoxml
|
|
||||||
|
|
||||||
/*
|
|
||||||
validate / sanity check / consistancy check the data
|
|
||||||
|
|
||||||
here is some code to do smart things like:
|
|
||||||
|
|
||||||
* check mac addresses are unique
|
|
||||||
* check uuid's are unique
|
|
||||||
* double check filenames are unique
|
|
||||||
* return a unique mac address
|
|
||||||
* return a unique uuid
|
|
||||||
|
|
||||||
*/
|
|
||||||
|
|
||||||
import (
|
|
||||||
"errors"
|
|
||||||
"os"
|
|
||||||
"path/filepath"
|
|
||||||
|
|
||||||
"github.com/google/uuid"
|
|
||||||
|
|
||||||
pb "go.wit.com/lib/protobuf/virtbuf"
|
|
||||||
"go.wit.com/log"
|
|
||||||
)
|
|
||||||
|
|
||||||
// will make sure the mac address is unique
|
|
||||||
func CheckUniqueMac(cluster *pb.Cluster, mac string) bool {
|
|
||||||
for _, d := range cluster.Droplets {
|
|
||||||
for _, n := range d.Networks {
|
|
||||||
if n.Mac == mac {
|
|
||||||
log.Info("duplicate MAC", n.Mac, "in droplet", d.Hostname)
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return true
|
|
||||||
}
|
|
||||||
|
|
||||||
// records all the known paths. this should go in the protobuf
|
|
||||||
func addClusterFilepath(cluster *pb.Cluster, dir string) *pb.Event {
|
|
||||||
var found bool = false
|
|
||||||
var e *pb.Event
|
|
||||||
for _, d := range cluster.Dirs {
|
|
||||||
if d == dir {
|
|
||||||
// found dir
|
|
||||||
found = true
|
|
||||||
break
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if !found {
|
|
||||||
if dir != "." {
|
|
||||||
// make a new Add Event
|
|
||||||
e = pb.NewAddEvent(nil, "Add Cluster Directory", dir)
|
|
||||||
cluster.Dirs = append(cluster.Dirs, dir)
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return e
|
|
||||||
}
|
|
||||||
|
|
||||||
// returns the droplet using a filename
|
|
||||||
func lookupFilename(cluster *pb.Cluster, filename string) *pb.Droplet {
|
|
||||||
filebase := filepath.Base(filename)
|
|
||||||
|
|
||||||
for _, d := range cluster.Droplets {
|
|
||||||
for _, disk := range d.Disks {
|
|
||||||
if filebase == disk.Filename {
|
|
||||||
return d
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func InsertFilename(cluster *pb.Cluster, d *pb.Droplet, filename string) (*pb.Event, error) {
|
|
||||||
dupd := lookupFilename(cluster, filename)
|
|
||||||
if dupd != nil {
|
|
||||||
log.Info("file", filename, "already on droplet", dupd.Hostname)
|
|
||||||
log.Info("file", filename, "on new droplet", d.Hostname)
|
|
||||||
if os.Getenv("VIRTIGO_IGNORE_DISKDUP") == "" {
|
|
||||||
log.Info("duplicate disk names (--xml-ignore-disk to ignore)")
|
|
||||||
return nil, errors.New("duplicate disk names")
|
|
||||||
} else {
|
|
||||||
log.Info("ignore duplicate disk names (--xml-ignore-disk=true)")
|
|
||||||
}
|
|
||||||
}
|
|
||||||
filebase := filepath.Base(filename)
|
|
||||||
dir := filepath.Dir(filename)
|
|
||||||
for _, disk := range d.Disks {
|
|
||||||
if disk.Filename == filebase {
|
|
||||||
log.Info("already have disk", filename)
|
|
||||||
return nil, nil
|
|
||||||
}
|
|
||||||
}
|
|
||||||
// make a new Add Event
|
|
||||||
e := d.NewChangeEvent("Add Disk", "", filename)
|
|
||||||
|
|
||||||
// add the disk protobuf entry
|
|
||||||
var disk *pb.Disk
|
|
||||||
disk = new(pb.Disk)
|
|
||||||
disk.Filename = filebase
|
|
||||||
disk.Filepath = dir
|
|
||||||
d.Disks = append(d.Disks, disk)
|
|
||||||
log.Info("New filename", filebase, dir)
|
|
||||||
return e, nil
|
|
||||||
}
|
|
||||||
|
|
||||||
func CheckUniqueFilenames(cluster *pb.Cluster) bool {
|
|
||||||
var ok bool = true
|
|
||||||
var disks map[string]string
|
|
||||||
disks = make(map[string]string)
|
|
||||||
|
|
||||||
for _, d := range cluster.Droplets {
|
|
||||||
for _, disk := range d.Disks {
|
|
||||||
filename := disk.Filename
|
|
||||||
addClusterFilepath(cluster, disk.Filepath)
|
|
||||||
if _, ok := disks[filename]; ok {
|
|
||||||
/*
|
|
||||||
if argv.IgnDisk {
|
|
||||||
log.Info("ignore dup disk", filename, disks[filename], d.Hostname)
|
|
||||||
} else {
|
|
||||||
}
|
|
||||||
*/
|
|
||||||
log.Info("file", filename, "on droplet", disks[filename])
|
|
||||||
log.Info("file", filename, "on droplet", d.Hostname)
|
|
||||||
log.Info("duplicate disk names (--xml-ignore-disk to ignore)")
|
|
||||||
ok = false
|
|
||||||
}
|
|
||||||
disks[filename] = d.Hostname
|
|
||||||
}
|
|
||||||
}
|
|
||||||
if ok {
|
|
||||||
log.Println("validated okay: no duplicate disk images")
|
|
||||||
}
|
|
||||||
return ok
|
|
||||||
}
|
|
||||||
|
|
||||||
func CheckDiskFilenames(cluster *pb.Cluster) []*pb.Event {
|
|
||||||
var alle []*pb.Event
|
|
||||||
|
|
||||||
for _, d := range cluster.Droplets {
|
|
||||||
for _, disk := range d.Disks {
|
|
||||||
filename := disk.Filename
|
|
||||||
filebase := filepath.Base(filename)
|
|
||||||
dir := filepath.Dir(filename)
|
|
||||||
addClusterFilepath(cluster, dir)
|
|
||||||
if disk.Filename != filebase {
|
|
||||||
// update filename
|
|
||||||
e := d.NewChangeEvent("Disk.Filename", disk.Filename, filebase)
|
|
||||||
alle = append(alle, e)
|
|
||||||
disk.Filename = filebase
|
|
||||||
}
|
|
||||||
if dir == "." {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if dir == "" {
|
|
||||||
continue
|
|
||||||
}
|
|
||||||
if disk.Filepath != dir {
|
|
||||||
// update filename
|
|
||||||
e := d.NewChangeEvent("Disk.Filepath", disk.Filepath, dir)
|
|
||||||
alle = append(alle, e)
|
|
||||||
disk.Filepath = dir
|
|
||||||
}
|
|
||||||
}
|
|
||||||
}
|
|
||||||
return alle
|
|
||||||
}
|
|
||||||
|
|
||||||
// this doesn't run often
|
|
||||||
func CheckDroplets(cluster *pb.Cluster, dump bool) bool {
|
|
||||||
// uuid map to check for duplicates
|
|
||||||
var umap map[string]string
|
|
||||||
umap = make(map[string]string)
|
|
||||||
|
|
||||||
// mac address map to check for duplicates
|
|
||||||
var macs map[string]string
|
|
||||||
macs = make(map[string]string)
|
|
||||||
|
|
||||||
for _, d := range cluster.Droplets {
|
|
||||||
// Generate a new UUID
|
|
||||||
if d.Uuid == "" {
|
|
||||||
u := uuid.New()
|
|
||||||
d.Uuid = u.String()
|
|
||||||
}
|
|
||||||
|
|
||||||
// seconds, ok := timeZone[tz]; ok {
|
|
||||||
if _, ok := umap[d.Uuid]; ok {
|
|
||||||
// UUID already exists
|
|
||||||
log.Info("duplicate UUID", d.Uuid, umap[d.Uuid])
|
|
||||||
log.Info("duplicate UUID", d.Uuid, d.Hostname)
|
|
||||||
os.Exit(-1)
|
|
||||||
}
|
|
||||||
umap[d.Uuid] = d.Hostname
|
|
||||||
|
|
||||||
for _, n := range d.Networks {
|
|
||||||
// log.Println("network:", n.Mac, d.Uuid, d.Hostname)
|
|
||||||
if _, ok := macs[n.Mac]; ok {
|
|
||||||
// UUID already exists
|
|
||||||
log.Info("duplicate MAC", n.Mac, macs[n.Mac], umap[macs[n.Mac]])
|
|
||||||
log.Info("duplicate MAC", n.Mac, d.Hostname)
|
|
||||||
os.Exit(-1)
|
|
||||||
}
|
|
||||||
macs[n.Mac] = d.Uuid
|
|
||||||
}
|
|
||||||
}
|
|
||||||
log.Println("validated okay: no duplicate MAC addr")
|
|
||||||
log.Println("validated okay: no duplicate UUID")
|
|
||||||
|
|
||||||
if dump {
|
|
||||||
for u, hostname := range umap {
|
|
||||||
log.Println("uuid:", u, "hostname:", hostname)
|
|
||||||
}
|
|
||||||
|
|
||||||
for mac, uuid := range macs {
|
|
||||||
log.Println("mac:", mac, "uuid", uuid, "hostname:", umap[uuid])
|
|
||||||
}
|
|
||||||
}
|
|
||||||
|
|
||||||
return false
|
|
||||||
}
|
|
||||||
Loading…
Reference in New Issue