자동화 강화: Golang의 비밀 무기

package main

import (
    "fmt"
    "io"
    "net/http"
    "os"
)

func main() {
    // 定义需要下载的文件URL
    urls := []string{"url1", "url2", "url3"}

    // 创建一个工作队列
    jobs := make(chan string)

    // 创建消费者Goroutine来下载文件
    for i := 0; i < 10; i++ {
        go func() {
            for {
                url := <-jobs
                downloadFile(url)
            }
        }()
    }

    // 将URL放入工作队列
    for _, url := range urls {
        jobs <- url
    }

    // 等待所有文件下载完成
    for len(jobs) > 0 {
        time.Sleep(100 * time.Millisecond)
    }
}

// downloadFile从URL下载文件
func downloadFile(url string) {
    resp, err := http.Get(url)
    if err != nil {
        fmt.Println(err)
        return
    }

    defer resp.Body.Close()

    filename := url[strings.LastIndexByte(url, '/')+1:]
    file, err := os.Create(filename)
    if err != nil {
        fmt.Println(err)
        return
    }

    defer file.Close()

    if _, err := io.Copy(file, resp.Body); err != nil {
        fmt.Println(err)
        return
    }

    fmt.Printf("Downloaded %s successfully\n", filename)
}

go run downloadFiles.go