root.go

package cmd

import (
	"os"

	"github.com/1995parham/saf/internal/cmd/consumer"
	"github.com/1995parham/saf/internal/cmd/producer"
	"github.com/1995parham/saf/internal/config"
	"github.com/1995parham/saf/internal/logger"
	"github.com/1995parham/saf/internal/telemetry/trace"
	"github.com/spf13/cobra"
	"go.uber.org/zap"
)

// ExitFailure status code.
const ExitFailure = 1

// Execute adds all child commands to the root command and sets flags appropriately.
// This is called by main.main(). It only needs to happen once to the rootCmd.
func Execute() {
	cfg := config.New()

	logger := logger.New(cfg.Logger)

	tracer := trace.New(cfg.Telemetry.Trace)

	// nolint: exhaustivestruct
	root := &cobra.Command{
		Use:   "saf",
		Short: "Queue with NATS Jetstream to remove all the erlangs from cloud",
	}

	producer.Register(root, cfg, logger, tracer)
	consumer.Register(root, cfg, logger, tracer)

	if err := root.Execute(); err != nil {
		logger.Error("failed to execute root command", zap.Error(err))

		os.Exit(ExitFailure)
	}
}