bugfix: initialize dfget downloader in dfdaemon

Signed-off-by: lowzj <[email protected]>
dragonflyoss · Apr 29, 2020 · bacc34b · bacc34b
1 parent 8cac49e
commit bacc34b
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/dfdaemon/proxy/proxy.go b/dfdaemon/proxy/proxy.go
@@ -207,6 +207,7 @@ func (proxy *Proxy) mirrorRegistry(w http.ResponseWriter, r *http.Request) {
 	reverseProxy := httputil.NewSingleHostReverseProxy(proxy.registry.Remote.URL)
 	t, err := transport.New(
 		transport.WithDownloader(proxy.downloadFactory()),
+		transport.WithStreamDownloader(proxy.streamDownloadFactory()),
 		transport.WithTLS(proxy.registry.TLSConfig()),
 		transport.WithCondition(proxy.shouldUseDfgetForMirror),
 	)
@@ -268,6 +269,7 @@ func (proxy *Proxy) handleHTTP(w http.ResponseWriter, req *http.Request) {
 
 func (proxy *Proxy) roundTripper(tlsConfig *tls.Config) http.RoundTripper {
 	rt, _ := transport.New(
+		transport.WithDownloader(proxy.downloadFactory()),
 		transport.WithStreamDownloader(proxy.streamDownloadFactory()),
 		transport.WithTLS(tlsConfig),
 		transport.WithCondition(proxy.shouldUseDfget),