• 企业400电话
  • 微网小程序
  • AI电话机器人
  • 电商代运营
  • 全 部 栏 目

    企业400电话 网络优化推广 AI电话机器人 呼叫中心 网站建设 商标✡知产 微网小程序 电商运营 彩铃•短信 增值拓展业务
    golang设置http response响应头与填坑记录

    1、 设置WriteHeader的顺序问题

    之前遇到个问题,在一段代码中这样设置WriteHeader,最后在header中取Name时怎么也取不到。

    w.WriteHeader(201)
    w.Header().Set("Name", "my name is smallsoup")

    用 golang 写 http server 时,可以很方便可通过 w.Header.Set(k, v) 来设置 http response 中 header 的内容。但是需要特别注意的是:某些时候不仅要修改 response的header ,还要修改 response的StatusCode。修改response的StatusCode 可以通过:w.WriteHeader(code) 来实现,例如:

    w.WriteHeader(404)

    如果这两种修改一起做,就必须让 w.WriteHeader 在所有的 w.Header.Set 之后,因为 w.WriteHeader 后 Set Header 是无效的。

    而且必须是在 w.Write([]byte("HelloWorld")) 之前,否则会报 http: multiple response.WriteHeader calls 因为其实调用w.Write的时候也会调用WriteHeader()方法,然后将w.wroteHeader置为true,再次调WriteHeader()则会判断wroteHeader,如果是true则会报错,而且本次调用不生效。

    可以看以下源码说明WriteHeader必须在Write之前调用。

    func (w *response) WriteHeader(code int) {
     if w.conn.hijacked() {
     w.conn.server.logf("http: response.WriteHeader on hijacked connection")
     return
     }
    //第二次WriteHeader()进来满足if条件就报错直接return
     if w.wroteHeader {
     w.conn.server.logf("http: multiple response.WriteHeader calls")
     return
     }
    //第一次write()进来这里会将w.wroteHeader置为true
     w.wroteHeader = true
     w.status = code
    
     if w.calledHeader  w.cw.header == nil {
     w.cw.header = w.handlerHeader.clone()
     }
    
     if cl := w.handlerHeader.get("Content-Length"); cl != "" {
     v, err := strconv.ParseInt(cl, 10, 64)
     if err == nil  v >= 0 {
     w.contentLength = v
     } else {
     w.conn.server.logf("http: invalid Content-Length of %q", cl)
     w.handlerHeader.Del("Content-Length")
     }
     }
    }

    2、 go会对Header中的key进行规范化处理

    go会对Header中的key进行规范化处理,所以在获取response的Header中的K,V值时一定要小心。

    reader.go中非导出方法canonicalMIMEHeaderKey中有这样一段,会将header的key进行规范化处理。

    1)reader.go中定义了isTokenTable数组,如果key的长度大于127或者包含不在isTokenTable中的字符,则该key不会被处理。

    2)将key的首字母大写,字符 - 后的单词的首字母也大写。

    分析如下源码,可以解释对key的大写处理:

    for i, c := range a {
     // 规范化:首字母大写
     // - 之后单子的首字母大写
     // 如:(Host, User-Agent, If-Modified-Since).
     if upper  'a' = c  c = 'z' {
     //大写转小写
     c -= toLower
     } else if !upper  'A' = c  c = 'Z' {
     //小写转大写
     c += toLower
     }
     //重新给key数组赋值
     a[i] = c
     //设置大小写标志位
     upper = c == '-' // for next time
    }

    正确的调用方式:

    服务器:myServer.go

    package main
    
    import (
     "net/http"
    )
    
    func main() {
    
     http.HandleFunc("/", func (w http.ResponseWriter, r *http.Request){
    
    
     w.Header().Set("name", "my name is smallsoup")
     w.WriteHeader(500)
     w.Write([]byte("hello world\n"))
    
     })
    
     http.ListenAndServe(":8080", nil)
    }

    客户端:

    myHttp.go:

    package main
    
    import (
     "fmt"
     "io/ioutil"
     "net/http"
    )
    
    func main() {
    
     myHttpGet()
    
    }
    
    func myHttpGet() {
    
     rsp, err := http.Get("http://localhost:8080")
     if err != nil {
     fmt.Println("myHttpGet error is ", err)
     return
     }
    
     defer rsp.Body.Close()
     body, err := ioutil.ReadAll(rsp.Body)
     if err != nil {
     fmt.Println("myHttpGet error is ", err)
     return
     }
    
     fmt.Println("response statuscode is ", rsp.StatusCode, 
     "\nhead[name]=", rsp.Header["Name"], 
     "\nbody is ", string(body))
    }

    1.运行服务器

    go run myServer.go

    2.运行客户端

    go run myHttp.go

    输出如下:statuscode是我们设置的500,Name也取到了值。

    golang处理http response碰到的问题和需要注意的点

    在处理http response的时候,偶然发现,body读取之后想再次读取的时候,发现读不到任何东西。见下方代码:

    response, err = ioutil.ReadAll(resp.Body)
    if err != nil {
    log.Println("ioutil ReadAll failed :", err.Error())
    return
    }

    之后如果想再次ioutil.ReadAll(resp.Body)的时候会发现读到的是空。于是我决定去看一下这个resp.Body,发现它是一个io.ReadCloser接口,包含了Reader和Closer接口:

    type ReadCloser interface {
    Reader
    Closer
    }

    于是我想到了文件,它也实现了io.Reader接口,所以用读文件试了下:

    func readFile(path string)string{
    fi,err := os.Open(path)
    if err != nil{panic(err)}
    defer fi.Close()
    
    byte1,err := ioutil.ReadAll(fi)
    fmt.Println(string(byte1))
    
    byte2,err := ioutil.ReadAll(fi)
    fmt.Println(string(byte2))
    
    return string(fd)
    }

    发现结果是一致的,fmt.Println(string(fd2))打印不出任何结果。我猜测应该是ioutil.ReadAll()是有记录偏移量,所以会出现第二次读取读不到的情况。作为client端处理response的时候会碰到这个问题,作为server端要处理request body的时候,一样会遇到此问题,那么该如何解决这个问题呢?
    有一个方法是再造一个io.ReadCloser,如下:

    fi2:= ioutil.NopCloser(bytes.NewBuffer(byte1))
    byte3,err := ioutil.ReadAll(fi2)
    fmt.Println(string(byte3))

    此外,作为client端处理response的时候,有一点要注意的是,body一定要close,否则会造成GC回收不到,继而产生内存泄露。其实在go的官方源码注释中,也明确注明了response body需要调用方进行手动关闭:It is the caller's responsibility to close Body.

    至于response body为什么需要进行关闭,这篇文章进行了解释: https://www.jb51.net/article/146275.htm

    那么作为client端生成的request body,需不需要手动关闭呢,答案是不需要的,net/http中的func (c *Client) Do(req *Request) (*Response, error)会调用Close()

    同样的,作为server端接收的request body,也是需要关闭,由Server自动进行关闭,The Server will close the request body. The ServeHTTP Handler does not need to.

    总结

    以上就是这篇文章的全部内容了,希望本文的内容对大家的学习或者工作具有一定的参考学习价值,如果有疑问大家可以留言交流,谢谢大家对脚本之家的支持。

    您可能感兴趣的文章:
    • 解决golang处理http response碰到的问题和需要注意的点
    • golang中为什么Response.Body需要被关闭详解
    • 解决Golang中ResponseWriter的一个坑
    上一篇:浅谈Go Slice 高级实践
    下一篇:golang中为什么Response.Body需要被关闭详解
  • 相关文章
  • 

    © 2016-2020 巨人网络通讯 版权所有

    《增值电信业务经营许可证》 苏ICP备15040257号-8

    golang设置http response响应头与填坑记录 golang,设置,http,response,响应,