major rewrite of code

Signed-off-by: Michael <michael.lindman@gmail.com>
This commit is contained in:
Michael 2022-10-04 03:28:14 +01:00
parent 61b3c71fca
commit 5f9797d00e
Signed by: michael
GPG Key ID: 523BD9EF68BDD44C

92
request.go Normal file
View File

@ -0,0 +1,92 @@
package request
import (
"context"
"fmt"
"io"
"log"
"net/http"
"net/url"
)
// Request HTTP request
type Request struct {
Method string
URL string
Headers map[string]string
Body io.Reader
}
// Response HTTP Response
type Response struct {
StatusCode int
URL *url.URL
Body []byte
}
// NewRequest initalises new http request
func NewRequest(method, url string, headers map[string]string, body io.Reader) (req *Request) {
return &Request{
Method: method,
URL: url,
Headers: headers,
Body: body,
}
}
func (request *Request) request(ctx context.Context, client *http.Client) (*http.Response, error) {
req, err := http.NewRequestWithContext(ctx, request.Method, request.URL, request.Body)
if err != nil {
return nil, err
}
for k, v := range request.Headers {
req.Header.Set(k, v)
}
return client.Do(req)
}
func response(statuscode int, path string, body []byte) *Response {
url, err := url.Parse(path)
if err != nil {
log.Println(err)
return &Response{statuscode, nil, body}
}
return &Response{statuscode, url, body}
}
// Do initiates http request
func (request *Request) Do(ctx context.Context, client *http.Client) (*Response, error) {
resp, err := request.request(ctx, client)
if err != nil {
return response(http.StatusInternalServerError, request.URL, nil), err
}
body, err := io.ReadAll(resp.Body)
if err != nil {
return response(http.StatusInternalServerError, request.URL, nil), err
}
defer resp.Body.Close()
if resp.StatusCode == http.StatusOK {
return response(resp.StatusCode, request.URL, body), nil
}
return response(resp.StatusCode, request.URL, nil), fmt.Errorf("request: %s - %d %s", request.URL, resp.StatusCode, http.StatusText(resp.StatusCode))
}
// Get sends GET request
func Get(ctx context.Context, url string, headers map[string]string) (*Response, error) {
return NewRequest(http.MethodGet, url, headers, nil).Do(ctx, &http.Client{})
}
// Post sends POST request
func Post(ctx context.Context, url string, headers map[string]string, data io.Reader) (*Response, error) {
return NewRequest(http.MethodPost, url, headers, data).Do(ctx, &http.Client{})
}
// Put sends PUT request
func Put(ctx context.Context, url string, headers map[string]string, data io.Reader) (*Response, error) {
return NewRequest(http.MethodPut, url, headers, data).Do(ctx, &http.Client{})
}
// Delete sends DELETE request
func Delete(ctx context.Context, url string, headers map[string]string, data io.Reader) (*Response, error) {
return NewRequest(http.MethodDelete, url, headers, data).Do(ctx, &http.Client{})
}