如何用Golang和goroutine并发处理一个目录下的所有图片文件

func isImageFile(filename string) bool {
    ext := strings.ToLower(filepath.Ext(filename))
    return ext == ".jpg" || ext == ".jpeg" || ext == ".png" || ext == ".gif" || ext == ".bmp" || ext == ".webp"
}
<p>func getImagesFromDir(dirPath string) ([]string, error) {
var imageFiles []string
entries, err := os.ReadDir(dirPath)
if err != nil {
return nil, err
}</p><pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">for _, entry := range entries {
    if !entry.IsDir() && isImageFile(entry.Name()) {
        imageFiles = append(imageFiles, filepath.Join(dirPath, entry.Name()))
    }
}
return imageFiles, nil

func processImagesConcurrently(imageFiles []string, workerCount int) {
    var wg sync.WaitGroup
    sem := make(chan struct{}, workerCount) // 控制并发数
<pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">for _, file := range imageFiles {
    wg.Add(1)
    go func(filePath string) {
        defer wg.Done()
        sem <- struct{}{}        // 获取信号
        defer func() { <-sem }() // 释放信号

        // 模拟图片处理（如压缩、加水印、转格式等）
        fmt.Printf("处理图片: %s\n", filePath)
        time.Sleep(500 * time.Millisecond) // 模拟耗时操作
        // 实际处理逻辑可调用 image.Decode 或第三方库
    }(file)
}

wg.Wait() // 等待所有任务完成

package main
<p>import (
"fmt"
"os"
"path/filepath"
"strings"
"sync"
"time"
)</p><p>func main() {
dirPath := "." // 默认当前目录，可改为 os.Args[1] 接收参数
if len(os.Args) > 1 {
dirPath = os.Args[1]
}</p><pre class="brush:php;toolbar:false;"><pre class="brush:php;toolbar:false;">imageFiles, err := getImagesFromDir(dirPath)
if err != nil {
    fmt.Fprintf(os.Stderr, "读取目录失败: %v\n", err)
    return
}

if len(imageFiles) == 0 {
    fmt.Println("未找到图片文件")
    return
}

fmt.Printf("找到 %d 个图片文件，开始并发处理...\n", len(imageFiles))

start := time.Now()
processImagesConcurrently(imageFiles, 5) // 最多5个并发
fmt.Printf("处理完成，耗时: %v\n", time.Since(start))