modify some file

pkg/validators/validators.go

@@ -3,6 +3,10 @@ package validators
 import (
 	"fmt"
 	"os/exec"
+
+	"jagacloud/node-agent/pkg/compute/libvirt"
+	"jagacloud/node-agent/pkg/config"
+	"jagacloud/node-agent/pkg/containers/lxc"
 )
 
 // CheckBridge returns nil if the bridge exists on the host.
@@ -44,3 +48,33 @@ func CheckStoragePool(name string) error {
 	// TODO: query configured pools
 	return nil
 }
+
+// CheckStoragePoolsVM ensures all disks reference known pools from config.
+func CheckStoragePoolsVM(disks []libvirt.DiskSpec, cfg config.Config) error {
+	for _, d := range disks {
+		if d.Pool == "" {
+			continue
+		}
+		if !poolExists(cfg, d.Pool) {
+			return fmt.Errorf("storage pool %s not configured", d.Pool)
+		}
+	}
+	return nil
+}
+
+// CheckStoragePoolsCT ensures rootfs pool exists.
+func CheckStoragePoolsCT(spec lxc.Spec, cfg config.Config) error {
+	if spec.RootfsPool != "" && !poolExists(cfg, spec.RootfsPool) {
+		return fmt.Errorf("storage pool %s not configured", spec.RootfsPool)
+	}
+	return nil
+}
+
+func poolExists(cfg config.Config, name string) bool {
+	for _, p := range cfg.StoragePools {
+		if p.Name == name {
+			return true
+		}
+	}
+	return false
+}