mirror of
https://github.com/go-admin-team/go-admin.git
synced 2026-09-21 10:13:01 +00:00
zhangwenjian
fcbd9ae02e
Each implementation keeps its provider client in an interface{} field that
Setup assigns, so an unconfigured store holds nil - and asserting nil to
the provider's client type panics:
panic: interface conversion: interface {} is nil, not *oss.Client
The upload endpoint reaches that path for any request naming a provider
the deployment never configured.
Three more things were wrong in the same files. OXS.Setup printed a
failure and returned the store anyway, handing back exactly the broken
object that panics. HuaWeiOBS.UpLoad printed the provider's error and
returned nil, so a failed upload reported success. Both it and
QiNiuKODO.UpLoad asserted the local path was a string without checking.
The tests asked the reader to paste their own credentials, so they failed
for everyone who did not. They now cover the guards and skip the part
that needs a provider unless credentials are in the environment.
122 lines
2.7 KiB
Go
122 lines
2.7 KiB
Go
package file_store
|
|
|
|
import (
|
|
"context"
|
|
"fmt"
|
|
"github.com/qiniu/go-sdk/v7/auth/qbox"
|
|
"github.com/qiniu/go-sdk/v7/storage"
|
|
)
|
|
|
|
type Zone string
|
|
|
|
const (
|
|
// HuaDong 华东
|
|
HuaDong Zone = "HuaDong"
|
|
// HuaBei 华北
|
|
HuaBei Zone = "HuaBei"
|
|
// HuaNan 华南
|
|
HuaNan Zone = "HuaNan"
|
|
// BeiMei 北美
|
|
BeiMei Zone = "BeiMei"
|
|
// XinJiaPo 新加坡
|
|
XinJiaPo Zone = "XinJiaPo"
|
|
)
|
|
|
|
type QiNiuKODO struct {
|
|
Client interface{}
|
|
BucketName string
|
|
cfg storage.Config
|
|
options []ClientOption
|
|
}
|
|
|
|
func (e *QiNiuKODO) getToken() (string, error) {
|
|
mac, ok := e.Client.(*qbox.Mac)
|
|
if !ok {
|
|
return "", notConfigured(QiNiuKodo)
|
|
}
|
|
putPolicy := storage.PutPolicy{
|
|
Scope: e.BucketName,
|
|
}
|
|
if len(e.options) > 0 {
|
|
if expires, ok := e.options[0]["Expires"].(uint64); ok {
|
|
putPolicy.Expires = expires
|
|
}
|
|
}
|
|
return putPolicy.UploadToken(mac), nil
|
|
}
|
|
|
|
//Setup 装载
|
|
//endpoint sss
|
|
func (e *QiNiuKODO) Setup(endpoint, accessKeyID, accessKeySecret, BucketName string, options ...ClientOption) error {
|
|
|
|
mac := qbox.NewMac(accessKeyID, accessKeySecret)
|
|
// 获取存储空间。
|
|
cfg := storage.Config{}
|
|
// 空间对应的机房
|
|
e.setZoneORDefault(cfg, options...)
|
|
// 是否使用https域名
|
|
cfg.UseHTTPS = true
|
|
// 上传是否使用CDN上传加速
|
|
cfg.UseCdnDomains = false
|
|
|
|
e.Client = mac
|
|
e.BucketName = BucketName
|
|
e.cfg = cfg
|
|
e.options = options
|
|
return nil
|
|
}
|
|
|
|
// setZoneORDefault 设置Zone或者默认华东
|
|
func (e *QiNiuKODO) setZoneORDefault(cfg storage.Config, options ...ClientOption) {
|
|
if len(options) > 0 && options[0]["Zone"] != nil {
|
|
if _, ok := options[0]["Zone"].(Zone); !ok {
|
|
cfg.Zone = &storage.ZoneHuadong
|
|
}
|
|
switch options[0]["Zone"].(Zone) {
|
|
case HuaDong:
|
|
cfg.Zone = &storage.ZoneHuadong
|
|
case HuaBei:
|
|
cfg.Zone = &storage.ZoneHuabei
|
|
case HuaNan:
|
|
cfg.Zone = &storage.ZoneHuanan
|
|
case BeiMei:
|
|
cfg.Zone = &storage.ZoneBeimei
|
|
case XinJiaPo:
|
|
cfg.Zone = &storage.ZoneXinjiapo
|
|
default:
|
|
cfg.Zone = &storage.ZoneHuadong
|
|
}
|
|
}
|
|
}
|
|
|
|
// UpLoad 文件上传
|
|
func (e *QiNiuKODO) UpLoad(yourObjectName string, localFile interface{}) error {
|
|
|
|
// 构建表单上传的对象
|
|
formUploader := storage.NewFormUploader(&e.cfg)
|
|
ret := storage.PutRet{}
|
|
// 可选配置
|
|
putExtra := storage.PutExtra{
|
|
Params: map[string]string{
|
|
"x:name": "github logo",
|
|
},
|
|
}
|
|
token, err := e.getToken()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
source, ok := localFile.(string)
|
|
if !ok {
|
|
return fmt.Errorf("kodo upload wants a path, got %T", localFile)
|
|
}
|
|
if err := formUploader.PutFile(context.Background(), &ret, token, yourObjectName, source, &putExtra); err != nil {
|
|
return fmt.Errorf("kodo upload: %w", err)
|
|
}
|
|
fmt.Println(ret.Key, ret.Hash)
|
|
return nil
|
|
}
|
|
|
|
func (e *QiNiuKODO) GetTempToken() (string, error) {
|
|
return e.getToken()
|
|
}
|