Merge pull request #55 from gatewayd-io/exit-on-startup-error

Add option whether to exit on startup error or not

Author: mostafa

gatewayd_plugin.yaml

@@ -28,4 +28,5 @@ plugins:
       - PERIODIC_INVALIDATOR_INTERVAL=1m
       - PERIODIC_INVALIDATOR_START_DELAY=1m
       - API_ADDRESS=localhost:18080
+      - EXIT_ON_STARTUP_ERROR=False
     checksum: 3988e10aefce2cd9b30888eddd2ec93a431c9018a695aea1cea0dac46ba91cae

main.go

@@ -33,6 +33,7 @@ func main() {
 	})
 
 	var metricsConfig *metrics.MetricsConfig
+	//nolint:nestif
 	if cfg := cast.ToStringMap(plugin.PluginConfig["config"]); cfg != nil {
 		metricsConfig = metrics.NewMetricsConfig(cfg)
 		if metricsConfig != nil && metricsConfig.Enabled {
@@ -43,20 +44,27 @@ func main() {
 		pluginInstance.Impl.Expiry = cast.ToDuration(cfg["expiry"])
 		pluginInstance.Impl.DefaultDBName = cast.ToString(cfg["defaultDBName"])
 		pluginInstance.Impl.ScanCount = cast.ToInt64(cfg["scanCount"])
+		pluginInstance.Impl.ExitOnStartupError = cast.ToBool(cfg["exitOnStartupError"])
 
 		redisConfig, err := redis.ParseURL(pluginInstance.Impl.RedisURL)
 		if err != nil {
 			logger.Error("Failed to parse Redis URL", "error", err)
-			os.Exit(1)
+			if pluginInstance.Impl.ExitOnStartupError {
+				logger.Info("Exiting due to startup error")
+				os.Exit(1)
+			}
 		}
 
 		pluginInstance.Impl.RedisClient = redis.NewClient(redisConfig)
 
 		// Ping the Redis server to check if it is available.
 		_, err = pluginInstance.Impl.RedisClient.Ping(context.Background()).Result()
 		if err != nil {
-			logger.Error("Failed to ping Redis server, plugin exited", "error", err)
-			os.Exit(1)
+			logger.Error("Failed to ping Redis server", "error", err)
+			if pluginInstance.Impl.ExitOnStartupError {
+				logger.Info("Exiting due to startup error")
+				os.Exit(1)
+			}
 		}
 
 		pluginInstance.Impl.PeriodicInvalidatorEnabled = cast.ToBool(

plugin/module.go

@@ -43,7 +43,8 @@ var (
 				"PERIODIC_INVALIDATOR_START_DELAY", "1m"),
 			"periodicInvalidatorInterval": sdkConfig.GetEnv(
 				"PERIODIC_INVALIDATOR_INTERVAL", "1m"),
-			"apiAddress": sdkConfig.GetEnv("API_ADDRESS", "localhost:8080"),
+			"apiAddress":         sdkConfig.GetEnv("API_ADDRESS", "localhost:8080"),
+			"exitOnStartupError": sdkConfig.GetEnv("EXIT_ON_STARTUP_ERROR", "false"),
 		},
 		"hooks": []interface{}{
 			int32(v1.HookName_HOOK_NAME_ON_CLOSED),

plugin/plugin.go

@@ -24,11 +24,12 @@ type Plugin struct {
 	Logger hclog.Logger
 
 	// Cache configuration.
-	RedisClient   *goRedis.Client
-	RedisURL      string
-	Expiry        time.Duration
-	DefaultDBName string
-	ScanCount     int64
+	RedisClient        *goRedis.Client
+	RedisURL           string
+	Expiry             time.Duration
+	DefaultDBName      string
+	ScanCount          int64
+	ExitOnStartupError bool
 
 	// Periodic invalidator configuration.
 	PeriodicInvalidatorEnabled    bool