fix(status): fix divide by zero (#136)

leptonai · Oct 26, 2024 · 6f30ce1 · 6f30ce1
1 parent 1a0ff53
commit 6f30ce1
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 2 deletions.
diff --git a/cmd/gpud/command/join.go b/cmd/gpud/command/join.go
@@ -53,7 +53,7 @@ func cmdJoin(cliContext *cli.Context) (retErr error) {
 		return fmt.Errorf("failed to get machine uid: %w", err)
 	}
 
-	cmd := exec.Command("nproc")
+	cmd := exec.Command("nproc", "--all")
 	var out bytes.Buffer
 	cmd.Stdout = &out
 	if err = cmd.Run(); err != nil {

diff --git a/cmd/gpud/command/status.go b/cmd/gpud/command/status.go
@@ -48,6 +48,10 @@ func cmdStatus(cliContext *cli.Context) error {
 			fmt.Printf("%s failed to get package status: %v\n", warningSign, err)
 			return err
 		}
+		if len(packageStatus) == 0 {
+			fmt.Printf("no packages found\n")
+			return nil
+		}
 		if statusWatch {
 			fmt.Print("\033[2J\033[H")
 		}
@@ -58,7 +62,11 @@ func cmdStatus(cliContext *cli.Context) error {
 			progress += status.TotalTime.Milliseconds() * int64(status.Progress) / 100
 		}
 
-		fmt.Printf("Total progress: %v%%, Estimate time left: %v\n", progress*100/totalTime, time.Duration(totalTime-progress)*time.Millisecond)
+		var totalProgress int64
+		if totalTime != 0 {
+			totalProgress = progress * 100 / totalTime
+		}
+		fmt.Printf("Total progress: %v%%, Estimate time left: %v\n", totalProgress, time.Duration(totalTime-progress)*time.Millisecond)
 		if !statusWatch {
 			break
 		}