码迷,mamicode.com
首页 > Web开发 > 详细

golang: multiple http.writeHeader calls

时间:2015-02-26 11:22:31      阅读:139      评论:0      收藏:0      [点我收藏+]

标签:

背景:
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))
 
 

golang: multiple http.writeHeader calls

标签:

原文地址:http://www.cnblogs.com/zhangqingping/p/4300808.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!