proxy.go 555 B

123456789101112131415161718192021222324252627282930313233
  1. package proxy
  2. import (
  3. "net"
  4. "net/http"
  5. "time"
  6. )
  7. const (
  8. dialTimeout = 30 * time.Second
  9. responseHeaderTimeout = 30 * time.Second
  10. )
  11. func NewHandler(endpoints []string) (http.Handler, error) {
  12. d, err := newDirector(endpoints)
  13. if err != nil {
  14. return nil, err
  15. }
  16. tr := http.Transport{
  17. Dial: func(network, address string) (net.Conn, error) {
  18. return net.DialTimeout(network, address, dialTimeout)
  19. },
  20. ResponseHeaderTimeout: responseHeaderTimeout,
  21. }
  22. rp := reverseProxy{
  23. director: d,
  24. transport: &tr,
  25. }
  26. return &rp, nil
  27. }