timeout.go

package gorequest

import (
	"errors"
	"net"
	"net/http"
	"time"
)

func (s *SuperAgent) Timeout(timeout time.Duration) *SuperAgent {
	s.safeModifyHttpClient()
	s.Client.Timeout = timeout
	return s
}

type Timeouts struct {
	Dial      time.Duration
	KeepAlive time.Duration

	TlsHandshake   time.Duration
	ResponseHeader time.Duration
	ExpectContinue time.Duration
	IdleConn       time.Duration
}

func (s *SuperAgent) Timeouts(timeouts *Timeouts) *SuperAgent {
	if timeouts == nil {
		s.Errors = append(s.Errors, errors.New("timeouts func: nil timeouts"))
		return s
	}

	s.safeModifyHttpClient()

	var transport *http.Transport
	if s.Client.Transport != nil {
		clientTransport, ok := s.Client.Transport.(*http.Transport)
		if !ok {
			return s
		}
		transport = clientTransport
		s.Transport = transport
		if s.isClone {
			s.safeModifyTransport()
			transport = s.Transport
		}
	} else {
		s.safeModifyTransport()
		transport = s.Transport
		s.Client.Transport = transport
	}

	transport.DialContext = (&net.Dialer{
		Timeout:   timeouts.Dial,
		KeepAlive: timeouts.KeepAlive,
	}).DialContext

	transport.TLSHandshakeTimeout = timeouts.TlsHandshake
	transport.ResponseHeaderTimeout = timeouts.ResponseHeader
	transport.ExpectContinueTimeout = timeouts.ExpectContinue
	transport.IdleConnTimeout = timeouts.IdleConn

	s.Client.Transport = transport

	return s
}