Swift国内社区: SwiftMic

Middleware 是现代 Web 框架中必不可少的一部分。通过 Middleware 可以单独处理 Request 和 Response，极其方便。

Middleware

基本用法

SampleMiddleware.swift

import Vapor
import HTTP

class SampleMiddleware: Middleware {

    func respond(to request: Request, chainingTo chain: Responder) throws -> Response {
        // You can manipulate the request before calling the handler
        // and abort early if necessary, a good injection point for
        // handling auth.

        // 此处可统一处理 Request 请求

        let response = try chain.respond(to: request)

        // 此处可统一处理 Response 响应

        // You can also manipulate the response to add headers
        // cookies, etc.

        return response

        // Vapor Middleware is based on S4 Middleware.
        // This means you can share it with any other project
        // that uses S4 Middleware. 
    }

}

main.swift

drop.middleware.append(SampleMiddleware())

每次 Request 和 Response 都将会响应到 SampleMiddleware 中的 respond 方法。

示例1

如果想给所有 API 的 Response 增加一个版本号字段，可直接在 SampleMiddleware 中添加代码： response.headers["Api_Version"] = "v1" 即可，而无需在每个 Response 中进行设置。

import Vapor
import HTTP

class SampleMiddleware: Middleware {

    func respond(to request: Request, chainingTo chain: Responder) throws -> Response {
        
        let response = try chain.respond(to: request)

        response.headers["Api_Version"] = "v1"

        return response
    }

}

示例2

有些 API 需要登录后才能调用，这里涉及到用户鉴权的问题，可通过 Middleware 统一处理 token 异常情况，而无需在每个 Response 中处理。

定义异常类型

enum ZException: Error {
    case ERROR_AUTH_TOKEN_INVALID
}

ERROR_AUTH_TOKEN_INVALID 异常类型统一处理

import Vapor
import HTTP

class SampleMiddleware: Middleware {

    func respond(to request: Request, chainingTo chain: Responder) throws -> Response {
        do {
            return try chain.respond(to: request)
        } catch ZException.ERROR_AUTH_TOKEN_INVALID {
            throw Abort.custom(status: .unauthorized, message: "ERROR_AUTH_TOKEN_INVALID")
        }
    }

}

当检测到 token 无效时，只需要抛出 ERROR_AUTH_TOKEN_INVALID 异常即可。

throw ZException.ERROR_AUTH_TOKEN_INVALID

Go to Vapor系列教程 - 目录