.

config/config.go

@@ -8,48 +8,42 @@ import (
 	"github.com/rs/zerolog"
 )
 
-// Environment variable names
+// Environment variable names.
 const (
-	EnvLogLevel        = "LOG_LEVEL"
-	EnvRootPath        = "ROOT_PATH"
-	EnvNumWorkers      = "NUM_OF_WORKERS"
-	EnvWorkerBatchSize = "WORKER_BATCH_SIZE"
-	EnvMaxResponseSize = "MAX_RESPONSE_SIZE"
-	EnvResponseTimeout = "RESPONSE_TIMEOUT"
+	EnvLogLevel               = "LOG_LEVEL"
+	EnvNumWorkers             = "NUM_OF_WORKERS"
+	EnvWorkerBatchSize        = "WORKER_BATCH_SIZE"
+	EnvMaxResponseSize        = "MAX_RESPONSE_SIZE"
+	EnvResponseTimeout        = "RESPONSE_TIMEOUT"
+	EnvPanicOnUnexpectedError = "PANIC_ON_UNEXPECTED_ERROR"
+	EnvBlacklistPath          = "BLACKLIST_PATH"
 )
 
-// Config holds the application configuration loaded from environment variables
+// Config holds the application configuration loaded from environment variables.
 type Config struct {
-	LogLevel        zerolog.Level // Logging level (debug, info, warn, error)
-	RootPath        string        // Root path for the application
-	MaxResponseSize int           // Maximum size of response in bytes
-	NumOfWorkers    int           // Number of concurrent workers
-	ResponseTimeout int           // Timeout for responses in seconds
-	WorkerBatchSize int           // Batch size for worker processing
+	LogLevel               zerolog.Level // Logging level (debug, info, warn, error)
+	MaxResponseSize        int           // Maximum size of response in bytes
+	NumOfWorkers           int           // Number of concurrent workers
+	ResponseTimeout        int           // Timeout for responses in seconds
+	WorkerBatchSize        int           // Batch size for worker processing
+	PanicOnUnexpectedError bool          // Panic on unexpected errors when visiting a URL
+	BlacklistPath          string        // File that has blacklisted strings of "host:port"
 }
 
-// String returns a string representation of the configuration
-func (c *Config) String() string {
-	return fmt.Sprintf(
-		"Config{LogLevel: %s, RootPath: %s, MaxResponseSize: %d, NumWorkers: %d, ResponseTimeout: %d, WorkerBatchSize: %d}",
-		c.LogLevel, c.RootPath, c.MaxResponseSize, c.NumOfWorkers, c.ResponseTimeout, c.WorkerBatchSize,
-	)
-}
+var CONFIG Config //nolint:gochecknoglobals
 
-var CONFIG Config
-
-// parsePositiveInt parses and validates positive integer values
+// parsePositiveInt parses and validates positive integer values.
 func parsePositiveInt(param, value string) (int, error) {
 	val, err := strconv.Atoi(value)
 	if err != nil {
-		return 0, &ValidationError{
+		return 0, ValidationError{
 			Param:  param,
 			Value:  value,
 			Reason: "must be a valid integer",
 		}
 	}
 	if val <= 0 {
-		return 0, &ValidationError{
+		return 0, ValidationError{
 			Param:  param,
 			Value:  value,
 			Reason: "must be positive",
@@ -58,6 +52,18 @@ func parsePositiveInt(param, value string) (int, error) {
 	return val, nil
 }
 
+func parseBool(param, value string) (bool, error) {
+	val, err := strconv.ParseBool(value)
+	if err != nil {
+		return false, ValidationError{
+			Param:  param,
+			Value:  value,
+			Reason: "cannot be converted to boolean",
+		}
+	}
+	return val, nil
+}
+
 // GetConfig loads and validates configuration from environment variables
 func GetConfig() *Config {
 	config := &Config{}
@@ -67,7 +73,7 @@ func GetConfig() *Config {
 		EnvLogLevel: func(v string) error {
 			level, err := zerolog.ParseLevel(v)
 			if err != nil {
-				return &ValidationError{
+				return ValidationError{
 					Param:  EnvLogLevel,
 					Value:  v,
 					Reason: "must be one of: debug, info, warn, error",
@@ -76,16 +82,6 @@ func GetConfig() *Config {
 			config.LogLevel = level
 			return nil
 		},
-		EnvRootPath: func(v string) error {
-			if _, err := os.Stat(v); err != nil {
-				return &ConfigError{
-					Param: EnvRootPath,
-					Err:   err,
-				}
-			}
-			config.RootPath = v
-			return nil
-		},
 		EnvNumWorkers: func(v string) error {
 			val, err := parsePositiveInt(EnvNumWorkers, v)
 			if err != nil {
@@ -118,6 +114,18 @@ func GetConfig() *Config {
 			config.ResponseTimeout = val
 			return nil
 		},
+		EnvPanicOnUnexpectedError: func(v string) error {
+			val, err := parseBool(EnvPanicOnUnexpectedError, v)
+			if err != nil {
+				return err
+			}
+			config.PanicOnUnexpectedError = val
+			return nil
+		},
+		EnvBlacklistPath: func(v string) error {
+			config.BlacklistPath = v
+			return nil
+		},
 	}
 
 	// Process each environment variable