clear
This commit is contained in:
parent
985181e942
commit
264dc51da4
@ -1,11 +1,8 @@
|
|||||||
package http
|
package http
|
||||||
|
|
||||||
import (
|
import (
|
||||||
"bufio"
|
|
||||||
"errors"
|
"errors"
|
||||||
"io"
|
|
||||||
"net"
|
"net"
|
||||||
"strconv"
|
|
||||||
"strings"
|
"strings"
|
||||||
)
|
)
|
||||||
|
|
||||||
@ -53,63 +50,3 @@ func Do(method string, url string, headers []Header) (*Response, error) {
|
|||||||
)
|
)
|
||||||
return ReadResponse(conn)
|
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