脚本之家,脚本语言编程技术及教程分享平台!
分类导航

Python|VBS|Ruby|Lua|perl|VBA|Golang|PowerShell|Erlang|autoit|Dos|bat|

服务器之家 - 脚本之家 - Golang - golang设置http response响应头与填坑记录

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

2020-05-19 09:27我的小碗汤 Golang

这篇文章主要给大家介绍了关于golang设置http response响应头与填坑记录的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,需要的朋友们下面随着小编来一起学习学习吧

1、 设置WriteHeader的顺序问题

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

?
1
2
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) 来实现,例如:

?
1
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之前调用。

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
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的大写处理:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
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

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
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:

?
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
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.运行服务器

?
1
go run myServer.go

2.运行客户端

?
1
go run myHttp.go

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

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

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

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

?
1
2
3
4
5
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接口:

?
1
2
3
4
type ReadCloser interface {
Reader
Closer
}

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

?
1
2
3
4
5
6
7
8
9
10
11
12
13
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,如下:

?
1
2
3
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为什么需要进行关闭,这篇文章进行了解释: http://www.zzvips.com/article/67352.html

那么作为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.

总结

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

原文链接:https://studygolang.com/articles/14431

延伸 · 阅读

精彩推荐
  • Golanggolang json.Marshal 特殊html字符被转义的解决方法

    golang json.Marshal 特殊html字符被转义的解决方法

    今天小编就为大家分享一篇golang json.Marshal 特殊html字符被转义的解决方法,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧 ...

    李浩的life12792020-05-27
  • Golanggo语言制作端口扫描器

    go语言制作端口扫描器

    本文给大家分享的是使用go语言编写的TCP端口扫描器,可以选择IP范围,扫描的端口,以及多线程,有需要的小伙伴可以参考下。 ...

    脚本之家3642020-04-25
  • GolangGolang通脉之数据类型详情

    Golang通脉之数据类型详情

    这篇文章主要介绍了Golang通脉之数据类型,在编程语言中标识符就是定义的具有某种意义的词,比如变量名、常量名、函数名等等,Go语言中标识符允许由...

    4272021-11-24
  • Golanggolang 通过ssh代理连接mysql的操作

    golang 通过ssh代理连接mysql的操作

    这篇文章主要介绍了golang 通过ssh代理连接mysql的操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...

    a165861639710342021-03-08
  • GolangGolang中Bit数组的实现方式

    Golang中Bit数组的实现方式

    这篇文章主要介绍了Golang中Bit数组的实现方式,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...

    天易独尊11682021-06-09
  • Golanggolang如何使用struct的tag属性的详细介绍

    golang如何使用struct的tag属性的详细介绍

    这篇文章主要介绍了golang如何使用struct的tag属性的详细介绍,从例子说起,小编觉得挺不错的,现在分享给大家,也给大家做个参考。一起跟随小编过来看...

    Go语言中文网11352020-05-21
  • Golanggolang的httpserver优雅重启方法详解

    golang的httpserver优雅重启方法详解

    这篇文章主要给大家介绍了关于golang的httpserver优雅重启的相关资料,文中通过示例代码介绍的非常详细,对大家的学习或者工作具有一定的参考学习价值,...

    helight2992020-05-14
  • Golanggo日志系统logrus显示文件和行号的操作

    go日志系统logrus显示文件和行号的操作

    这篇文章主要介绍了go日志系统logrus显示文件和行号的操作,具有很好的参考价值,希望对大家有所帮助。一起跟随小编过来看看吧...

    SmallQinYan12302021-02-02