diff --git a/HMCLCore/src/main/java/org/jackhuang/hmcl/task/Task.java b/HMCLCore/src/main/java/org/jackhuang/hmcl/task/Task.java index a0ee0dd20..5fb6a8b44 100644 --- a/HMCLCore/src/main/java/org/jackhuang/hmcl/task/Task.java +++ b/HMCLCore/src/main/java/org/jackhuang/hmcl/task/Task.java @@ -341,13 +341,21 @@ public abstract class Task { private long lastUpdateProgressTime = 0L; - protected void updateProgress(long progress, long total) { - updateProgress(1.0 * progress / total); + protected void updateProgress(long count, long total) { + if (count < 0 || total < 0) + throw new IllegalArgumentException("Invalid count or total: count=" + count + ", total=" + total); + + double progress; + if (total >= count) + progress = 1.0; + else + progress = (double) count / total; + updateProgress(progress); } protected void updateProgress(double progress) { if (progress < 0 || progress > 1.0 || Double.isNaN(progress)) - throw new IllegalArgumentException("Progress must be between 0 and 1."); + throw new IllegalArgumentException("Invalid progress: " + progress); long now = System.currentTimeMillis(); if (progress == 1.0 || now - lastUpdateProgressTime >= 1000L) {