e10a9aff7d
These interfaces allow us to preserve both the checking of error "cause" as well as messages returned from the gRPC API so that the client gets full error reason instead of a default "metadata: not found" in the case of a missing image. Signed-off-by: Phil Estes <estesp@linux.vnet.ibm.com>
47 lines
841 B
Go
47 lines
841 B
Go
package docker
|
|
|
|
import (
|
|
"sync"
|
|
|
|
"github.com/containerd/containerd/content"
|
|
)
|
|
|
|
type Status struct {
|
|
content.Status
|
|
|
|
// UploadUUID is used by the Docker registry to reference blob uploads
|
|
UploadUUID string
|
|
}
|
|
|
|
type StatusTracker interface {
|
|
GetStatus(string) (Status, error)
|
|
SetStatus(string, Status)
|
|
}
|
|
|
|
type memoryStatusTracker struct {
|
|
statuses map[string]Status
|
|
m sync.Mutex
|
|
}
|
|
|
|
func NewInMemoryTracker() StatusTracker {
|
|
return &memoryStatusTracker{
|
|
statuses: map[string]Status{},
|
|
}
|
|
}
|
|
|
|
func (t *memoryStatusTracker) GetStatus(ref string) (Status, error) {
|
|
t.m.Lock()
|
|
defer t.m.Unlock()
|
|
status, ok := t.statuses[ref]
|
|
if !ok {
|
|
return Status{}, content.ErrNotFound("")
|
|
}
|
|
return status, nil
|
|
}
|
|
|
|
func (t *memoryStatusTracker) SetStatus(ref string, status Status) {
|
|
t.m.Lock()
|
|
t.statuses[ref] = status
|
|
t.m.Unlock()
|
|
}
|