add stream response body (#44)

Author: joyqi

reverseproxy.go

@@ -68,6 +68,8 @@ func (p *ReverseProxy) init() error {
 				IsTLS:                  p.opt.tlsConfig != nil,
 				TLSConfig:              p.opt.tlsConfig,
 				DisablePathNormalizing: p.opt.disablePathNormalizing,
+				MaxResponseBodySize:    p.opt.maxResponseBodySize,
+				StreamResponseBody:     p.opt.streamResponseBody,
 			}
 			p.clients = append(p.clients, client)
 		}
@@ -84,6 +86,8 @@ func (p *ReverseProxy) init() error {
 		IsTLS:                  p.opt.tlsConfig != nil,
 		TLSConfig:              p.opt.tlsConfig,
 		DisablePathNormalizing: p.opt.disablePathNormalizing,
+		MaxResponseBodySize:    p.opt.maxResponseBodySize,
+		StreamResponseBody:     p.opt.streamResponseBody,
 		MaxConnDuration:        p.opt.maxConnDuration,
 	}
 	p.clients = append(p.clients, client)

reverseproxy_option.go

@@ -40,6 +40,12 @@ type buildOption struct {
 	// disableVirtualHost disable virtual host.
 	disableVirtualHost bool
 
+	// maxResponseBodySize is the maximum response body size in bytes.
+	maxResponseBodySize int
+
+	// streamResponseBody denotes whether stream response body or not.
+	streamResponseBody bool
+
 	// maxConnDuration of hostClient
 	maxConnDuration time.Duration
 }
@@ -135,6 +141,14 @@ func WithDisableVirtualHost(isDisableVirtualHost bool) Option {
 	})
 }
 
+// WithStreamResponseBody sets whether stream response body or not.
+func WithStreamResponseBody(size int) Option {
+	return newFuncBuildOption(func(o *buildOption) {
+		o.streamResponseBody = true
+		o.maxResponseBodySize = size
+	})
+}
+
 // WithMaxConnDuration sets maxConnDuration of hostClient, which
 // means keep-alive connections are closed after this duration.
 func WithMaxConnDuration(d time.Duration) Option {