修复更新下载失败重试

Author: orange-juzipi

cmd/update.go

@@ -8,6 +8,7 @@ import (
 	"path/filepath"
 	"time"
 
+	"github.com/https-cert/deploy/internal/config"
 	"github.com/https-cert/deploy/internal/updater"
 	"github.com/spf13/cobra"
 )
@@ -21,6 +22,10 @@ func CreateCheckUpdateCmd() *cobra.Command {
 		RunE: func(cmd *cobra.Command, args []string) error {
 			ctx := context.Background()
 
+			if err := config.Init(ConfigFile); err != nil {
+				return fmt.Errorf("初始化配置失败: %w", err)
+			}
+
 			fmt.Println("正在检查更新...")
 			info, err := updater.CheckUpdate(ctx)
 			if err != nil {
@@ -50,6 +55,10 @@ func CreateUpdateCmd() *cobra.Command {
 		RunE: func(cmd *cobra.Command, args []string) error {
 			ctx := context.Background()
 
+			if err := config.Init(ConfigFile); err != nil {
+				return fmt.Errorf("初始化配置失败: %w", err)
+			}
+
 			// 获取当前可执行文件的真实路径
 			execPath, err := os.Executable()
 			if err != nil {

internal/updater/updater.go

@@ -28,6 +28,7 @@ const (
 	githubRepo      = "https-cert/deploy"
 	githubAPIURL    = "https://api.github.com/repos/" + githubRepo + "/releases/latest"
 	downloadTimeout = 10 * time.Minute
+	downloadRetries = 3
 )
 
 // 常见的 GitHub 镜像加速服务
@@ -436,6 +437,27 @@ func getBinaryName() string {
 
 // downloadFile 下载文件
 func downloadFile(ctx context.Context, downloadURL, filepath string) error {
+	var lastErr error
+
+	for attempt := 1; attempt <= downloadRetries; attempt++ {
+		if attempt > 1 {
+			time.Sleep(time.Duration(attempt-1) * time.Second)
+			os.Remove(filepath)
+		}
+
+		if err := downloadFileOnce(ctx, downloadURL, filepath); err != nil {
+			lastErr = err
+			logger.Warn("下载失败，准备重试", "attempt", attempt, "max", downloadRetries, "error", err)
+			continue
+		}
+
+		return nil
+	}
+
+	return lastErr
+}
+
+func downloadFileOnce(ctx context.Context, downloadURL, filepath string) error {
 	ctx, cancel := context.WithTimeout(ctx, downloadTimeout)
 	defer cancel()