golang: multiple http.writeHeader calls

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

背景:
golang的http服务,读取文件,提供给client下载时候。
出现 multiple http.writeHeader calls 错误。

func DownloadFile(w http.ResponseWriter, r *http.Request, sequence uint64, userid string) {
    userkey := userid
    filename := r.FormValue("filename")
    if len(userkey) <= 0 || len(filename) <= 0 {
        //........
    }
    fp := fmt.Sprintf("%s%s/%s", g_sc.Rootdir, userkey, filename)
    fd, err := os.Open(fp)
    if err == nil {
        defer fd.Close()
        buf, err := ioutil.ReadAll(fd)
        if err != nil {
            //.........
        } else {
            size := len(buf)
            w.Header().Add("Content-Length", fmt.Sprintf("%d", size))
            fmt.Fprint(w, string(buf))
        }
    }
}

 

问题出现在这几行代码:

size := len(buf)
w.Header().Add("Content-Length", fmt.Sprintf("%d", size))
fmt.Fprint(w, string(buf))

 
可以做如下修改:
 
buf := new(bytes.Buffer)
buf.Write(...)
 
if err {
  // you can use http.Error here, no response has been written yet
  http.Error(w, err.String, http.StatusInternalServerError)
  return
}
 
if err := buf.WriteTo(w); err != nil {
  log.Printf("WriteTo: %v", err)
  // you can not use http.Error here anymore. So just log the message (e.g. "broken pipe...")
}

 

 或者注释这两行代码
 //size := len(buf)
 //w.Header().Add("Content-Length", fmt.Sprintf("%d", size))
fmt.Fprint(w, string(buf))
 
 

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

本文来自:博客园

感谢作者:zhangqingping

查看原文:golang: multiple http.writeHeader calls

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

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