xinyin025/geekai
1
0
Fork 0
mirror of https://github.com/yangjian102621/geekai.git synced 2025-09-21 18:56:38 +08:00
Code Issues Actions Packages Projects Releases Wiki Activity

fix: fix bug for token expired with QiNiu oss upload file

Browse Source
This commit is contained in:
RockYang 2023-11-08 21:14:09 +08:00
parent 8bafe72434
commit ef87487f60
1 changed files with 16 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
api/service/oss/qiniu_oss.go
View File

@ -15,12 +15,13 @@ import (
) )
type QinNiuOss struct { type QinNiuOss struct {
config *types.QiNiuOssConfig config *types.QiNiuOssConfig
token string mac *qbox.Mac
uploader *storage.FormUploader putPolicy storage.PutPolicy
manager *storage.BucketManager uploader *storage.FormUploader
proxyURL string manager *storage.BucketManager
dir string proxyURL string
dir string
} }
func NewQiNiuOss(appConfig *types.AppConfig) QinNiuOss { func NewQiNiuOss(appConfig *types.AppConfig) QinNiuOss {
@ -38,12 +39,13 @@ func NewQiNiuOss(appConfig *types.AppConfig) QinNiuOss {
Scope: config.Bucket, Scope: config.Bucket,
} }
return QinNiuOss{ return QinNiuOss{
config: config, config: config,
token: putPolicy.UploadToken(mac), mac: mac,
uploader: formUploader, putPolicy: putPolicy,
manager: storage.NewBucketManager(mac, &storeConfig), uploader: formUploader,
proxyURL: appConfig.ProxyURL, manager: storage.NewBucketManager(mac, &storeConfig),
dir: "chatgpt-plus", proxyURL: appConfig.ProxyURL,
dir: "chatgpt-plus",
} }
} }
@ -65,7 +67,7 @@ func (s QinNiuOss) PutFile(ctx *gin.Context, name string) (string, error) {
// 上传文件 // 上传文件
ret := storage.PutRet{} ret := storage.PutRet{}
extra := storage.PutExtra{} extra := storage.PutExtra{}
err = s.uploader.Put(ctx, &ret, s.token, key, src, file.Size, &extra) err = s.uploader.Put(ctx, &ret, s.putPolicy.UploadToken(s.mac), key, src, file.Size, &extra)
if err != nil { if err != nil {
return "", err return "", err
} }
@ -93,7 +95,7 @@ func (s QinNiuOss) PutImg(imageURL string, useProxy bool) (string, error) {
ret := storage.PutRet{} ret := storage.PutRet{}
extra := storage.PutExtra{} extra := storage.PutExtra{}
// 上传文件字节数据 // 上传文件字节数据
err = s.uploader.Put(context.Background(), &ret, s.token, key, bytes.NewReader(imageData), int64(len(imageData)), &extra) err = s.uploader.Put(context.Background(), &ret, s.putPolicy.UploadToken(s.mac), key, bytes.NewReader(imageData), int64(len(imageData)), &extra)
if err != nil { if err != nil {
return "", err return "", err
} }