Merge pull request #12512 from ywk253100/200717_summary

Include the registry info in the project summary API for proxy cache project
This commit is contained in:
stonezdj(Daojun Zhang) 2020-07-29 14:39:30 +08:00 committed by GitHub
commit 5b1b94f25c
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
6 changed files with 23 additions and 13 deletions

View File

@ -3877,6 +3877,8 @@ definitions:
used:
$ref: "#/definitions/ResourceList"
description: The used status of the quota
registry:
$ref: "#/definitions/Registry"
User:
type: object
properties:

View File

@ -19,6 +19,7 @@ import (
"time"
"github.com/goharbor/harbor/src/pkg/quota/types"
"github.com/goharbor/harbor/src/replication/model"
)
const (
@ -212,4 +213,5 @@ type ProjectSummary struct {
LimitedGuestCount int64 `json:"limited_guest_count"`
Quota *QuotaSummary `json:"quota,omitempty"`
Registry *model.Registry `json:"registry"`
}

View File

@ -593,6 +593,18 @@ func (p *ProjectAPI) Summary() {
}
wg.Wait()
if p.project.RegistryID > 0 {
registry, err := replication.RegistryMgr.Get(p.project.RegistryID)
if err != nil {
log.Warningf("failed to get registry %d: %v", p.project.RegistryID, err)
} else {
if registry != nil {
registry.Credential = nil
summary.Registry = registry
}
}
}
p.Data["json"] = summary
p.ServeJSON()
}

View File

@ -20,7 +20,6 @@ import (
"net/http"
"strconv"
common_model "github.com/goharbor/harbor/src/common/models"
"github.com/goharbor/harbor/src/replication"
"github.com/goharbor/harbor/src/replication/dao/models"
"github.com/goharbor/harbor/src/replication/event"
@ -58,10 +57,8 @@ func (r *ReplicationPolicyAPI) List() {
// TODO: support more query
query := &model.PolicyQuery{
Name: r.GetString("name"),
Pagination: common_model.Pagination{
Page: page,
Size: size,
},
}
total, policies, err := replication.PolicyCtl.List(query)

View File

@ -3,7 +3,6 @@ package dao
import (
"testing"
common_models "github.com/goharbor/harbor/src/common/models"
"github.com/goharbor/harbor/src/replication/dao/models"
"github.com/goharbor/harbor/src/replication/model"
"github.com/stretchr/testify/assert"
@ -106,11 +105,9 @@ func TestGetPolicies(t *testing.T) {
{
Name: tt.args.name,
Namespace: tt.args.namespace,
Pagination: common_models.Pagination{
Page: tt.args.page,
Size: tt.args.pageSize,
},
},
}...)
if tt.wantErr {
require.NotNil(t, err, "wantErr: %s", err)

View File

@ -17,7 +17,6 @@ package model
import (
"fmt"
"github.com/astaxie/beego/validation"
"github.com/goharbor/harbor/src/common/models"
"github.com/robfig/cron"
"time"
)
@ -168,5 +167,6 @@ type PolicyQuery struct {
Namespace string
SrcRegistry int64
DestRegistry int64
models.Pagination
Page int64
Size int64
}