Created utility copy function and only update overlays if changed
This commit is contained in:
@@ -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
|
||||
}
|
||||
|
||||
@@ -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)
|
||||
|
||||
}
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user