fix #1669 write empty body panic error

This commit is contained in:
ysqi
2016-02-12 11:27:59 +08:00
parent 2b23764ee0
commit 810f6db8d2
3 changed files with 31 additions and 2 deletions

View File

@@ -24,11 +24,13 @@ package context
import (
"bufio"
"bytes"
"crypto/hmac"
"crypto/sha1"
"encoding/base64"
"errors"
"fmt"
"io"
"net"
"net/http"
"strconv"
@@ -184,6 +186,14 @@ func (w *Response) Write(p []byte) (int, error) {
return w.ResponseWriter.Write(p)
}
// Write writes the data to the connection as part of an HTTP reply,
// and sets `started` to true.
// started means the response has sent out.
func (w *Response) Copy(buf *bytes.Buffer) (int64, error) {
w.Started = true
return io.Copy(w.ResponseWriter, buf)
}
// WriteHeader sends an HTTP response header with status code,
// and sets `started` to true.
func (w *Response) WriteHeader(code int) {