Web应用中用户下载后自动删除服务器生成文件的策略与实现

import "os"

// 假设 filename 是要删除的文件的完整路径
err := os.Remove(filename)
if err != nil {
    // 处理错误，例如文件不存在或权限不足
    // log.Printf("Failed to delete file %s: %v", filename, err)
} else {
    // log.Printf("File %s deleted successfully", filename)
}

$.fileDownload('/url/to/download.zip', {
    successCallback: function (url) {
        // 文件下载成功后执行此回调
        // 在这里，我们可以发起另一个Ajax请求到服务器，通知它删除文件
        console.log('File downloaded successfully from: ' + url);

        // 提取文件名或唯一标识符
        const filenameOrId = new URL(url).searchParams.get('fileId'); // 假设URL中包含文件标识符
        if (filenameOrId) {
            $.ajax({
                url: '/api/delete-temp-file', // 服务器端删除文件的API端点
                type: 'POST',
                data: { fileId: filenameOrId }, // 将文件标识符发送给服务器
                success: function(response) {
                    console.log('Server acknowledged file deletion: ', response);
                },
                error: function(xhr, status, error) {
                    console.error('Error notifying server for file deletion: ', error);
                }
            });
        }
    },
    failCallback: function (html, url) {
        // 文件下载失败后执行此回调
        console.error('File download failed from: ' + url, html);
        // 可以选择在这里也通知服务器，或者让服务器通过定时任务清理
    }
});

package controllers

import (
    "os"
    "path/filepath"
    "github.com/beego/beego/v2/server/web"
)

type FileController struct {
    web.Controller
}

// PostDeleteTempFile 处理客户端发来的删除临时文件的请求
// @router /api/delete-temp-file [post]
func (c *FileController) PostDeleteTempFile() {
    fileId := c.GetString("fileId") // 获取客户端发送的文件标识符

    if fileId == "" {
        c.Data["json"] = map[string]string{"status": "error", "message": "fileId is required"}
        c.ServeJSON()
        return
    }

    // **安全注意事项：** 必须对 fileId 进行严格的验证和清理，
    // 防止目录遍历攻击 (e.g., ../../sensitive_file) 或删除非预期文件。
    // 理想情况下，fileId 应该是一个不直接暴露文件路径的唯一令牌。
    // 这里仅为示例，实际应用中需更严谨。

    // 假设临时文件都存储在 /tmp/downloads 目录下
    tempDir := "/tmp/downloads" // 你的临时文件存储路径

    // 构造完整的文件路径
    // 再次强调：这里需要确保 fileId 不包含路径分隔符，或者进行严格的路径清理
    fullPath := filepath.Join(tempDir, filepath.Base(fileId)) 

    // 检查文件是否存在且确实在允许的临时目录内
    if !filepath.HasPrefix(fullPath, tempDir) {
        c.Data["json"] = map[string]string{"status": "error", "message": "Invalid file path"}
        c.ServeJSON()
        return
    }

    err := os.Remove(fullPath)
    if err != nil {
        c.Data["json"] = map[string]string{"status": "error", "message": "Failed to delete file", "details": err.Error()}
        c.ServeJSON()
        return
    }

    c.Data["json"] = map[string]string{"status": "success", "message": "File deleted successfully"}
    c.ServeJSON()
}

// 在你的 routers/router.go 中注册路由
/*
import (
    "your_project_path/controllers"
)
func init() {
    web.Router("/api/delete-temp-file", &controllers.FileController{}, "post:PostDeleteTempFile")
}
*/

package main

import (
    "archive/zip"
    "bytes"
    "net/http"
    "time"
)

func generateAndServeZip(w http.ResponseWriter, r *http.Request) {
    buf := new(bytes.Buffer)
    zipWriter := zip.NewWriter(buf)

    // 往ZIP文件中添加内容
    file1, _ := zipWriter.Create("hello.txt")
    file1.Write([]byte("Hello, world!"))

    file2, _ := zipWriter.Create("data.json")
    file2.Write([]byte(`{"key": "value"}`))

    zipWriter.Close() // 非常重要，完成ZIP写入

    w.Header().Set("Content-Type", "application/zip")
    w.Header().Set("Content-Disposition", "attachment; filename=\"generated_files.zip\"")
    w.Header().Set("Content-Length", string(buf.Len())) // 设置Content-Length

    // 直接将 buffer 的内容写入响应
    _, err := w.Write(buf.Bytes())
    if err != nil {
        http.Error(w, "Failed to write zip content", http.StatusInternalServerError)
        return
    }
}

func main() {
    http.HandleFunc("/download-on-the-fly", generateAndServeZip)
    http.ListenAndServe(":8080", nil)
}