Use project controller rather than the manager in API handlers (#10946)

Use project controller rather than the manager in API handlers

Signed-off-by: Wenkai Yin <yinw@vmware.com>
This commit is contained in:
He Weiwei 2020-03-05 18:58:47 +08:00 committed by GitHub
parent 84bd30c570
commit d21318dfcf
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 4 additions and 8 deletions

View File

@ -32,7 +32,6 @@ import (
"github.com/goharbor/harbor/src/common/rbac"
"github.com/goharbor/harbor/src/common/utils"
ierror "github.com/goharbor/harbor/src/internal/error"
"github.com/goharbor/harbor/src/pkg/project"
"github.com/goharbor/harbor/src/pkg/q"
v1 "github.com/goharbor/harbor/src/pkg/scan/rest/v1"
"github.com/goharbor/harbor/src/server/v2.0/handler/assembler"
@ -48,7 +47,6 @@ const (
func newArtifactAPI() *artifactAPI {
return &artifactAPI{
artCtl: artifact.Ctl,
proMgr: project.Mgr,
repoCtl: repository.Ctl,
scanCtl: scan.DefaultController,
tagCtl: tag.Ctl,
@ -58,7 +56,6 @@ func newArtifactAPI() *artifactAPI {
type artifactAPI struct {
BaseAPI
artCtl artifact.Controller
proMgr project.Manager
repoCtl repository.Controller
scanCtl scan.Controller
tagCtl tag.Controller

View File

@ -19,11 +19,11 @@ import (
"fmt"
"github.com/go-openapi/runtime/middleware"
"github.com/goharbor/harbor/src/api/artifact"
"github.com/goharbor/harbor/src/api/project"
"github.com/goharbor/harbor/src/api/repository"
cmodels "github.com/goharbor/harbor/src/common/models"
"github.com/goharbor/harbor/src/common/rbac"
"github.com/goharbor/harbor/src/common/utils/log"
"github.com/goharbor/harbor/src/pkg/project"
"github.com/goharbor/harbor/src/pkg/q"
"github.com/goharbor/harbor/src/server/v2.0/models"
operation "github.com/goharbor/harbor/src/server/v2.0/restapi/operations/repository"
@ -31,7 +31,7 @@ import (
func newRepositoryAPI() *repositoryAPI {
return &repositoryAPI{
proMgr: project.Mgr,
proCtl: project.Ctl,
repoCtl: repository.Ctl,
artCtl: artifact.Ctl,
}
@ -39,8 +39,7 @@ func newRepositoryAPI() *repositoryAPI {
type repositoryAPI struct {
BaseAPI
// TODO replace proMgr with proCtl
proMgr project.Manager
proCtl project.Controller
repoCtl repository.Controller
artCtl artifact.Controller
}
@ -49,7 +48,7 @@ func (r *repositoryAPI) ListRepositories(ctx context.Context, params operation.L
if err := r.RequireProjectAccess(ctx, params.ProjectName, rbac.ActionList, rbac.ResourceRepository); err != nil {
return r.SendError(ctx, err)
}
project, err := r.proMgr.Get(params.ProjectName)
project, err := r.proCtl.GetByName(ctx, params.ProjectName)
if err != nil {
return r.SendError(ctx, err)
}