Skip to content

Commit d5ac556

Browse files
authored
Merge pull request #3 from MahdiBM/mmbm-fix-thread-crash
Fix a crash when streaming responses
2 parents 7f923de + 5e11583 commit d5ac556

File tree

1 file changed

+9
-3
lines changed

1 file changed

+9
-3
lines changed

Sources/OpenAPIVapor/VaporTransport.swift

+9-3
Original file line numberDiff line numberDiff line change
@@ -147,11 +147,17 @@ extension Vapor.Response.Body {
147147
_ = writer.eventLoop.makeFutureWithTask {
148148
do {
149149
for try await chunk in body {
150-
try await writer.write(.buffer(ByteBuffer(bytes: chunk))).get()
150+
try await writer.eventLoop.flatSubmit {
151+
writer.write(.buffer(ByteBuffer(bytes: chunk)))
152+
}.get()
151153
}
152-
try await writer.write(.end).get()
154+
try await writer.eventLoop.flatSubmit {
155+
writer.write(.end)
156+
}.get()
153157
} catch {
154-
try await writer.write(.error(error)).get()
158+
try await writer.eventLoop.flatSubmit {
159+
writer.write(.error(error))
160+
}.get()
155161
}
156162
}
157163
}

0 commit comments

Comments
 (0)