-
-
Notifications
You must be signed in to change notification settings - Fork 78
/
config_clean.go
120 lines (103 loc) · 3.18 KB
/
config_clean.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
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
package cmd
import (
cfg "github.com/jenkins-zh/jenkins-cli/app/config"
"github.com/jenkins-zh/jenkins-cli/app/helper"
"github.com/jenkins-zh/jenkins-cli/app/i18n"
"github.com/jenkins-zh/jenkins-cli/client"
"github.com/spf13/cobra"
"net"
"net/http"
"time"
)
// ConfigCleanOption option for config list command
type ConfigCleanOption struct {
Timeout int
CleanTimeout bool
Logger helper.Printer
}
var configCleanOption ConfigCleanOption
func init() {
configCmd.AddCommand(configCleanCmd)
configCleanCmd.Flags().IntVarP(&configCleanOption.Timeout, "timeout", "t", 5,
i18n.T("Timeout in second value when checking with the Jenkins URL"))
configCleanCmd.Flags().BoolVarP(&configCleanOption.CleanTimeout, "clean-timeout", "", false,
i18n.T("Clean the config items when timeout with API request"))
}
// CheckResult is the result of checking
type CheckResult struct {
Name string
StatusCode int
Timeout bool
}
var configCleanCmd = &cobra.Command{
Use: "clean",
Short: i18n.T("Clean up some unavailable config items"),
Long: i18n.T("Clean up some unavailable config items"),
RunE: configCleanOption.Run,
}
// Run is the main entry point of command config-clean
func (o *ConfigCleanOption) Run(cmd *cobra.Command, args []string) (err error) {
if config := getConfig(); config == nil {
cmd.Println("cannot found config file")
}
o.Logger = cmd
itemCount := len(config.JenkinsServers)
checkResult := make(chan CheckResult, itemCount)
for _, jenkins := range config.JenkinsServers {
go func(target cfg.JenkinsServer) {
checkResult <- o.Check(target)
}(jenkins)
}
checkResultList := make([]CheckResult, itemCount)
for i := range config.JenkinsServers {
checkResultList[i] = <-checkResult
}
// do the clean work
err = o.CleanByCondition(checkResultList)
cmd.Println()
return
}
// Check check the target JenkinsServer config
// make a request to a Jenkins API
func (o *ConfigCleanOption) Check(jenkins cfg.JenkinsServer) (result CheckResult) {
result.Name = jenkins.Name
jClient := &client.PluginManager{
JenkinsCore: client.JenkinsCore{
RoundTripper: pluginListOption.RoundTripper,
Timeout: time.Duration(o.Timeout),
},
}
jClient.URL = jenkins.URL
jClient.UserName = jenkins.UserName
jClient.Token = jenkins.Token
jClient.Proxy = jenkins.Proxy
jClient.ProxyAuth = jenkins.ProxyAuth
jClient.InsecureSkipVerify = jenkins.InsecureSkipVerify
var statusCode int
var err error
if statusCode, _, err = jClient.Request(http.MethodGet, "/api/json", nil, nil); err != nil {
if err, ok := err.(net.Error); ok {
result.Timeout = err.Timeout()
} else {
o.Logger.Println("check request failed, error is", err)
}
}
result.StatusCode = statusCode
return
}
// CleanByCondition do the clean work by conditions
func (o *ConfigCleanOption) CleanByCondition(resultList []CheckResult) (err error) {
if len(resultList) == 0 {
return
}
for _, result := range resultList {
if o.CleanTimeout && result.Timeout {
if err = removeJenkins(result.Name); err == nil {
o.Logger.Printf("removed invalid item %s due to timeout reason\n", result.Name)
}
} else {
o.Logger.Printf("status code of %s is %d\n", result.Name, result.StatusCode)
}
}
return
}