Getting rid of IsTestEnv

DiceDB · Dec 10, 2024 · 10d70d9 · 10d70d9
1 parent e448d78
commit 10d70d9
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 4 deletions.
diff --git a/config/config.go b/config/config.go
@@ -28,7 +28,7 @@ type Config struct {
 	}
 	Server struct {
 		Port                 string // Field for the server port
-		IsTestEnv            bool
+		Environment          string
 		RequestLimitPerMin   int64         // Field for the request limit
 		RequestWindowSec     float64       // Field for the time window in float64
 		AllowedOrigins       []string      // Field for the allowed origins
@@ -64,14 +64,14 @@ func LoadConfig() *Config {
 		},
 		Server: struct {
 			Port                 string
-			IsTestEnv            bool
+			Environment          string
 			RequestLimitPerMin   int64
 			RequestWindowSec     float64
 			AllowedOrigins       []string
 			CronCleanupFrequency time.Duration
 		}{
 			Port:                 getEnv("PORT", ":8080"),
-			IsTestEnv:            getEnv("ENVIRONMENT", "local") != "production",                            // Default server port
+			Environment:          getEnv("ENVIRONMENT", "local"),
 			RequestLimitPerMin:   getEnvInt("REQUEST_LIMIT_PER_MIN", 1000),                                  // Default request limit
 			RequestWindowSec:     getEnvFloat64("REQUEST_WINDOW_SEC", 60),                                   // Default request window in float64
 			AllowedOrigins:       getEnvArray("ALLOWED_ORIGINS", []string{"http://localhost:3000"}),         // Default allowed origins

diff --git a/util/helpers.go b/util/helpers.go
@@ -55,7 +55,7 @@ func ParseHTTPRequest(r *http.Request) (*cmds.CommandRequest, error) {
 
 	configValue := config.LoadConfig()
 	// Check if the command is blocklisted
-	if err := BlockListedCommand(command); err != nil && !configValue.Server.IsTestEnv {
+	if err := BlockListedCommand(command); err != nil && configValue.Server.Environment == "production" {
 		return nil, err
 	}