146 lines
4.3 KiB
Go
146 lines
4.3 KiB
Go
package backupEtcFromNE
|
||
|
||
import (
|
||
"encoding/json"
|
||
"fmt"
|
||
"os"
|
||
"strings"
|
||
"time"
|
||
|
||
"ems.agt/lib/dborm"
|
||
"ems.agt/lib/global"
|
||
"ems.agt/lib/log"
|
||
"ems.agt/restagent/config"
|
||
"ems.agt/src/framework/cron"
|
||
)
|
||
|
||
var NewProcessor = &BarProcessor{
|
||
progress: 0,
|
||
count: 0,
|
||
}
|
||
|
||
// bar 队列任务处理
|
||
type BarProcessor struct {
|
||
// 任务进度
|
||
progress int
|
||
// 执行次数
|
||
count int
|
||
}
|
||
|
||
type BarParams struct {
|
||
Duration int `json:"duration"`
|
||
TableName string `json:"tableName"`
|
||
ColName string `json:"colName"` // column name of time string
|
||
Extras string `json:"extras"` // extras condition for where
|
||
}
|
||
|
||
func (s *BarProcessor) Execute(data any) (any, error) {
|
||
log.Infof("execute %d，last progress： %d ", s.count, s.progress)
|
||
s.count++
|
||
|
||
options := data.(cron.JobData)
|
||
sysJob := options.SysJob
|
||
var params BarParams
|
||
|
||
err := json.Unmarshal([]byte(sysJob.TargetParams), ¶ms)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
log.Infof("Repeat %v Job ID %s", options.Repeat, sysJob.JobID)
|
||
|
||
var nes []dborm.NeInfo
|
||
_, err = dborm.XormGetAllNeInfo(&nes)
|
||
if err != nil {
|
||
return nil, err
|
||
}
|
||
|
||
var successfulNEs, failureNEs []string
|
||
for _, neInfo := range nes {
|
||
neTypeUpper := strings.ToUpper(neInfo.NeType)
|
||
neTypeLower := strings.ToLower(neInfo.NeType)
|
||
nePath := fmt.Sprintf("%s/etc/%s", config.GetYamlConfig().OMC.Backup, neTypeLower)
|
||
isExist, err := global.PathExists(nePath)
|
||
if err != nil {
|
||
log.Errorf("Failed to PathExists:", err)
|
||
failureNEs = append(failureNEs, neInfo.NeType+"/"+neInfo.NeId)
|
||
continue
|
||
}
|
||
if isExist {
|
||
err = os.RemoveAll(nePath)
|
||
if err != nil {
|
||
log.Errorf("Failed to RemoveAll:", err)
|
||
failureNEs = append(failureNEs, neInfo.NeType+"/"+neInfo.NeId)
|
||
continue
|
||
}
|
||
}
|
||
err = os.MkdirAll(nePath, os.ModePerm)
|
||
if err != nil {
|
||
log.Errorf("Failed to MkdirAll:", err)
|
||
failureNEs = append(failureNEs, neInfo.NeType+"/"+neInfo.NeId)
|
||
continue
|
||
}
|
||
|
||
var scpCmd string
|
||
ipType := global.ParseIPAddr(neInfo.Ip)
|
||
if neTypeLower != "omc" {
|
||
if ipType == global.IsIPv4 {
|
||
scpCmd = fmt.Sprintf("scp -r %s@%s:%s/%s/*.yaml %s/etc/%s", config.GetYamlConfig().NE.User,
|
||
neInfo.Ip, config.GetYamlConfig().NE.EtcDir,
|
||
neTypeLower, config.GetYamlConfig().OMC.Backup, neTypeLower)
|
||
} else {
|
||
scpCmd = fmt.Sprintf("scp -r %s@[%s]:%s/%s/*.yaml %s/etc/%s", config.GetYamlConfig().NE.User,
|
||
neInfo.Ip, config.GetYamlConfig().NE.EtcDir,
|
||
neTypeLower, config.GetYamlConfig().OMC.Backup, neTypeLower)
|
||
}
|
||
} else {
|
||
if ipType == global.IsIPv4 {
|
||
scpCmd = fmt.Sprintf("scp -r %s@%s:%s/etc/*.yaml %s/etc/%s", config.GetYamlConfig().NE.User,
|
||
neInfo.Ip, config.GetYamlConfig().NE.OmcDir, config.GetYamlConfig().OMC.Backup, neTypeLower)
|
||
} else {
|
||
scpCmd = fmt.Sprintf("scp -r %s@[%s]:%s/etc/*.yaml %s/etc/%s", config.GetYamlConfig().NE.User,
|
||
neInfo.Ip, config.GetYamlConfig().NE.OmcDir, config.GetYamlConfig().OMC.Backup, neTypeLower)
|
||
}
|
||
}
|
||
|
||
zipFile := fmt.Sprintf("%s-%s-etc-%s.zip", neTypeLower, strings.ToLower(neInfo.NeId), time.Now().Format(global.DateData))
|
||
zipFilePath := config.GetYamlConfig().OMC.Backup + "/" + zipFile
|
||
zipCmd := fmt.Sprintf("cd %s/etc && zip -r %s %s/*", config.GetYamlConfig().OMC.Backup, zipFilePath, neTypeLower)
|
||
|
||
command := fmt.Sprintf("%s&&%s", scpCmd, zipCmd)
|
||
|
||
log.Trace("command:", command)
|
||
out, err := global.ExecCmd(command)
|
||
if err != nil {
|
||
log.Error("Faile to exec command:", err)
|
||
failureNEs = append(failureNEs, neInfo.NeType+"/"+neInfo.NeId)
|
||
continue
|
||
}
|
||
log.Trace("command output:", out)
|
||
|
||
md5Sum, err := global.GetFileMD5Sum(zipFilePath)
|
||
if err != nil {
|
||
log.Error("Faile to md5sum:", err)
|
||
failureNEs = append(failureNEs, neInfo.NeType+"/"+neInfo.NeId)
|
||
continue
|
||
}
|
||
//log.Debug("md5Str:", md5Sum)
|
||
path := config.GetYamlConfig().OMC.Backup
|
||
neBackup := dborm.NeBackup{NeType: neTypeUpper, NeId: neInfo.NeId, FileName: zipFile, Path: path, Md5Sum: md5Sum}
|
||
_, err = dborm.XormInsertTableOne("ne_backup", neBackup)
|
||
if err != nil {
|
||
log.Error("Faile to XormInsertTableOne:", err)
|
||
failureNEs = append(failureNEs, neInfo.NeType+"/"+neInfo.NeId)
|
||
continue
|
||
}
|
||
successfulNEs = append(successfulNEs, neInfo.NeType+"/"+neInfo.NeId)
|
||
}
|
||
|
||
log.Infof("successfulNEs: %s failureNEs: %s", successfulNEs, failureNEs)
|
||
// result
|
||
return map[string]any{
|
||
"successfulNEs": successfulNEs,
|
||
"failureNEs": failureNEs,
|
||
}, nil
|
||
}
|