fix: 七牛云私有 bucket 下载 URL 签名支持

对私有读 bucket,通过 HMAC-SHA1 签名生成带 e(过期时间)和 token(下载凭证)
参数的临时下载 URL,前端获取的素材 URL 和下载接口均返回签名 URL。

新增配置项 GEN2D_QINIU_URL_EXPIRE,默认 3600 秒。
This commit is contained in:
2026-05-25 19:05:55 +08:00
parent 641c53124b
commit 5100a52bca
6 changed files with 65 additions and 12 deletions
+3
View File
@@ -71,6 +71,7 @@ type QiniuConfig struct {
Bucket string `mapstructure:"bucket"`
CDNHost string `mapstructure:"cdn_host"`
UseHTTPS bool `mapstructure:"use_https"`
URLExpire int64 `mapstructure:"url_expire"` // 私有下载签名 URL 有效期(秒),默认 3600
}
// Load 从 YAML 配置文件和环境变量加载配置。
@@ -142,6 +143,7 @@ func setDefaults(v *viper.Viper) {
v.SetDefault("qiniu.bucket", "")
v.SetDefault("qiniu.cdn_host", "")
v.SetDefault("qiniu.use_https", true)
v.SetDefault("qiniu.url_expire", int64(3600))
}
func bindEnvVars(v *viper.Viper) {
@@ -179,4 +181,5 @@ func bindEnvVars(v *viper.Viper) {
v.BindEnv("qiniu.bucket", "GEN2D_QINIU_BUCKET")
v.BindEnv("qiniu.cdn_host", "GEN2D_QINIU_CDN_HOST")
v.BindEnv("qiniu.use_https", "GEN2D_QINIU_USE_HTTPS")
v.BindEnv("qiniu.url_expire", "GEN2D_QINIU_URL_EXPIRE")
}
+8
View File
@@ -31,3 +31,11 @@ image_gen:
width: 1024
height: 1024
quality: "low"
qiniu:
access_key: ""
secret_key: ""
bucket: ""
cdn_host: ""
use_https: true
url_expire: 3600 # 私有下载签名 URL 有效期(秒)
+1 -1
View File
@@ -207,7 +207,7 @@ func GetAssets(c *gin.Context) {
for _, a := range assets {
response = append(response, model.AssetResponse{
Key: a.Key,
URL: a.URL,
URL: storageSvc.GetSignedURL(a.Key),
Format: a.Format,
})
}
+44 -5
View File
@@ -3,6 +3,9 @@ package service
import (
"bytes"
"context"
"crypto/hmac"
"crypto/sha1"
"encoding/base64"
"fmt"
"net/url"
"path/filepath"
@@ -65,15 +68,43 @@ func (s *StorageService) Upload(ctx context.Context, key string, data []byte) (s
return "", fmt.Errorf("qiniu upload: %w", err)
}
cdnURL := s.buildURL(key)
cdnURL := s.buildBaseURL(key)
l.Info("upload succeeded", "key", key, "url", cdnURL)
return cdnURL, nil
}
// GetDownloadURL 生成素材的下载 URL。
// 公开 bucket 直接返回 CDN URL;私有 bucket 生成带签名的临时 URL。
// 私有 bucket 生成带签名和过期时间的临时 URL。
func (s *StorageService) GetDownloadURL(_ context.Context, key string) (string, error) {
return s.buildURL(key), nil
return s.signDownloadURL(key), nil
}
// GetSignedURL 获取私有资源签名 URL(公开方法,供 handler 层使用)。
func (s *StorageService) GetSignedURL(key string) string {
return s.signDownloadURL(key)
}
// signDownloadURL 生成七牛云私有资源下载签名 URL。
// 格式:http://<domain>/<key>?e=<deadline>&token=<downloadToken>
func (s *StorageService) signDownloadURL(key string) string {
baseURL := s.buildBaseURL(key)
expire := s.cfg.URLExpire
if expire <= 0 {
expire = 3600
}
deadline := time.Now().Unix() + expire
signStr := fmt.Sprintf("%s?e=%d", baseURL, deadline)
mac := hmac.New(sha1.New, []byte(s.cfg.SecretKey))
mac.Write([]byte(signStr))
signature := mac.Sum(nil)
encodedSign := base64.URLEncoding.EncodeToString(signature)
token := fmt.Sprintf("%s:%s", s.cfg.AccessKey, encodedSign)
return fmt.Sprintf("%s?e=%d&token=%s", baseURL, deadline, token)
}
// Delete 从七牛云删除指定对象。
@@ -91,9 +122,17 @@ func (s *StorageService) Delete(ctx context.Context, key string) error {
return nil
}
// buildURL 根据配置构建完整的 CDN URL。
func (s *StorageService) buildURL(key string) string {
// buildBaseURL 根据配置构建完整的基础 CDN URL(不含签名参数)。
func (s *StorageService) buildBaseURL(key string) string {
host := strings.TrimRight(s.cfg.CDNHost, "/")
scheme := "http"
if s.cfg.UseHTTPS {
scheme = "https"
}
// 如果 host 已经包含 scheme,直接使用;否则添加
if !strings.HasPrefix(host, "http://") && !strings.HasPrefix(host, "https://") {
host = scheme + "://" + host
}
segments := strings.Split(key, "/")
for i, seg := range segments {
segments[i] = url.PathEscape(seg)