|
- // Copyright 2020 The Gitea Authors. All rights reserved.
- // Use of this source code is governed by a MIT-style
- // license that can be found in the LICENSE file.
-
- package storage
-
- import (
- "github.com/unknwon/com"
- "path"
- "strconv"
- "strings"
-
- "code.gitea.io/gitea/modules/log"
- "code.gitea.io/gitea/modules/obs"
- "code.gitea.io/gitea/modules/setting"
- )
-
- //check if has the object
- //todo:修改查询方式
- func ObsHasObject(path string) (bool, error) {
- hasObject := false
- output, err := ObsCli.ListObjects(&obs.ListObjectsInput{Bucket: setting.Bucket})
- if err != nil {
- log.Error("ListObjects failed:%v", err)
- return hasObject, err
- }
-
- for _, obj := range output.Contents {
- //obj.Key:attachment/0/1/019fd24e-4ef7-41cc-9f85-4a7b8504d958
- if path == obj.Key {
- hasObject = true
- break
- }
- }
-
- return hasObject, nil
- }
-
- func GetObsPartInfos(uuid string, uploadID string) (string, error) {
- key := strings.TrimPrefix(path.Join(setting.BasePath, path.Join(uuid[0:1], uuid[1:2], uuid, uuid)), "/")
-
- output, err := ObsCli.ListParts(&obs.ListPartsInput{
- Bucket: setting.Bucket,
- Key: key,
- UploadId: uploadID,
- })
- if err != nil {
- log.Error("ListParts failed:", err.Error())
- return "", err
- }
-
- var chunks string
- for _, partInfo := range output.Parts {
- chunks += strconv.Itoa(partInfo.PartNumber) + "-" + partInfo.ETag + ","
- }
-
- return chunks, nil
- }
-
- func NewObsMultiPartUpload(uuid, fileName string) (string, error) {
- input := &obs.InitiateMultipartUploadInput{}
- input.Bucket = setting.Bucket
- input.Key = strings.TrimPrefix(path.Join(setting.BasePath, path.Join(uuid[0:1], uuid[1:2], uuid, fileName)), "/")
-
- output, err := ObsCli.InitiateMultipartUpload(input)
- if err != nil {
- log.Error("InitiateMultipartUpload failed:", err.Error())
- return "", err
- }
-
- return output.UploadId, nil
- }
-
- func CompleteObsMultiPartUpload(uuid, uploadID, fileName string) error {
- input := &obs.CompleteMultipartUploadInput{}
- input.Bucket = setting.Bucket
- input.Key = strings.TrimPrefix(path.Join(setting.BasePath, path.Join(uuid[0:1], uuid[1:2], uuid, fileName)), "/")
- input.UploadId = uploadID
- output, err := ObsCli.ListParts(&obs.ListPartsInput{
- Bucket: setting.Bucket,
- Key: input.Key,
- UploadId: uploadID,
- })
- if err != nil {
- log.Error("ListParts failed:", err.Error())
- return err
- }
-
- for _, partInfo := range output.Parts {
- input.Parts = append(input.Parts, obs.Part{
- PartNumber: partInfo.PartNumber,
- ETag: partInfo.ETag,
- })
- }
-
- _, err = ObsCli.CompleteMultipartUpload(input)
- if err != nil {
- log.Error("CompleteMultipartUpload failed:", err.Error())
- return err
- }
-
- return nil
- }
-
- func ObsGenMultiPartSignedUrl(uuid string, uploadId string, partNumber int, fileName string) (string, error) {
-
- input := &obs.CreateSignedUrlInput{}
- input.Bucket = setting.Bucket
- input.Key = strings.TrimPrefix(path.Join(setting.BasePath, path.Join(uuid[0:1], uuid[1:2], uuid, fileName)), "/")
- input.Expires = 60 * 60
- input.Method = obs.HttpMethodPut
-
- input.QueryParams = map[string]string{
- "partNumber": com.ToStr(partNumber, 10),
- "uploadId": uploadId,
- //"partSize": com.ToStr(partSize,10),
- }
-
- output, err := ObsCli.CreateSignedUrl(input)
- if err != nil {
- log.Error("CreateSignedUrl failed:", err.Error())
- return "", err
- }
-
- return output.SignedUrl, nil
- }
-
- func ObsGetPreSignedUrl(uuid, fileName string) (string, error) {
- input := &obs.CreateSignedUrlInput{}
- input.Method = obs.HttpMethodGet
- input.Key = strings.TrimPrefix(path.Join(setting.BasePath, path.Join(uuid[0:1], uuid[1:2], uuid, fileName)), "/")
- input.Bucket = setting.Bucket
- input.Expires = 60 * 60
-
- reqParams := make(map[string]string)
- reqParams["response-content-disposition"] = "attachment; filename=\"" + fileName + "\""
- input.QueryParams = reqParams
- output, err := ObsCli.CreateSignedUrl(input)
- if err != nil {
- log.Error("CreateSignedUrl failed:", err.Error())
- return "", err
- }
-
- return output.SignedUrl, nil
- }
|