Add disk space check to deploy

bravetools · Oct 5, 2022 · ab8809a · ab8809a
1 parent f3a7403
commit ab8809a
Showing 1 changed file with 10 additions and 0 deletions.
diff --git a/platform/host_api.go b/platform/host_api.go
@@ -1087,9 +1087,19 @@ func (bh *BraveHost) InitUnit(backend Backend, unitParams shared.Service) (err e
 	if err != nil {
 		return fmt.Errorf("failed to obtain image hash %q", unitParams.Image)
 	}
+	imgSize, err := localImageSize(imageStruct)
+	if err != nil {
+		return fmt.Errorf("failed to get image size for image %q", imageStruct.String())
+	}
 	defer DeleteImageByFingerprint(lxdServer, fingerprint)
 
 	// Resource checks
+	if unitParams.Storage != "" {
+		err = CheckStoragePoolSpace(lxdServer, unitParams.Storage, imgSize)
+		if err != nil {
+			return err
+		}
+	}
 	err = CheckMemory(lxdServer, unitParams.Resources.RAM)
 	if err != nil {
 		return err