sashakoshka/go-gemini
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

Use bufio.Reader in (*Client).Do

Browse Source
This commit is contained in:
adnano 2020-09-21 17:58:36 -04:00
parent e6f70cf697
commit fd74b8fbe9
1 changed files with 8 additions and 7 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

15
client.go
View File

@ -1,6 +1,7 @@
package gemini
import (
"bufio"
"crypto/tls"
"errors"
"io/ioutil"
@ -99,9 +100,11 @@ func (c *Client) Do(req *Request) (*Response, error) {
return nil, err
}
buf := bufio.NewReader(conn)
// Read the response header
code := make([]byte, 2)
if _, err := conn.Read(code); err != nil {
if _, err := buf.Read(code); err != nil {
return nil, err
}
status, err := strconv.Atoi(string(code))
@ -110,22 +113,20 @@ func (c *Client) Do(req *Request) (*Response, error) {
}
// Read one space
space := make([]byte, 1)
if _, err := conn.Read(space); err != nil {
if space, err := buf.ReadByte(); err != nil {
return nil, err
}
if space[0] != ' ' {
} else if space != ' ' {
return nil, ErrProtocol
}
// Read the meta
meta, err := readLine(conn)
meta, err := readLine(buf)
if err != nil {
return nil, err
}
// Read the response body
body, err := ioutil.ReadAll(conn)
body, err := ioutil.ReadAll(buf)
if err != nil {
return nil, err
}