golang 设置 http response 响应头与坑

我的小碗汤 · · 6560 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。

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

image

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

最后,csdn资源,收集了海量学习资料,如果你准备入IT坑,励志成为优秀的程序猿,那么这些资源很适合你,包括java、go、python、springcloud、elk、嵌入式 、大数据、面试资料、前端等资源。同时我们组建了一个技术交流群,里面有很多大佬,会不定时分享技术文章,如果你想来一起学习提高,可以关注以下公众号后回复【2】,获取。


我是小碗汤,我们一起学习,扫码关注,精彩内容第一时间推给你

欢迎扫码关注


有疑问加站长微信联系(非本文作者)

本文来自:简书

感谢作者:我的小碗汤

查看原文:golang 设置 http response 响应头与坑

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:692541889

6560 次点击  ∙  1 赞  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传