wehub-resource-sync
e071084ebe
govulncheck / govulncheck (push) Has been cancelled
Lint / golangci-lint (push) Has been cancelled
Run Tests / Unit Tests (push) Has been cancelled
Run Tests / Etcd Integration Tests (push) Has been cancelled
Harness (E2E) / Harnesses (mock LLM) (push) Has been cancelled
Harness (E2E) / Provider harnesses (live LLM conformance) (push) Has been cancelled
40 lines
622 B
Go
40 lines
622 B
Go
package http
|
|
|
|
import (
|
|
"errors"
|
|
"net/http"
|
|
|
|
"go-micro.dev/v6/selector"
|
|
)
|
|
|
|
type roundTripper struct {
|
|
rt http.RoundTripper
|
|
st selector.Strategy
|
|
opts Options
|
|
}
|
|
|
|
func (r *roundTripper) RoundTrip(req *http.Request) (*http.Response, error) {
|
|
s, err := r.opts.Registry.GetService(req.URL.Host)
|
|
if err != nil {
|
|
return nil, err
|
|
}
|
|
|
|
next := r.st(s)
|
|
|
|
// rudimentary retry 3 times
|
|
for i := 0; i < 3; i++ {
|
|
n, err := next()
|
|
if err != nil {
|
|
continue
|
|
}
|
|
req.URL.Host = n.Address
|
|
w, err := r.rt.RoundTrip(req)
|
|
if err != nil {
|
|
continue
|
|
}
|
|
return w, nil
|
|
}
|
|
|
|
return nil, errors.New("failed request")
|
|
}
|