configure.swift 1.3 KB

123456789101112131415161718192021222324252627282930313233343536373839
  1. import PostgreSQL
  2. import Vapor
  3. class ServerHeaderMiddleWare : Middleware, Service {
  4. func respond(to request: Request, chainingTo next: Responder) throws -> EventLoopFuture<Response> {
  5. return try next.respond(to: request).map { response in
  6. response.http.headers.add(name: HTTPHeaderName.server, value: "Vapor")
  7. return response
  8. }
  9. }
  10. }
  11. public func configure(_ config: inout Config, _ env: inout Environment, _ services: inout Services) throws {
  12. let router = EngineRouter.default()
  13. try routes(router)
  14. services.register(router, as: Router.self)
  15. services.register(ServerHeaderMiddleWare.self) { _ in
  16. return ServerHeaderMiddleWare()
  17. }
  18. var middlewares = MiddlewareConfig()
  19. middlewares.use(ErrorMiddleware.self)
  20. middlewares.use(ServerHeaderMiddleWare.self)
  21. services.register(middlewares)
  22. try services.register(PostgreSQLProvider())
  23. let postgresql = PostgreSQLDatabase(
  24. config: PostgreSQLDatabaseConfig(
  25. hostname: "tfb-database",
  26. username: "benchmarkdbuser",
  27. database: "hello_world",
  28. password: "benchmarkdbpass"
  29. ))
  30. var databases = DatabasesConfig()
  31. databases.add(database: postgresql, as: .psql)
  32. services.register(databases)
  33. }