/
ticket.go
94 lines (83 loc) · 2.83 KB
/
ticket.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
90
91
92
93
94
// Copyright (c) 2021 Terminus, Inc.
//
// This program is free software: you can use, redistribute, and/or modify
// it under the terms of the GNU Affero General Public License, version 3
// or later ("AGPL"), as published by the Free Software Foundation.
//
// This program is distributed in the hope that it will be useful, but WITHOUT
// ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
// FITNESS FOR A PARTICULAR PURPOSE.
//
// You should have received a copy of the GNU Affero General Public License
// along with this program. If not, see <http://www.gnu.org/licenses/>.
package dao
import (
"github.com/jinzhu/gorm"
"github.com/erda-project/erda/apistructs"
"github.com/erda-project/erda/modules/dop/model"
)
// CreateTicket 创建工单
func (client *DBClient) CreateTicket(ticket *model.Ticket) error {
return client.Create(ticket).Error
}
// UpdateTicket 更新工单
func (client *DBClient) UpdateTicket(ticket *model.Ticket) error {
return client.Save(ticket).Error
}
// GetTicket 获取工单详情
func (client *DBClient) GetTicket(ticketID int64) (*model.Ticket, error) {
var ticket model.Ticket
if err := client.Where("id = ?", ticketID).Find(&ticket).Error; err != nil {
return nil, err
}
return &ticket, nil
}
// GetOpenTicketByKey 获取open状态的工单
func (client *DBClient) GetOpenTicketByKey(key string) (*model.Ticket, error) {
var ticket model.Ticket
if err := client.Where("`key` = ?", key).Where("status = ?", apistructs.TicketOpen).
Find(&ticket).Error; err != nil {
if gorm.IsRecordNotFoundError(err) {
return nil, nil
}
return nil, err
}
return &ticket, nil
}
// GetTicketByRequestID 根据requestID header获取记录
func (client *DBClient) GetTicketByRequestID(requestID string) (*model.Ticket, error) {
var ticket model.Ticket
if err := client.Where("request_id = ?", requestID).Find(&ticket).Error; err != nil {
if gorm.IsRecordNotFoundError(err) {
return nil, nil
}
return nil, err
}
return &ticket, nil
}
// GetClusterOpenTicketsNum 获取指定集群open状态的工单总数
func (client *DBClient) GetClusterOpenTicketsNum(ticketType, targetType, targetID string) (uint64, error) {
var ticketsCount uint64
if err := client.Model(&model.Ticket{}).
Where("type = ?", ticketType).
Where("target_id = ?", targetID).
Where("target_type = ?", targetType).
Where("status = ?", apistructs.TicketOpen).
Count(&ticketsCount).
Error; err != nil {
return 0, err
}
return ticketsCount, nil
}
// DeleteTicket 删除工单
func (client *DBClient) DeleteTicket(targetID, targetType, ticketType string) error {
var m model.Ticket
//err := client.Model(&model.Ticket{}).
client.LogMode(true)
err := client.Debug().Table("ps_tickets").
Where("type = ?", ticketType).
Where("target_id = ?", targetID).
Where("target_type = ?", targetType).
Delete(&m).Error
return err
}