mirror of
https://github.com/goharbor/harbor.git
synced 2024-11-24 03:05:39 +01:00
424 lines
10 KiB
Go
424 lines
10 KiB
Go
/*
|
|
Copyright (c) 2016 VMware, Inc. All Rights Reserved.
|
|
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 dao
|
|
|
|
import (
|
|
"fmt"
|
|
"time"
|
|
|
|
"strings"
|
|
|
|
"github.com/astaxie/beego/orm"
|
|
"github.com/vmware/harbor/models"
|
|
)
|
|
|
|
// AddRepTarget ...
|
|
func AddRepTarget(target models.RepTarget) (int64, error) {
|
|
o := GetOrmer()
|
|
return o.Insert(&target)
|
|
}
|
|
|
|
// GetRepTarget ...
|
|
func GetRepTarget(id int64) (*models.RepTarget, error) {
|
|
o := GetOrmer()
|
|
t := models.RepTarget{ID: id}
|
|
err := o.Read(&t)
|
|
if err == orm.ErrNoRows {
|
|
return nil, nil
|
|
}
|
|
return &t, err
|
|
}
|
|
|
|
// GetRepTargetByName ...
|
|
func GetRepTargetByName(name string) (*models.RepTarget, error) {
|
|
o := GetOrmer()
|
|
t := models.RepTarget{Name: name}
|
|
err := o.Read(&t, "Name")
|
|
if err == orm.ErrNoRows {
|
|
return nil, nil
|
|
}
|
|
return &t, err
|
|
}
|
|
|
|
// GetRepTargetByConnInfo ...
|
|
func GetRepTargetByConnInfo(endpoint, username string) (*models.RepTarget, error) {
|
|
o := GetOrmer()
|
|
t := models.RepTarget{
|
|
URL: endpoint,
|
|
Username: username,
|
|
}
|
|
err := o.Read(&t, "URL", "Username")
|
|
if err == orm.ErrNoRows {
|
|
return nil, nil
|
|
}
|
|
return &t, err
|
|
}
|
|
|
|
// DeleteRepTarget ...
|
|
func DeleteRepTarget(id int64) error {
|
|
o := GetOrmer()
|
|
_, err := o.Delete(&models.RepTarget{ID: id})
|
|
return err
|
|
}
|
|
|
|
// UpdateRepTarget ...
|
|
func UpdateRepTarget(target models.RepTarget) error {
|
|
o := GetOrmer()
|
|
_, err := o.Update(&target, "URL", "Name", "Username", "Password")
|
|
return err
|
|
}
|
|
|
|
// FilterRepTargets filters targets by name
|
|
func FilterRepTargets(name string) ([]*models.RepTarget, error) {
|
|
o := GetOrmer()
|
|
|
|
var args []interface{}
|
|
|
|
sql := `select * from replication_target `
|
|
if len(name) != 0 {
|
|
sql += `where name like ? `
|
|
args = append(args, "%"+name+"%")
|
|
}
|
|
sql += `order by creation_time`
|
|
|
|
var targets []*models.RepTarget
|
|
|
|
if _, err := o.Raw(sql, args).QueryRows(&targets); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return targets, nil
|
|
}
|
|
|
|
// AddRepPolicy ...
|
|
func AddRepPolicy(policy models.RepPolicy) (int64, error) {
|
|
o := GetOrmer()
|
|
sqlTpl := `insert into replication_policy (name, project_id, target_id, enabled, description, cron_str, start_time, creation_time, update_time ) values (?, ?, ?, ?, ?, ?, %s, NOW(), NOW())`
|
|
var sql string
|
|
if policy.Enabled == 1 {
|
|
sql = fmt.Sprintf(sqlTpl, "NOW()")
|
|
} else {
|
|
sql = fmt.Sprintf(sqlTpl, "NULL")
|
|
}
|
|
p, err := o.Raw(sql).Prepare()
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
r, err := p.Exec(policy.Name, policy.ProjectID, policy.TargetID, policy.Enabled, policy.Description, policy.CronStr)
|
|
if err != nil {
|
|
return 0, err
|
|
}
|
|
id, err := r.LastInsertId()
|
|
return id, err
|
|
}
|
|
|
|
// GetRepPolicy ...
|
|
func GetRepPolicy(id int64) (*models.RepPolicy, error) {
|
|
o := GetOrmer()
|
|
sql := `select * from replication_policy where id = ?`
|
|
|
|
var policy models.RepPolicy
|
|
|
|
if err := o.Raw(sql, id).QueryRow(&policy); err != nil {
|
|
if err == orm.ErrNoRows {
|
|
return nil, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
return &policy, nil
|
|
}
|
|
|
|
// FilterRepPolicies filters policies by name and project ID
|
|
func FilterRepPolicies(name string, projectID int64) ([]*models.RepPolicy, error) {
|
|
o := GetOrmer()
|
|
|
|
var args []interface{}
|
|
|
|
sql := `select rp.id, rp.project_id, p.name as project_name, rp.target_id,
|
|
rt.name as target_name, rp.name, rp.enabled, rp.description,
|
|
rp.cron_str, rp.start_time, rp.creation_time, rp.update_time,
|
|
count(rj.status) as error_job_count
|
|
from replication_policy rp
|
|
left join project p on rp.project_id=p.project_id
|
|
left join replication_target rt on rp.target_id=rt.id
|
|
left join replication_job rj on rp.id=rj.policy_id and (rj.status="error"
|
|
or rj.status="retrying") `
|
|
|
|
if len(name) != 0 && projectID != 0 {
|
|
sql += `where rp.name like ? and rp.project_id = ? `
|
|
args = append(args, "%"+name+"%")
|
|
args = append(args, projectID)
|
|
} else if len(name) != 0 {
|
|
sql += `where rp.name like ? `
|
|
args = append(args, "%"+name+"%")
|
|
} else if projectID != 0 {
|
|
sql += `where rp.project_id = ? `
|
|
args = append(args, projectID)
|
|
}
|
|
|
|
sql += `group by rp.id order by rp.creation_time`
|
|
|
|
var policies []*models.RepPolicy
|
|
if _, err := o.Raw(sql, args).QueryRows(&policies); err != nil {
|
|
return nil, err
|
|
}
|
|
return policies, nil
|
|
}
|
|
|
|
// GetRepPolicyByName ...
|
|
func GetRepPolicyByName(name string) (*models.RepPolicy, error) {
|
|
o := GetOrmer()
|
|
sql := `select * from replication_policy where name = ?`
|
|
|
|
var policy models.RepPolicy
|
|
|
|
if err := o.Raw(sql, name).QueryRow(&policy); err != nil {
|
|
if err == orm.ErrNoRows {
|
|
return nil, nil
|
|
}
|
|
return nil, err
|
|
}
|
|
|
|
return &policy, nil
|
|
}
|
|
|
|
// GetRepPolicyByProject ...
|
|
func GetRepPolicyByProject(projectID int64) ([]*models.RepPolicy, error) {
|
|
o := GetOrmer()
|
|
sql := `select * from replication_policy where project_id = ?`
|
|
|
|
var policies []*models.RepPolicy
|
|
|
|
if _, err := o.Raw(sql, projectID).QueryRows(&policies); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return policies, nil
|
|
}
|
|
|
|
// GetRepPolicyByTarget ...
|
|
func GetRepPolicyByTarget(targetID int64) ([]*models.RepPolicy, error) {
|
|
o := GetOrmer()
|
|
sql := `select * from replication_policy where target_id = ?`
|
|
|
|
var policies []*models.RepPolicy
|
|
|
|
if _, err := o.Raw(sql, targetID).QueryRows(&policies); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return policies, nil
|
|
}
|
|
|
|
// GetRepPolicyByProjectAndTarget ...
|
|
func GetRepPolicyByProjectAndTarget(projectID, targetID int64) ([]*models.RepPolicy, error) {
|
|
o := GetOrmer()
|
|
sql := `select * from replication_policy where project_id = ? and target_id = ?`
|
|
|
|
var policies []*models.RepPolicy
|
|
|
|
if _, err := o.Raw(sql, projectID, targetID).QueryRows(&policies); err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
return policies, nil
|
|
}
|
|
|
|
// UpdateRepPolicy ...
|
|
func UpdateRepPolicy(policy *models.RepPolicy) error {
|
|
o := GetOrmer()
|
|
_, err := o.Update(policy, "TargetID", "Name", "Enabled", "Description", "CronStr")
|
|
return err
|
|
}
|
|
|
|
// DeleteRepPolicy ...
|
|
func DeleteRepPolicy(id int64) error {
|
|
o := GetOrmer()
|
|
_, err := o.Delete(&models.RepPolicy{ID: id})
|
|
return err
|
|
}
|
|
|
|
// UpdateRepPolicyEnablement ...
|
|
func UpdateRepPolicyEnablement(id int64, enabled int) error {
|
|
o := GetOrmer()
|
|
p := models.RepPolicy{
|
|
ID: id,
|
|
Enabled: enabled,
|
|
}
|
|
|
|
var err error
|
|
if enabled == 1 {
|
|
p.StartTime = time.Now()
|
|
_, err = o.Update(&p, "Enabled", "StartTime")
|
|
} else {
|
|
_, err = o.Update(&p, "Enabled")
|
|
}
|
|
|
|
return err
|
|
}
|
|
|
|
// EnableRepPolicy ...
|
|
func EnableRepPolicy(id int64) error {
|
|
return UpdateRepPolicyEnablement(id, 1)
|
|
}
|
|
|
|
// DisableRepPolicy ...
|
|
func DisableRepPolicy(id int64) error {
|
|
return UpdateRepPolicyEnablement(id, 0)
|
|
}
|
|
|
|
// AddRepJob ...
|
|
func AddRepJob(job models.RepJob) (int64, error) {
|
|
o := GetOrmer()
|
|
if len(job.Status) == 0 {
|
|
job.Status = models.JobPending
|
|
}
|
|
if len(job.TagList) > 0 {
|
|
job.Tags = strings.Join(job.TagList, ",")
|
|
}
|
|
return o.Insert(&job)
|
|
}
|
|
|
|
// GetRepJob ...
|
|
func GetRepJob(id int64) (*models.RepJob, error) {
|
|
o := GetOrmer()
|
|
j := models.RepJob{ID: id}
|
|
err := o.Read(&j)
|
|
if err == orm.ErrNoRows {
|
|
return nil, nil
|
|
}
|
|
genTagListForJob(&j)
|
|
return &j, nil
|
|
}
|
|
|
|
// GetRepJobByPolicy ...
|
|
func GetRepJobByPolicy(policyID int64) ([]*models.RepJob, error) {
|
|
var res []*models.RepJob
|
|
_, err := repJobPolicyIDQs(policyID).All(&res)
|
|
genTagListForJob(res...)
|
|
return res, err
|
|
}
|
|
|
|
// FilterRepJobs filters jobs by repo and policy ID
|
|
func FilterRepJobs(policyID int64, repository, status string, startTime,
|
|
endTime *time.Time, limit int) ([]*models.RepJob, error) {
|
|
o := GetOrmer()
|
|
|
|
qs := o.QueryTable(new(models.RepJob))
|
|
if policyID != 0 {
|
|
qs = qs.Filter("PolicyID", policyID)
|
|
}
|
|
if len(repository) != 0 {
|
|
qs = qs.Filter("Repository__icontains", repository)
|
|
}
|
|
if len(status) != 0 {
|
|
qs = qs.Filter("Status__icontains", status)
|
|
}
|
|
|
|
if startTime != nil {
|
|
fmt.Printf("%v\n", startTime)
|
|
qs = qs.Filter("CreationTime__gte", startTime)
|
|
}
|
|
|
|
if endTime != nil {
|
|
fmt.Printf("%v\n", endTime)
|
|
qs = qs.Filter("CreationTime__lte", endTime)
|
|
}
|
|
|
|
if limit != 0 {
|
|
qs = qs.Limit(limit)
|
|
}
|
|
|
|
qs = qs.OrderBy("-CreationTime")
|
|
|
|
var jobs []*models.RepJob
|
|
_, err := qs.All(&jobs)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
genTagListForJob(jobs...)
|
|
|
|
return jobs, nil
|
|
}
|
|
|
|
// GetRepJobToStop get jobs that are possibly being handled by workers of a certain policy.
|
|
func GetRepJobToStop(policyID int64) ([]*models.RepJob, error) {
|
|
var res []*models.RepJob
|
|
_, err := repJobPolicyIDQs(policyID).Filter("status__in", models.JobPending, models.JobRunning).All(&res)
|
|
genTagListForJob(res...)
|
|
return res, err
|
|
}
|
|
|
|
func repJobQs() orm.QuerySeter {
|
|
o := GetOrmer()
|
|
return o.QueryTable("replication_job")
|
|
}
|
|
|
|
func repJobPolicyIDQs(policyID int64) orm.QuerySeter {
|
|
return repJobQs().Filter("policy_id", policyID)
|
|
}
|
|
|
|
// DeleteRepJob ...
|
|
func DeleteRepJob(id int64) error {
|
|
o := GetOrmer()
|
|
_, err := o.Delete(&models.RepJob{ID: id})
|
|
return err
|
|
}
|
|
|
|
// UpdateRepJobStatus ...
|
|
func UpdateRepJobStatus(id int64, status string) error {
|
|
o := GetOrmer()
|
|
j := models.RepJob{
|
|
ID: id,
|
|
Status: status,
|
|
}
|
|
num, err := o.Update(&j, "Status")
|
|
if num == 0 {
|
|
err = fmt.Errorf("Failed to update replication job with id: %d %s", id, err.Error())
|
|
}
|
|
return err
|
|
}
|
|
|
|
// ResetRunningJobs update all running jobs status to pending
|
|
func ResetRunningJobs() error {
|
|
o := GetOrmer()
|
|
sql := fmt.Sprintf("update replication_job set status = '%s' where status = '%s'", models.JobPending, models.JobRunning)
|
|
_, err := o.Raw(sql).Exec()
|
|
return err
|
|
}
|
|
|
|
// GetRepJobByStatus get jobs of certain statuses
|
|
func GetRepJobByStatus(status ...string) ([]*models.RepJob, error) {
|
|
var res []*models.RepJob
|
|
var t []interface{}
|
|
for _, s := range status {
|
|
t = append(t, interface{}(s))
|
|
}
|
|
_, err := repJobQs().Filter("status__in", t...).All(&res)
|
|
genTagListForJob(res...)
|
|
return res, err
|
|
}
|
|
|
|
func genTagListForJob(jobs ...*models.RepJob) {
|
|
for _, j := range jobs {
|
|
if len(j.Tags) > 0 {
|
|
j.TagList = strings.Split(j.Tags, ",")
|
|
}
|
|
}
|
|
}
|