Replace panic calls with proper error handling

cmd/es-index-cleaner/app/flags.go

@@ -52,7 +52,7 @@ func (*Config) AddFlags(flags *flag.FlagSet) {
 }
 
 // InitFromViper initializes config from viper.Viper.
-func (c *Config) InitFromViper(v *viper.Viper) {
+func (c *Config) InitFromViper(v *viper.Viper) error {
 	c.IndexPrefix = v.GetString(indexPrefix)
 	if c.IndexPrefix != "" {
 		c.IndexPrefix += "-"
@@ -66,7 +66,8 @@ func (c *Config) InitFromViper(v *viper.Viper) {
 	c.Password = v.GetString(password)
 	tlsCfg, err := tlsFlagsCfg.InitFromViper(v)
 	if err != nil {
-		panic(err)
+		return err
 	}
 	c.TLSConfig = tlsCfg
+	return nil
 }

cmd/es-index-cleaner/app/flags_test.go

@@ -33,7 +33,8 @@ func TestBindFlags(t *testing.T) {
 	})
 	require.NoError(t, err)
 
-	c.InitFromViper(v)
+	err = c.InitFromViper(v)
+	require.NoError(t, err)
 	assert.Equal(t, "tenant1-", c.IndexPrefix)
 	assert.True(t, c.Rollover)
 	assert.True(t, c.Archive)

cmd/es-index-cleaner/main.go

@@ -54,7 +54,9 @@ func main() {
 				return fmt.Errorf("could not parse NUM_OF_DAYS argument: %w", err)
 			}
 
-			cfg.InitFromViper(v)
+			if err := cfg.InitFromViper(v); err != nil {
+				return fmt.Errorf("failed to initialize config: %w", err)
+			}
 
 			ctx := context.Background()
 			tlscfg, err := cfg.TLSConfig.LoadTLSConfig(ctx)

cmd/es-rollover/app/actions.go

@@ -49,7 +49,9 @@ type ActionCreatorFunction func(client.Client, Config) Action
 // ExecuteAction execute the action returned by the createAction function
 func ExecuteAction(opts ActionExecuteOptions, createAction ActionCreatorFunction) error {
 	cfg := Config{}
-	cfg.InitFromViper(opts.Viper)
+	if err := cfg.InitFromViper(opts.Viper); err != nil {
+		return fmt.Errorf("failed to initialize config: %w", err)
+	}
 
 	ctx := context.Background()
 	tlsCfg, err := cfg.TLSConfig.LoadTLSConfig(ctx)

cmd/es-rollover/app/flags.go

@@ -56,7 +56,7 @@ func AddFlags(flags *flag.FlagSet) {
 }
 
 // InitFromViper initializes config from viper.Viper.
-func (c *Config) InitFromViper(v *viper.Viper) {
+func (c *Config) InitFromViper(v *viper.Viper) error {
 	c.IndexPrefix = v.GetString(indexPrefix)
 	if c.IndexPrefix != "" {
 		c.IndexPrefix += "-"
@@ -71,7 +71,8 @@ func (c *Config) InitFromViper(v *viper.Viper) {
 	c.AdaptiveSampling = v.GetBool(adaptiveSampling)
 	tlsCfg, err := tlsFlagsCfg.InitFromViper(v)
 	if err != nil {
-		panic(err)
+		return err
 	}
 	c.TLSConfig = tlsCfg
+	return nil
 }

cmd/es-rollover/app/flags_test.go

@@ -35,7 +35,8 @@ func TestBindFlags(t *testing.T) {
 	})
 	require.NoError(t, err)
 
-	c.InitFromViper(v)
+	err = c.InitFromViper(v)
+	require.NoError(t, err)
 	assert.Equal(t, "tenant1-", c.IndexPrefix)
 	assert.True(t, c.Archive)
 	assert.Equal(t, 150, c.Timeout)

cmd/internal/flags/flags.go

@@ -39,16 +39,16 @@ func TryLoadConfigFile(v *viper.Viper) error {
 }
 
 // ParseJaegerTags parses the Jaeger tags string into a map.
-func ParseJaegerTags(jaegerTags string) map[string]string {
+func ParseJaegerTags(jaegerTags string) (map[string]string, error) {
 	if jaegerTags == "" {
-		return nil
+		return nil, nil
 	}
 	tagPairs := strings.Split(string(jaegerTags), ",")
 	tags := make(map[string]string)
 	for _, p := range tagPairs {
 		kv := strings.SplitN(p, "=", 2)
 		if len(kv) != 2 {
-			panic(fmt.Sprintf("invalid Jaeger tag pair %q, expected key=value", p))
+			return nil, fmt.Errorf("invalid Jaeger tag pair %q, expected key=value", p)
 		}
 		k, v := strings.TrimSpace(kv[0]), strings.TrimSpace(kv[1])
 
@@ -77,7 +77,7 @@ func ParseJaegerTags(jaegerTags string) map[string]string {
 		tags[k] = v
 	}
 
-	return tags
+	return tags, nil
 }
 
 // SharedFlags holds flags configuration

cmd/internal/flags/flags_test.go

@@ -9,10 +9,13 @@ import (
 	"testing"
 
 	"github.com/stretchr/testify/assert"
+	"github.com/stretchr/testify/require"
 )
 
 func TestParseJaegerTags(t *testing.T) {
-	assert.Nil(t, ParseJaegerTags(""))
+	tags, err := ParseJaegerTags("")
+	require.NoError(t, err)
+	assert.Nil(t, tags)
 
 	jaegerTags := fmt.Sprintf("%s,%s,%s,%s,%s,%s",
 		"key=value",
@@ -34,14 +37,13 @@ func TestParseJaegerTags(t *testing.T) {
 		"envVar5": "",
 	}
 
-	assert.Equal(t, expectedTags, ParseJaegerTags(jaegerTags))
+	tags, err = ParseJaegerTags(jaegerTags)
+	require.NoError(t, err)
+	assert.Equal(t, expectedTags, tags)
 }
 
-func TestParseJaegerTagsPanic(t *testing.T) {
-	assert.PanicsWithValue(t,
-		"invalid Jaeger tag pair \"no-equals-sign\", expected key=value",
-		func() {
-			ParseJaegerTags("no-equals-sign")
-		},
-	)
+func TestParseJaegerTagsError(t *testing.T) {
+	_, err := ParseJaegerTags("no-equals-sign")
+	require.Error(t, err)
+	assert.ErrorContains(t, err, "invalid Jaeger tag pair")
 }

cmd/tracegen/main.go

@@ -8,6 +8,7 @@ import (
 	"errors"
 	"flag"
 	"fmt"
+	"os"
 	"time"
 
 	"github.com/go-logr/zapr"
@@ -37,7 +38,8 @@ func main() {
 	zc.Level = zap.NewAtomicLevelAt(zapcore.Level(-8)) // level used by OTEL's Debug()
 	logger, err := zc.Build()
 	if err != nil {
-		panic(err)
+		fmt.Fprintf(os.Stderr, "failed to create logger: %v\n", err)
+		os.Exit(1)
 	}
 	otel.SetLogger(zapr.NewLogger(logger))