Go标准库--net/http学习

Go中对网络的支持提供了标准库，net包提供了可移植的网络I/O接口，包括TCP/IP、UDP、域名解析和Unix域socket。 http包提供了HTTP客户端和服务端的实现。 一般我们用http肯定多一些，下面来看一下http的使用方式。

1. Post和Get请求的使用

创建一个最简单的get请求：

package main

import (
        "fmt"
        "io/ioutil"
        "net/http"
)

func main() {
        resp, err := http.Get("http://www.baidu.com")
        if err != nil {
                fmt.Print("err",err)
        }
        closer := resp.Body
        bytes, err := ioutil.ReadAll(closer)
        fmt.Println(string(bytes))
}

执行程序可以获取百度首页的源码。

再看一个post请求的格式：

package main

import (
        "bytes"
        "fmt"
        "io/ioutil"
        "net/http"
)

func main() {
        
        url := "http://www.baidu.com/comment/add"
        body := "{\"userId\":223446,\"articleId\":443567,\"comment\":\"我是一条评论\"}"
        response, err := http.Post(url, "•application/x-www-form-urlencoded", bytes.NewBuffer([]byte(body)))
        if err != nil {
                fmt.Println("err",err)
        }
        b1, err := ioutil.ReadAll(response.Body)
        fmt.Println(string(b1))

}

上面的post请求第一个参数是url地址,第二个参数是Content-Type ，第三个参数是post请求参数的字节码。

Http Header里的Content-Type一般有这三种：

• application/x-www-form-urlencoded：数据被编码为名称/值对。这是标准的编码格式。
• multipart/form-data： 数据被编码为一条消息，页上的每个控件对应消息中的一个部分。
• text/plain： 数据以纯文本形式(text/json/xml/html)进行编码，其中不含任何控件或格式字符。postman软件里标的是RAW。

网页中form的enctype属性为编码方式，常用有两种：

• application/x-www-form-urlencoded，默认编码方式
• multipart/form-data

Go还提供了另一种Post请求的api：

func PostForm(url string, data url.Values) (resp *Response, err error) {
        return DefaultClient.PostForm(url, data)
}

//DefaultClient.PostForm(url, data)的实现如下：
func (c *Client) PostForm(url string, data url.Values) (resp *Response, err error) {
        return c.Post(url, "application/x-www-form-urlencoded", strings.NewReader(data.Encode()))
}

PostForm()方法默认“application/x-www-form-urlencoded”的Content-Type类型，不用你手动去填写Content-Type了。

2. 如果想在请求中手动set header头信息该怎么做

http包提供了Header类型：

type Header map[string][]string

用于请求头信息的获取和填充。

你可以自己定义Header：

headers := http.Header{"token": {"fdase34532534fwr324brfh3urhf839hf349h"}}
headers.Add("Accept-Charset","UTF-8")
headers.Set("Host","www.baidu.com")

注意：Header是 map[string][]string类型的，value为字符数字。

3.NewRequest 和Client

在http包中也提供了一个叫做Client的结构体，它实现了Get,Post,等方法。并且还提供了一个默认的变量可以直接使用：

var DefaultClient = &Client{}

NewRequest是一个方法：

func NewRequest(method, url string, body io.Reader) (*Request, error){}

第一个参数为请求类型，“GET”，“POST”，“PUT”，“DELETE”，等等。

如果body参数实现了io.Closer接口，Request返回值的Body 字段会被设置为body，并会被Client类型的Do、Post和PostFOrm方法以及Transport.RoundTrip方法关闭。

观看源码我们可以发现，因为NewRequest是一个通用方法，我们调用Get,Post其实是Go帮我们在底层传了相应的method去调用NewRequest，

所以最核心的http请求源码就是NewRequest。

另外，Client的Get和Post方实现里面去调用NewRequest，而http里面的Get和Post方法其实是分别调用Client的Get和Post方法的。

http的Post方法:

func Post(url, contentType string, body io.Reader) (resp *Response, err error) {
        return DefaultClient.Post(url, contentType, body)
}

上面使用了DefaultClient变量。

Client的Post方法：

func (c *Client) Post(url, contentType string, body io.Reader) (resp *Response, err error) {
        req, err := NewRequest("POST", url, body)
        if err != nil {
                return nil, err
        }
        req.Header.Set("Content-Type", contentType)
        return c.Do(req)
}

直接使用NewRequest方法。

4. 服务端的实现

前面说的这些发送Get或者Post请求，其实是相当于我们在模拟客户端的实现调用服务端接口。那么类比Java 的Controller层，使用Go应该如何实现呢？http.HandlerFunc我们提供了路由注册功能。

type HandlerFunc func(ResponseWriter, *Request)

