- 配置版本:fsutil 版本化备份(内容相同自动去重),主 nginx.conf 与 VPS vhost 的版本列表/逐行差异/一键回滚,回滚先校验语法与 nginx -t 再写回 - agent 0.12.2:写 vhost 前留版本;nginx -t 校验失败改为恢复上一个可用版本(此前直接删文件),新增域名 vhost 版本端点,双架构二进制已重编 - 通知:补企业微信 / 钉钉 / 飞书机器人,切换预设时服务地址跟随该预设默认值 - 阈值告警:新增 internal/monitor 巡检(CPU/内存/磁盘持续超阈值、隧道连续 N 次未注册),默认关闭,带静默期与恢复通知 - API Token:api_tokens 表 + Authorization Bearer 认证,只读/读写两种范围;令牌管理与改密/退出仅限登录会话
This commit is contained in:
@@ -5,8 +5,15 @@
|
||||
package fsutil
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"crypto/rand"
|
||||
"encoding/hex"
|
||||
"fmt"
|
||||
"os"
|
||||
"path/filepath"
|
||||
"sort"
|
||||
"strings"
|
||||
"time"
|
||||
)
|
||||
|
||||
// WriteFileAtomic 先写同目录的临时文件再 rename:读者要么看到旧内容、要么看到新内容。
|
||||
@@ -53,3 +60,115 @@ func Backup(path string) error {
|
||||
}
|
||||
return WriteFileAtomic(path+".bak", data, 0o600)
|
||||
}
|
||||
|
||||
// VersionsDir 返回某文件的版本目录(<path>.versions),供列表 / 回滚读取。
|
||||
func VersionsDir(path string) string {
|
||||
return path + ".versions"
|
||||
}
|
||||
|
||||
// BackupVersioned 把现有文件留存为 <path>.versions/<时间戳>-<随机>.conf,并只保留最近 keep 份。
|
||||
// 文件名前缀是 UTC 时间戳(同一秒内用随机后缀避开冲突),因此目录内按文件名倒序就是版本新旧顺序。
|
||||
// 文件不存在视为无需备份,返回空串;keep <= 0 表示不剪枝。
|
||||
func BackupVersioned(path string, keep int) (string, error) {
|
||||
data, err := os.ReadFile(path)
|
||||
if os.IsNotExist(err) {
|
||||
return "", nil
|
||||
}
|
||||
if err != nil {
|
||||
return "", err
|
||||
}
|
||||
dir := VersionsDir(path)
|
||||
// 内容与最新版本一致时不重复记录:Apply 每次保存都会调用,去重后保留期才对应「真实的 N 次变更」
|
||||
if versions, err := ListVersions(path); err == nil && len(versions) > 0 {
|
||||
if newest, err := ReadVersion(path, versions[0].Name); err == nil && bytes.Equal(newest, data) {
|
||||
return filepath.Join(dir, versions[0].Name), nil
|
||||
}
|
||||
}
|
||||
if err := os.MkdirAll(dir, 0o700); err != nil {
|
||||
return "", err
|
||||
}
|
||||
name := time.Now().UTC().Format("20060102T150405.000") + "-" + randSuffix() + filepath.Ext(path)
|
||||
target := filepath.Join(dir, name)
|
||||
if err := WriteFileAtomic(target, data, 0o600); err != nil {
|
||||
return "", err
|
||||
}
|
||||
if keep > 0 {
|
||||
pruneVersions(dir, keep)
|
||||
}
|
||||
return target, nil
|
||||
}
|
||||
|
||||
// VersionEntry 描述一个已留存的版本(名称前缀是 UTC 时间戳,因此按名称倒序就是新旧顺序)。
|
||||
type VersionEntry struct {
|
||||
Name string `json:"name"`
|
||||
Size int64 `json:"size"`
|
||||
CreatedAt time.Time `json:"created_at"`
|
||||
}
|
||||
|
||||
// ListVersions 按新的在前列出版本;版本目录不存在时返回空列表(首次备份前是正常状态)。
|
||||
func ListVersions(path string) ([]VersionEntry, error) {
|
||||
entries, err := os.ReadDir(VersionsDir(path))
|
||||
if err != nil {
|
||||
if os.IsNotExist(err) {
|
||||
return nil, nil
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
out := make([]VersionEntry, 0, len(entries))
|
||||
for _, entry := range entries {
|
||||
if entry.IsDir() || strings.HasPrefix(entry.Name(), ".") {
|
||||
continue
|
||||
}
|
||||
info, err := entry.Info()
|
||||
if err != nil {
|
||||
continue
|
||||
}
|
||||
out = append(out, VersionEntry{Name: entry.Name(), Size: info.Size(), CreatedAt: info.ModTime().UTC()})
|
||||
}
|
||||
sort.Slice(out, func(i, j int) bool { return out[i].Name > out[j].Name })
|
||||
return out, nil
|
||||
}
|
||||
|
||||
// ReadVersion 读取指定版本的内容。版本名虽然来自 ListVersions,这里仍校验一次:
|
||||
// 带路径分隔符 / 上跳的名称直接拒绍,避免接口被用来读版本目录以外的文件。
|
||||
func ReadVersion(path, name string) ([]byte, error) {
|
||||
if name == "" || name != filepath.Base(name) || name == "." || name == ".." {
|
||||
return nil, fmt.Errorf("版本名无效")
|
||||
}
|
||||
data, err := os.ReadFile(filepath.Join(VersionsDir(path), name))
|
||||
if err != nil {
|
||||
if os.IsNotExist(err) {
|
||||
return nil, fmt.Errorf("版本不存在")
|
||||
}
|
||||
return nil, err
|
||||
}
|
||||
return data, nil
|
||||
}
|
||||
|
||||
// pruneVersions 只保留最新的 keep 份(按文件名倒序,前缀是时间戳)。
|
||||
// 剪枝失败不影响本次备份结果,因此只忽略错误。
|
||||
func pruneVersions(dir string, keep int) {
|
||||
entries, err := os.ReadDir(dir)
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
names := make([]string, 0, len(entries))
|
||||
for _, entry := range entries {
|
||||
if entry.IsDir() || strings.HasPrefix(entry.Name(), ".") {
|
||||
continue
|
||||
}
|
||||
names = append(names, entry.Name())
|
||||
}
|
||||
sort.Sort(sort.Reverse(sort.StringSlice(names)))
|
||||
for _, name := range names[min(keep, len(names)):] {
|
||||
_ = os.Remove(filepath.Join(dir, name))
|
||||
}
|
||||
}
|
||||
|
||||
func randSuffix() string {
|
||||
var buf [2]byte
|
||||
if _, err := rand.Read(buf[:]); err != nil {
|
||||
return "0000"
|
||||
}
|
||||
return hex.EncodeToString(buf[:])
|
||||
}
|
||||
|
||||
@@ -69,3 +69,132 @@ func TestBackupCopiesContent(t *testing.T) {
|
||||
t.Fatalf("备份内容不符:%q", data)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBackupVersionedStoresContent(t *testing.T) {
|
||||
dir := t.TempDir()
|
||||
path := filepath.Join(dir, "nginx.conf")
|
||||
if err := os.WriteFile(path, []byte("worker_processes 1;"), 0o600); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
version, err := BackupVersioned(path, 5)
|
||||
if err != nil || version == "" {
|
||||
t.Fatalf("应返回版本路径:%q %v", version, err)
|
||||
}
|
||||
if !strings.Contains(version, VersionsDir(path)) {
|
||||
t.Fatalf("版本应落在版本目录内:%s", version)
|
||||
}
|
||||
data, err := os.ReadFile(version)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if string(data) != "worker_processes 1;" {
|
||||
t.Fatalf("版本内容不符:%q", data)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBackupVersionedPrunesToKeep(t *testing.T) {
|
||||
dir := t.TempDir()
|
||||
path := filepath.Join(dir, "nginx.conf")
|
||||
// 每次写入不同内容:内容相同会被去重,剪枝也就无从检验
|
||||
for _, content := range []string{"v1", "v2", "v3", "v4", "v5"} {
|
||||
if err := os.WriteFile(path, []byte(content), 0o600); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if _, err := BackupVersioned(path, 2); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
}
|
||||
entries, err := os.ReadDir(VersionsDir(path))
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if len(entries) != 2 {
|
||||
t.Fatalf("应只保留 2 份版本,实际 %d", len(entries))
|
||||
}
|
||||
}
|
||||
|
||||
func TestBackupVersionedSkipsMissingFile(t *testing.T) {
|
||||
version, err := BackupVersioned(filepath.Join(t.TempDir(), "nope.conf"), 5)
|
||||
if err != nil || version != "" {
|
||||
t.Fatalf("文件不存在时应返回空路径且无错误:%q %v", version, err)
|
||||
}
|
||||
}
|
||||
|
||||
func TestListVersionsNewestFirst(t *testing.T) {
|
||||
dir := t.TempDir()
|
||||
path := filepath.Join(dir, "nginx.conf")
|
||||
if err := os.WriteFile(path, []byte("v1"), 0o600); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if _, err := BackupVersioned(path, 5); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if err := os.WriteFile(path, []byte("v2"), 0o600); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if _, err := BackupVersioned(path, 5); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
|
||||
versions, err := ListVersions(path)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if len(versions) != 2 || versions[0].Name <= versions[1].Name {
|
||||
t.Fatalf("版本应按名称倒序(新的在前)列出:%v", versions)
|
||||
}
|
||||
seen := map[string]bool{}
|
||||
for _, version := range versions {
|
||||
data, err := ReadVersion(path, version.Name)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if version.Size != int64(len(data)) {
|
||||
t.Fatalf("版本大小不符:%d vs %d", version.Size, len(data))
|
||||
}
|
||||
seen[string(data)] = true
|
||||
}
|
||||
if !seen["v1"] || !seen["v2"] {
|
||||
t.Fatalf("两个版本的内容都应在:%v", seen)
|
||||
}
|
||||
}
|
||||
|
||||
func TestBackupVersionedSkipsIdenticalContent(t *testing.T) {
|
||||
dir := t.TempDir()
|
||||
path := filepath.Join(dir, "nginx.conf")
|
||||
if err := os.WriteFile(path, []byte("same"), 0o600); err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
first, err := BackupVersioned(path, 5)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
second, err := BackupVersioned(path, 5)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if first != second {
|
||||
t.Fatalf("内容未变时不应新增版本:%q vs %q", first, second)
|
||||
}
|
||||
versions, err := ListVersions(path)
|
||||
if err != nil {
|
||||
t.Fatal(err)
|
||||
}
|
||||
if len(versions) != 1 {
|
||||
t.Fatalf("内容未变时应只有 1 个版本,实际 %d", len(versions))
|
||||
}
|
||||
}
|
||||
|
||||
func TestReadVersionRejectsTraversal(t *testing.T) {
|
||||
path := filepath.Join(t.TempDir(), "nginx.conf")
|
||||
if _, err := ReadVersion(path, "../nginx.conf"); err == nil {
|
||||
t.Fatal("带路径上跳的版本名应被拒绝")
|
||||
}
|
||||
if _, err := ReadVersion(path, "missing.conf"); err == nil {
|
||||
t.Fatal("不存在的版本应报错")
|
||||
}
|
||||
versions, err := ListVersions(path)
|
||||
if err != nil || len(versions) != 0 {
|
||||
t.Fatalf("版本目录不存在时应返回空列表且无错误:%v %v", versions, err)
|
||||
}
|
||||
}
|
||||
|
||||
Reference in New Issue
Block a user