call onFinish() too

MahdiBM · MahdiBM · commit a40b9dddf99f · 2025-03-02T21:21:14.000+03:30
diff --git a/Sources/PostgresNIO/New/PSQLRowStream.swift b/Sources/PostgresNIO/New/PSQLRowStream.swift
@@ -313,8 +313,9 @@ final class PSQLRowStream: @unchecked Sendable {
                 self.downstreamState = .consumed(.failure(error))
                 return self.eventLoop.makeFailedFuture(error)
             }
-        case .asyncSequence(let consumer, let dataSource, _):
+        case .asyncSequence(let consumer, let dataSource, let onFinish):
             consumer.finish()
+            onFinish()
 
             let promise = self.eventLoop.makePromise(of: [PostgresRow].self)
 

Original file line number	Diff line number	Diff line change
`@@ -313,8 +313,9 @@ final class PSQLRowStream: @unchecked Sendable {`
`313`	`313`	`self.downstreamState = .consumed(.failure(error))`
`314`	`314`	`return self.eventLoop.makeFailedFuture(error)`
`315`	`315`	`}`
`316`		`- case .asyncSequence(let consumer, let dataSource, _):`
	`316`	`+ case .asyncSequence(let consumer, let dataSource, let onFinish):`
`317`	`317`	`consumer.finish()`
	`318`	`+ onFinish()`
`318`	`319`
`319`	`320`	`let promise = self.eventLoop.makePromise(of: [PostgresRow].self)`
`320`	`321`