moby/pkg/plugins/transport/http.go
Sebastiaan van Stijn dfd331b2c8
pkg/plugins/transport: export httpTransport, and return concrete type
Make NewHTTPTransport return a concrete type.

Signed-off-by: Sebastiaan van Stijn <github@gone.nl>
2023-07-25 22:38:37 +02:00

41 lines
1,017 B
Go

package transport // import "github.com/docker/docker/pkg/plugins/transport"
import (
"io"
"net/http"
"strings"
)
// HTTPTransport holds an [http.RoundTripper]
// and information about the scheme and address the transport
// sends request to.
type HTTPTransport struct {
http.RoundTripper
scheme string
addr string
}
// NewHTTPTransport creates a new HTTPTransport.
func NewHTTPTransport(r http.RoundTripper, scheme, addr string) *HTTPTransport {
return &HTTPTransport{
RoundTripper: r,
scheme: scheme,
addr: addr,
}
}
// NewRequest creates a new http.Request and sets the URL
// scheme and address with the transport's fields.
func (t HTTPTransport) NewRequest(path string, data io.Reader) (*http.Request, error) {
if !strings.HasPrefix(path, "/") {
path = "/" + path
}
req, err := http.NewRequest(http.MethodPost, path, data)
if err != nil {
return nil, err
}
req.Header.Add("Accept", VersionMimetype)
req.URL.Scheme = t.scheme
req.URL.Host = t.addr
return req, nil
}