65 lines
1.6 KiB
Go
65 lines
1.6 KiB
Go
package controller
|
|
|
|
import (
|
|
"io"
|
|
"strings"
|
|
|
|
"gitea.loveuer.com/loveuer/cluster/internal/registry/storage"
|
|
)
|
|
|
|
// StartBlobUpload 开始 blob 上传
|
|
func StartBlobUpload(store storage.Storage, repo string) (string, error) {
|
|
return store.StartBlobUpload(repo)
|
|
}
|
|
|
|
// PatchBlobUpload 上传 blob 数据块
|
|
func PatchBlobUpload(store storage.Storage, uuid string, data io.Reader) (int64, error) {
|
|
if err := store.PutBlobUploadChunk(uuid, data); err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
// 获取当前上传大小
|
|
size, err := store.GetBlobUpload(uuid)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
|
|
return size, nil
|
|
}
|
|
|
|
// PutBlobUpload 完成 blob 上传
|
|
func PutBlobUpload(store storage.Storage, repo, uuid, digest string, data interface{}, requestPath string) (string, error) {
|
|
// 如果有请求体，先追加数据
|
|
if data != nil {
|
|
if reader, ok := data.(io.Reader); ok {
|
|
if err := store.PutBlobUploadChunk(uuid, reader); err != nil {
|
|
return "", err
|
|
}
|
|
}
|
|
}
|
|
|
|
// 完成上传
|
|
if err := store.CompleteBlobUpload(repo, uuid, digest); err != nil {
|
|
return "", err
|
|
}
|
|
|
|
// 清理上传文件
|
|
store.DeleteBlobUpload(uuid)
|
|
|
|
// 返回 blob 位置
|
|
// 从 /v2/{name}/blobs/uploads/{uuid} 转换为 /v2/{name}/blobs/{digest}
|
|
pathParts := strings.Split(requestPath, "/")
|
|
if len(pathParts) >= 4 {
|
|
// 构建新的路径: /v2/{name}/blobs/{digest}
|
|
location := "/v2/" + pathParts[2] + "/blobs/" + digest
|
|
return location, nil
|
|
}
|
|
|
|
return requestPath, nil
|
|
}
|
|
|
|
// GetBlobUpload 获取上传状态
|
|
func GetBlobUpload(store storage.Storage, uuid string) (int64, error) {
|
|
return store.GetBlobUpload(uuid)
|
|
}
|