fanhuadesenlinnn/cloudpods
1
0
Fork 0
mirror of https://hubproxy.babadafafafafa.cn/https://github.com/yunionio/cloudpods.git synced 2026-09-20 08:03:53 +08:00
Code Issues Packages Projects Releases Wiki Activity

feat(devtool): support batch applying for virtual machines

Browse Source
This commit is contained in:
rainzm
2021-08-05 14:47:32 +08:00
parent da1114898f
commit 39d93ebb92
4 changed files with 60 additions and 4 deletions
Download Patch File Download Diff File Expand all files Collapse all files

30
pkg/devtool/models/script.go
View File

@@ -152,6 +152,36 @@ func (s *SScript) PerformApply(ctx context.Context, userCred mcclient.TokenCrede
return output, nil
}
func (s *SScript) AllowPerformBatchApply(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject) bool {
return s.IsOwner(userCred) || db.IsAdminAllowPerform(userCred, s, "batch-apply")
}
func (s *SScript) PerformBatchApply(ctx context.Context, userCred mcclient.TokenCredential, query jsonutils.JSONObject, input api.ScriptBatchApplyInput) (api.ScriptBatchApplyOutput, error) {
output := api.ScriptBatchApplyOutput{
Results: make([]api.ScriptBatchApplyResult, len(input.ServerIds)),
}
var argsGenerator string
if s.Name == MonitorAgent {
argsGenerator = MonitorAgent
}
for i, serverId := range input.ServerIds {
output.Results[i].ServerId = serverId
sa, err := ScriptApplyManager.createScriptApply(ctx, s.Id, serverId, nil, argsGenerator)
if err != nil {
output.Results[i].Reason = err.Error()
continue
}
err = sa.StartApply(ctx, userCred)
if err != nil {
output.Results[i].Reason = err.Error()
continue
}
output.Results[i].Succeed = true
output.Results[i].ScriptApplyId = sa.Id
}
return output, nil
}
type sServerInfo struct {
ServerId string
VpcId string