fasthttp_http_client.go

package main

import (
	"errors"
	"fmt"
	"net/url"
	"time"

	"github.com/valyala/fasthttp"
)

type fasthttpHTTPClient struct {
	client          *fasthttp.Client
	maxRedirections int
	timeout         time.Duration
}

func newFasthttpHTTPClient(c *fasthttp.Client, maxRedirections int, timeout time.Duration) httpClient {
	return &fasthttpHTTPClient{c, maxRedirections, timeout}
}

func (c *fasthttpHTTPClient) Get(u *url.URL, headers map[string]string) (httpResponse, error) {
	req, res := fasthttp.Request{}, fasthttp.Response{}
	req.SetRequestURI(u.String())
	req.SetConnectionClose()

	for k, v := range headers {
		req.Header.Add(k, v)
	}

	i := 0

	for {
		err := c.client.DoTimeout(&req, &res, c.timeout)
		if err != nil {
			return nil, err
		}

		switch res.StatusCode() / 100 {
		case 2:
			return newFasthttpHTTPResponse(req.URI(), &res), nil
		case 3:
			i++

			if i > c.maxRedirections {
				return nil, errors.New("too many redirections")
			}

			u := res.Header.Peek("Location")

			if len(u) == 0 {
				return nil, errors.New("location header not found")
			}

			req.URI().UpdateBytes(u)
		default:
			return nil, fmt.Errorf("%v", res.StatusCode())
		}
	}
}