ThinkChat2.0新版上线,更智能更精彩,支持会话、画图、阅读、搜索等,送10W Token,即刻开启你的AI之旅 广告
## 封装http请求客户端 在2015年,TomásSenart就使用接口包装http.Client结构进行了精彩的探讨,你可以在https//github.com/gophercon/2015-talks上找到更多相关信息。 本节从他的想法中获取并演示了一个与http.Client结构的Transport接口相同的示例,类似于我们之前的示例,为REST API编写客户端。 本节将为标准的http.Client结构实现日志记录和基本身份验证中间件。 它还包括一个decorate函数,可以在需要使用各种中间件时使用。 ### 实践 1. 建立 config.go: ``` package decorator import ( "log" "net/http" "os" ) // Setup 初始化客户端接口 func Setup() *http.Client { c := http.Client{} t := Decorate(&http.Transport{}, Logger(log.New(os.Stdout, "", 0)), BasicAuth("username", "password"), ) c.Transport = t return &c } ``` 2. 建立 decorator.go: ``` package decorator import "net/http" // TransportFunc 实现了 RountTripper 接口 type TransportFunc func(*http.Request) (*http.Response, error) // RoundTrip 仅调用原始的函数 func (tf TransportFunc) RoundTrip(r *http.Request) (*http.Response, error) { return tf(r) } // Decorator 是一个方便的函数来表示我们的中间件内部功能 type Decorator func(http.RoundTripper) http.RoundTripper // Decorate 用于包装所有中间件 func Decorate(t http.RoundTripper, rts ...Decorator) http.RoundTripper { decorated := t for _, rt := range rts { decorated = rt(decorated) } return decorated } ``` 3. 建立 middleware.go: ``` package decorator import ( "log" "net/http" "time" ) // Logger 日志中间件 func Logger(l *log.Logger) Decorator { return func(c http.RoundTripper) http.RoundTripper { return TransportFunc(func(r *http.Request) (*http.Response, error) { start := time.Now() l.Printf("started request to %s at %s", r.URL, start.Format("2006-01-02 15:04:05")) resp, err := c.RoundTrip(r) l.Printf("completed request to %s in %s", r.URL, time.Since(start)) return resp, err }) } } // BasicAuth 基础身份认证中间件 func BasicAuth(username, password string) Decorator { return func(c http.RoundTripper) http.RoundTripper { return TransportFunc(func(r *http.Request) (*http.Response, error) { r.SetBasicAuth(username, password) resp, err := c.RoundTrip(r) return resp, err }) } } ``` 4. 建立 exec.go: ``` package decorator import "fmt" // Exec 创建一个客户端,调用google.com然后打印响应 func Exec() error { c := Setup() resp, err := c.Get("https://www.google.com") if err != nil { return err } fmt.Println("Response code:", resp.StatusCode) return nil } ``` 5. 建立 main.go: ``` package main import "github.com/agtorre/go-cookbook/chapter6/decorator" func main() { if err := decorator.Exec(); err != nil { panic(err) } } ``` 6. 这会输出: ``` started request to https://www.google.com at 2017-01-01 13:38:42 completed request to https://www.google.com in 194.013054ms Response code: 200 ``` 本节将闭包作为一流的公民和接口。主要技巧是具有功能实现接口。这允许封装具有由函数实现的接口的结构实现的接口。 middleware.go文件包含两个示例客户端中间件函数。 这些可以扩展为包含其他中间件,例如更复杂的身份验证。 本示例也可以与之前的示例结合使用,以生成可以通过其他中间件进行扩展的OAuth2客户端。 Decorator函数是一种便利的封装函数,可以这样比较下: ``` Decorate(RoundTripper, Middleware1, Middleware2, etc) vs var t RoundTripper t = Middleware1(t) t = Middleware2(t) etc ``` 与封装客户端相比,这种方法的优点是我们可以保持接口解耦。如果你想要一个功能齐全的客户端,你还需要实现GET,POST和PostForm等方法。 ### 说明 * * * * 学识浅薄,错误在所难免。欢迎在群中就本书提出修改意见,以飨后来者,长风拜谢。 Golang中国(211938256) beego实战(258969317) Go实践(386056972)