Question

我有一个http客户端，可以创建与主机的多个连接。我想设置它可以设置到特定主机的最大连接数。在go的请求中没有这样的选项.Transport。我的代码看起来像

package main 

import (
  "fmt"
  "net/http"
  "net/url"
)

const (
  endpoint_url_fmt      = "https://blah.com/api1?%s"
)


func main() {

  transport := http.Transport{ DisableKeepAlives : false }

  outParams := url.Values{}
  outParams.Set("method", "write")
  outParams.Set("message", "BLAH")

  for {
    // Encode as part of URI.
    outboundRequest, err := http.NewRequest(
      "GET",
      fmt.Sprintf(endpoint_url_fmt, outParams.Encode()),
      nil
    )
    outboundRequest.Close = false
    _ , err = transport.RoundTrip(outboundRequest)
    if err != nil {
      fmt.Println(err)
    }
  }

}

我希望这能创建1个连接。因为我在for循环中调用它。但这会不断创建无限数量的连接。

使用请求库的类似python代码只创建一个连接。

#!/usr/bin/env python
import requests
endpoint_url_fmt      = "https://something.com/restserver.php"
params = {}
params['method'] = 'write'
params['category'] = category_errors_scuba
params['message'] = "blah"
while True:
  r = requests.get(endpoint_url_fmt, params = params)

由于某种原因，go代码没有重用http连接。

编辑： go代码需要关闭主体以重用连接。

 resp , err = transport.RoundTrip(outboundRequest)
 resp.Close() //  This allows the connection to be reused

Answer 1

基于OP的进一步澄清。默认客户端重用连接。

请务必关闭response。

调用者在完成阅读后应该关闭resp.Body。如果resp.Body未关闭，则客户端的基础RoundTripper（通常为Transport）可能无法重新使用到服务器的持久TCP连接以用于后续“保持活动”请求。

此外，我发现在调用Close（）之前，我还需要阅读，直到响应完成。

e.g。

res, _ := client.Do(req)
io.Copy(ioutil.Discard, res.Body)
res.Body.Close()

要确保http.Client连接重用，请务必执行以下两项操作：

阅读直至回复完成（即ioutil.ReadAll(resp.Body)）
致电Body.Close()

旧答案，对速率限制很有用，但不是OP之后的内容：

我不认为通过golang 1.1 http API设置最大连接。这意味着如果你不小心，你可以通过大量的TCP连接（直到用完文件描述符或其他任何东西）射击自己。

那就是说，你可以通过time.Tick限制特定主机（以及出站请求和连接）调用go例程的速率。

例如：

import "time"

requests_per_second := 5
throttle := time.Tick(1000000000 / requests_per_second)

for i := 0; i < 16; i += 1 {
  <-throttle  
  go serveQueue()
}

Answer 2

http.Transport中有一些有趣的改进：

// DisableKeepAlives, if true, disables HTTP keep-alives and
// will only use the connection to the server for a single
// HTTP request.
//
// This is unrelated to the similarly named TCP keep-alives.
DisableKeepAlives bool

// ...

// MaxIdleConns controls the maximum number of idle (keep-alive)
// connections across all hosts. Zero means no limit.
MaxIdleConns int // Go 1.7

// MaxIdleConnsPerHost, if non-zero, controls the maximum idle
// (keep-alive) connections to keep per-host. If zero,
// DefaultMaxIdleConnsPerHost is used.
MaxIdleConnsPerHost int

// MaxConnsPerHost optionally limits the total number of
// connections per host, including connections in the dialing,
// active, and idle states. On limit violation, dials will block.
//
// Zero means no limit.
MaxConnsPerHost int // Go 1.11

为什么Go HTTPS Client不重用连接？

2 个答案: