fieldproject/fieldproject_com/main.go

package main

import (
	"fieldproject_common/config"
	"github.com/spf13/cobra"
	"go.uber.org/zap"
	"os"
	"utils/log"
)

func main() {
	rootCmd := &cobra.Command{Use: "org"}
	rootCmd.AddCommand(Buyer())
	rootCmd.AddCommand(Winner())
	if err := rootCmd.Execute(); err != nil {
		log.Fatal("rootCmd.Execute failed", zap.Error(err))
	}
}

// buyer 处理
func Buyer() *cobra.Command {
	var (
		cfg string
	)
	cmdClient := &cobra.Command{
		Use:   "buyer",
		Short: "Start Run buyer...",
		Run: func(cmd *cobra.Command, args []string) {
			if len(cfg) == 0 {
				_ = cmd.Help()
				os.Exit(0)
			}
			config.Init(cfg)
			InitLog()
			err := InitDb()
			if err != nil {
				log.Fatal("InitDb failed", zap.Error(err))
			}
			log.Info("success")
		},
	}
	cmdClient.Flags().StringVarP(&cfg, "conf", "c", "", "config")
	return cmdClient
}

// Winner @Description
// @Author J 2022/7/26 11:48
func Winner() *cobra.Command {
	var (
		cfg string
	)
	cmdServer := &cobra.Command{
		Use:   "winner",
		Short: "Start Run winner...",
		Run: func(cmd *cobra.Command, args []string) {
			if len(cfg) == 0 {
				_ = cmd.Help()
				os.Exit(0)
			}

		},
	}
	//cmdServer.Flags().StringVarP(&cfg, "conf", "c", "", "config")
	return cmdServer
}