feat: create directory before check

Signed-off-by: mlycore <[email protected]>
apache · Dec 18, 2023 · 05739d0 · 05739d0
1 parent 5ea5152
commit 05739d0
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/pitr/agent/internal/handler/diskspace.go b/pitr/agent/internal/handler/diskspace.go
@@ -40,6 +40,10 @@ func DiskSpace(ctx *fiber.Ctx) error {
 		return fmt.Errorf("invalid parameter, err wrap: %w", err)
 	}
 
+	if err := os.MkdirAll(in.DiskPath, 0755); err != nil {
+		return fmt.Errorf("mkdir [%s] failure, err wrap: %w", in.DiskPath, err)
+	}
+
 	// show disk space
 	cmd := fmt.Sprintf("df -h %s", in.DiskPath)
 	output, err := cmds.Exec(os.Getenv("SHELL"), cmd)