HandlerFunc type是一个适配器，通过类型转换让我们可以将普通的函数作为HTTP处理器使用。如果f是一个具有适当签名的函数，HandlerFunc(f)通过调用f实现了Handler接口（因为HandlerFunc实现了ServeHTTP函数），其实就是将函数f显示转换成HandlerFunc类型。

4.1 如何创建web服务端

package main

import (
    "net/http"
)

func SayHello(w http.ResponseWriter, req *http.Request) {
    w.Write([]byte("Hello"))
}

func main() {
    http.HandleFunc("/hello", SayHello)
    http.ListenAndServe(":8080", nil)

}

首先调用Http.HandleFunc

按顺序做了几件事：

• 调用了DefaultServerMux的HandleFunc
• 调用了DefaultServerMux的Handle
• 往DefaultServeMux的map[string]muxEntry中增加对应的handler和路由规则

其次调用http.ListenAndServe(“:8080”, nil)

按顺序做了几件事情：

• 实例化Server
• 调用Server的ListenAndServe()
• 调用net.Listen(“tcp”, addr)监听端口
• 启动一个for循环，在循环体中Accept请求
• 对每个请求实例化一个Conn，并且开启一个goroutine为这个请求进行服务go c.serve()
• 读取每个请求的内容w, err := c.readRequest()
• 判断header是否为空，如果没有设置handler（这个例子就没有设置handler），handler就设置为DefaultServeMux
• 调用handler的ServeHttp
• 在这个例子中，下面就进入到DefaultServerMux.ServeHttp
• 根据request选择handler，并且进入到这个handler的ServeHTTP

   mux.handler(r).ServeHTTP(w, r)

• 选择handler：

A 判断是否有路由能满足这个request（循环遍历ServerMux的muxEntry）

B 如果有路由满足，调用这个路由handler的ServeHttp

C 如果没有路由满足，调用NotFoundHandler的ServeHttp

上面我们注意到：

ListenAndServe(addr string, handler Handler)

其实是有两个参数的：当前监听端口号，事件处理器handler。

Handler接口的定义方式：

type Handler interface {
   ServeHTTP(ResponseWriter, *Request)
}

我们只需要实现这个接口就可以定义自己的handler，Go语言在自带包中已经帮我们提供了实现这个接口的公共方法：

type HandlerFunc func(ResponseWriter, *Request)

// ServeHTTP calls f(w, r).
func (f HandlerFunc) ServeHTTP(w ResponseWriter, r *Request) {
        f(w, r)
}

Go语言将func(ResponseWriter, *Request)这种类型的函数直接定义了类型HandlerFunc，而且还实现了ServeHTTP这个方法，但是这个方法本身并没有实现任何逻辑，需要我们自己来实现。

package main

import (
        "net/http"
)

func myHandler(w http.ResponseWriter, r *http.Request) {
        w.Write([]byte("Hello World"))
}

func main() {
        http.HandleFunc("/hello", SayHello)
        http.ListenAndServe(":8080", http.HandlerFunc(myHandler))

}

类比的话，handler机制类似于Java SpringMVC中的Interceptor，是一个拦截器的性质。它发生在http.HandleFunc处理逻辑之前。

我们可以来实现一个小功能：只有指定referer头来的请求才能调用服务，否则返回403。

我们先定义一个结构体:

type SpecialRefer struct {
        handler     http.Handler
        referer string
}

包含两个对象：handler和自定义的referer。

因为我们需要将这个SpecialRefer实例化并传递给ListenAndServe这个方法，因此它必须实现ServeHTTP这个方法，所以在ServeHTTP里面可以直接定义我们用来实现中间件的逻辑。

func (this *SpecialRefer) ServeHTTP(w http.ResponseWriter, r *http.Request) {
        if r.Referer() == this.referer {
                this.handler.ServeHTTP(w, r)
        } else {
                w.WriteHeader(403)
        }
}

取出当前请求头中的referer信息，如果跟我们约定的不同则拦截请求。

完整代码如下：

package main

import (
        "net/http"
)

type SpecialRefer struct {
        handler     http.Handler
        referer string
}

func (this *SpecialRefer) ServeHTTP(w http.ResponseWriter, r *http.Request) {
        if r.Referer() == this.referer {
                this.handler.ServeHTTP(w, r)
        } else {
                w.WriteHeader(403)
        }
}

func myHandler(w http.ResponseWriter, r *http.Request) {
        w.Write([]byte("this is handler"))
}

func hello(w http.ResponseWriter, r *http.Request)  {
        w.Write([]byte("hello"))
}

func main() {
        referer := &SpecialRefer{
                handler:http.HandlerFunc(myHandler),
                referer:"www.baidu.com",
        }
        http.HandleFunc("/hello",hello)
        http.ListenAndServe(":8080", referer)
}