harbor/dao/replication_job.go

234 lines
5.2 KiB
Go
Raw Normal View History

2016-05-10 13:38:50 +02:00
package dao
import (
"fmt"
2016-05-16 07:57:30 +02:00
2016-05-27 09:04:20 +02:00
"strings"
2016-05-10 13:38:50 +02:00
"github.com/astaxie/beego/orm"
"github.com/vmware/harbor/models"
)
2016-05-27 12:46:07 +02:00
// AddRepTarget ...
2016-05-10 13:38:50 +02:00
func AddRepTarget(target models.RepTarget) (int64, error) {
2016-06-07 07:41:02 +02:00
o := GetOrmer()
2016-05-10 13:38:50 +02:00
return o.Insert(&target)
}
2016-05-27 12:46:07 +02:00
// GetRepTarget ...
2016-05-10 13:38:50 +02:00
func GetRepTarget(id int64) (*models.RepTarget, error) {
2016-06-07 07:41:02 +02:00
o := GetOrmer()
2016-05-10 13:38:50 +02:00
t := models.RepTarget{ID: id}
err := o.Read(&t)
if err == orm.ErrNoRows {
return nil, nil
}
return &t, err
}
2016-05-27 12:46:07 +02:00
// GetRepTargetByName ...
func GetRepTargetByName(name string) (*models.RepTarget, error) {
2016-06-07 07:41:02 +02:00
o := GetOrmer()
t := models.RepTarget{Name: name}
err := o.Read(&t, "Name")
if err == orm.ErrNoRows {
return nil, nil
}
return &t, err
}
2016-05-27 12:46:07 +02:00
// DeleteRepTarget ...
2016-05-10 13:38:50 +02:00
func DeleteRepTarget(id int64) error {
2016-06-07 07:41:02 +02:00
o := GetOrmer()
2016-05-10 13:38:50 +02:00
_, err := o.Delete(&models.RepTarget{ID: id})
return err
}
2016-05-27 12:46:07 +02:00
// UpdateRepTarget ...
func UpdateRepTarget(target models.RepTarget) error {
2016-06-07 07:41:02 +02:00
o := GetOrmer()
_, err := o.Update(&target, "URL", "Name", "Username", "Password")
return err
}
2016-05-27 12:46:07 +02:00
// GetAllRepTargets ...
2016-05-23 12:41:48 +02:00
func GetAllRepTargets() ([]*models.RepTarget, error) {
2016-06-07 07:41:02 +02:00
o := GetOrmer()
2016-06-13 02:31:32 +02:00
2016-05-23 10:48:55 +02:00
qs := o.QueryTable(&models.RepTarget{})
2016-05-23 12:41:48 +02:00
var targets []*models.RepTarget
_, err := qs.All(&targets)
2016-05-23 10:48:55 +02:00
return targets, err
}
2016-06-13 02:31:32 +02:00
// FilterRepTargets filters targets by name
func FilterRepTargets(name string) ([]*models.RepTarget, error) {
if len(name) == 0 {
return GetAllRepTargets()
}
o := GetOrmer()
var targets []*models.RepTarget
sql := "select * from replication_target where name like ?"
if _, err := o.Raw(sql, "%"+name+"%").QueryRows(targets); err != nil {
return nil, err
}
return targets, nil
}
2016-05-27 12:46:07 +02:00
// AddRepPolicy ...
2016-05-10 13:38:50 +02:00
func AddRepPolicy(policy models.RepPolicy) (int64, error) {
2016-06-07 07:41:02 +02:00
o := GetOrmer()
2016-05-10 13:38:50 +02:00
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
}
2016-05-27 12:46:07 +02:00
// GetRepPolicy ...
2016-05-10 13:38:50 +02:00
func GetRepPolicy(id int64) (*models.RepPolicy, error) {
2016-06-07 07:41:02 +02:00
o := GetOrmer()
2016-05-10 13:38:50 +02:00
p := models.RepPolicy{ID: id}
err := o.Read(&p)
if err == orm.ErrNoRows {
return nil, nil
}
return &p, err
}
2016-05-27 12:46:07 +02:00
// GetRepPolicyByName ...
func GetRepPolicyByName(name string) (*models.RepPolicy, error) {
2016-06-07 07:41:02 +02:00
o := GetOrmer()
p := models.RepPolicy{Name: name}
err := o.Read(&p, "Name")
if err == orm.ErrNoRows {
return nil, nil
}
return &p, err
}
2016-05-27 12:46:07 +02:00
// GetRepPolicyByProject ...
func GetRepPolicyByProject(projectID int64) ([]*models.RepPolicy, error) {
var res []*models.RepPolicy
2016-06-07 07:41:02 +02:00
o := GetOrmer()
_, err := o.QueryTable("replication_policy").Filter("project_id", projectID).All(&res)
return res, err
}
2016-05-27 12:46:07 +02:00
// DeleteRepPolicy ...
2016-05-10 13:38:50 +02:00
func DeleteRepPolicy(id int64) error {
2016-06-07 07:41:02 +02:00
o := GetOrmer()
2016-05-10 13:38:50 +02:00
_, err := o.Delete(&models.RepPolicy{ID: id})
return err
}
2016-05-27 12:46:07 +02:00
// UpdateRepPolicyEnablement ...
2016-05-27 09:04:20 +02:00
func UpdateRepPolicyEnablement(id int64, enabled int) error {
2016-06-07 07:41:02 +02:00
o := GetOrmer()
2016-05-10 13:38:50 +02:00
p := models.RepPolicy{
ID: id,
Enabled: enabled}
2016-05-27 09:04:20 +02:00
_, err := o.Update(&p, "Enabled")
2016-05-10 13:38:50 +02:00
return err
}
2016-05-27 12:46:07 +02:00
// EnableRepPolicy ...
2016-05-10 13:38:50 +02:00
func EnableRepPolicy(id int64) error {
2016-05-27 09:04:20 +02:00
return UpdateRepPolicyEnablement(id, 1)
2016-05-10 13:38:50 +02:00
}
2016-05-27 12:46:07 +02:00
// DisableRepPolicy ...
2016-05-10 13:38:50 +02:00
func DisableRepPolicy(id int64) error {
2016-05-27 09:04:20 +02:00
return UpdateRepPolicyEnablement(id, 0)
2016-05-10 13:38:50 +02:00
}
2016-05-27 12:46:07 +02:00
// AddRepJob ...
2016-05-10 13:38:50 +02:00
func AddRepJob(job models.RepJob) (int64, error) {
2016-06-07 07:41:02 +02:00
o := GetOrmer()
2016-05-10 13:38:50 +02:00
if len(job.Status) == 0 {
job.Status = models.JobPending
}
if len(job.TagList) > 0 {
job.Tags = strings.Join(job.TagList, ",")
}
2016-05-10 13:38:50 +02:00
return o.Insert(&job)
}
2016-05-27 12:46:07 +02:00
// GetRepJob ...
2016-05-10 13:38:50 +02:00
func GetRepJob(id int64) (*models.RepJob, error) {
2016-06-07 07:41:02 +02:00
o := GetOrmer()
2016-05-10 13:38:50 +02:00
j := models.RepJob{ID: id}
err := o.Read(&j)
if err == orm.ErrNoRows {
return nil, nil
}
genTagListForJob(&j)
return &j, nil
2016-05-10 13:38:50 +02:00
}
2016-05-27 12:46:07 +02:00
// GetRepJobByPolicy ...
func GetRepJobByPolicy(policyID int64) ([]*models.RepJob, error) {
var res []*models.RepJob
_, err := repJobPolicyIDQs(policyID).All(&res)
genTagListForJob(res...)
return res, err
}
// 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 repJobPolicyIDQs(policyID int64) orm.QuerySeter {
2016-06-07 07:41:02 +02:00
o := GetOrmer()
return o.QueryTable("replication_job").Filter("policy_id", policyID)
}
2016-05-27 12:46:07 +02:00
// DeleteRepJob ...
2016-05-10 13:38:50 +02:00
func DeleteRepJob(id int64) error {
2016-06-07 07:41:02 +02:00
o := GetOrmer()
2016-05-10 13:38:50 +02:00
_, err := o.Delete(&models.RepJob{ID: id})
return err
}
2016-05-27 12:46:07 +02:00
// UpdateRepJobStatus ...
2016-05-10 13:38:50 +02:00
func UpdateRepJobStatus(id int64, status string) error {
2016-06-07 07:41:02 +02:00
o := GetOrmer()
2016-05-10 13:38:50 +02:00
j := models.RepJob{
ID: id,
Status: status,
}
num, err := o.Update(&j, "Status")
if num == 0 {
2016-05-16 07:57:30 +02:00
err = fmt.Errorf("Failed to update replication job with id: %d %s", id, err.Error())
2016-05-10 13:38:50 +02:00
}
return err
}
func genTagListForJob(jobs ...*models.RepJob) {
for _, j := range jobs {
if len(j.Tags) > 0 {
j.TagList = strings.Split(j.Tags, ",")
}
}
}