refactor: cmd server

labasubagia · Sep 15, 2023 · 53ee1ec · 53ee1ec
1 parent b3dfdb8
commit 53ee1ec
Showing 1 changed file with 6 additions and 20 deletions.
diff --git a/cmd/server.go b/cmd/server.go
@@ -12,18 +12,16 @@ import (
 	"github.com/spf13/cobra"
 )
 
-var (
-	dbTypeStr  = strings.Join(repository.Keys(), ", ")
-	logTypeStr = strings.Join(logger.Keys(), ", ")
-)
-
 func init() {
+	dbTypeStr := strings.Join(repository.Keys(), ", ")
+	logTypeStr := strings.Join(logger.Keys(), ", ")
+
 	rootCmd.AddCommand(serverCmd)
 
 	serverCmd.Flags().Bool("prod", false, "use for production")
-	serverCmd.Flags().IntP("port", "p", config.HTTPServerPort, "server port number")
-	serverCmd.Flags().StringP("database", "d", repository.DefaultType, fmt.Sprintf("database type in (%s)", dbTypeStr))
-	serverCmd.Flags().StringP("log", "l", logger.DefaultType, fmt.Sprintf("log type in (%s)", logTypeStr))
+	serverCmd.Flags().IntVarP(&config.HTTPServerPort, "port", "p", config.HTTPServerPort, "server port number")
+	serverCmd.Flags().StringVarP(&config.DBType, "database", "d", repository.DefaultType, fmt.Sprintf("database type in (%s)", dbTypeStr))
+	serverCmd.Flags().StringVarP(&config.LogType, "log", "l", logger.DefaultType, fmt.Sprintf("log type in (%s)", logTypeStr))
 }
 
 var serverCmd = &cobra.Command{
@@ -37,18 +35,6 @@ var serverCmd = &cobra.Command{
 		if err == nil && isProduction {
 			config.Environment = util.EnvProduction
 		}
-
-		// port
-		port, err := cmd.Flags().GetInt("port")
-		if err == nil {
-			config.HTTPServerPort = port
-		}
-
-		// logger
-		logType, err := cmd.Flags().GetString("log")
-		if err == nil {
-			config.LogType = logType
-		}
 		logger := logger.NewLogger(config)
 		logger.Info().Msgf("use logger %s", config.LogType)