mirror of https://github.com/goharbor/harbor.git
102 lines
2.9 KiB
Go
102 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 retention
|
|
|
|
import (
|
|
"context"
|
|
"encoding/json"
|
|
"fmt"
|
|
"time"
|
|
|
|
"github.com/beego/beego/v2/client/orm"
|
|
|
|
"github.com/goharbor/harbor/src/pkg/retention/dao"
|
|
"github.com/goharbor/harbor/src/pkg/retention/dao/models"
|
|
"github.com/goharbor/harbor/src/pkg/retention/policy"
|
|
)
|
|
|
|
// Manager defines operations of managing policy
|
|
type Manager interface {
|
|
// Create new policy and return ID
|
|
CreatePolicy(ctx context.Context, p *policy.Metadata) (int64, error)
|
|
// Update the existing policy
|
|
// Full update
|
|
UpdatePolicy(ctx context.Context, p *policy.Metadata) error
|
|
// Delete the specified policy
|
|
// No actual use so far
|
|
DeletePolicy(ctx context.Context, id int64) error
|
|
// Get the specified policy
|
|
GetPolicy(ctx context.Context, id int64) (*policy.Metadata, error)
|
|
}
|
|
|
|
// DefaultManager ...
|
|
type DefaultManager struct {
|
|
}
|
|
|
|
// CreatePolicy Create Policy
|
|
func (d *DefaultManager) CreatePolicy(ctx context.Context, p *policy.Metadata) (int64, error) {
|
|
p1 := &models.RetentionPolicy{}
|
|
p1.ScopeLevel = p.Scope.Level
|
|
p1.ScopeReference = p.Scope.Reference
|
|
p1.TriggerKind = p.Trigger.Kind
|
|
data, _ := json.Marshal(p)
|
|
p1.Data = string(data)
|
|
p1.CreateTime = time.Now()
|
|
p1.UpdateTime = p1.CreateTime
|
|
return dao.CreatePolicy(ctx, p1)
|
|
}
|
|
|
|
// UpdatePolicy Update Policy
|
|
func (d *DefaultManager) UpdatePolicy(ctx context.Context, p *policy.Metadata) error {
|
|
p1 := &models.RetentionPolicy{}
|
|
p1.ID = p.ID
|
|
p1.ScopeLevel = p.Scope.Level
|
|
p1.ScopeReference = p.Scope.Reference
|
|
p1.TriggerKind = p.Trigger.Kind
|
|
p.ID = 0
|
|
data, _ := json.Marshal(p)
|
|
p.ID = p1.ID
|
|
p1.Data = string(data)
|
|
p1.UpdateTime = time.Now()
|
|
return dao.UpdatePolicy(ctx, p1, "scope_level", "trigger_kind", "data", "update_time")
|
|
}
|
|
|
|
// DeletePolicy Delete Policy
|
|
func (d *DefaultManager) DeletePolicy(ctx context.Context, id int64) error {
|
|
return dao.DeletePolicy(ctx, id)
|
|
}
|
|
|
|
// GetPolicy Get Policy
|
|
func (d *DefaultManager) GetPolicy(ctx context.Context, id int64) (*policy.Metadata, error) {
|
|
p1, err := dao.GetPolicy(ctx, id)
|
|
if err != nil {
|
|
if err == orm.ErrNoRows {
|
|
return nil, fmt.Errorf("no such Retention policy with id %v", id)
|
|
}
|
|
return nil, err
|
|
}
|
|
p := &policy.Metadata{}
|
|
if err = json.Unmarshal([]byte(p1.Data), p); err != nil {
|
|
return nil, err
|
|
}
|
|
p.ID = id
|
|
return p, nil
|
|
}
|
|
|
|
// NewManager ...
|
|
func NewManager() Manager {
|
|
return &DefaultManager{}
|
|
}
|