mirror of
https://github.com/go-admin-team/go-admin.git
synced 2026-09-22 02:27:57 +00:00
zhangwenjian
5ecb1e6e4c
`gofmt -l` listed 26 files. Seventeen of them were missing the newline at the end of the file; the rest are indentation that used spaces where the file uses tabs, a handful of call sites written `f(a,b)`, and the doc comment spacing gofmt has rewritten since 1.19 (`//X` to `// X`). Nothing here changes behaviour: `go build ./...` and `go vet ./...` are clean and `go test ./common/...` passes, which is the half of the tree these files are concentrated in. Only the files gofmt named are touched, so the diff reads line by line rather than as a reflow of the whole repository. `gofmt -l` is now empty, which is the precondition for gating it in CI -- worth doing, but a separate change.
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()
|
|
}
|