add client and transport

http/client.go

@@ -1,11 +1,6 @@
 package http
 
 import (
-	"bufio"
-	"errors"
-	"io"
-	"net"
-	"strconv"
 	"strings"
 )
 
@@ -14,14 +9,12 @@ var (
 	SP   = []byte(" ")
 )
 
-func getIP(domain string) string {
-	if domain == "test.ru" {
-		return "127.0.0.1:8081"
-	}
-	return ""
+type Client struct {
+	DNS       DNS
+	Transport RoundTripper
 }
 
-func Do(method string, url string, headers []Header) (*Response, error) {
+func (c *Client) Do(method string, url string, headers []Header) (*Response, error) {
 	url = strings.TrimPrefix(url, "http://")
 	url = strings.TrimPrefix(url, "https://")
 	path := "/"
@@ -33,87 +26,23 @@ func Do(method string, url string, headers []Header) (*Response, error) {
 		path = "/" + path
 	}
 
-	connectPath := getIP(arr[0])
-	conn, err := net.Dial("tcp", connectPath)
-	// conn, err := tls.Dial("tcp", connectPath, nil)
-	if err != nil {
-		return nil, errors.New("connect " + connectPath)
+	connectPath := ""
+	if c.DNS != nil {
+		var err error
+		connectPath, err = c.DNS.GetIP(arr[0])
+		if err != nil {
+			return nil, err
+		}
 	}
-	defer conn.Close()
 
-	DebugWrap(
-		conn,
+	resp, err := c.Transport.RoundTrip(
 		&Request{
-			Method:   method,
-			Path:     path,
-			Protocol: "HTTP/1.0",
-			Headers:  headers,
+			ConnectPath: connectPath,
+			Method:      method,
+			Path:        path,
+			Protocol:    "HTTP/1.0",
+			Headers:     headers,
 		},
-		WriteRequest,
 	)
-	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")
-	}
-	statusCodeStr, err := b.ReadString(' ')
-	if err != nil {
-		return nil, errors.New("read statusCode")
-	}
-	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")
-	}
-
-	statusCode, err := strconv.Atoi(statusCodeStr[:len(statusCodeStr)-1])
-	if err != nil {
-		return nil, errors.New("read statusCode")
-	}
-
-	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
+	return resp, err
 }