123456789101112131415161718192021222324252627282930313233343536373839 |
- import PostgreSQL
- import Vapor
- class ServerHeaderMiddleWare : Middleware, Service {
- func respond(to request: Request, chainingTo next: Responder) throws -> EventLoopFuture<Response> {
- return try next.respond(to: request).map { response in
- response.http.headers.add(name: HTTPHeaderName.server, value: "Vapor")
- return response
- }
- }
- }
- public func configure(_ config: inout Config, _ env: inout Environment, _ services: inout Services) throws {
- let router = EngineRouter.default()
- try routes(router)
- services.register(router, as: Router.self)
- services.register(ServerHeaderMiddleWare.self) { _ in
- return ServerHeaderMiddleWare()
- }
- var middlewares = MiddlewareConfig()
- middlewares.use(ErrorMiddleware.self)
- middlewares.use(ServerHeaderMiddleWare.self)
- services.register(middlewares)
- try services.register(PostgreSQLProvider())
- let postgresql = PostgreSQLDatabase(
- config: PostgreSQLDatabaseConfig(
- hostname: "tfb-database",
- username: "benchmarkdbuser",
- database: "hello_world",
- password: "benchmarkdbpass"
- ))
- var databases = DatabasesConfig()
- databases.add(database: postgresql, as: .psql)
- services.register(databases)
- }
|