component.go

package edge_apis

import (
	"crypto/x509"
	"github.com/openziti/edge-api/rest_util"
	"net/http"
	"net/http/cookiejar"
	"net/url"
	"time"
)

// Components provides the basic shared lower level pieces used to assemble go-swagger/openapi clients. These
// components are interconnected and have references to each other. This struct is used to set, move, and manage
// them as a set.
type Components struct {
	HttpClient    *http.Client
	HttpTransport *http.Transport
	CaPool        *x509.CertPool
}

type ComponentsConfig struct {
	Proxy func(*http.Request) (*url.URL, error)
}

// NewComponents assembles a new set of components with reasonable production defaults.
func NewComponents() *Components {
	return NewComponentsWithConfig(&ComponentsConfig{
		Proxy: http.ProxyFromEnvironment,
	})
}

// NewComponentsWithConfig assembles a new set of components with reasonable production defaults.
func NewComponentsWithConfig(cfg *ComponentsConfig) *Components {
	tlsClientConfig, _ := rest_util.NewTlsConfig()

	httpTransport := &http.Transport{
		TLSClientConfig:       tlsClientConfig,
		ForceAttemptHTTP2:     true,
		MaxIdleConns:          10,
		IdleConnTimeout:       10 * time.Second,
		TLSHandshakeTimeout:   10 * time.Second,
		ExpectContinueTimeout: 1 * time.Second,
	}

	if cfg != nil && cfg.Proxy != nil {
		httpTransport.Proxy = cfg.Proxy
	}

	jar, _ := cookiejar.New(nil)

	httpClient := &http.Client{
		Transport:     httpTransport,
		CheckRedirect: nil,
		Jar:           jar,
		Timeout:       10 * time.Second,
	}

	return &Components{
		HttpClient:    httpClient,
		HttpTransport: httpTransport,
	}
}