Arjun Nayini
2c4336880d
* [Code Style] Some code style changes Minor changes to make some code Swiftier * Fixes for PR feedback
38 lines
1.3 KiB
Swift
38 lines
1.3 KiB
Swift
// This source file is part of the Swift.org Server APIs open source project
|
|
//
|
|
// Copyright (c) 2017 Swift Server API project authors
|
|
// Licensed under Apache License v2.0 with Runtime Library Exception
|
|
//
|
|
// See http://swift.org/LICENSE.txt for license information
|
|
//
|
|
|
|
import Foundation
|
|
import HTTP
|
|
|
|
/// Simple `HTTPRequestHandler` that prints "Hello, World" as per K&R
|
|
class AbortAndSendHelloHandler: HTTPRequestHandling {
|
|
|
|
var chunkCalledCount = 0
|
|
var chunkLength = 0
|
|
|
|
func handle(request: HTTPRequest, response: HTTPResponseWriter ) -> HTTPBodyProcessing {
|
|
//Assume the router gave us the right request - at least for now
|
|
response.writeHeader(status: .ok, headers: [.transferEncoding: "chunked", "X-foo": "bar"])
|
|
return .processBody { (chunk, stop) in
|
|
switch chunk {
|
|
case .chunk(let data, let finishedProcessing):
|
|
stop = true
|
|
self.chunkCalledCount += 1
|
|
self.chunkLength += data.count
|
|
finishedProcessing()
|
|
case .end:
|
|
response.writeBody("Hello, World!")
|
|
response.done()
|
|
default:
|
|
stop = true /* don't call us anymore */
|
|
response.abort()
|
|
}
|
|
}
|
|
}
|
|
}
|