mirror of
https://github.com/gofiber/fiber.git
synced 2025-02-23 07:24:04 +00:00
93dc33ad89
* add tls config for http client in proxy middleware * update exmaple and make tls.config also affects to proxy.Balancer * add Test_Proxy_Forward_WithTlsConfig * add a variable of type *tls.Config * add a RWMutex for clientTlsConfig * remove global tls config * Update README.md * Update proxy.go Co-authored-by: RW <rene@gofiber.io>
83 lines
1.8 KiB
Go
83 lines
1.8 KiB
Go
package proxy
|
|
|
|
import (
|
|
"crypto/tls"
|
|
"time"
|
|
|
|
"github.com/gofiber/fiber/v2"
|
|
"github.com/valyala/fasthttp"
|
|
)
|
|
|
|
// Config defines the config for middleware.
|
|
type Config struct {
|
|
// Next defines a function to skip this middleware when returned true.
|
|
//
|
|
// Optional. Default: nil
|
|
Next func(c *fiber.Ctx) bool
|
|
|
|
// Servers defines a list of <scheme>://<host> HTTP servers,
|
|
//
|
|
// which are used in a round-robin manner.
|
|
// i.e.: "https://foobar.com, http://www.foobar.com"
|
|
//
|
|
// Required
|
|
Servers []string
|
|
|
|
// ModifyRequest allows you to alter the request
|
|
//
|
|
// Optional. Default: nil
|
|
ModifyRequest fiber.Handler
|
|
|
|
// ModifyResponse allows you to alter the response
|
|
//
|
|
// Optional. Default: nil
|
|
ModifyResponse fiber.Handler
|
|
|
|
// Timeout is the request timeout used when calling the proxy client
|
|
//
|
|
// Optional. Default: 1 second
|
|
Timeout time.Duration
|
|
|
|
// Per-connection buffer size for requests' reading.
|
|
// This also limits the maximum header size.
|
|
// Increase this buffer if your clients send multi-KB RequestURIs
|
|
// and/or multi-KB headers (for example, BIG cookies).
|
|
ReadBufferSize int
|
|
|
|
// Per-connection buffer size for responses' writing.
|
|
WriteBufferSize int
|
|
|
|
// tls config for the http client
|
|
TlsConfig *tls.Config
|
|
}
|
|
|
|
// ConfigDefault is the default config
|
|
var ConfigDefault = Config{
|
|
Next: nil,
|
|
ModifyRequest: nil,
|
|
ModifyResponse: nil,
|
|
Timeout: fasthttp.DefaultLBClientTimeout,
|
|
}
|
|
|
|
// configDefault function to set default values
|
|
func configDefault(config ...Config) Config {
|
|
// Return default config if nothing provided
|
|
if len(config) < 1 {
|
|
return ConfigDefault
|
|
}
|
|
|
|
// Override default config
|
|
cfg := config[0]
|
|
|
|
// Set default values
|
|
if cfg.Timeout <= 0 {
|
|
cfg.Timeout = ConfigDefault.Timeout
|
|
}
|
|
|
|
// Set default values
|
|
if len(cfg.Servers) == 0 {
|
|
panic("Servers cannot be empty")
|
|
}
|
|
return cfg
|
|
}
|