Golang自动备份与恢复实现方法

func backupDir(src, dst string) error {
    return filepath.Walk(src, func(path string, info os.FileInfo, err error) error {
        if err != nil {
            return err
        }
        // 计算相对路径
        relPath, _ := filepath.Rel(src, path)
        destPath := filepath.Join(dst, relPath)
    if info.IsDir() {
        return os.MkdirAll(destPath, info.Mode())
    }

    // 打开源文件
    srcFile, err := os.Open(path)
    if err != nil {
        return err
    }
    defer srcFile.Close()

    // 创建目标文件
    destFile, err := os.Create(destPath)
    if err != nil {
        return err
    }
    defer destFile.Close()

    // 复制内容
    _, err = io.Copy(destFile, srcFile)
    return err
})
}

err := backupDir("/data/app", "/backup/app_20241010")
if err != nil {
    log.Fatal("Backup failed:", err)
}

func createTarGz(src, tarFile string) error {
    outFile, err := os.Create(tarFile)
    if err != nil {
        return err
    }
    defer outFile.Close()
gzipWriter := gzip.NewWriter(outFile)
defer gzipWriter.Close()

tarWriter := tar.NewWriter(gzipWriter)
defer tarWriter.Close()

return filepath.Walk(src, 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, _ := filepath.Rel(src, path)
    header.Name = relPath

    if err := tarWriter.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(tarWriter, file)
    return err
})
}

c := cron.New()
c.AddFunc("0 0 * * *", func() {
    timestamp := time.Now().Format("20060102_150405")
    backupPath := fmt.Sprintf("/backup/app_%s.tar.gz", timestamp)
    err := createTarGz("/data/app", backupPath)
    if err != nil {
        log.Println("Backup failed:", err)
    } else {
        log.Println("Backup completed:", backupPath)
    }
})
c.Start()

func restoreFromTarGz(tarFile, dest string) error {
    inFile, err := os.Open(tarFile)
    if err != nil {
        return err
    }
    defer inFile.Close()
gzipReader, err := gzip.NewReader(inFile)
if err != nil {
    return err
}
defer gzipReader.Close()

tarReader := tar.NewReader(gzipReader)

for {
    header, err := tarReader.Next()
    if err == io.EOF {
        break
    }
    if err != nil {
        return err
    }

    targetPath := filepath.Join(dest, header.Name)
    switch header.Typeflag {
    case tar.TypeDir:
        if _, err := os.Stat(targetPath); err != nil {
            os.MkdirAll(targetPath, 0755)
        }
    case tar.TypeReg:
        outFile, err := os.Create(targetPath)
        if err != nil {
            return err
        }
        _, err = io.Copy(outFile, tarReader)
        outFile.Close()
        if err != nil {
            return err
        }
    }
}
return nil
}

err := restoreFromTarGz("/backup/app_20241010.tar.gz", "/data/app")
if err != nil {
    log.Fatal("Restore failed:", err)
}