golang http连接复用方法

Go  /  管理员 发布于 5年前   822

server端

golang httpserver 默认开启keepalive连接复用选项

handler函数需要完整读body数据，构造返回消息，否则当数据不能一次发送完成时，连接复用就会失效。

示例如下

package main import ("fmt""io/ioutil""log""net/http""os""strconv""strings""time") func connHandler(w http.ResponseWriter, r *http.Request) {// parser.ParseForm()response_time := r.Header.Get("sleep-time")// <= NOTEif _, err := ioutil.ReadAll(r.Body); err != nil {http.Error(w, err.Error(), 500)return}defer r.Body.Close()// sleep for some timeresp_time := 1if response_time != "" {ms, _ := strconv.ParseFloat(response_time, 64)resp_time = (int)(ms * 1000)}time.Sleep(time.Duration(resp_time) * time.Millisecond)// parepare responsestatus := 200body := ""w.Header().Set("Content-Type", "text/plain")w.Header().Set("Content-Length", strconv.Itoa(len(body)))w.WriteHeader(status)w.Write([]byte(body))} func main() {http.HandleFunc("/", connHandler)if err := http.ListenAndServe(":server_port", nil); err != nil {log.Fatal("ListenAndServe: ", err)}}

client 端

客户端需要构建全局client，完整读 response body，并关闭body

package mainimport (  "bytes"  "fmt"  "io"  "io/ioutil"  "log"  "net/http"  "time")var (  httpClient *http.Client)const (  MaxIdleConnections int = 20  RequestTimeout   int = 30)// init HTTPClientfunc init() {  httpClient = createHTTPClient()}// createHTTPClient for connection re-usefunc createHTTPClient() *http.Client {  client := &http.Client{   Transport: &http.Transport{      MaxIdle  ConnsPerHost: MaxIdleConnections, }, Timeout: time.Duration(RequestTimeout) * time.Second,  }  return client}func conn_reuse_post(conn_reuse_times int) {  var endPoint string = "http://server_ip:server_port/"  data := []byte{}  // fill data   for i := 0; i < conn_reuse_times; i++ { // use global httpClient to send request resp, err := httpClient.Post(endPoint, "application/x-www-form-urlencoded", bytes.NewBuffer([]byte(data))) fmt.Println(resp) if err != nil {   log.Println("err", err)   return } io.Copy(ioutil.Discard, resp.Body) // <= NOTE resp.Body.Close()  // <= NOTE  }}func main() {  conn_reuse_post(5)}

以上这篇golang http连接复用方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持。