使用http.Client

标准库的net/http包的http.Client结构十分灵活，可用户处理HTTP API。此结构具有独立的传输函数，并且相对简单，可以短路请求，修改每个客户端操作的请求头以及处理REST操作。 创建http请求是常见的操作，本节将从操作http.Client对象的基础开始。

实践

建立 client.go：

package client

import (
    "crypto/tls"
    "net/http"
)

// Setup 设置http.Client并重新定义全局DefaultClient
func Setup(isSecure, nop bool) *http.Client {
    c := http.DefaultClient

    // 有时为了方便测试，我们需要关闭SSL验证
    if !isSecure {
        c.Transport = &http.Transport{
            TLSClientConfig: &tls.Config{
                InsecureSkipVerify: false,
            },
        }
    }
    if nop {
        c.Transport = &NopTransport{}
    }
    http.DefaultClient = c
    return c
}

// NopTransport 没有任何操作的传输
type NopTransport struct {
}

// RoundTrip 实现了 RoundTripper 接口
func (n *NopTransport) RoundTrip(*http.Request) (*http.Response, error) {
    // 注意这里只为StatusCode赋值
    return &http.Response{StatusCode: http.StatusTeapot}, nil
}

建立 exec.go：

package client

import (
    "fmt"
    "net/http"
)

// DoOps 接收client参数 然后请求 google.com
func DoOps(c *http.Client) error {
    resp, err := c.Get("http://www.google.com")
    if err != nil {
        return err
    }
    fmt.Println("results of DoOps:", resp.StatusCode)

    return nil
}

// DefaultGetGolang 使用默认的client请求 golang.org
func DefaultGetGolang() error {
    resp, err := http.Get("https://www.golang.org")
    if err != nil {
        return err
    }
    fmt.Println("results of DefaultGetGolang:", resp.StatusCode)
    return nil
}

建立 storage.go：

package client

import (
    "fmt"
    "net/http"
)

// Controller 包含有匿名字段*http.Client
type Controller struct {
    *http.Client
}

func (c *Controller) DoOps() error {
    resp, err := c.Client.Get("http://www.google.com")
    if err != nil {
        return err
    }
    fmt.Println("results of client.DoOps", resp.StatusCode)
    return nil
}

建立 main.go：

package main

import "github.com/agtorre/go-cookbook/chapter6/client"

func main() {

    cli := client.Setup(true, false)

    if err := client.DefaultGetGolang(); err != nil {
        panic(err)
    }

    if err := client.DoOps(cli); err != nil {
        panic(err)
    }

    c := client.Controller{Client: cli}
    if err := c.DoOps(); err != nil {
        panic(err)
    }

    client.Setup(true, true)

    if err := client.DefaultGetGolang(); err != nil {
        panic(err)
    }
}

这会输出：

results of DefaultGetGolang: 200
results of DoOps: 200
results of client.DoOps 200
results of DefaultGetGolang: 418

说明

net/http包公开了一个DefaultClient包变量，该变量使用默认配置操作Do，GET，POST等。我们的Setup函数返回一个客户端，该客户端在设置时，灵活的修改可以通过调整实现的RoundTripper接口实现。

通过调整Setup函数的nop参数，可以方便的对client的配置进行调整，这对测试来说很重要。