TechEmpower.FrameworkBenchmarks/frameworks/Swift/hummingbird2/src-postgres/Sources/server/main.swift

import Foundation
import Hummingbird
import PostgresNIO

// postgresql.conf specifies max_connections = 2000
// https://github.com/TechEmpower/FrameworkBenchmarks/wiki/Project-Information-Environment#citrine-self-hosted
// https://github.com/TechEmpower/FrameworkBenchmarks/blob/master/toolset/databases/postgres/postgresql.conf#L64

extension Int {
    func bound(_ minValue: Int, _ maxValue: Int) -> Int {
        return Swift.min(maxValue, Swift.max(minValue, self))
    }
}

struct TechFrameworkRequestContext: RequestContext {
    var coreContext: CoreRequestContextStorage

    init(source: ApplicationRequestContextSource) {
        self.coreContext = CoreRequestContextStorage(source: source)
    }
}

func runApp() async throws {
    let env = Environment()
    let serverHostName = env.get("SERVER_HOSTNAME") ?? "127.0.0.1"
    let serverPort = env.get("SERVER_PORT", as: Int.self) ?? 8080

    var postgresConfiguration = PostgresClient.Configuration(
        host: "tfb-database",
        username: "benchmarkdbuser",
        password: "benchmarkdbpass",
        database: "hello_world",
        tls: .disable
    )
    postgresConfiguration.options.maximumConnections = 200
    let postgresClient = PostgresClient(
        configuration: postgresConfiguration,
        eventLoopGroup: MultiThreadedEventLoopGroup.singleton
    )
    let router = Router(context: TechFrameworkRequestContext.self)
    router.addRoutes(WorldController(postgresClient: postgresClient).routes)
    router.addRoutes(FortunesController(postgresClient: postgresClient).routes)
    var app = Application(
        router: router,
        configuration: .init(
            address: .hostname(serverHostName, port: serverPort),
            serverName: "HB2",
            backlog: 8192
        )
    )
    app.addServices(postgresClient)
    try await app.runService()
}

try await runApp()