Add support for 'server_flavor' revproxy config option

Author: vcarvajal

config.go

@@ -33,6 +33,8 @@ var (
 	DefaultTimeout = 100 * time.Millisecond
 	// DefaultServerIdentifier is the default value
 	DefaultServerIdentifier = runtime.Version()
+	// DefaultServerFlavor is the default value
+	DefaultServerFlavor = ""
 )
 
 // HeaderExtractorFunc is a header extraction function
@@ -53,6 +55,7 @@ type ModuleConfig struct {
 	rpcAddress                string
 	rpcNetwork                string
 	serverIdentifier          string
+	serverFlavor              string
 	timeout                   time.Duration
 }
 
@@ -72,6 +75,7 @@ func NewModuleConfig(options ...ModuleConfigOption) (*ModuleConfig, error) {
 		rpcAddress:                DefaultRPCAddress,
 		rpcNetwork:                DefaultRPCNetwork,
 		serverIdentifier:          DefaultServerIdentifier,
+		serverFlavor:              DefaultServerFlavor,
 		timeout:                   DefaultTimeout,
 	}
 	if err := c.SetOptions(options...); err != nil {
@@ -185,6 +189,11 @@ func (c *ModuleConfig) ServerIdentifier() string {
 	return c.serverIdentifier
 }
 
+// ServerFlavor returns the configuration value
+func (c *ModuleConfig) ServerFlavor() string {
+	return c.serverFlavor
+}
+
 // Timeout returns the configuration value
 func (c *ModuleConfig) Timeout() time.Duration {
 	return c.timeout
@@ -335,6 +344,15 @@ func ServerIdentifier(id string) ModuleConfigOption {
 	}
 }
 
+// ServerFlavor is a function argument that sets the server
+// flavor for custom setups using revproxy.
+func ServerFlavor(serverModule string) ModuleConfigOption {
+	return func(c *ModuleConfig) error {
+		c.serverFlavor = serverModule
+		return nil
+	}
+}
+
 // FromModuleConfig allow cloning the config
 func FromModuleConfig(mcfg *ModuleConfig) ModuleConfigOption {
 	return func(c *ModuleConfig) error {

module.go

@@ -63,7 +63,7 @@ func NewModule(h http.Handler, options ...ModuleConfigOption) (*Module, error) {
 	in := RPCMsgIn{
 		ModuleVersion: m.config.ModuleIdentifier(),
 		ServerVersion: m.config.ServerIdentifier(),
-		ServerFlavor:  "",
+		ServerFlavor:  m.config.ServerFlavor(),
 		Timestamp:     now.Unix(),
 		NowMillis:     now.UnixNano() / 1e6,
 	}