feat: fix logger output; add debug option

Author: yeqown

logger.go

@@ -11,9 +11,9 @@ type __Logger interface {
 type nopLogger struct{}
 
 func (n *nopLogger) Printf(format string, args ...interface{}) {
-	// if format not end with '/n', then append it
-	if format[len(format)-1] != 0x0a {
-		format += "/n"
+	// if format not end with '\n', then append it
+	if format[len(format)-1] != '\n' {
+		format += "\n"
 	}
 
 	fmt.Printf(format, args...)

reverseproxy.go

@@ -58,17 +58,18 @@ func (p *ReverseProxy) init() error {
 
 	if p.opt.openBalance {
 		// config balancer
-		p.clients = make([]*fasthttp.HostClient, len(p.opt.addresses))
+		p.clients = make([]*fasthttp.HostClient, 0, len(p.opt.addresses))
 		p.bla = NewBalancer(p.opt.weights)
 
-		for idx, addr := range p.opt.addresses {
-			p.clients[idx] = &fasthttp.HostClient{
+		for _, addr := range p.opt.addresses {
+			client := &fasthttp.HostClient{
 				Addr:                   addr,
 				Name:                   _fasthttpHostClientName,
 				IsTLS:                  p.opt.tlsConfig != nil,
 				TLSConfig:              p.opt.tlsConfig,
 				DisablePathNormalizing: p.opt.disablePathNormalizing,
 			}
+			p.clients = append(p.clients, client)
 		}
 
 		return nil
@@ -133,10 +134,7 @@ func (p *ReverseProxy) ServeHTTP(ctx *fasthttp.RequestCtx) {
 	}
 
 	c := p.getClient()
-
-	if p.opt.debug && p.opt.logger != nil {
-		debugF(p.opt.debug, p.opt.logger, "rev request headers to proxy, addr = %s, headers = %s", c.Addr, req.Header.String())
-	}
+	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)
@@ -154,9 +152,7 @@ func (p *ReverseProxy) ServeHTTP(ctx *fasthttp.RequestCtx) {
 	}
 
 	// deal with response headers
-	if p.opt.debug && p.opt.logger != nil {
-		debugF(p.opt.debug, p.opt.logger, "rev response headers from proxy, addr = %s, headers = %s", c.Addr, res.Header.String())
-	}
+	debugF(p.opt.debug, p.opt.logger, "rev response headers from proxy, addr = %s, headers = %s", c.Addr, res.Header.String())
 
 	for _, h := range hopHeaders {
 		res.Header.Del(h)

reverseproxy_option.go

@@ -100,6 +100,12 @@ func WithBalancer(addrWeights map[string]Weight) Option {
 	})
 }
 
+func WithDebug() Option {
+	return newFuncBuildOption(func(o *buildOption) {
+		o.debug = true
+	})
+}
+
 // WithTimeout specify the timeout of each request
 func WithTimeout(d time.Duration) Option {
 	return newFuncBuildOption(func(o *buildOption) {

ws_reverseporxy_option.go

@@ -82,6 +82,13 @@ func WithURL_OptionWS(u string) OptionWS {
 	})
 }
 
+// WithDebug_OptionWS is used to enable debug mode.
+func WithDebug_OptionWS() OptionWS {
+	return newFuncBuildOptionWS(func(o *buildOptionWS) {
+		o.debug = true
+	})
+}
+
 // WithDialer_OptionWS use specified dialer
 func WithDialer_OptionWS(dialer *websocket.Dialer) OptionWS {
 	return newFuncBuildOptionWS(func(o *buildOptionWS) {