feat: 媒体机直传 + 向量回填

- media: spider 下载图片后按内容 sha1 生成 key,经后端预签名直传 S4,上报只带 key
  (spider 不持有 AK/SK;单张失败自动回退为上报 URL)
- embed: 新增 embed 子命令,从后端领取缺向量的图片→下载→本地 CLIP 编码→回传
  (媒体拆分后后端拿不到图片字节,向量只能靠回填补齐)
- serve: 服务化常驻模式(领取 crawl 任务 → 抓取 → 回报)
- config: 新增 media.upload/concurrency、vision.endpoint/concurrency
This commit is contained in:
toom1996
2026-10-25 00:15:22 +08:00
parent 6020426815
commit 3fe0b86e11
11 changed files with 1305 additions and 2 deletions

113
cmd/embed.go Normal file
View File

@ -0,0 +1,113 @@
package cmd
import (
"context"
"log"
"net/http"
"strings"
"time"
"github.com/spf13/cobra"
"my-spiders/internal/config"
"my-spiders/internal/embed"
)
var (
embedKinds string
embedLimit int
embedOnce bool
embedSleep int
embedVision string
embedCo int
)
var embedCmd = &cobra.Command{
Use: "embed",
Short: "向量回填:拉取缺向量的图片,本地 CLIP 编码后回传后端",
Long: `媒体拆分后由 spider 直接传图,后端拿不到图片字节、算不出向量。
本命令跑在有 GPU 的机器上,把缺向量的图补上:
后端领取(缺向量的图) → 从存储下载 → 本地 CLIP 编码 → 回传向量
常驻运行(默认):spider embed
只跑一轮: spider embed --once`,
Run: func(cmd *cobra.Command, args []string) {
if IngestClient == nil {
log.Fatal("[embed] 入库管线未启用(config.ingest.enabled=false),无法领取回填任务")
}
vision := embedVision
if vision == "" {
vision = config.GlobalConfig.Vision.Endpoint
}
if vision == "" {
vision = "http://127.0.0.1:8900"
}
// 视觉服务没起来就白跑:先探活,明确报错比反复超时好排查。
if err := visionReady(vision); err != nil {
log.Fatalf("[embed] 视觉服务不可用(%s): %v\n 先把 CLIP 服务起来:app-cli\\vision\\vision.ps1 start", vision, err)
}
kinds := []string{"runway", "street"}
if strings.TrimSpace(embedKinds) != "" {
kinds = nil
for _, k := range strings.Split(embedKinds, ",") {
if k = strings.TrimSpace(k); k != "" {
kinds = append(kinds, k)
}
}
}
conc := embedCo
if conc <= 0 {
conc = config.GlobalConfig.Vision.Concurrency
}
w := embed.New(IngestClient, vision, conc, 8)
log.Printf("[embed] 启动:视觉=%s 类别=%v 单批=%d 并发=%d", vision, kinds, embedLimit, conc)
ctx := context.Background()
for {
idle := 0
for _, kind := range kinds {
st, err := w.RunOnce(ctx, kind, embedLimit)
if err != nil {
log.Printf("[embed] %s 本轮失败: %v", kind, err)
continue
}
if st.Fetched == 0 {
idle++
}
}
if embedOnce {
return
}
if idle == len(kinds) {
// 没有待办:睡一会儿再看,避免空转打后端
time.Sleep(time.Duration(embedSleep) * time.Second)
}
}
},
}
// visionReady 探活视觉服务(/health),失败返回错误。
func visionReady(endpoint string) error {
client := &http.Client{Timeout: 5 * time.Second}
resp, err := client.Get(endpoint + "/health")
if err != nil {
return err
}
defer resp.Body.Close()
if resp.StatusCode != http.StatusOK {
return err
}
return nil
}
func init() {
rootCmd.AddCommand(embedCmd)
embedCmd.Flags().StringVar(&embedKinds, "kind", "", "只处理指定类别(runway,street),默认两类都做")
embedCmd.Flags().IntVar(&embedLimit, "limit", 32, "每轮向后端领取的张数(1~200)")
embedCmd.Flags().BoolVar(&embedOnce, "once", false, "只跑一轮就退出(用于验证/定时任务)")
embedCmd.Flags().IntVar(&embedSleep, "sleep", 30, "空闲时休眠秒数(常驻模式)")
embedCmd.Flags().StringVar(&embedVision, "vision", "", "视觉服务地址(默认取 config.vision.endpoint,再兜底 127.0.0.1:8900)")
embedCmd.Flags().IntVar(&embedCo, "concurrency", 0, "下载并发(默认取 config.vision.concurrency)")
}