feedback

Package.swift

@@ -250,7 +250,7 @@ let package = Package(
         .library(name: "NIOHTTPResponsiveness", targets: ["NIOHTTPResponsiveness"]),
     ],
     dependencies: [
-        .package(url: "https://github.com/apple/swift-nio.git", from: "2.67.0"),
+        .package(url: "https://github.com/apple/swift-nio.git", from: "2.77.0"),
         .package(url: "https://github.com/apple/swift-nio-http2.git", from: "1.27.0"),
         .package(url: "https://github.com/apple/swift-http-types.git", from: "1.3.0"),
         .package(url: "https://github.com/apple/swift-http-structured-headers.git", from: "1.1.0"),

Sources/NIOHTTPResponsiveness/HTTPDrippingDownloadHandler.swift

@@ -50,6 +50,7 @@ public final class HTTPDrippingDownloadHandler: ChannelDuplexHandler {
 
     private var phase = Phase.waitingOnHead
     private var scheduled: Scheduled<Void>?
+    private var scheduledCallbackHandler: HTTPDrippingDownloadHandlerScheduledCallbackHandler?
     private var pendingRead = false
     private var pendingWrite = false
     private var activelyWritingChunk = false
@@ -156,6 +157,7 @@ public final class HTTPDrippingDownloadHandler: ChannelDuplexHandler {
     public func channelInactive(context: ChannelHandlerContext) {
         self.phase = .done
         self.scheduled?.cancel()
+        context.fireChannelInactive()
     }
 
     public func channelWritabilityChanged(context: ChannelHandlerContext) {
@@ -219,10 +221,22 @@ public final class HTTPDrippingDownloadHandler: ChannelDuplexHandler {
         // More chunks to write.. Kick off timer
         drippingState.currentChunkBytesLeft = self.size
         self.phase = .dripping(drippingState)
-        let this = NIOLoopBound(self, eventLoop: context.eventLoop)
-        let loopBoundContext = NIOLoopBound(context, eventLoop: context.eventLoop)
-        self.scheduled = context.eventLoop.scheduleTask(in: self.frequency) {
-            this.value.writeChunk(context: loopBoundContext.value)
+        if self.scheduledCallbackHandler == nil {
+            let this = NIOLoopBound(self, eventLoop: context.eventLoop)
+            let loopBoundContext = NIOLoopBound(context, eventLoop: context.eventLoop)
+            self.scheduledCallbackHandler = HTTPDrippingDownloadHandlerScheduledCallbackHandler(handler: this, context: loopBoundContext)
+        }
+        // SAFTEY: scheduling the callback only potentially throws when invoked off eventloop
+        try! context.eventLoop.scheduleCallback(in: self.frequency, handler: self.scheduledCallbackHandler!)
+    }
+
+    private struct HTTPDrippingDownloadHandlerScheduledCallbackHandler: NIOScheduledCallbackHandler & Sendable {
+        var handler: NIOLoopBound<HTTPDrippingDownloadHandler>
+        var context: NIOLoopBound<ChannelHandlerContext>
+
+        func handleScheduledCallback(eventLoop: some EventLoop) {
+            self.handler.value.writeChunk(context: self.context.value)
         }
     }
 }
+

Sources/NIOHTTPResponsiveness/HTTPReceiveDiscardHandler.swift

@@ -70,7 +70,7 @@ public final class HTTPReceiveDiscardHandler: ChannelInboundHandler {
             string:
                 "Received in excess of expectation; expected(\(expectation)) received(\(self.received))"
         )
-        self.writeSimpleResponse(context: context, status: .ok, body: body)
+        self.writeSimpleResponse(context: context, status: .badRequest, body: body)
     }
 
     private func writeSimpleResponse(

Sources/NIOHTTPResponsiveness/SimpleResponsivenessRequestMux.swift

@@ -70,8 +70,8 @@ public final class SimpleResponsivenessRequestMux: ChannelInboundHandler {
                     status: .ok,
                     body: self.responsivenessConfigBuffer
                 )
-            case (.get, .some(""), .some("responsiveness"), .some("download"), .some(let count)):
-                self.addHandlerOrInternalError(context: context, handler: HTTPDrippingDownloadHandler(count: 1, size: count))
+            case (.get, .some(""), .some("responsiveness"), .some("download"), .some(let size)):
+                self.addHandlerOrInternalError(context: context, handler: HTTPDrippingDownloadHandler(count: 1, size: size))
             case (.post, .some(""), .some("responsiveness"), .some("upload"), .none):
                 // Check if we should expect a certain count
                 var expectation: Int?