85 lines
2.0 KiB
Go
85 lines
2.0 KiB
Go
package rainyun
|
|
|
|
import (
|
|
"encoding/json"
|
|
"fmt"
|
|
|
|
"github.com/ssdomei232/goodBaby/handler/db"
|
|
"github.com/ssdomei232/goodBaby/model"
|
|
"github.com/ssdomei232/rainyun-go-sdk/v2/rainyun/public"
|
|
"github.com/ssdomei232/rainyun-go-sdk/v2/rainyun/rcs"
|
|
"github.com/ssdomei232/rainyun-go-sdk/v2/rainyun/rgs"
|
|
)
|
|
|
|
// 从 Rule 中获取 RainyunConfig
|
|
func getRainyunConfigFromRule(rule *model.Rule) (*RainyunWorkOrderRule, error) {
|
|
var rainyunConfig RainyunWorkOrderRule
|
|
if err := json.Unmarshal([]byte(rule.ConfigJson), &rainyunConfig); err != nil {
|
|
return nil, fmt.Errorf("解析Rainyun规则配置失败: %w", err)
|
|
}
|
|
return &rainyunConfig, nil
|
|
}
|
|
|
|
// 从 Rule 中获取 RainyunAccount
|
|
func getRainyunAccountFromRule(rule *model.Rule) (*RainyunAccount, error) {
|
|
var rainyunAccount RainyunAccount
|
|
if err := db.LoadAccountConfig(rule.AccountID, &rainyunAccount); err != nil {
|
|
return nil, err
|
|
}
|
|
return &rainyunAccount, nil
|
|
}
|
|
|
|
func reinstallOneRCS(rcsID int, rcsClient *rcs.Client) error {
|
|
rcsDetail, err := rcsClient.GetRcsDetails(rcsID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
rcsOSList, err := public.GetRcsOSList()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, os := range rcsOSList.Data {
|
|
if os.Region == rcsDetail.Data.Data.Node.Region && os.OsType == "linux" {
|
|
rcsReinstallRequest := &rcs.ReinstallRcsRequest{
|
|
OsID: os.ID,
|
|
ResetOsd: true,
|
|
}
|
|
_, err := rcsClient.ReinstallRcs(rcsID, rcsReinstallRequest)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
break
|
|
}
|
|
}
|
|
return nil
|
|
}
|
|
|
|
func reinstallOneRGS(rgsID int, rgsClient *rgs.Client) error {
|
|
rgsDetail, err := rgsClient.GetRgsDetails(rgsID)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
rgsOSList, err := public.GetRgsOSList()
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
for _, os := range rgsOSList.Data {
|
|
if os.Region == rgsDetail.Data.Data.Node.Region && os.OsType == "linux" {
|
|
rgsReinstallRequest := &rcs.ReinstallRcsRequest{
|
|
OsID: os.ID,
|
|
ResetOsd: true,
|
|
}
|
|
_, err := rgsClient.Reinstallgs(rgsID, rgsReinstallRequest)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
break
|
|
}
|
|
}
|
|
return nil
|
|
}
|