add disable virtual host (#43)

Author: joyqi

reverseproxy.go

@@ -138,7 +138,9 @@ func (p *ReverseProxy) ServeHTTP(ctx *fasthttp.RequestCtx) {
 	debugF(p.opt.debug, p.opt.logger, "rev request headers to proxy, addr = %s, headers = %s", c.Addr, req.Header.String())
 
 	// assign the host to support virtual hosting, aka shared web hosting (one IP, multiple domains)
-	req.SetHost(c.Addr)
+	if !p.opt.disableVirtualHost {
+		req.SetHost(c.Addr)
+	}
 
 	// execute the request and rev response with timeout
 	if err := p.doWithTimeout(c, req, res); err != nil {

reverseproxy_option.go

@@ -37,6 +37,9 @@ type buildOption struct {
 	// disablePathNormalizing disable path normalizing.
 	disablePathNormalizing bool
 
+	// disableVirtualHost disable virtual host.
+	disableVirtualHost bool
+
 	// maxConnDuration of hostClient
 	maxConnDuration time.Duration
 }
@@ -51,6 +54,7 @@ func defaultBuildOption() *buildOption {
 		tlsConfig:              nil,
 		timeout:                0,
 		disablePathNormalizing: false,
+		disableVirtualHost:     false,
 		maxConnDuration:        0,
 	}
 }
@@ -124,6 +128,13 @@ func WithDisablePathNormalizing(isDisablePathNormalizing bool) Option {
 	})
 }
 
+// WithDisableVirtualHost sets whether disable virtual host.
+func WithDisableVirtualHost(isDisableVirtualHost bool) Option {
+	return newFuncBuildOption(func(o *buildOption) {
+		o.disableVirtualHost = isDisableVirtualHost
+	})
+}
+
 // WithMaxConnDuration sets maxConnDuration of hostClient, which
 // means keep-alive connections are closed after this duration.
 func WithMaxConnDuration(d time.Duration) Option {