handle race in chunked response handling

handle condition where output channel is closed already before the check and wait to detect it starts.
JuliaComputing · Oct 31, 2021 · 80aabed · 80aabed
1 parent c5dc1a3
commit 80aabed
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 1 deletion.
diff --git a/Project.toml b/Project.toml
@@ -4,7 +4,7 @@ keywords = ["Swagger", "OpenApi", "REST"]
 license = "MIT"
 desc = "Swagger (OpenAPI) helper and code generator for Julia"
 authors = ["Tanmay Mohapatra <[email protected]>"]
-version = "0.3.1"
+version = "0.3.2"
 
 [deps]
 Dates = "ade2ca70-3891-5945-98fb-dc099432e06a"

diff --git a/src/client.jl b/src/client.jl
@@ -317,15 +317,18 @@ function do_request(ctx::Ctx, stream::Bool=false; stream_to::Union{Channel,Nothi
                     end
                 end
                 @async begin
+                    interrupted = false
                     while isopen(stream_to)
                         try
                             wait(stream_to)
                             yield()
                         catch ex
                             isa(ex, InvalidStateException) || rethrow(ex)
+                            interrupted = true
                             istaskdone(download_task) || schedule(download_task, InterruptException(), error=true)
                         end
                     end
+                    interrupted || istaskdone(download_task) || schedule(download_task, InterruptException(), error=true)
                 end
             end
         else