clear
This commit is contained in:
parent
985181e942
commit
264dc51da4
@ -1,11 +1,8 @@
|
||||
package http
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"errors"
|
||||
"io"
|
||||
"net"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
@ -53,63 +50,3 @@ func Do(method string, url string, headers []Header) (*Response, error) {
|
||||
)
|
||||
return ReadResponse(conn)
|
||||
}
|
||||
|
||||
func ReadResponse(r io.Reader) (*Response, error) {
|
||||
b := bufio.NewReader(r)
|
||||
|
||||
if _, err := b.ReadString(' '); err != nil {
|
||||
return nil, errors.New("read httpVersion")
|
||||
}
|
||||
statusCode, err := b.ReadString(' ')
|
||||
if err != nil {
|
||||
return nil, errors.New("read statusCode")
|
||||
}
|
||||
statusCode = statusCode[:len(statusCode)-1]
|
||||
if _, err := b.ReadString('\r'); err != nil {
|
||||
return nil, errors.New("read statusName")
|
||||
}
|
||||
if _, err := b.ReadString('\n'); err != nil {
|
||||
return nil, errors.New("read LF")
|
||||
}
|
||||
|
||||
headerStr := ""
|
||||
contentLengthStr := ""
|
||||
headers := []Header{}
|
||||
for {
|
||||
headerStr, err = b.ReadString('\n')
|
||||
if err != nil {
|
||||
return nil, errors.New("read header")
|
||||
}
|
||||
if headerStr == "\r\n" {
|
||||
break
|
||||
}
|
||||
arr := strings.Split(headerStr, ": ")
|
||||
value := strings.TrimSpace(arr[1])
|
||||
header := Header{Name: arr[0], Value: value}
|
||||
if header.Name == "Content-Length" {
|
||||
contentLengthStr = header.Value
|
||||
}
|
||||
headers = append(headers, header)
|
||||
}
|
||||
|
||||
length, err := strconv.Atoi(contentLengthStr)
|
||||
if err != nil {
|
||||
return nil, errors.New("read Content-Length")
|
||||
}
|
||||
|
||||
body := make([]byte, 0, length)
|
||||
|
||||
for i := 0; i < length; i++ {
|
||||
r, err := b.ReadByte()
|
||||
if err != nil {
|
||||
return nil, errors.New("read body")
|
||||
}
|
||||
body = append(body, r)
|
||||
}
|
||||
|
||||
return &Response{
|
||||
StatusCode: statusCode,
|
||||
Headers: headers,
|
||||
Body: body,
|
||||
}, nil
|
||||
}
|
||||
|
@ -1,46 +0,0 @@
|
||||
package http
|
||||
|
||||
import (
|
||||
"bytes"
|
||||
"strings"
|
||||
"testing"
|
||||
|
||||
"github.com/stretchr/testify/assert"
|
||||
)
|
||||
|
||||
func Test_WriteRequest(t *testing.T) {
|
||||
b := &strings.Builder{}
|
||||
|
||||
WriteRequest(
|
||||
b,
|
||||
&Request{
|
||||
Method: "GET",
|
||||
Path: "/",
|
||||
Protocol: "HTTP/1.0",
|
||||
},
|
||||
)
|
||||
|
||||
assert.Equal(t, "GET / HTTP/1.0\r\n\r\n", b.String())
|
||||
}
|
||||
|
||||
func Test_ReadResponse(t *testing.T) {
|
||||
b := bytes.NewBuffer(
|
||||
[]byte("HTTP/1.1 200 OK\r\nDate: Sat, 03 Feb 2024 16:40:29 GMT\r\nContent-Length: 12\r\nContent-Type: text/plain; charset=utf-8\r\n\r\nHello World!"),
|
||||
)
|
||||
|
||||
r, err := ReadResponse(b)
|
||||
assert.Nil(t, err)
|
||||
|
||||
assert.Equal(t, 200, r.StatusCode)
|
||||
|
||||
assert.Equal(t, "Date", r.Headers[0].Name)
|
||||
assert.Equal(t, "Sat, 03 Feb 2024 16:40:29 GMT", r.Headers[0].Value)
|
||||
|
||||
assert.Equal(t, "Content-Length", r.Headers[1].Name)
|
||||
assert.Equal(t, "12", r.Headers[1].Value)
|
||||
|
||||
assert.Equal(t, "Content-Type", r.Headers[2].Name)
|
||||
assert.Equal(t, "text/plain; charset=utf-8", r.Headers[2].Value)
|
||||
|
||||
assert.Equal(t, []byte("Hello World!"), r.Body)
|
||||
}
|
69
http/read_response.go
Normal file
69
http/read_response.go
Normal file
@ -0,0 +1,69 @@
|
||||
package http
|
||||
|
||||
import (
|
||||
"bufio"
|
||||
"errors"
|
||||
"io"
|
||||
"strconv"
|
||||
"strings"
|
||||
)
|
||||
|
||||
func ReadResponse(r io.Reader) (*Response, error) {
|
||||
b := bufio.NewReader(r)
|
||||
|
||||
if _, err := b.ReadString(' '); err != nil {
|
||||
return nil, errors.New("read httpVersion")
|
||||
}
|
||||
statusCode, err := b.ReadString(' ')
|
||||
if err != nil {
|
||||
return nil, errors.New("read statusCode")
|
||||
}
|
||||
statusCode = statusCode[:len(statusCode)-1]
|
||||
if _, err := b.ReadString('\r'); err != nil {
|
||||
return nil, errors.New("read statusName")
|
||||
}
|
||||
if _, err := b.ReadString('\n'); err != nil {
|
||||
return nil, errors.New("read LF")
|
||||
}
|
||||
|
||||
headerStr := ""
|
||||
contentLengthStr := ""
|
||||
headers := []Header{}
|
||||
for {
|
||||
headerStr, err = b.ReadString('\n')
|
||||
if err != nil {
|
||||
return nil, errors.New("read header")
|
||||
}
|
||||
if headerStr == "\r\n" {
|
||||
break
|
||||
}
|
||||
arr := strings.Split(headerStr, ": ")
|
||||
value := strings.TrimSpace(arr[1])
|
||||
header := Header{Name: arr[0], Value: value}
|
||||
if header.Name == "Content-Length" {
|
||||
contentLengthStr = header.Value
|
||||
}
|
||||
headers = append(headers, header)
|
||||
}
|
||||
|
||||
length, err := strconv.Atoi(contentLengthStr)
|
||||
if err != nil {
|
||||
return nil, errors.New("read Content-Length")
|
||||
}
|
||||
|
||||
body := make([]byte, 0, length)
|
||||
|
||||
for i := 0; i < length; i++ {
|
||||
r, err := b.ReadByte()
|
||||
if err != nil {
|
||||
return nil, errors.New("read body")
|
||||
}
|
||||
body = append(body, r)
|
||||
}
|
||||
|
||||
return &Response{
|
||||
StatusCode: statusCode,
|
||||
Headers: headers,
|
||||
Body: body,
|
||||
}, nil
|
||||
}
|
@ -1,18 +0,0 @@
|
||||
package http
|
||||
|
||||
import (
|
||||
"errors"
|
||||
)
|
||||
|
||||
var (
|
||||
ErrHeaderNotFound = errors.New("header not found")
|
||||
)
|
||||
|
||||
func (r *Response) GetHeader(name string) (Header, error) {
|
||||
for _, h := range r.Headers {
|
||||
if h.Name == name {
|
||||
return h, nil
|
||||
}
|
||||
}
|
||||
return Header{}, ErrHeaderNotFound
|
||||
}
|
Loading…
x
Reference in New Issue
Block a user