mirror of
https://github.com/DullJZ/s3-balance.git
synced 2026-09-06 16:27:38 +08:00
Multipart error message
This commit is contained in:
@@ -3,10 +3,12 @@ package api
|
|||||||
import (
|
import (
|
||||||
"context"
|
"context"
|
||||||
"encoding/xml"
|
"encoding/xml"
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"log"
|
"log"
|
||||||
"net/http"
|
"net/http"
|
||||||
|
"sort"
|
||||||
"strconv"
|
"strconv"
|
||||||
"time"
|
"time"
|
||||||
|
|
||||||
@@ -14,6 +16,7 @@ import (
|
|||||||
"github.com/aws/aws-sdk-go-v2/aws"
|
"github.com/aws/aws-sdk-go-v2/aws"
|
||||||
"github.com/aws/aws-sdk-go-v2/service/s3"
|
"github.com/aws/aws-sdk-go-v2/service/s3"
|
||||||
"github.com/aws/aws-sdk-go-v2/service/s3/types"
|
"github.com/aws/aws-sdk-go-v2/service/s3/types"
|
||||||
|
"github.com/aws/smithy-go"
|
||||||
"github.com/gorilla/mux"
|
"github.com/gorilla/mux"
|
||||||
)
|
)
|
||||||
|
|
||||||
@@ -518,6 +521,10 @@ func (h *S3Handler) handleCompleteMultipartUpload(w http.ResponseWriter, r *http
|
|||||||
|
|
||||||
// 完成分片上传
|
// 完成分片上传
|
||||||
ctx := context.Background()
|
ctx := context.Background()
|
||||||
|
sort.SliceStable(completeReq.Parts, func(i, j int) bool {
|
||||||
|
return completeReq.Parts[i].PartNumber < completeReq.Parts[j].PartNumber
|
||||||
|
})
|
||||||
|
|
||||||
var parts []types.CompletedPart
|
var parts []types.CompletedPart
|
||||||
for _, part := range completeReq.Parts {
|
for _, part := range completeReq.Parts {
|
||||||
parts = append(parts, types.CompletedPart{
|
parts = append(parts, types.CompletedPart{
|
||||||
@@ -537,7 +544,11 @@ func (h *S3Handler) handleCompleteMultipartUpload(w http.ResponseWriter, r *http
|
|||||||
})
|
})
|
||||||
if err != nil {
|
if err != nil {
|
||||||
log.Printf("CompleteMultipartUpload failed: %v", err)
|
log.Printf("CompleteMultipartUpload failed: %v", err)
|
||||||
h.sendS3Error(w, "InternalError", "Failed to complete multipart upload", key)
|
if apiErr, ok := getAPIError(err); ok {
|
||||||
|
h.sendS3Error(w, apiErr.ErrorCode(), apiErr.ErrorMessage(), key)
|
||||||
|
} else {
|
||||||
|
h.sendS3Error(w, "InternalError", "Failed to complete multipart upload", key)
|
||||||
|
}
|
||||||
return
|
return
|
||||||
}
|
}
|
||||||
|
|
||||||
@@ -580,6 +591,22 @@ func (h *S3Handler) handleCompleteMultipartUpload(w http.ResponseWriter, r *http
|
|||||||
h.sendXMLResponse(w, http.StatusOK, result)
|
h.sendXMLResponse(w, http.StatusOK, result)
|
||||||
}
|
}
|
||||||
|
|
||||||
|
func getAPIError(err error) (smithy.APIError, bool) {
|
||||||
|
var apiErr smithy.APIError
|
||||||
|
if errors.As(err, &apiErr) {
|
||||||
|
return apiErr, true
|
||||||
|
}
|
||||||
|
|
||||||
|
var opErr *smithy.OperationError
|
||||||
|
if errors.As(err, &opErr) {
|
||||||
|
if apiErr, ok := opErr.Err.(smithy.APIError); ok {
|
||||||
|
return apiErr, true
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
return nil, false
|
||||||
|
}
|
||||||
|
|
||||||
// handleAbortMultipartUpload 中止分片上传
|
// handleAbortMultipartUpload 中止分片上传
|
||||||
func (h *S3Handler) handleAbortMultipartUpload(w http.ResponseWriter, r *http.Request) {
|
func (h *S3Handler) handleAbortMultipartUpload(w http.ResponseWriter, r *http.Request) {
|
||||||
vars := mux.Vars(r)
|
vars := mux.Vars(r)
|
||||||
|
|||||||
Reference in New Issue
Block a user