1 //go:build !go1.8 2 // +build !go1.8 3 4 // Copyright 2017 Microsoft Corporation 5 // 6 // Licensed under the Apache License, Version 2.0 (the "License"); 7 // you may not use this file except in compliance with the License. 8 // You may obtain a copy of the License at 9 // 10 // http://www.apache.org/licenses/LICENSE-2.0 11 // 12 // Unless required by applicable law or agreed to in writing, software 13 // distributed under the License is distributed on an "AS IS" BASIS, 14 // WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. 15 // See the License for the specific language governing permissions and 16 // limitations under the License. 17 18 package autorest 19 20 import ( 21 "bytes" 22 "io/ioutil" 23 "net/http" 24 ) 25 26 // RetriableRequest provides facilities for retrying an HTTP request. 27 type RetriableRequest struct { 28 req *http.Request 29 br *bytes.Reader 30 } 31 32 // Prepare signals that the request is about to be sent. 33 func (rr *RetriableRequest) Prepare() (err error) { 34 // preserve the request body; this is to support retry logic as 35 // the underlying transport will always close the reqeust body 36 if rr.req.Body != nil { 37 if rr.br != nil { 38 _, err = rr.br.Seek(0, 0 /*io.SeekStart*/) 39 rr.req.Body = ioutil.NopCloser(rr.br) 40 } 41 if err != nil { 42 return err 43 } 44 if rr.br == nil { 45 // fall back to making a copy (only do this once) 46 err = rr.prepareFromByteReader() 47 } 48 } 49 return err 50 } 51 52 func removeRequestBody(req *http.Request) { 53 req.Body = nil 54 req.ContentLength = 0 55 } 56