mirror of
https://github.com/goharbor/harbor
synced 2025-04-13 12:33:51 +00:00
ee4b16ccdb
Compare the local digest and the remote digest when pull by tag Use HEAD request (ManifestExist) instead of GET request (GetManifest) to avoid been throttled For manifest list, it can avoid GET request because cached manifest list maybe different with the original manifest list Make RemoteInterface public Fixes #13112 Signed-off-by: stonezdj <stonezdj@gmail.com>
92 lines
2.7 KiB
Go
92 lines
2.7 KiB
Go
// Copyright Project Harbor Authors
|
|
//
|
|
// Licensed under the Apache License, Version 2.0 (the "License");
|
|
// you may not use this file except in compliance with the License.
|
|
// You may obtain a copy of the License at
|
|
//
|
|
// http://www.apache.org/licenses/LICENSE-2.0
|
|
//
|
|
// Unless required by applicable law or agreed to in writing, software
|
|
// distributed under the License is distributed on an "AS IS" BASIS,
|
|
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
|
// See the License for the specific language governing permissions and
|
|
// limitations under the License.
|
|
|
|
package proxy
|
|
|
|
import (
|
|
"fmt"
|
|
"github.com/docker/distribution"
|
|
"github.com/goharbor/harbor/src/replication/adapter"
|
|
"github.com/goharbor/harbor/src/replication/model"
|
|
"github.com/goharbor/harbor/src/replication/registry"
|
|
"io"
|
|
)
|
|
|
|
// RemoteInterface defines operations related to remote repository under proxy
|
|
type RemoteInterface interface {
|
|
// BlobReader create a reader for remote blob
|
|
BlobReader(repo, dig string) (int64, io.ReadCloser, error)
|
|
// Manifest get manifest by reference
|
|
Manifest(repo string, ref string) (distribution.Manifest, string, error)
|
|
// ManifestExist checks manifest exist, if exist, return digest
|
|
ManifestExist(repo string, ref string) (bool, string, error)
|
|
}
|
|
|
|
// remoteHelper defines operations related to remote repository under proxy
|
|
type remoteHelper struct {
|
|
regID int64
|
|
registry adapter.ArtifactRegistry
|
|
registryMgr registry.Manager
|
|
}
|
|
|
|
// NewRemoteHelper create a remote interface
|
|
func NewRemoteHelper(regID int64) (RemoteInterface, error) {
|
|
r := &remoteHelper{
|
|
regID: regID,
|
|
registryMgr: registry.NewDefaultManager()}
|
|
if err := r.init(); err != nil {
|
|
return nil, err
|
|
}
|
|
return r, nil
|
|
}
|
|
|
|
func (r *remoteHelper) init() error {
|
|
|
|
if r.registry != nil {
|
|
return nil
|
|
}
|
|
reg, err := r.registryMgr.Get(r.regID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
if reg == nil {
|
|
return fmt.Errorf("failed to get registry, registryID: %v", r.regID)
|
|
}
|
|
if reg.Status != model.Healthy {
|
|
return fmt.Errorf("current registry is unhealthy, regID:%v, Name:%v, Status: %v", reg.ID, reg.Name, reg.Status)
|
|
}
|
|
factory, err := adapter.GetFactory(reg.Type)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
adp, err := factory.Create(reg)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
r.registry = adp.(adapter.ArtifactRegistry)
|
|
return nil
|
|
}
|
|
|
|
func (r *remoteHelper) BlobReader(repo, dig string) (int64, io.ReadCloser, error) {
|
|
return r.registry.PullBlob(repo, dig)
|
|
}
|
|
|
|
func (r *remoteHelper) Manifest(repo string, ref string) (distribution.Manifest, string, error) {
|
|
return r.registry.PullManifest(repo, ref)
|
|
}
|
|
|
|
func (r *remoteHelper) ManifestExist(repo string, ref string) (bool, string, error) {
|
|
return r.registry.ManifestExist(repo, ref)
|
|
}
|