feat: add status cache

Author: alexfrs69

cmd/mc-router/main.go

@@ -74,6 +74,9 @@ type Config struct {
 	FakeOnline     bool   `default:"false" usage:"Enable fake online MOTD when backend is offline and auto-scale-up is enabled"`
 	FakeOnlineMOTD string `default:"Server is sleeping\nJoin to wake it up" usage:"Custom MOTD to show when backend is offline and auto-scale-up is enabled"`
 
+	CacheStatus         bool   `default:"false" usage:"Cache status response for backends"`
+	CacheStatusInterval string `default:"30s" usage:"Interval to update the status cache"`
+
 	Webhook WebhookConfig `usage:"Webhook configuration"`
 }
 
@@ -141,7 +144,6 @@ func main() {
 	// Only one instance should be created
 	server.DownScaler = server.NewDownScaler(ctx, downScalerEnabled, downScalerDelay)
 
-
 	c := make(chan os.Signal, 1)
 	signal.Notify(c, syscall.SIGINT, syscall.SIGTERM)
 
@@ -175,10 +177,11 @@ func main() {
 		ReceiveProxyProto:          config.ReceiveProxyProtocol,
 		TrustedProxyNets:           trustedIpNets,
 		RecordLogins:               config.RecordLogins,
-		AutoScaleUpAllowDenyConfig: autoScaleUpAllowDenyConfig,
-		AutoScaleUp:                config.AutoScaleUp,
+		AutoScaleUpAllowDenyConfig: autoScaleAllowDenyConfig,
+		AutoScaleUp:                config.AutoScale.Up,
 		FakeOnline:                 config.FakeOnline,
 		FakeOnlineMOTD:             config.FakeOnlineMOTD,
+		CacheStatus:                config.CacheStatus,
 	}
 
 	connector := server.NewConnector(metricsBuilder.BuildConnectorMetrics(), connectorConfig)
@@ -198,6 +201,15 @@ func main() {
 			server.NewWebhookNotifier(config.Webhook.Url, config.Webhook.RequireUser))
 	}
 
+	var cacheInterval time.Duration
+	if config.CacheStatus {
+		cacheInterval, err = time.ParseDuration(config.CacheStatusInterval)
+		if err != nil {
+			logrus.WithError(err).Fatal("Unable to parse cache status interval")
+		}
+		logrus.WithField("interval", config.CacheStatusInterval).Info("Using cache status interval")
+	}
+
 	if config.NgrokToken != "" {
 		connector.UseNgrok(config.NgrokToken)
 	}
@@ -254,6 +266,15 @@ func main() {
 		logrus.WithError(err).Fatal("Unable to start metrics reporter")
 	}
 
+	if config.CacheStatus {
+		logrus.Info("Starting status cache updater")
+		connector.StatusCache.StartUpdater(connector, cacheInterval, func() map[string]string {
+			mappings := server.Routes.GetMappings()
+			logrus.WithField("mappings", mappings).Debug("Status cache updater")
+			return mappings
+		})
+	}
+
 	// wait for process-stop signal
 	<-c
 	logrus.Info("Stopping. Waiting for connections to complete...")

go.mod

@@ -23,6 +23,7 @@ require (
 )
 
 require (
+	github.com/Raqbit/mc-pinger v0.2.4 // indirect
 	github.com/beorn7/perks v1.0.1 // indirect
 	github.com/cespare/xxhash/v2 v2.3.0 // indirect
 	github.com/containerd/log v0.1.0 // indirect