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-05-14

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

发表于

我来说两句

0 条评论
登录 后参与评论

相关文章

来自专栏jeremy的技术点滴

JVM的Finalization Delay引起的OOM

3808
来自专栏coder修行路

Go实现短url项目

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

4385
来自专栏大内老A

采用一个自创的"验证框架"实现对数据实体的验证[编程篇]

昨天晚上突发奇想,弄了一个简易版的验证框架,用于进行数据实体的验证。目前仅仅实现基于属性的声明式的验证,即通过自定义特性(Custom Attribute)的方...

2596
来自专栏finleyMa

PHP7 新语法总结,更新7.2注意事项

太空船操作符用于比较两个表达式。当$a小于、等于或大于$b时它分别返回-1、0或1

3462
来自专栏我的小碗汤

golang 设置 http response 响应头与坑

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

832
来自专栏SDNLAB

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

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

37012
来自专栏Spark生态圈

[Spark SQL] 源码解析之Analyzer

Analyzer模块将Unresolved LogicalPlan结合元数据catalog进行绑定,最终转化为Resolved LogicalPlan。跟着代码...

1232
来自专栏Android 研究

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

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

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

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

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

3536
来自专栏Android相关

Android中的Proguard使用

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

1073

扫码关注云+社区