/
pipeline.go
80 lines (68 loc) · 2.09 KB
/
pipeline.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
/*
* Copyright (c) 2018, 奶爸<1@5.nu>
* All rights reserved.
*/
package sqlite3
import (
"github.com/jinzhu/gorm"
"github.com/naiba/nocd"
)
//PipelineService Sqlite3的用户服务实现
type PipelineService struct {
DB *gorm.DB
}
//Create 创建用户
func (ps *PipelineService) Create(p *nocd.Pipeline) error {
return ps.DB.Create(p).Error
}
//Update 更新用户
func (ps *PipelineService) Update(p *nocd.Pipeline) error {
return ps.DB.Save(p).Error
}
//Delete 删除用户
func (ps *PipelineService) Delete(pid uint) error {
db := ps.DB.Begin()
err := db.Where("pipeline_id = ?", pid).Delete(nocd.PipeLog{}).Error
if err != nil {
db.Rollback()
return err
}
err = db.Where("id = ?", pid).Delete(nocd.Pipeline{}).Error
if err != nil {
db.Rollback()
return err
}
db.Commit()
return nil
}
//UserPipelines 获取用户的所有部署流
func (ps *PipelineService) UserPipelines(u *nocd.User) []nocd.Pipeline {
var p []nocd.Pipeline
ps.DB.Model(u).Related(&p)
return p
}
//UserPipeline 根据用户ID和部署流ID获取部署流,通常用来检测用户对部署流的操作权限
func (ps *PipelineService) UserPipeline(uid, pid uint) (nocd.Pipeline, error) {
var p nocd.Pipeline
err := ps.DB.Where("user_id = ? AND id = ?", uid, pid).First(&p).Error
return p, err
}
//RepoPipelines 获取项目下面的所有部署流
func (ps *PipelineService) RepoPipelines(r *nocd.Repository) []nocd.Pipeline {
var p []nocd.Pipeline
ps.DB.Model(r).Related(&p)
return p
}
//GetPipelinesByRidAndEventAndBranch 根据流ID和事件及分支获取部署流
func (ps *PipelineService) GetPipelinesByRidAndEventAndBranch(rid uint, event string, branch string) (p []nocd.Pipeline, err error) {
err = ps.DB.Where("repository_id = ? AND events LIKE ? AND branch = ?", rid, "%\""+event+"\"%", branch).Find(&p).Error
return
}
//Server 读取所属服务器信息
func (ps *PipelineService) Server(p *nocd.Pipeline) error {
return ps.DB.Model(p).Related(&p.Server).Error
}
//User 读取所属用户信息
func (ps *PipelineService) User(p *nocd.Pipeline) error {
return ps.DB.Model(p).Related(&p.User).Error
}