Request.WriteTo: return int64

This commit is contained in:
Adnan Maolood 2021-02-28 22:20:43 -05:00
parent 768ec6c17b
commit f7012b38da

View File

@ -79,20 +79,20 @@ func ReadRequest(r io.Reader) (*Request, error) {
// WriteTo writes r to w in the Gemini request format. // WriteTo writes r to w in the Gemini request format.
// This method consults the request URL only. // This method consults the request URL only.
func (r *Request) WriteTo(w io.Writer) (int, error) { func (r *Request) WriteTo(w io.Writer) (int64, error) {
bw := bufio.NewWriterSize(w, 1026) bw := bufio.NewWriterSize(w, 1026)
url := r.URL.String() url := r.URL.String()
if len(url) > 1024 { if len(url) > 1024 {
return 0, ErrInvalidRequest return 0, ErrInvalidRequest
} }
var wrote int var wrote int64
n, err := bw.WriteString(url) n, err := bw.WriteString(url)
wrote += n wrote += int64(n)
if err != nil { if err != nil {
return wrote, err return wrote, err
} }
n, err = bw.Write(crlf) n, err = bw.Write(crlf)
wrote += n wrote += int64(n)
if err != nil { if err != nil {
return wrote, err return wrote, err
} }