用 Go + Cobra 写了个轻量 CLI，搞定前端静态资源发布与 CDN 预热

# 扫描 dist 目录，上传到指定 bucket，并自动预热关键资源
shipit deploy --dir ./dist --bucket my-web-assets --warmup

mkdir shipit && cd shipit
go mod init shipit
go get -u github.com/spf13/cobra@latest

package cmd

import (
	"fmt"
	"os"

	"github.com/spf13/cobra"
)

var rootCmd = &cobra.Command{
	Use:   "shipit",
	Short: "前端静态资源发布与 CDN 预热工具",
	Long:  `一个专为前端构建流水线设计的静态文件部署工具，支持并发上传与自动化 CDN 预热。`,
}

func Execute() {
	if err := rootCmd.Execute(); err != nil {
		fmt.Println(err)
		os.Exit(1)
	}
}

package cmd

import (
	"fmt"
	"path/filepath"
	"strings"
	"sync"
	"time"

	"github.com/spf13/cobra"
)

var (
	distDir     string
	bucket      string
	concurrency int
	enableWarm  bool
)

func init() {
	deployCmd.Flags().StringVarP(&distDir, "dir", "d", "./dist", "待上传的构建目录")
	deployCmd.Flags().StringVarP(&bucket, "bucket", "b", "", "对象存储 Bucket 名称")
	deployCmd.Flags().IntVarP(&concurrency, "concurrency", "c", 10, "并发上传数量")
	deployCmd.Flags().BoolVar(&enableWarm, "warmup", true, "是否自动预热首屏关键资源")

	_ = deployCmd.MarkFlagRequired("bucket")
	rootCmd.AddCommand(deployCmd)
}

var deployCmd = &cobra.Command{
	Use:   "deploy",
	Short: "部署静态资源到 OSS 并触发 CDN 预热",
	Run: func(cmd *cobra.Command, args []string) {
		start := time.Now()
		fmt.Printf("开始部署目录: %s 到 Bucket: %s\n", distDir, bucket)

		// 1. 扫描文件
		files, err := scanFiles(distDir)
		if err != nil {
			fmt.Printf("扫描文件失败: %v\n", err)
			return
		}
		fmt.Printf("共发现 %d 个静态文件\n", len(files))

		// 2. 并发上传
		uploadAssets(files, concurrency)

		// 3. 预热处理
		if enableWarm {
			warmupKeyAssets(files)
		}

		fmt.Printf("部署完成，耗时: %v\n", time.Since(start))
	},
}

func scanFiles(root string) ([]string, error) {
	var files []string
	err := filepath.Walk(root, func(path string, info os.FileInfo, err error) error {
		if err != nil {
			return err
		}
		if !info.IsDir() {
			files = append(files, path)
		}
		return nil
	})
	return files, err
}

func uploadAssets(files []string, workers int) {
	jobs := make(chan string, len(files))
	var wg sync.WaitGroup

	// 启动 Worker
	for w := 1; w <= workers; w++ {
		wg.Add(1)
		go func(workerId int) {
			defer wg.Done()
			for file := range jobs {
				// 模拟调用对象存储 SDK 上传文件
				uploadSingleFile(file)
			}
		}(w)
	}

	// 派发任务
	for _, file := range files {
		jobs <- file
	}
	close(jobs)

	wg.Wait()
}

func uploadSingleFile(filePath string) {
	// 这里接入实际的阿里云 OSS / 腾讯云 COS / AWS S3 SDK
	// 针对 .html 设置 Cache-Control: no-cache
	// 针对 带 hash 的 js/css 设置 Cache-Control: max-age=31536000, immutable
	time.Sleep(50 * time.Millisecond) // 模拟网络延迟
	fmt.Printf("[UPLOAD] 成功: %s\n", filePath)
}

func warmupKeyAssets(files []string) {
	var warmupList []string
	var purgeList []string

	for _, file := range files {
		ext := filepath.Ext(file)
		// 找出需要刷新的入口 HTML
		if ext == ".html" {
			purgeList = append(purgeList, file)
		}
		// 找出需要预热的核心静态资源（过滤掉小图标等非关键资源）
		if ext == ".js" || ext == ".css" {
			if strings.Contains(file, "main") || strings.Contains(file, "vendor") {
				warmupList = append(warmupList, file)
			}
		}
	}

	fmt.Printf("\n[CDN] 正在刷新入口文件 (%d 个)...\n", len(purgeList))
	// mockCDNCall("PURGE", purgeList)

	fmt.Printf("[CDN] 正在预热核心资源 (%d 个)...\n", len(warmupList))
	// mockCDNCall("WARMUP", warmupList)
}

package main

import "shipit/cmd"

func main() {
	cmd.Execute()
}

# 编译给 Linux CI Runner 用
CGO_ENABLED=0 GOOS=linux GOARCH=amd64 go build -ldflags="-s -w" -o shipit-linux-amd64 main.go

# 编译给公司 M 系列芯片的 Mac 机器用
CGO_ENABLED=0 GOOS=darwin GOARCH=arm64 go build -ldflags="-s -w" -o shipit-darwin-arm64 main.go

# .gitlab-ci.yml 示意
deploy_job:
  stage: deploy
  script:
    - pnpm build
    - ./bin/shipit-linux-amd64 deploy --dir ./dist --bucket my-prod-bucket --concurrency 20