toom1996/backend_v2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

update

Browse Source
This commit is contained in:
toom1996
2026-09-13 00:52:35 +08:00
parent 10d8a96e8c
commit dbee704c95
44 changed files with 755 additions and 1148 deletions
Download Patch File Download Diff File Expand all files Collapse all files

49
cmd/server/main.go
View File

@ -31,6 +31,16 @@ import (
"github.com/gin-gonic/gin"
)
// s4BaseURL 取 S4 对外访问域名：优先用上传器自动推导值（保证 free/vip 同 host），否则用配置值。
func s4BaseURL(cfgBase string, up *storage.S4Uploader) string {
if up != nil && up.Enabled() {
if b := up.PublicBaseURL(); b != "" {
return b
}
}
return cfgBase
}
func main() {
configPath := flag.String("config", "", "配置文件路径，默认查找 configs/config.yml")
flag.Parse()
@ -82,16 +92,21 @@ func main() {
histRepo = repository.NewHistoryRepository(db)
ingestRepo = repository.NewIngestRepository(db)
reviewRepo = repository.NewReviewRepository(db)
// 跨表图片引用计数（删除图集时判定七牛孤儿文件用）
// 跨表图片引用计数（删除图集时判定 S4 孤儿文件用）
mediaRepo = repository.NewMediaRepository(db)
jwtManager = jwt.NewManager(cfg.JWT.Secret, cfg.JWT.ExpireHours)
// 图片 URL 拼装器：库里只存七牛 key，渲染时按 tier 拼 base_url + 样式。
imgComposer = imgurl.New(cfg.Qiniu.BaseURL, cfg.Qiniu.StyleNormal, cfg.Qiniu.StyleVip, cfg.Qiniu.LocalBase, cfg.Qiniu.AK, cfg.Qiniu.SK, cfg.Qiniu.SignTTLNormal, cfg.Qiniu.SignTTLHD)
// S4 上传器（凭证齐全才真正启用；否则为零值，图片回落本地兜底）。
s4Up = storage.NewS4Uploader(cfg.S4.AK, cfg.S4.SK, cfg.S4.Bucket, cfg.S4.Endpoint, cfg.S4.BaseURL, cfg.S4.Region)
// 图片存储：七牛云优先（若启用），失败兜底本地 uploads。
// localUp 同时实现 Uploader 与 Deleter（本地兜底删除）；启用七牛时两者均为七牛实现。
// 图片 URL 拼装器：库里只存 S4 key，渲染时拼 base_url + 展示样式（VIP 与免费同质量）。
// base_url 优先取 S4 上传器自动推导值（保证展示图 URL 同源）。
// StyleThumb 供列表/卡片等小尺寸场景使用（详情页与大图仍走 StyleDisplay）。
imgComposer = imgurl.New(s4BaseURL(cfg.S4.BaseURL, s4Up), cfg.S4.StyleDisplay, cfg.S4.StyleThumb)
// 图片存储：缤纷云 S4 优先（若启用），失败兜底本地 uploads。
// localUp 同时实现 Uploader 与 Deleter（本地兜底删除）；启用 S4 时两者均为 S4 实现。
localUp *storage.LocalUploader = storage.NewLocalUploader(uploadDir, cfg.Upload.URLPrefix)
imgUp storage.Uploader = localUp
imgDel storage.Deleter = localUp
@ -108,23 +123,21 @@ func main() {
)
// 爬虫入库：入队 + 异步 worker（去重 / 补季节码 / 下载图 / 写草稿表）
// 图片存储：七牛云优先（若启用），失败兜底本地 uploads。
// imgUp / imgDel 已在上方 var 块声明，这里仅在启用七牛时切换为七牛实现。
if cfg.Qiniu.Enabled {
// 注意：上传器不传 base_url（末参传空），Upload 只返回七牛 key；
// 完整可访问 URL 由 imgComposer 在读取时按 tier 拼 base_url + 样式，
// 这样既方便换 CDN 域名，又能做普通/高清分级（见 internal/pkg/imgurl）。
qu := storage.NewQiniuUploader(cfg.Qiniu.AK, cfg.Qiniu.SK, cfg.Qiniu.Bucket, cfg.Qiniu.Zone, "")
imgUp = qu
imgDel = qu
log.Printf("✓ 图片存储: 七牛云 bucket=%s zone=%s（库只存 key，渲染时拼 base_url=%s）", cfg.Qiniu.Bucket, cfg.Qiniu.Zone, cfg.Qiniu.BaseURL)
// 图片存储：缤纷云 S4 优先（若启用），失败兜底本地 uploads。
// imgUp / imgDel 已在上方 var 块声明，这里仅在启用 S4 且上传器真正可用时切换为 S4 实现。
if cfg.S4.Enabled && s4Up != nil && s4Up.Enabled() {
// Upload 只返回 S4 key（不拼 base_url），完整可访问 URL 由 imgComposer 在读取时按 tier 拼；
// free 走公开 CoreIX 样式、VIP 走预签名原图，既方便换域名又能做分级（见 internal/pkg/imgurl）。
imgUp = s4Up
imgDel = s4Up
log.Printf("✓ 图片存储: 缤纷云 S4 bucket=%s base=%s（库只存 key，渲染时拼 base_url）", cfg.S4.Bucket, s4Up.PublicBaseURL())
} else {
log.Printf("✓ 图片存储: 本地 %s", cfg.Upload.URLPrefix)
}
// 爬虫入库服务：注入 mediaRepo（引用计数删孤儿）+ imgDel（七牛/本地删除器）。
// 必须在 articleSvc / snapSvc 之前创建——图集服务删除时需调用它把「清理七牛」异步入队。
// 爬虫入库服务：注入 mediaRepo（引用计数删孤儿）+ imgDel（S4/本地删除器）。
// 必须在 articleSvc / snapSvc 之前创建——图集服务删除时需调用它把「清理存储孤儿图」异步入队。
ingestSvc := service.NewIngestService(ingestRepo, brandRepo, mediaRepo, imgUp, imgDel, localUp)
// 图集服务：注入 ingestSvc 作为「清理七牛孤儿图」的入队器，使下架/删除异步触发存储清理。
// 图集服务：注入 ingestSvc 作为「清理孤儿图」的入队器，使下架/删除异步触发存储清理。
articleSvc = service.NewArticleService(articleRepo, mediaRepo, imgComposer, imgDel, ingestSvc)
snapSvc = service.NewStreetSnapService(snapRepo, mediaRepo, imgComposer, imgDel, ingestSvc)