This commit is contained in:
Владимир Фёдоров 2025-07-21 02:04:21 +07:00
parent 985181e942
commit 264dc51da4
4 changed files with 69 additions and 127 deletions

View File

@ -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
}

View File

@ -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
View 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
}

View File

@ -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
}