mirror of https://github.com/goharbor/harbor.git
100 lines
2.9 KiB
Go
100 lines
2.9 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 pkg
|
|
|
|
import (
|
|
"github.com/goharbor/harbor/src/lib/config"
|
|
"github.com/goharbor/harbor/src/pkg/artifact"
|
|
cachedArtifact "github.com/goharbor/harbor/src/pkg/cached/artifact/redis"
|
|
cachedManifest "github.com/goharbor/harbor/src/pkg/cached/manifest/redis"
|
|
cachedProject "github.com/goharbor/harbor/src/pkg/cached/project/redis"
|
|
cachedProjectMeta "github.com/goharbor/harbor/src/pkg/cached/project_metadata/redis"
|
|
cachedRepo "github.com/goharbor/harbor/src/pkg/cached/repository/redis"
|
|
"github.com/goharbor/harbor/src/pkg/project"
|
|
"github.com/goharbor/harbor/src/pkg/project/metadata"
|
|
"github.com/goharbor/harbor/src/pkg/repository"
|
|
|
|
// init the db config
|
|
_ "github.com/goharbor/harbor/src/pkg/config/db"
|
|
// init the rest config
|
|
_ "github.com/goharbor/harbor/src/pkg/config/rest"
|
|
)
|
|
|
|
// Define global resource manager.
|
|
var (
|
|
// ArtifactMgr is the manager for artifact.
|
|
ArtifactMgr artifact.Manager
|
|
// ProjectMgr is the manager for project.
|
|
ProjectMgr project.Manager
|
|
// ProjectMetaMgr is the manager for project metadata.
|
|
ProjectMetaMgr metadata.Manager
|
|
// RepositoryMgr is the manager for repository.
|
|
RepositoryMgr repository.Manager
|
|
// ManifestMgr is the manager for manifest.
|
|
ManifestMgr cachedManifest.CachedManager
|
|
)
|
|
|
|
// init initialize mananger for resources
|
|
func init() {
|
|
cacheEnabled := config.CacheEnabled()
|
|
initArtifactMgr(cacheEnabled)
|
|
initProjectMgr(cacheEnabled)
|
|
initProjectMetaMgr(cacheEnabled)
|
|
initRepositoryMgr(cacheEnabled)
|
|
initManifestManager(cacheEnabled)
|
|
}
|
|
|
|
func initArtifactMgr(cacheEnabled bool) {
|
|
artMgr := artifact.NewManager()
|
|
// check cache enable
|
|
if cacheEnabled {
|
|
ArtifactMgr = cachedArtifact.NewManager(artMgr)
|
|
} else {
|
|
ArtifactMgr = artMgr
|
|
}
|
|
}
|
|
|
|
func initProjectMgr(cacheEnabled bool) {
|
|
projectMgr := project.New()
|
|
// check cache enable
|
|
if cacheEnabled {
|
|
ProjectMgr = cachedProject.NewManager(projectMgr)
|
|
} else {
|
|
ProjectMgr = projectMgr
|
|
}
|
|
}
|
|
|
|
func initProjectMetaMgr(cacheEnabled bool) {
|
|
projectMetaMgr := metadata.New()
|
|
if cacheEnabled {
|
|
ProjectMetaMgr = cachedProjectMeta.NewManager(projectMetaMgr)
|
|
} else {
|
|
ProjectMetaMgr = projectMetaMgr
|
|
}
|
|
}
|
|
|
|
func initRepositoryMgr(cacheEnabled bool) {
|
|
repoMgr := repository.New()
|
|
if cacheEnabled {
|
|
RepositoryMgr = cachedRepo.NewManager(repoMgr)
|
|
} else {
|
|
RepositoryMgr = repoMgr
|
|
}
|
|
}
|
|
|
|
func initManifestManager(_ bool) {
|
|
ManifestMgr = cachedManifest.NewManager()
|
|
}
|