feat(distribution): 站点归属兜底——未带站点编码时按访问域名识别分站

- 新增 middleware.SiteDomainFallback:显式头/query 优先,两者都为空时用 Request.Host / X-Forwarded-Host 反查 distribution_sites.domain,并回填 X-Distribution-Site-Code 供后续 sitecode.FromRequest 使用
- 新增 distribution.Repository.ResolveSiteCodeByDomain(仅取 status=active 的站点)
- 路由层在所有业务路由之前注册;域名到站点映射走 1 分钟进程内缓存(含负缓存);查库异常不阻断请求、退化为按主站处理;回环/本地地址跳过查询
- 背景:站点归属此前完全依赖前端注入头,首屏探测失败或漏带就会把分站写入记成主站(2026-10-10 前已因此丢失一批分站注册归属)
- 新增单测:中间件(显式优先、域名解析、本地地址跳过、转发域兜底、缓存与容错)与仓储域名查询边界(大小写/空白/主站域名/停用站点)
This commit is contained in:
yml2213
2026-10-13 17:33:16 +08:00
parent f33dc97f6f
commit 2e13443805
5 changed files with 357 additions and 0 deletions
@@ -2679,6 +2679,34 @@ func (r *Repository) evictSiteInfoCache(ctx context.Context, site *model.Distrib
}
}
// ResolveSiteCodeByDomain 按访问域名解析启用中的分销站点编码,找不到返回空串。
//
// 用途:站点归属兜底。前端(SPA)会显式带 X-Distribution-Site-Code 头,
// 但首屏探测、脚本调用、分享链接直达等场景可能没带;此时按请求域名判定站点,
// 避免注册等写入把分站用户记成主站(历史上就丢过一批)。
func (r *Repository) ResolveSiteCodeByDomain(ctx context.Context, domain string) (string, error) {
if r == nil || r.db == nil {
return "", ErrDependencyUnavailable
}
domain = strings.ToLower(strings.TrimSpace(domain))
if domain == "" {
return "", nil
}
var site model.DistributionSite
err := r.db.WithContext(ctx).
Select("site_code").
Where("status = 'active' AND domain = ?", domain).
Order("id ASC").
First(&site).Error
if err != nil {
if errors.Is(err, gorm.ErrRecordNotFound) {
return "", nil
}
return "", err
}
return site.SiteCode, nil
}
func (r *Repository) GetPublicSiteInfo(ctx context.Context, code, domain string) (*PublicSiteInfoDTO, error) {
if r.db == nil {
return nil, ErrDependencyUnavailable