c55a4ac779
The io/ioutil package has been deprecated in Go 1.16. This commit replaces the existing io/ioutil functions with their new definitions in io and os packages. Signed-off-by: Eng Zer Jun <engzerjun@gmail.com>
77 lines
2 KiB
Go
77 lines
2 KiB
Go
package request
|
|
|
|
import (
|
|
"bytes"
|
|
"encoding/json"
|
|
"io"
|
|
"net/http"
|
|
"strings"
|
|
)
|
|
|
|
// Options defines request options, like request modifiers and which host to target
|
|
type Options struct {
|
|
host string
|
|
requestModifiers []func(*http.Request) error
|
|
}
|
|
|
|
// Host creates a modifier that sets the specified host as the request URL host
|
|
func Host(host string) func(*Options) {
|
|
return func(o *Options) {
|
|
o.host = host
|
|
}
|
|
}
|
|
|
|
// With adds a request modifier to the options
|
|
func With(f func(*http.Request) error) func(*Options) {
|
|
return func(o *Options) {
|
|
o.requestModifiers = append(o.requestModifiers, f)
|
|
}
|
|
}
|
|
|
|
// Method creates a modifier that sets the specified string as the request method
|
|
func Method(method string) func(*Options) {
|
|
return With(func(req *http.Request) error {
|
|
req.Method = method
|
|
return nil
|
|
})
|
|
}
|
|
|
|
// RawString sets the specified string as body for the request
|
|
func RawString(content string) func(*Options) {
|
|
return RawContent(io.NopCloser(strings.NewReader(content)))
|
|
}
|
|
|
|
// RawContent sets the specified reader as body for the request
|
|
func RawContent(reader io.ReadCloser) func(*Options) {
|
|
return With(func(req *http.Request) error {
|
|
req.Body = reader
|
|
return nil
|
|
})
|
|
}
|
|
|
|
// ContentType sets the specified Content-Type request header
|
|
func ContentType(contentType string) func(*Options) {
|
|
return With(func(req *http.Request) error {
|
|
req.Header.Set("Content-Type", contentType)
|
|
return nil
|
|
})
|
|
}
|
|
|
|
// JSON sets the Content-Type request header to json
|
|
func JSON(o *Options) {
|
|
ContentType("application/json")(o)
|
|
}
|
|
|
|
// JSONBody creates a modifier that encodes the specified data to a JSON string and set it as request body. It also sets
|
|
// the Content-Type header of the request.
|
|
func JSONBody(data interface{}) func(*Options) {
|
|
return With(func(req *http.Request) error {
|
|
jsonData := bytes.NewBuffer(nil)
|
|
if err := json.NewEncoder(jsonData).Encode(data); err != nil {
|
|
return err
|
|
}
|
|
req.Body = io.NopCloser(jsonData)
|
|
req.Header.Set("Content-Type", "application/json")
|
|
return nil
|
|
})
|
|
}
|