Golang代码优化04-Http响应处理

介绍读取http的response内容。

读取方式

• 1.使用ioutil.ReadAll()读取response

  func main () {
      // send get request
      resp, err := http.Get("https://api.ipify.org/?format=json")
  	if err != nil {
  		fmt.Println(err)
  		return
      }
      
      // defer must execute after check error,otherwise will panic when resp is nil
      defer resp.Body.Close()
      
      // read body
  	body, err := ioutil.ReadAll(resp.Body)
  	if err != nil {
  		fmt.Println(err)
  		return
  	}
  	fmt.Println(string(body))
  }

  但是ioutil.ReadAll()方法有弊端,有时候却会导致一些性能问题。比如http大量请求时轻则导致内存浪费严重，重则导致内存泄漏影响业务。

• 2.使用buffer读取

  func main () {
  	resp, err := http.Get("https://api.ipify.org/?format=json")
  	if err != nil {
  		fmt.Println(err)
  		return
  	}
      defer resp.Body.Close()
      
      // new buffer
  	buffer := bytes.NewBuffer(make([]byte, 4056))
  
  	if _, err := io.Copy(buffer, resp.Body);err != nil {
  		fmt.Println(err)
  		return
  	}
  	fmt.Println(buffer.String())
  }

• 3.BufferPool
  当需要处理较多的httpResponse时，可以选择先初始化一个Buffer池，提升效率。

  // BufferPool use sync.pool to build a buffer pool
  type BufferPool struct {
  	pool sync.Pool
  }
  
  // NewBufferPool return BufferPool instance by buffer length
  func NewBufferPool(bufLen int) *BufferPool {
  	return &BufferPool{pool:sync.Pool{New: func() interface{} {
  		return bytes.NewBuffer(make([]byte, bufLen))
  	}}}
  }
  
  // StringifyResponse return response body as string
  func (bp *BufferPool) StringifyResponse(response *http.Response) (string, error) {
  	if response == nil {
  		return "", fmt.Errorf("response is empty")
  	}
  
  	// close response
  	defer func() {
  		_ = response.Body.Close()
  	}()
  
  	if response.StatusCode != http.StatusOK {
  		return "", fmt.Errorf("response status code is:%d", response.StatusCode)
  	}
      
  	buffer := bp.pool.Get().(*bytes.Buffer) // get buffer from pool
  	buffer.Reset() // reset buffer
  	defer func() {
  		if buffer != nil {
  			bp.pool.Put(buffer) // return back
  			buffer = nil // close 
  		}
  	}()
  	_, err := io.Copy(buffer, response.Body)
  
  	if err != nil {
  		return "", fmt.Errorf("failed to read respone:%s", err.Error())
  	}
  
  	return buffer.String(), nil
  }

• 本文作者： Hongker
• 本文链接： https://hongker.github.io/2020/03/31/http-response/
• 版权声明： 本博客所有文章除特别声明外，均采用 MIT 许可协议。转载请注明出处！