allow configuring backend on a per-request basis

JuliaCloud · Jul 12, 2021 · 57141e4 · 57141e4
1 parent 16419cd
commit 57141e4
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 7 deletions.
diff --git a/src/utilities/request.jl b/src/utilities/request.jl
@@ -1,3 +1,9 @@
+abstract type AbstractBackend end
+struct HTTPBackend <: AbstractBackend end
+struct DownloadsBackend <: AbstractBackend end
+
+default_backend() = DownloadsBackend()
+
 Base.@kwdef mutable struct Request
     service::String
     api_version::String
@@ -14,15 +20,10 @@ Base.@kwdef mutable struct Request
     return_raw::Bool=false
     response_dict_type::Type{<:AbstractDict}=LittleDict
     downloader::Union{Nothing, Downloads.Downloader}=nothing
+    backend::AbstractBackend=default_backend()
 end
 
-abstract type AbstractBackend end
-struct HTTPBackend <: AbstractBackend end
-struct DownloadsBackend <: AbstractBackend end
-
-default_backend() = DownloadsBackend()
-
-submit_request(aws::AbstractAWSConfig, request::Request; return_headers::Bool=false) = submit_request(default_backend(), aws, request; return_headers)
+submit_request(aws::AbstractAWSConfig, request::Request; return_headers::Bool=false) = submit_request(request.backend, aws, request; return_headers)
 
 const AWS_DOWNLOADER = Ref{Union{Nothing, Downloader}}(nothing)
 const AWS_DOWNLOAD_LOCK = ReentrantLock()

diff --git a/src/utilities/utilities.jl b/src/utilities/utilities.jl
@@ -69,6 +69,7 @@ function _extract_common_kw_args(service, args)
         http_options=_pop!(args, "http_options", LittleDict{Symbol, String}()),
         response_dict_type=_pop!(args, "response_dict_type", LittleDict),
         downloader=_pop!(args, "downloader", nothing),
+        backend=_pop!(args, "backend", default_backend()),
     )
 end