...

Source file src/edge-infra.dev/pkg/edge/client/transport.go

Documentation: edge-infra.dev/pkg/edge/client

     1  package client
     2  
     3  import (
     4  	"net/http"
     5  )
     6  
     7  const (
     8  	BearerToken   = "bearer"
     9  	TotpToken     = "totp"
    10  	EdgeVersion   = "edge-version"
    11  	Authorization = "Authorization"
    12  )
    13  
    14  type Transport struct {
    15  	T       http.RoundTripper
    16  	Headers map[string]string
    17  }
    18  
    19  func (t *Transport) RoundTrip(req *http.Request) (*http.Response, error) {
    20  	for key, val := range t.Headers {
    21  		req.Header.Add(key, val)
    22  	}
    23  	return t.T.RoundTrip(req)
    24  }
    25  
    26  func NewTokenClient(headers map[string]string) *http.Client {
    27  	return &http.Client{
    28  		Transport: &Transport{
    29  			T:       http.DefaultTransport,
    30  			Headers: headers,
    31  		},
    32  	}
    33  }
    34  

View as plain text