feat: 删除不需要文件夹

2023-10-10 10:56:44 +08:00
parent ce7c3cae68
commit d173205528
154 changed files with 32276 additions and 1 deletions
--- a/features/sm/backup.go
+++ b/features/sm/backup.go
@@ -0,0 +1,109 @@
+package sm
+
+import (
+	"database/sql"
+	"fmt"
+	"os"
+	"os/exec"
+	"time"
+
+	"ems.agt/lib/log"
+	"ems.agt/restagent/config"
+	_ "github.com/go-sql-driver/mysql"
+)
+
+var dbConfig = config.GetYamlConfig().Database
+
+func DatabaseWhoreBackup() {
+	// MySQL数据库连接信息
+	sqlStr := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8&parseTime=true&loc=Local",
+		dbConfig.User, dbConfig.Password, dbConfig.Host, dbConfig.Port, dbConfig.Name)
+	db, err := sql.Open("mysql", sqlStr)
+	if err != nil {
+		log.Error("Failed to connect to database:", err)
+		return
+	}
+	defer db.Close()
+
+	// 备份SQL文件路径
+	backupFile := dbConfig.Backup + "/" + "whore_backup_" + dbConfig.Name + ".sql"
+
+	// 执行mysqldump命令进行备份
+	cmd := exec.Command("mysqldump", "-u", dbConfig.User, "-p"+dbConfig.Password, "-h", dbConfig.Host, dbConfig.Name)
+	output, err := cmd.Output()
+	if err != nil {
+		log.Error("Failed to execute mysqldump command:", err)
+		return
+	}
+
+	// 将备份结果写入SQL文件
+	file, err := os.Create(backupFile)
+	if err != nil {
+		log.Error("Failed to create backup file:", err)
+		return
+	}
+	defer file.Close()
+
+	_, err = file.Write(output)
+	if err != nil {
+		log.Error("Failed to write backup file:", err)
+		return
+	}
+
+	log.Info("Backup completed successfully.")
+}
+
+func DatabaseIncrementalBackup() {
+	// MySQL数据库连接信息
+	sqlStr := fmt.Sprintf("%s:%s@tcp(%s:%s)/%s?charset=utf8&parseTime=true&loc=Local",
+		dbConfig.User, dbConfig.Password, dbConfig.Host, dbConfig.Port, dbConfig.Name)
+	db, err := sql.Open("mysql", sqlStr)
+	if err != nil {
+		log.Error("Failed to connect to database:", err)
+		return
+	}
+	defer db.Close()
+
+	// 备份SQL文件路径
+	backupFile := dbConfig.Backup + "/" + "incremental_backup_" + dbConfig.Name + ".sql"
+
+	// 上次备份的时间点
+	lastBackupTime := time.Date(2022, time.January, 1, 0, 0, 0, 0, time.Local)
+
+	// 构建增量备份SQL语句
+	query := fmt.Sprintf("SELECT * FROM table WHERE modified_at > '%s'", lastBackupTime.Format("2006-01-02 15:04:05"))
+
+	// 执行查询
+	rows, err := db.Query(query)
+	if err != nil {
+		log.Error("Failed to execute query:", err)
+		return
+	}
+	defer rows.Close()
+
+	// 创建增量备份SQL文件
+	file, err := os.Create(backupFile)
+	if err != nil {
+		log.Error("Failed to create backup file:", err)
+		return
+	}
+	defer file.Close()
+
+	// 将查询结果写入SQL文件
+	for rows.Next() {
+		var data string
+		err := rows.Scan(&data)
+		if err != nil {
+			log.Error("Failed to scan row:", err)
+			return
+		}
+
+		_, err = file.WriteString(data + "\n")
+		if err != nil {
+			log.Error("Failed to write backup file:", err)
+			return
+		}
+	}
+
+	log.Info("Incremental backup completed successfully.")
+}