X-Git-Url: https://git.arvados.org/arvados.git/blobdiff_plain/d53271e587b7bdbfe37b8ae8eaf890dd69c2796b..ac2426ef6ba7c2c722da0f7b0add2c5040529bd3:/services/keepproxy/keepproxy.go?ds=sidebyside diff --git a/services/keepproxy/keepproxy.go b/services/keepproxy/keepproxy.go index f2a93f1e3a..7900096caf 100644 --- a/services/keepproxy/keepproxy.go +++ b/services/keepproxy/keepproxy.go @@ -515,16 +515,15 @@ func (handler IndexHandler) ServeHTTP(resp http.ResponseWriter, req *http.Reques kc := *handler.KeepClient - var pass bool - var tok string - if pass, tok = CheckAuthorizationHeader(kc, handler.ApiTokenCache, req); !pass { + ok, token := CheckAuthorizationHeader(kc, handler.ApiTokenCache, req) + if !ok { status, err = http.StatusForbidden, BadAuthorizationHeader return } // Copy ArvadosClient struct and use the client's API token arvclient := *kc.Arvados - arvclient.ApiToken = tok + arvclient.ApiToken = token kc.Arvados = &arvclient // Only GET method is supported @@ -533,6 +532,7 @@ func (handler IndexHandler) ServeHTTP(resp http.ResponseWriter, req *http.Reques return } + // Get index from all LocalRoots and write to resp var reader io.Reader for uuid := range kc.LocalRoots() { reader, err = kc.GetIndex(uuid, prefix) @@ -541,15 +541,7 @@ func (handler IndexHandler) ServeHTTP(resp http.ResponseWriter, req *http.Reques return } - var readBytes []byte - readBytes, err = ioutil.ReadAll(reader) - if err != nil { - status = http.StatusBadGateway - return - } - - // Got index for this server; write to resp - _, err := resp.Write(readBytes) + _, err = io.Copy(resp, reader) if err != nil { status = http.StatusBadGateway return