golang 设置 http response 响应头与坑

Love The Way You Lie 2022-01-07 07:13 296阅读 0赞

## 正文 ##

**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也取到了值。

![在这里插入图片描述][2019101920032475.png]

**关于go、docker、k8s相关学习资源关注文末公众号后回复【1】获取**

--------------------

# 作者简洁 #

> 作者：小碗汤，一位热爱、认真写作的小伙，目前维护原创公众号：『我的小碗汤』，专注于写golang、docker、kubernetes等知识等提升硬实力的文章，期待你的关注。 转载说明：务必注明来源（注明：来源于公众号：我的小碗汤， 作者：小碗汤）

[2019101920032475.png]: /images/20211227/3e17a3d0b3ec40c7a59d362aa7bd5d1e.png