opensource/cloudpods
3
0
Fork 0
mirror of https://github.com/yunionio/cloudpods.git synced 2026-09-24 16:03:43 +08:00
Code Issues Packages Projects Releases Wiki Activity

proxysetting: add class level test action

Browse Source
This commit is contained in:
Yousong Zhou
2020-04-09 17:38:47 +08:00
parent d07d8f9652
commit 58db9ddf0f
2 changed files with 24 additions and 9 deletions
Download Patch File Download Diff File Expand all files Collapse all files
pkg/apis/cloudcommon/proxy/proxy.go
+5
View File
@@ -39,3 +39,8 @@ func (ps *SProxySetting) IsZero() bool {
}
return false
}
type ProxySettingTestInput struct {
HttpProxy string
HttpsProxy string
}
pkg/cloudcommon/db/proxy/proxysetting.go
+19 -9
View File
@@ -106,19 +106,13 @@ func (ps *SProxySetting) AllowPerformTest(ctx context.Context, userCred mcclient
return db.IsAdminAllowPerform(userCred, ps, "test")
}
func (ps *SProxySetting) PerformTest(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) (jsonutils.JSONObject, error) {
func (man *SProxySettingManager) test(ctx context.Context, urls map[string]string) (jsonutils.JSONObject, error) {
type TestURLResult struct {
Ok bool `json:"ok"`
Reason string `json:"reason"`
}
var (
r = map[string]TestURLResult{}
m = map[string]string{
"http_proxy": ps.HTTPProxy,
"https_proxy": ps.HTTPSProxy,
}
)
for k, v := range m {
r := map[string]TestURLResult{}
for k, v := range urls {
if v == "" {
r[k] = TestURLResult{Ok: true}
continue
@@ -165,6 +159,22 @@ func (ps *SProxySetting) PerformTest(ctx context.Context, userCred mcclient.Toke
return jsonutils.Marshal(r), nil
}
func (ps *SProxySetting) PerformTest(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data jsonutils.JSONObject) (jsonutils.JSONObject, error) {
urls := map[string]string{
"http_proxy": ps.HTTPProxy,
"https_proxy": ps.HTTPSProxy,
}
return ProxySettingManager.test(ctx, urls)
}
func (man *SProxySettingManager) PerformTest(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, data proxyapi.ProxySettingTestInput) (jsonutils.JSONObject, error) {
urls := map[string]string{
"http_proxy": data.HttpProxy,
"https_proxy": data.HttpsProxy,
}
return man.test(ctx, urls)
}
func (man *SProxySettingManager) InitializeData() error {
_, err := man.FetchById(proxyapi.ProxySettingId_DIRECT)
if err == nil {