93 lines
2.3 KiB
Go
93 lines
2.3 KiB
Go
|
package service
|
|||
|
|
|
|||
|
|
import (
|
|||
|
|
"context"
|
|||
|
|
|
|||
|
|
"go-common/app/admin/ep/melloi/model"
|
|||
|
|
"go-common/library/ecode"
|
|||
|
|
"go-common/library/log"
|
|||
|
|
)
|
|||
|
|
|
|||
|
|
// AddLabel create new label
|
|||
|
|
func (s *Service) AddLabel(label *model.Label) (err error) {
|
|||
|
|
if _, err = s.dao.QueryLabel(label); err == nil {
|
|||
|
|
err = ecode.MelloiLabelExistErr
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
label.Active = 1
|
|||
|
|
return s.dao.AddLabel(label)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// QueryLabel query all labels
|
|||
|
|
func (s *Service) QueryLabel(c context.Context) ([]*model.Label, error) {
|
|||
|
|
return s.dao.QueryLabels(c)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// DeleteLabel delete label by id
|
|||
|
|
func (s *Service) DeleteLabel(id int64) error {
|
|||
|
|
if id <= 0 {
|
|||
|
|
return ecode.RequestErr
|
|||
|
|
}
|
|||
|
|
return s.dao.DeleteLabel(id)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// AddLabelRelation create new label relation
|
|||
|
|
func (s *Service) AddLabelRelation(lr *model.LabelRelation) (err error) {
|
|||
|
|
label := &model.Label{ID: lr.LabelID}
|
|||
|
|
if label, err = s.dao.QueryLabel(label); err != nil {
|
|||
|
|
return ecode.RequestErr
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 存在相同的记录
|
|||
|
|
if _, err = s.dao.QueryLabelExist(lr); err != nil {
|
|||
|
|
return ecode.MelloiLabelExistErr
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// 每个脚本|任务，最多有2个label
|
|||
|
|
var lre []*model.LabelRelation
|
|||
|
|
relation := model.LabelRelation{Type: lr.Type, TargetID: lr.TargetID}
|
|||
|
|
if lre, err = s.dao.QueryLabelRelation(&relation); err != nil {
|
|||
|
|
return err
|
|||
|
|
}
|
|||
|
|
if len(lre) >= 2 {
|
|||
|
|
return ecode.MelloiLabelCountErr
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
lr.Description = label.Description
|
|||
|
|
lr.Color = label.Color
|
|||
|
|
lr.LabelName = label.Name
|
|||
|
|
lr.Active = 1
|
|||
|
|
if err = s.dao.AddLabelRelation(lr); err != nil {
|
|||
|
|
log.Error("s.dao.AddLabelRelation err :(%v)", err)
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// DeleteLabelRelation delete label relation by id
|
|||
|
|
func (s *Service) DeleteLabelRelation(id int64) (err error) {
|
|||
|
|
if id <= 0 {
|
|||
|
|
return ecode.RequestErr
|
|||
|
|
}
|
|||
|
|
// 标签不存在
|
|||
|
|
var exist bool
|
|||
|
|
if exist, err = s.dao.CheckLabelRelationExist(id); err != nil {
|
|||
|
|
return
|
|||
|
|
}
|
|||
|
|
if !exist {
|
|||
|
|
return ecode.MelloiLabelRelationNotExist
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
return s.dao.DeleteLabelRelation(id)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// QueryLabelRelation query label relation by id , type, targetid
|
|||
|
|
func (s *Service) QueryLabelRelation(lre *model.LabelRelation) (lr []*model.LabelRelation, err error) {
|
|||
|
|
return s.dao.QueryLabelRelation(lre)
|
|||
|
|
}
|
|||
|
|
|
|||
|
|
// QueryLabelRelationByIDs query label relation by ids
|
|||
|
|
func (s *Service) QueryLabelRelationByIDs(ids []int64) (lr []*model.LabelRelation, err error) {
|
|||
|
|
return s.dao.QueryLabelRelationByIDs(ids)
|
|||
|
|
}
|