@@ -37,7 +37,7 @@ func withMockServer<Result>(
3737 result = . failure( error)
3838 }
3939
40- try ? await server. stop ( ) . get ( )
40+ try ? await server. stop ( )
4141 return try result. get ( )
4242}
4343
@@ -50,7 +50,7 @@ final class MockLambdaServer<Behavior: LambdaServerBehavior> {
5050 private let group : EventLoopGroup
5151
5252 private var channel : Channel ?
53- private var shutdown = false , n , m
53+ private var shutdown = false
5454
5555 init (
5656 behavior: Behavior ,
@@ -70,7 +70,7 @@ final class MockLambdaServer<Behavior: LambdaServerBehavior> {
7070 assert ( shutdown)
7171 }
7272
73- func start( ) async throws -> Int {
73+ fileprivate func start( ) async throws -> Int {
7474 let logger = self . logger
7575 let keepAlive = self . keepAlive
7676 let behavior = self . behavior
@@ -99,15 +99,12 @@ final class MockLambdaServer<Behavior: LambdaServerBehavior> {
9999 return localAddress. port!
100100 }
101101
102- func stop( ) -> EventLoopFuture < Void > {
102+ fileprivate func stop( ) async throws {
103103 self . logger. info ( " stopping \( self ) " )
104- guard let channel = self . channel else {
105- return self . group. next ( ) . makeFailedFuture ( ServerError . notReady)
106- }
107- return channel. close ( ) . always { _ in
108- self . shutdown = true
109- self . logger. info ( " \( self ) stopped " )
110- }
104+ let channel = self . channel!
105+ try ? await channel. close ( ) . get ( )
106+ self . shutdown = true
107+ self . logger. info ( " \( self ) stopped " )
111108 }
112109}
113110
@@ -232,28 +229,30 @@ final class HTTPHandler: ChannelInboundHandler {
232229 }
233230 let head = HTTPResponseHead ( version: HTTPVersion ( major: 1 , minor: 1 ) , status: status, headers: headers)
234231
232+ let logger = self . logger
235233 context. write ( wrapOutboundOut ( . head( head) ) ) . whenFailure { error in
236- self . logger. error ( " \( self ) write error \( error) " )
234+ logger. error ( " write error \( error) " )
237235 }
238236
239237 if let b = body {
240238 var buffer = context. channel. allocator. buffer ( capacity: b. utf8. count)
241239 buffer. writeString ( b)
242240 context. write ( wrapOutboundOut ( . body( . byteBuffer( buffer) ) ) ) . whenFailure { error in
243- self . logger. error ( " \( self ) write error \( error) " )
241+ logger. error ( " write error \( error) " )
244242 }
245243 }
246244
247245 let loopBoundContext = NIOLoopBound ( context, eventLoop: context. eventLoop)
248246
247+ let keepAlive = self . keepAlive
249248 context. writeAndFlush ( wrapOutboundOut ( . end( nil ) ) ) . whenComplete { result in
250249 if case . failure( let error) = result {
251- self . logger. error ( " \( self ) write error \( error) " )
250+ logger. error ( " write error \( error) " )
252251 }
253- if !self . keepAlive {
252+ if !keepAlive {
254253 let context = loopBoundContext. value
255254 context. close ( ) . whenFailure { error in
256- self . logger. error ( " \( self ) close error \( error) " )
255+ logger. error ( " close error \( error) " )
257256 }
258257 }
259258 }
0 commit comments