perf: 修 spider 内存暴涨(整场图入内存 + 场次无限并发)

线上现象:spider 进程 RSS 1.3GB(33% 内存),单机 3.7G 只剩 141MB 可用,
服务器 1.5 小时内重启两次。

- media.UploadAll 改为按批(8 张)下载→上传→立即释放;
  原来先把整场 97 张(约 116MB)全读进内存,多场并发时线性叠加
- SpiderStart 对同品牌场次加 2 路并发上限(原来对全部场次无上限 go 出去)
- systemd 单元加 GOMEMLIMIT=700MiB / MemoryHigh=800M / MemoryMax=1200M 硬护栏,
  即使再退化也只 OOM 掉这一个进程,不会拖死整机
This commit is contained in:
toom1996
2026-10-25 18:01:46 +08:00
parent 3fe0b86e11
commit 6a8b7f0704
2 changed files with 72 additions and 45 deletions

View File

@ -63,41 +63,62 @@ func (u *Uploader) UploadAll(ctx context.Context, urls []string) []Result {
return results return results
} }
// ① 并发下载 // 分批处理:**不能**先把整场的图全读进内存。
data := make([][]byte, len(urls)) // 一场 100 张 × 1.2MB ≈ 120MB;而 SpiderStart 对同一品牌的多个场次是并发抓的,
errs := make([]error, len(urls)) // 峰值会翻好几倍——线上实测 spider 进程一度涨到 1.3GB,把小机器(3.7G)压到只剩 141MB 可用。
u.parallel(len(urls), func(i int) { // 按批(默认 8 张)下载→上传→立即释放,峰值稳定在 ~10MB。
body, err := u.download(ctx, urls[i]) const batchSize = 8
for start := 0; start < len(urls); start += batchSize {
end := start + batchSize
if end > len(urls) {
end = len(urls)
}
u.uploadBatch(ctx, urls, results, start, end)
}
return results
}
// uploadBatch 处理 urls[start:end] 这一段:下载 → 生成 key → 预签名 → PUT。
// 返回时本批的图片字节已可回收(不再被引用)。
func (u *Uploader) uploadBatch(ctx context.Context, urls []string, results []Result, start, end int) {
n := end - start
data := make([][]byte, n)
errs := make([]error, n)
// ① 并发下载本批
u.parallel(n, func(i int) {
body, err := u.download(ctx, urls[start+i])
if err != nil { if err != nil {
errs[i] = err errs[i] = err
return return
} }
data[i] = body data[i] = body
}) })
log.Printf("[media] 下载完成 %d 张(失败 %d)", countOK(data), len(urls)-countOK(data)) log.Printf("[media] 下载完成 %d 张(失败 %d)", countOK(data), n-countOK(data))
// ② 生成本批 key(内容 sha1 + 原扩展名) // ② 生成本批 key(内容 sha1 + 原扩展名)
for i := range urls { for i := 0; i < n; i++ {
if errs[i] != nil { if errs[i] != nil {
results[i] = Result{Err: errs[i]} results[start+i] = Result{Err: errs[i]}
continue continue
} }
results[i] = Result{ results[start+i] = Result{
Key: u.Prefix + "/" + sha1Hex(data[i]) + extOf(urls[i]), Key: u.Prefix + "/" + sha1Hex(data[i]) + extOf(urls[start+i]),
Size: int64(len(data[i])), Size: int64(len(data[i])),
} }
} }
// ③ 批量申请预签名地址(同一张图可能在同一场里出现多次,按 key 去重) // ③ 申请预签名(同一张图可能在一场里重复出现,按 key 去重)
want := map[string]ingest.UploadItem{} want := make(map[string]ingest.UploadItem, n)
for i := range urls { for i := 0; i < n; i++ {
if results[i].Key == "" { key := results[start+i].Key
if key == "" {
continue continue
} }
if _, ok := want[results[i].Key]; ok { if _, ok := want[key]; ok {
continue continue
} }
want[results[i].Key] = ingest.UploadItem{Key: results[i].Key, ContentType: contentTypeOf(urls[i])} want[key] = ingest.UploadItem{Key: key, ContentType: contentTypeOf(urls[start+i])}
} }
items := make([]ingest.UploadItem, 0, len(want)) items := make([]ingest.UploadItem, 0, len(want))
for _, it := range want { for _, it := range want {
@ -105,44 +126,37 @@ func (u *Uploader) UploadAll(ctx context.Context, urls []string) []Result {
} }
presigned := make(map[string]string, len(items)) presigned := make(map[string]string, len(items))
const chunk = 100 // 后端单次上限 200,这里留一半余量 got, err := u.Client.RequestUploadURLs(ctx, items)
for start := 0; start < len(items); start += chunk {
end := start + chunk
if end > len(items) {
end = len(items)
}
got, err := u.Client.RequestUploadURLs(ctx, items[start:end])
if err != nil { if err != nil {
// 申请失败:整批回退成 URL 上报(后端自己下载),不中断整场抓取。 // 申请失败:本批回退成 URL 上报(后端自己下载),不中断整场抓取。
for i := range results { for i := 0; i < n; i++ {
if results[i].Key != "" { if results[start+i].Key != "" {
results[i] = Result{Err: fmt.Errorf("申请上传地址失败: %w", err)} results[start+i] = Result{Err: fmt.Errorf("申请上传地址失败: %w", err)}
} }
} }
return results return
} }
for _, g := range got { for _, g := range got {
presigned[g.Key] = g.URL presigned[g.Key] = g.URL
} }
}
log.Printf("[media] 已拿到 %d 个预签名地址,开始上传", len(presigned)) // ④ 并发 PUT 本批
u.parallel(n, func(i int) {
// ④ 并发 PUT(内容已在内存里,直接传字节) res := results[start+i]
u.parallel(len(urls), func(i int) { if res.Key == "" {
if results[i].Key == "" {
return return
} }
putURL, ok := presigned[results[i].Key] putURL, ok := presigned[res.Key]
if !ok { if !ok {
results[i] = Result{Err: fmt.Errorf("未拿到 %s 的上传地址", results[i].Key)} results[start+i] = Result{Err: fmt.Errorf("未拿到 %s 的上传地址", res.Key)}
return return
} }
if err := u.put(ctx, putURL, urls[i], data[i]); err != nil { if perr := u.put(ctx, putURL, urls[start+i], data[i]); perr != nil {
results[i] = Result{Err: err} results[start+i] = Result{Err: perr}
} }
}) })
return results
log.Printf("[media] 本批 %d 张上传完成(%d/%d)", n, end, len(urls))
} }
// parallel 以 Concurrency 路并发执行 fn(0..n-1)。 // parallel 以 Concurrency 路并发执行 fn(0..n-1)。

