运行 Vapor 4 文件流后的操作

Question

我运行一个处理大文件（上传和下载）的网站。目前我正在将网站从 Perfect 过渡到 Vapor。在 Perfect 中，流式传输文件给用户，然后在文件完成后执行一些操作是相对简单的。

我正在尝试在 Vapor 中做同样的事情，但我似乎无法弄清楚如何在流结束时设置回调，以及流结束时是否已完全由用户下载或在那里是一个中断。

有人知道怎么做吗？以下是我尝试过的一些方法。

这是基本结构

func downloadFile(request: Request) -> EventLoopFuture<Response> {
    //do some authentication stuff here
    
    let promise = request.eventLoop.makePromise(of: Response.self)
    let response = request.fileio.streamFile(at: "somePath")
    promise.succeed(response)
    let future = promise.futureResult
    return future
}

第一次修改

func downloadFile(request: Request) -> EventLoopFuture<Response> {
    //do some authentication stuff here
    
    let promise = request.eventLoop.makePromise(of: Response.self)
    let response = request.fileio.streamFile(at: "somePath")
    promise.succeed(response)
    let future = promise.futureResult
    
    future.eventLoop.next().execute {
        //run some post-download processing here.
        //ideally I would like to know if the download completed or not.
    }
    
    return future
}

二次修改

func downloadFile(request: Request) -> EventLoopFuture<Response> {
    //do some authentication stuff here
    
    let promise = request.eventLoop.makePromise(of: Response.self)
    let response = request.fileio.streamFile(at: "somePath")
    promise.succeed(response)
    let future = promise.futureResult
    
    future.whenComplete { result in
        switch result {
            case .succeed:
            //run succeed processing here.
            case .failure:
            //run failure code here
        }
    }
    
    return future
}

在上述所有情况下，所有处理都是立即完成的（可能是因为我们处理的是 futures 和 promises 并且一切都是异步的）。

我的问题是，在 Vapor 中有什么方法可以知道大文件何时完成流式传输以及流式传输的结果是什么？可能是一些中间件？

Answer 1

从最近发布的 Vapor 开始，现在有一个可选的处理程序，您可以使用它来执行下载成功/失败的操作。

let response = request.fileio.streamFile(at: somepath) { result in
    switch result {
        case .success(): //do success processing
        case .failure(let error): // do failure processing and can print(error)
    }
}

运行 Vapor 4 文件流后的操作

Running actions after file stream in Vapor 4

swift

vapor

server-side-swift

swift-nio