Created utility copy function and only update overlays if changed

This commit is contained in:
Gregory Kurtzer
2020-11-01 21:19:09 -08:00
parent fe4bcb8bbd
commit b4eb68de61
3 changed files with 52 additions and 27 deletions

View File

@@ -3,7 +3,7 @@ package main
import (
"bufio"
"fmt"
"io"
"github.com/hpcng/warewulf/internal/pkg/util"
"os"
"path/filepath"
"strings"
@@ -60,19 +60,8 @@ func BuildOverlayDir(sourceDir string, destDir string, replace map[string]string
sourceFD.Close()
destFD.Close()
} else {
sourceFD, err := os.Open(sourceDir + "/" + path)
if err != nil {
return err
}
defer sourceFD.Close()
err := util.CopyFile(sourceDir + "/" + path, destDir + "/" + path)
destFD, err := os.OpenFile(destDir + "/" + path, os.O_RDWR|os.O_CREATE, info.Mode())
if err != nil {
return err
}
defer destFD.Close()
_, err = io.Copy(destFD, sourceFD)
if err != nil {
return err
}

View File

@@ -75,7 +75,6 @@ func main(){
fmt.Printf("note: This needs to create an overlay for each node with macro expansions\n")
for _, node := range assets.FindAllNodes() {
fmt.Printf("BUILDING OVERLAY: %s\n", node.Fqdn)
overlayDir := fmt.Sprintf("/etc/warewulf/overlays/%s", node.Overlay)
replace := make(map[string]string)
@@ -93,17 +92,32 @@ func main(){
replace[fmt.Sprintf("%s:GATEWAY", key)] = dev.Gateway
}
overlayDest := "/tmp/.overlay-" + util.RandomString(16)
BuildOverlayDir(overlayDir, overlayDest, replace)
destFile := fmt.Sprintf("%s/provision/overlays/%s.img", LocalStateDir, node.Fqdn)
cmd := fmt.Sprintf("cd %s; find . | cpio --quiet -o -H newc -F \"%s/provision/overlays/%s.img\"", overlayDest, LocalStateDir, node.Fqdn)
err := exec.Command("/bin/sh", "-c", cmd).Run()
if err != nil {
fmt.Printf("%s", err)
destMod, _ := os.Stat(destFile)
destModTime := destMod.ModTime()
configMod, _ := os.Stat("/etc/warewulf/nodes.yaml")
configModTime := configMod.ModTime()
sourceModTime := util.DirModTime(overlayDir)
if sourceModTime.After(destModTime) || configModTime.After(destModTime) {
fmt.Printf("BUILDING OVERLAY: %s\n", node.Fqdn)
overlayDest := "/tmp/.overlay-" + util.RandomString(16)
BuildOverlayDir(overlayDir, overlayDest, replace)
cmd := fmt.Sprintf("cd %s; find . | cpio --quiet -o -H newc -F \"%s\"", overlayDest, destFile)
err := exec.Command("/bin/sh", "-c", cmd).Run()
if err != nil {
fmt.Printf("%s", err)
}
os.RemoveAll(overlayDest)
} else {
fmt.Printf("Skipping overlay (nothing changed): %s\n", node.Fqdn)
}
os.RemoveAll(overlayDest)
}
}
}