【随记】golang 设置 http response 响应头与坑

今天在二开B站kartos框架时，发现怎么设置http的response Header都无法生效。故发现了设置 http response 响应头与坑。

1、 设置WriteHeader的顺序问题

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 在所有的 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")
   }
 }
}