127 lines
3.1 KiB
Go
127 lines
3.1 KiB
Go
package dbrest
|
|
|
|
import (
|
|
"archive/tar"
|
|
"compress/gzip"
|
|
"io"
|
|
"net/http"
|
|
"os"
|
|
"os/exec"
|
|
"path/filepath"
|
|
"time"
|
|
|
|
"ems.agt/lib/dborm"
|
|
"ems.agt/lib/log"
|
|
"ems.agt/lib/services"
|
|
"ems.agt/restagent/config"
|
|
_ "github.com/go-sql-driver/mysql"
|
|
)
|
|
|
|
var (
|
|
// 系统备份-数据库备份
|
|
UriDbBackup = config.DefaultUriPrefix + "/dataManagement/{apiVersion}/dbBackup"
|
|
// 系统备份-文件备份
|
|
UriConfBackup = config.DefaultUriPrefix + "/dataManagement/{apiVersion}/confBackup"
|
|
)
|
|
|
|
// DbBackup 系统备份-数据库备份1
|
|
func DbBackup(w http.ResponseWriter, r *http.Request) {
|
|
dbConfig := config.GetYamlConfig().Database
|
|
|
|
// 备份SQL文件路径
|
|
fileName := "database_backup_" + time.Now().Format("20060102150405") + ".sql"
|
|
backupFile := dbConfig.Backup + "/" + fileName
|
|
|
|
// 执行mysqldump命令进行备份
|
|
cmd := exec.Command("mysqldump", "-u", dbConfig.User, "-p"+dbConfig.Password, "-h", dbConfig.Host, "--all-databases", ">", backupFile)
|
|
err := cmd.Start()
|
|
if err != nil {
|
|
services.ResponseErrorWithJson(w, 400, err.Error())
|
|
return
|
|
}
|
|
|
|
// 备份结果
|
|
res, err := dborm.DbClient.XEngine.Exec("INSERT INTO `sys_backup`(`id`, `created_at`, `updated_at`, `backup_type`, `backup_way`, `name`, `path`) VALUES (null, ?, ?, '1', '0', ?, ?);", time.Now(), time.Now(), fileName, backupFile)
|
|
if err != nil {
|
|
services.ResponseErrorWithJson(w, 500, err.Error())
|
|
return
|
|
}
|
|
services.ResponseWithJson(w, 200, res)
|
|
log.Info("Backup completed successfully.")
|
|
}
|
|
|
|
// 系统备份-文件备份0
|
|
func ConfBackup(w http.ResponseWriter, r *http.Request) {
|
|
dbConfig := config.GetYamlConfig().Database
|
|
etcDir := config.GetYamlConfig().NE.EtcDir
|
|
|
|
// 文件名
|
|
fileName := "conf_backup_" + time.Now().Format("20060102150405") + ".tar.gz"
|
|
backupFile := dbConfig.Backup + "/" + fileName
|
|
|
|
err := createTarGz(etcDir, backupFile)
|
|
if err != nil {
|
|
services.ResponseErrorWithJson(w, 500, err.Error())
|
|
return
|
|
}
|
|
|
|
// 备份结果
|
|
res, err := dborm.DbClient.XEngine.Exec("INSERT INTO `sys_backup`(`id`, `created_at`, `updated_at`, `backup_type`, `backup_way`, `name`, `path`) VALUES (null, ?, ?, '1', '0', ?, ?);", time.Now(), time.Now(), fileName, backupFile)
|
|
if err != nil {
|
|
services.ResponseErrorWithJson(w, 500, err.Error())
|
|
return
|
|
}
|
|
services.ResponseWithJson(w, 200, res)
|
|
log.Info("Backup completed successfully.")
|
|
}
|
|
|
|
// 打压缩
|
|
func createTarGz(source, target string) error {
|
|
tarFile, err := os.Create(target)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer tarFile.Close()
|
|
|
|
gw := gzip.NewWriter(tarFile)
|
|
defer gw.Close()
|
|
|
|
tw := tar.NewWriter(gw)
|
|
defer tw.Close()
|
|
|
|
return filepath.Walk(source, func(path string, info os.FileInfo, err error) error {
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
header, err := tar.FileInfoHeader(info, "")
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
relPath, err := filepath.Rel(source, path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
|
|
header.Name = relPath
|
|
|
|
if err := tw.WriteHeader(header); err != nil {
|
|
return err
|
|
}
|
|
|
|
if !info.Mode().IsRegular() {
|
|
return nil
|
|
}
|
|
|
|
file, err := os.Open(path)
|
|
if err != nil {
|
|
return err
|
|
}
|
|
defer file.Close()
|
|
|
|
_, err = io.Copy(tw, file)
|
|
return err
|
|
})
|
|
}
|