chore: separate encoder type

internal/app/app.go

@@ -147,9 +147,9 @@ func getPublisher(c *cli.Context, redisClient *redis.Client, topic string) (publ
 }
 
 func getMessageEncoder(c *cli.Context) encoder.Encoder {
-	publisherType := c.String(publisherTypeFlag.Name)
-	switch publisherType {
-	case publisherpkg.PublisherTypeKafka:
+	encoderType := c.String(encoderTypeFlag.Name)
+	switch encoderType {
+	case encoder.EncoderTypeProtobuf:
 		return encoder.NewProtobufEncoder()
 	default:
 		return encoder.NewJSONEncoder()

internal/app/flags.go

@@ -106,12 +106,20 @@ var (
 		Usage:   "A list of address for connecting to kafka. Default: localhost:9092",
 	}
 
+	encoderTypeFlag = &cli.StringFlag{
+		Name:     "encoder-type",
+		EnvVars:  []string{"ENCODER_TYPE"},
+		Value:    "",
+		Required: true,
+		Usage:    "Type of encoder. Supports: `protobuf`, `json` (default)",
+	}
+
 	publisherTypeFlag = &cli.StringFlag{
 		Name:     "publisher-type",
 		EnvVars:  []string{"PUBLISHER_TYPE"},
 		Value:    "",
 		Required: true,
-		Usage:    "Type of publisher. Supports: `kafka`, `redis-stream`",
+		Usage:    "Type of publisher. Supports: `kafka`, `redis-stream` (default)",
 	}
 	publisherTopicFlag = &cli.StringFlag{
 		Name:     "publisher-topic",
@@ -160,6 +168,11 @@ func NewKafkaFlags() []cli.Flag {
 	return []cli.Flag{kafkaAddrsFlag}
 }
 
+// NewEncoderFlags returns flags for encoder.
+func NewEncoderFlags() []cli.Flag {
+	return []cli.Flag{encoderTypeFlag}
+}
+
 // NewPublisherFlags returns flags for publishers.
 func NewPublisherFlags() []cli.Flag {
 	return []cli.Flag{publisherTypeFlag, publisherMaxLenFlag, publisherTopicFlag}
@@ -181,6 +194,7 @@ func NewFlags() []cli.Flag {
 	}
 	flags = append(flags, NewSentryFlags()...)
 	flags = append(flags, NewRedisFlags()...)
+	flags = append(flags, NewEncoderFlags()...)
 	flags = append(flags, NewKafkaFlags()...)
 	flags = append(flags, NewPublisherFlags()...)
 	flags = append(flags, NewBlockKeeperFlags()...)

pkg/encoder/config.go

@@ -0,0 +1,6 @@
+package encoder
+
+const (
+	EncoderTypeProtobuf = "protobuf"
+	EncoderTypeJSON     = "json"
+)