golang 设置 http response 响应头与坑

01 设置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")
    }
  }
}

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

原文发布于微信公众号 - 我的小碗汤(mysmallsoup)

原文发表时间:2018-06-29

本文参与腾讯云自媒体分享计划,欢迎正在阅读的你也加入,一起分享。

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏coder修行路

Go实现短url项目

首先说一下这种业务的应用场景: 把一个长url转换为一个短url网址 主要用于微博,二维码,等有字数限制的场景 主要实现的功能分析: 把长url的地址转换为短u...

4505
来自专栏JMCui

Netty 系列六(编解码器).

    网络传输的单位是字节,如何将应用程序的数据转换为字节,以及将字节转换为应用程序的数据,就要说到到我们该篇介绍的编码器和解码器。

1111
来自专栏Java帮帮-微信公众号-技术文章全总结

Java类加载器(用户自定义类加载器实现)

java类加载器主要分为如下几种: jvm提供的类加载器 根类加载器:底层实现,主要加载java核心类库(如:java.lang.*) 扩展类加载器:使用jav...

3596
来自专栏SDNLAB

POF技术分享(三):Packet处理流程

前言: 之前对POF基本原理、POF交换机源码结构进行解读,但是,要想完成POF交换机的二次开发和拓展,有必要对POF交换机特有的数据包处理流程、POF交换机和...

38612
来自专栏Hongten

JSP 九大内置对象

① out - javax.servlet.jsp.jspWriter    out对象用于把结果输出到网页上。

5012
来自专栏Android相关

Android中的Proguard使用

之前介绍了如何使用命令行将Jar包根据配置文件进行ProGuard,以及ProGuard的过程,会遇到的问题等。接下来会介绍常用的ProGuard如何配置参数。...

1193
来自专栏androidBlog

java 解决文件名重复问题的两种方法

版权声明:本文为博主原创文章,未经博主允许不得转载。 https://blog.csdn.net/gdutxiaoxu/article/de...

3991
来自专栏Alice

demo3同通讯录展示的方式分组排序

按A-Z顺序分组展示 有些项目中会需要这样的需求。形成类似于上述的界面。类似于通讯录里边的排序。实现的效果:所有的数据展示的时候,能够分组展示。顺序按照A-Z的...

2299
来自专栏Android 研究

APK安装流程详解5——Installer、InstallerConnection和Installd守护进程

因为Installer继承自SystemService,所以我们看下Installer的onStart方法 代码在Installer.java 396行

1531
来自专栏芋道源码1024

Spring Webflux —— 源码阅读之 handler 包

查找给定请求的handler,如果找不到特定的请求,则返回一个空的Mono。这个方法被getHandler(org.springframework.web.se...

3685

扫码关注云+社区

领取腾讯云代金券