You can not select more than 25 topics
Topics must start with a letter or number, can include dashes ('-') and can be up to 35 characters long.
104 lines
2.1 KiB
104 lines
2.1 KiB
package main
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"fmt"
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"path/filepath"
|
|
"sync"
|
|
"time"
|
|
)
|
|
|
|
func downloadImages(images map[string]string, dir string) (map[string]bool, map[string]error) {
|
|
var mu sync.Mutex
|
|
var wg sync.WaitGroup
|
|
|
|
success := make(map[string]bool)
|
|
errors := make(map[string]error)
|
|
|
|
// 并发控制：最多同时下载5张图片
|
|
semaphore := make(chan struct{}, 5)
|
|
|
|
for key, url := range images {
|
|
wg.Add(1)
|
|
|
|
go func(key, url string) {
|
|
defer wg.Done()
|
|
|
|
// 获取信号量
|
|
semaphore <- struct{}{}
|
|
defer func() { <-semaphore }()
|
|
|
|
// 下载图片
|
|
err := downloadImage(url, dir, key)
|
|
|
|
mu.Lock()
|
|
if err != nil {
|
|
errors[key] = err
|
|
} else {
|
|
success[key] = true
|
|
}
|
|
mu.Unlock()
|
|
}(key, url)
|
|
}
|
|
|
|
wg.Wait()
|
|
return success, errors
|
|
}
|
|
|
|
func downloadImage(url, dir, filename string) error {
|
|
// 创建HTTP客户端
|
|
client := &http.Client{
|
|
Timeout: 30 * time.Second,
|
|
Transport: &http.Transport{
|
|
TLSClientConfig: &tls.Config{InsecureSkipVerify: true},
|
|
},
|
|
}
|
|
|
|
// 创建请求
|
|
req, err := http.NewRequest("GET", url, nil)
|
|
if err != nil {
|
|
return fmt.Errorf("创建请求失败: %v", err)
|
|
}
|
|
|
|
// 设置请求头（模拟浏览器）
|
|
req.Header.Set("User-Agent", "Mozilla/5.0 (Windows NT 10.0; Win64; x64) AppleWebKit/537.36")
|
|
req.Header.Set("Accept", "image/webp,image/apng,image/*,*/*;q=0.8")
|
|
req.Header.Set("Accept-Language", "zh-CN,zh;q=0.9,en;q=0.8")
|
|
req.Header.Set("Referer", "https://hd4k.com/")
|
|
|
|
// 发送请求
|
|
resp, err := client.Do(req)
|
|
if err != nil {
|
|
return fmt.Errorf("请求失败: %v", err)
|
|
}
|
|
defer resp.Body.Close()
|
|
|
|
if resp.StatusCode != http.StatusOK {
|
|
return fmt.Errorf("HTTP错误: %s", resp.Status)
|
|
}
|
|
|
|
// 获取文件扩展名
|
|
ext := getFileExtension(url)
|
|
fullFilename := filename + ext
|
|
filePath := filepath.Join(dir, fullFilename)
|
|
|
|
// 创建文件
|
|
file, err := os.Create(filePath)
|
|
if err != nil {
|
|
return fmt.Errorf("创建文件失败: %v", err)
|
|
}
|
|
defer file.Close()
|
|
|
|
// 下载并保存
|
|
_, err = io.Copy(file, resp.Body)
|
|
if err != nil {
|
|
// 删除可能已损坏的文件
|
|
os.Remove(filePath)
|
|
return fmt.Errorf("保存文件失败: %v", err)
|
|
}
|
|
|
|
return nil
|
|
}
|
|
|