forked from TrueCloudLab/distribution
f141480d98
Several error codes are generally useful but tied to the v2 specification definitions. This change moves these error code definitions into the common package for use by the health package, which is not tied to the v2 API. Signed-off-by: Stephen J Day <stephen.day@docker.com>
93 lines
2.6 KiB
Go
93 lines
2.6 KiB
Go
package handlers
|
|
|
|
import (
|
|
"net/http"
|
|
|
|
"github.com/docker/distribution"
|
|
"github.com/docker/distribution/context"
|
|
"github.com/docker/distribution/digest"
|
|
"github.com/docker/distribution/registry/api/errcode"
|
|
"github.com/docker/distribution/registry/api/v2"
|
|
"github.com/gorilla/handlers"
|
|
)
|
|
|
|
// blobDispatcher uses the request context to build a blobHandler.
|
|
func blobDispatcher(ctx *Context, r *http.Request) http.Handler {
|
|
dgst, err := getDigest(ctx)
|
|
if err != nil {
|
|
|
|
if err == errDigestNotAvailable {
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
ctx.Errors = append(ctx.Errors, v2.ErrorCodeDigestInvalid.WithDetail(err))
|
|
})
|
|
}
|
|
|
|
return http.HandlerFunc(func(w http.ResponseWriter, r *http.Request) {
|
|
ctx.Errors = append(ctx.Errors, v2.ErrorCodeDigestInvalid.WithDetail(err))
|
|
})
|
|
}
|
|
|
|
blobHandler := &blobHandler{
|
|
Context: ctx,
|
|
Digest: dgst,
|
|
}
|
|
|
|
return handlers.MethodHandler{
|
|
"GET": http.HandlerFunc(blobHandler.GetBlob),
|
|
"HEAD": http.HandlerFunc(blobHandler.GetBlob),
|
|
"DELETE": http.HandlerFunc(blobHandler.DeleteBlob),
|
|
}
|
|
}
|
|
|
|
// blobHandler serves http blob requests.
|
|
type blobHandler struct {
|
|
*Context
|
|
|
|
Digest digest.Digest
|
|
}
|
|
|
|
// GetBlob fetches the binary data from backend storage returns it in the
|
|
// response.
|
|
func (bh *blobHandler) GetBlob(w http.ResponseWriter, r *http.Request) {
|
|
context.GetLogger(bh).Debug("GetBlob")
|
|
blobs := bh.Repository.Blobs(bh)
|
|
desc, err := blobs.Stat(bh, bh.Digest)
|
|
if err != nil {
|
|
if err == distribution.ErrBlobUnknown {
|
|
bh.Errors = append(bh.Errors, v2.ErrorCodeBlobUnknown.WithDetail(bh.Digest))
|
|
} else {
|
|
bh.Errors = append(bh.Errors, errcode.ErrorCodeUnknown.WithDetail(err))
|
|
}
|
|
return
|
|
}
|
|
|
|
if err := blobs.ServeBlob(bh, w, r, desc.Digest); err != nil {
|
|
context.GetLogger(bh).Debugf("unexpected error getting blob HTTP handler: %v", err)
|
|
bh.Errors = append(bh.Errors, errcode.ErrorCodeUnknown.WithDetail(err))
|
|
return
|
|
}
|
|
}
|
|
|
|
// DeleteBlob deletes a layer blob
|
|
func (bh *blobHandler) DeleteBlob(w http.ResponseWriter, r *http.Request) {
|
|
context.GetLogger(bh).Debug("DeleteBlob")
|
|
|
|
blobs := bh.Repository.Blobs(bh)
|
|
err := blobs.Delete(bh, bh.Digest)
|
|
if err != nil {
|
|
switch err {
|
|
case distribution.ErrBlobUnknown:
|
|
w.WriteHeader(http.StatusNotFound)
|
|
bh.Errors = append(bh.Errors, v2.ErrorCodeBlobUnknown)
|
|
case distribution.ErrUnsupported:
|
|
w.WriteHeader(http.StatusMethodNotAllowed)
|
|
bh.Errors = append(bh.Errors, errcode.ErrorCodeUnsupported)
|
|
default:
|
|
bh.Errors = append(bh.Errors, errcode.ErrorCodeUnknown)
|
|
}
|
|
return
|
|
}
|
|
|
|
w.Header().Set("Content-Length", "0")
|
|
w.WriteHeader(http.StatusAccepted)
|
|
}
|