Merge pull request #37 from hankji/master

add MaxConnDuration option for hostClient

Author: yeqown

reverseproxy.go

@@ -84,6 +84,7 @@ func (p *ReverseProxy) init() error {
 		IsTLS:                  p.opt.tlsConfig != nil,
 		TLSConfig:              p.opt.tlsConfig,
 		DisablePathNormalizing: p.opt.disablePathNormalizing,
+		MaxConnDuration:        p.opt.maxConnDuration,
 	}
 	p.clients = append(p.clients, client)
 	return nil

reverseproxy_option.go

@@ -36,6 +36,9 @@ type buildOption struct {
 
 	// disablePathNormalizing disable path normalizing.
 	disablePathNormalizing bool
+
+	// maxConnDuration of hostClient
+	maxConnDuration time.Duration
 }
 
 func defaultBuildOption() *buildOption {
@@ -48,6 +51,7 @@ func defaultBuildOption() *buildOption {
 		tlsConfig:              nil,
 		timeout:                0,
 		disablePathNormalizing: false,
+		maxConnDuration:        0,
 	}
 }
 
@@ -119,3 +123,11 @@ func WithDisablePathNormalizing(isDisablePathNormalizing bool) Option {
 		o.disablePathNormalizing = isDisablePathNormalizing
 	})
 }
+
+// WithMaxConnDuration sets maxConnDuration of hostClient, which
+// means keep-alive connections are closed after this duration.
+func WithMaxConnDuration(d time.Duration) Option {
+	return newFuncBuildOption(func(o *buildOption) {
+		o.maxConnDuration = d
+	})
+}