在 Vapor 4 中的文件流之后运行操作

MAH*_*MAH 5 swift vapor server-side-swift swift-nio

我运行一个处理大文件(上传和下载)的网站。目前我正在将网站从 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
}
Run Code Online (Sandbox Code Playgroud)

第一次修改

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
}
Run Code Online (Sandbox Code Playgroud)

第二次修改

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
}
Run Code Online (Sandbox Code Playgroud)

在上述所有情况下,所有处理都立即完成(可能是因为我们正在处理期货和承诺,并且一切都是异步的)。

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

MAH*_*MAH 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)
    }
}
Run Code Online (Sandbox Code Playgroud)