View File

@ -79,6 +79,12 @@ func (s *VogueSpider) RunBrand(brandUID, brandName string, maxShows int) int {
return int(atomic.LoadInt32(&s.submitted)) return int(atomic.LoadInt32(&s.submitted))
} }
// maxShowConcurrency 同一品牌内同时抓取的场次数。
//
// 调大能更快跑完一个品牌,但每场都会占用一批图片内存(媒体直传按批读入),
// 实测无上限并发会把进程顶到 GB 级;2 路在带宽与内存之间比较平衡。
const maxShowConcurrency = 2
// uploadLookMedia 把一批 look 的图片直传到对象存储,回填 main_key / detail_keys。 // uploadLookMedia 把一批 look 的图片直传到对象存储,回填 main_key / detail_keys。
// //
// 返回 true 表示全部成功;false 表示有失败项 —— 失败的图仍保留原始 URL, // 返回 true 表示全部成功;false 表示有失败项 —— 失败的图仍保留原始 URL,
@ -226,15 +232,22 @@ func (s *VogueSpider) SpiderStart(task ingest.CrawlBrand) {
showsList = showsList[:s.MaxShows] showsList = showsList[:s.MaxShows]
} }
// 详情页并发抓取:这里不再自建信号量,HTTP 并发与频控统一由 s.Fetcher 承担 // 详情页并发抓取:**限制同时抓几场**。
// (避免与品牌级信号量叠加导致并发放大)。 //
// 一个品牌可能几十场(Acne Studios 70+),每场的图片在媒体直传时按批驻留内存。
// 早先这里对全部场次无上限并发,实测会把 spider 进程顶到 1.3GB,
// 在 3.7G 的小机器上直接把系统压到只剩百来兆可用内存 → 整机卡死重启。
// 2 路并发足够喂满带宽(单场内部还有 6 路图片并发),内存峰值可控。
var wg sync.WaitGroup var wg sync.WaitGroup
showSem := make(chan struct{}, maxShowConcurrency)
for _, list := range showsList { for _, list := range showsList {
wg.Add(1) wg.Add(1)
go func(info gjson.Result) { go func(info gjson.Result) {
defer wg.Done() defer wg.Done()
showSem <- struct{}{}
defer func() { <-showSem }()
s.getDetail(task.BrandUID, info) s.getDetail(task.BrandUID, info)
}(list) }(list)
} }