package qiniu import ( "context" "fmt" "git.bvbej.com/bvbej/base-golang/pkg/md5" "github.com/qiniu/go-sdk/v7/auth/qbox" "github.com/qiniu/go-sdk/v7/storage" "net/http" "net/url" "path" "strings" "time" ) var _ QiNiu = (*qiNiu)(nil) type QiNiu interface { i() SetDefaultUploadTokenTTL(ttl uint64) GetCallbackUploadToken(ttl uint64, callbackURL string) string GetUploadToken(ttl uint64) string GetPrivateURL(key string, ttl uint64) string VerifyCallback(req *http.Request) (bool, error) UploadFile(key, localFile string) (*PutRet, error) ResumeUploadFile(key, localFile string) (*PutRet, error) DelFile(key string) error TimestampSecuritySign(path string, ttl time.Duration) string GetFileInfo(key string) *storage.FileInfo } type qiNiu struct { mac *qbox.Mac conf *storage.Config bucket string domain string securityKey string md5 md5.MD5 uploadTokenTTL uint64 } type PutRet struct { Key string `json:"key"` Hash string `json:"hash"` Fsize string `json:"fsize"` Fname string `json:"fname"` Ext string `json:"ext"` Unique string `json:"unique"` User string `json:"user"` } func New(accessKey, secretKey, bucket, domain, securityKey string) QiNiu { region, _ := storage.GetRegion(accessKey, bucket) return &qiNiu{ mac: qbox.NewMac(accessKey, secretKey), bucket: bucket, domain: domain, securityKey: securityKey, conf: &storage.Config{ Region: region, //空间所在的存储区域 UseHTTPS: true, //是否使用https域名 UseCdnDomains: true, //上传是否使用CDN上传加速 }, md5: md5.New(), uploadTokenTTL: 3600, } } func (q *qiNiu) i() {} func (q *qiNiu) SetDefaultUploadTokenTTL(ttl uint64) { q.uploadTokenTTL = ttl } func (q *qiNiu) GetUploadToken(ttl uint64) string { putPolicy := storage.PutPolicy{ Scope: q.bucket, Expires: ttl, } return putPolicy.UploadToken(q.mac) } func (q *qiNiu) GetCallbackUploadToken(ttl uint64, callbackURL string) string { putPolicy := storage.PutPolicy{ Scope: q.bucket, CallbackURL: callbackURL, CallbackBody: `{"key":"$(key)","hash":"$(etag)","fname":"$(fname)","fsize":"$(fsize)","ext":"$(ext)","unique":"$(x:unique)","user":"$(x:user)"}`, CallbackBodyType: "application/json", Expires: ttl, } return putPolicy.UploadToken(q.mac) } func (q *qiNiu) GetPrivateURL(key string, ttl uint64) string { deadline := time.Now().Add(time.Second * time.Duration(ttl)).Unix() return storage.MakePrivateURL(q.mac, q.domain, key, deadline) } func (q *qiNiu) VerifyCallback(req *http.Request) (bool, error) { return q.mac.VerifyCallback(req) } func (q *qiNiu) UploadFile(key, localFile string) (*PutRet, error) { upToken := q.GetUploadToken(q.uploadTokenTTL) //构建表单上传的对象 formUploader := storage.NewFormUploader(q.conf) //请求参数 filename := path.Base(key) fileSuffix := path.Ext(key) filePrefix := filename[0 : len(filename)-len(fileSuffix)] putExtra := &storage.PutExtra{ Params: map[string]string{ "x:unique": filePrefix, "x:user": "-", }, } //自定义返回body ret := new(PutRet) err := formUploader.PutFile(context.Background(), ret, upToken, key, localFile, putExtra) if err != nil { return nil, err } return ret, nil } func (q *qiNiu) ResumeUploadFile(key, localFile string) (*PutRet, error) { upToken := q.GetUploadToken(q.uploadTokenTTL) //构建分片上传的对象 resumeUploader := storage.NewResumeUploaderV2(q.conf) //请求参数 filename := path.Base(key) fileSuffix := path.Ext(key) filePrefix := filename[0 : len(filename)-len(fileSuffix)] putExtra := &storage.RputV2Extra{ CustomVars: map[string]string{ "x:unique": filePrefix, "x:user": "-", }, } //自定义返回body ret := new(PutRet) err := resumeUploader.PutFile(context.Background(), ret, upToken, key, localFile, putExtra) if err != nil { return nil, err } return ret, nil } func (q *qiNiu) DelFile(key string) error { bucketManager := storage.NewBucketManager(q.mac, q.conf) err := bucketManager.Delete(q.bucket, key) if err != nil { return err } return nil } func (q *qiNiu) TimestampSecuritySign(path string, ttl time.Duration) string { sep := "/" path = strings.Trim(path, sep) splits := strings.Split(path, sep) for i, split := range splits { splits[i] = url.QueryEscape(split) } path = sep + strings.Join(splits, sep) unix := time.Now().Add(ttl).Unix() hex := fmt.Sprintf("%x", unix) encrypt := q.md5.Encrypt(q.securityKey + path + hex) param := make(url.Values) param.Set("sign", encrypt) param.Set("t", hex) return param.Encode() } func (q *qiNiu) GetFileInfo(key string) *storage.FileInfo { bucketManager := storage.NewBucketManager(q.mac, q.conf) fileInfo, sErr := bucketManager.Stat(q.bucket, key) if sErr != nil { return nil } return &fileInfo }