/
label_relation.go
89 lines (77 loc) · 2.97 KB
/
label_relation.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
// Copyright (c) 2021 Terminus, Inc.
//
// 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 "github.com/erda-project/erda/apistructs"
// LabelRelation 标签关联关系
type LabelRelation struct {
BaseModel
LabelID uint64 `json:"label_id"` // 标签 id
RefType apistructs.ProjectLabelType `json:"ref_type"` // 标签作用类型, eg: issue
RefID uint64 `json:"ref_id"` // 标签关联目标 id
}
// TableName 表名
func (LabelRelation) TableName() string {
return "dice_label_relations"
}
// CreateLabelRelation 创建标签关联关系
func (client *DBClient) CreateLabelRelation(lr *LabelRelation) error {
return client.Create(lr).Error
}
// DeleteLabelRelations 删除标签关联关系
func (client *DBClient) DeleteLabelRelations(refType apistructs.ProjectLabelType, refID uint64) error {
return client.Where("ref_type = ?", refType).Where("ref_id = ?", refID).
Delete(LabelRelation{}).Error
}
// DeleteLabelRelationsByLabel 根据 labelID 删除标签
func (client *DBClient) DeleteLabelRelationsByLabel(labelID uint64) error {
return client.Where("label_id = ?", labelID).Delete(LabelRelation{}).Error
}
// GetLabelRelationsByRef 获取标签关联关系列表
func (client *DBClient) GetLabelRelationsByRef(refType apistructs.ProjectLabelType, refID uint64) ([]LabelRelation, error) {
var lrs []LabelRelation
if err := client.Where("ref_type = ?", refType).Where("ref_id = ?", refID).
Find(&lrs).Error; err != nil {
return nil, err
}
return lrs, nil
}
// GetLabelRelationsByLabels 获取标签关联关系列表
func (client *DBClient) GetLabelRelationsByLabels(refType apistructs.ProjectLabelType, labelIDs []uint64) ([]LabelRelation, error) {
var lrs []LabelRelation
if err := client.Debug().Where("ref_type = ?", refType).Where("label_id in (?)", labelIDs).
Find(&lrs).Error; err != nil {
return nil, err
}
return lrs, nil
}
// BatchQueryIssueLabelIDMap 批量查询 issue label id
func (client *DBClient) BatchQueryIssueLabelIDMap(issueIDs []int64) (map[uint64][]uint64, error) {
if len(issueIDs) == 0 {
return nil, nil
}
var refs []LabelRelation
sql := client.Where("`ref_type` = ?", "issue").Where("`ref_id` IN (?)", issueIDs).Find(&refs)
if err := sql.Error; err != nil {
return nil, err
}
if len(refs) == 0 {
return nil, nil
}
// key: issueID, value: labelIDs
m := make(map[uint64][]uint64, len(issueIDs))
for _, ref := range refs {
m[ref.RefID] = append(m[ref.RefID], ref.LabelID)
}
return m, nil
}