区分命令信息

cmd/cmd/export-all-job.go

@@ -0,0 +1,56 @@
+package cmd
+
+import (
+	"excel_export/biz/config"
+	"fmt"
+	"github.com/spf13/cobra"
+	"strconv"
+	"time"
+)
+
+func init() {
+	exportAllJobCmd.Flags().IntP("query", "q", 10000, "单次查询数")
+
+	exportAllJobCmd.Flags().StringP("system", "s", "", "需要导出的系统")
+	exportAllJobCmd.Flags().StringP("begin", "b", "", "开始时间，如：2006-01-02 15:04:05")
+	exportAllJobCmd.Flags().StringP("end", "e", "", "结束时间，如：2006-01-02 15:04:05")
+
+	rootCmd.AddCommand(exportAllJobCmd)
+}
+
+var exportAllJobCmd = &cobra.Command{
+	Use:           "export-all-job",
+	Short:         "导出数据",
+	SilenceUsage:  true,
+	SilenceErrors: true,
+	Run:           exportAllJobRun,
+}
+
+func exportAllJobRun(cmd *cobra.Command, args []string) {
+	c := config.DefaultConfig
+
+	sName := cmd.Flag("system").Value.String()
+	sys, err := c.GetSystem(sName)
+	if err != nil {
+		CmdError(cmd, "%s", err.Error())
+	}
+
+	begin := MustFlagsDateTime(cmd, "begin")
+	end := MustFlagsDateTime(cmd, "end")
+
+	query := cmd.Flag("query").Value.String()
+	batch, err := strconv.Atoi(query)
+	if err != nil {
+		CmdError(cmd, "无效的参数：%s", err.Error())
+	}
+
+	for _, job := range sys.Jobs {
+		fmt.Printf("[%s]执行【%s】【%s】导出\n", time.Now().Format("2006-01-02 15:04:05"), sName, job.Name)
+		b := time.Now()
+		ee := NewCsv(config.DefaultConfig)
+		ee.Export(sName, job.Name, begin, end, batch)
+		e := time.Now()
+		fmt.Println("导出耗时：" + e.Sub(b).String())
+	}
+
+}

cmd/cmd/export-all.go

@@ -0,0 +1,49 @@
+package cmd
+
+import (
+	"excel_export/biz/config"
+	"fmt"
+	"github.com/spf13/cobra"
+	"strconv"
+	"time"
+)
+
+func init() {
+	exportAllCmd.Flags().IntP("query", "q", 10000, "单次查询数")
+	exportAllCmd.Flags().StringP("begin", "b", "", "开始时间，如：2006-01-02 15:04:05")
+	exportAllCmd.Flags().StringP("end", "e", "", "结束时间，如：2006-01-02 15:04:05")
+
+	rootCmd.AddCommand(exportAllCmd)
+}
+
+var exportAllCmd = &cobra.Command{
+	Use:           "export-all",
+	Short:         "导出数据",
+	SilenceUsage:  true,
+	SilenceErrors: true,
+	Run:           exportAllRun,
+}
+
+func exportAllRun(cmd *cobra.Command, args []string) {
+	c := config.DefaultConfig
+
+	begin := MustFlagsDateTime(cmd, "begin")
+	end := MustFlagsDateTime(cmd, "end")
+
+	query := cmd.Flag("query").Value.String()
+	batch, err := strconv.Atoi(query)
+	if err != nil {
+		CmdError(cmd, "无效的参数：%s", err.Error())
+	}
+	for _, sys := range c.Systems {
+		for _, job := range sys.Jobs {
+			fmt.Printf("[%s]执行【%s】【%s】导出\n", time.Now().Format("2006-01-02 15:04:05"), sys.Name, job.Name)
+			b := time.Now()
+			ee := NewCsv(config.DefaultConfig)
+			ee.Export(sys.Name, job.Name, begin, end, batch)
+			e := time.Now()
+			fmt.Println("导出耗时：" + e.Sub(b).String())
+		}
+	}
+
+}

cmd/cmd/export.go

@@ -0,0 +1,61 @@
+package cmd
+
+import (
+	"excel_export/biz/config"
+	"fmt"
+	"github.com/spf13/cobra"
+	"strconv"
+	"time"
+)
+
+func init() {
+	exportCmd.Flags().IntP("query", "q", 10000, "单次查询数")
+
+	exportCmd.Flags().StringP("system", "s", "", "需要导出的系统")
+	exportCmd.Flags().StringP("job", "j", "", "需要导出的任务")
+	exportCmd.Flags().StringP("begin", "b", "", "开始时间，如：2006-01-02 15:04:05")
+	exportCmd.Flags().StringP("end", "e", "", "结束时间，如：2006-01-02 15:04:05")
+
+	rootCmd.AddCommand(exportCmd)
+}
+
+var exportCmd = &cobra.Command{
+	Use:           "export",
+	Short:         "导出数据",
+	SilenceUsage:  true,
+	SilenceErrors: true,
+	Run:           exportRun,
+}
+
+func exportRun(cmd *cobra.Command, args []string) {
+	c := config.DefaultConfig
+
+	sName := cmd.Flag("system").Value.String()
+	sys, err := c.GetSystem(sName)
+	if err != nil {
+		CmdError(cmd, "%s", err.Error())
+	}
+
+	jName := cmd.Flag("job").Value.String()
+
+	_, err = sys.GetJob(jName)
+	if err != nil {
+		CmdError(cmd, "%s", err.Error())
+	}
+
+	fmt.Printf("[%s]执行【%s】【%s】导出\n", time.Now().Format("2006-01-02 15:04:05"), sName, jName)
+
+	begin := MustFlagsDateTime(cmd, "begin")
+	end := MustFlagsDateTime(cmd, "end")
+
+	query := cmd.Flag("query").Value.String()
+	batch, err := strconv.Atoi(query)
+	if err != nil {
+		CmdError(cmd, "无效的参数：%s", err.Error())
+	}
+	b := time.Now()
+	ee := NewCsv(config.DefaultConfig)
+	ee.Export(sName, jName, begin, end, batch)
+	e := time.Now()
+	fmt.Println("耗时：" + e.Sub(b).String())
+}

cmd/cmd/job.go

@@ -0,0 +1,36 @@
+package cmd
+
+import (
+	"excel_export/biz/config"
+	"github.com/spf13/cobra"
+)
+
+func init() {
+	jobCmd.Flags().StringP("system", "s", "", "系统名称")
+	rootCmd.AddCommand(jobCmd)
+}
+
+var jobCmd = &cobra.Command{
+	Use:           "job",
+	Short:         "支持的系统",
+	SilenceUsage:  true,
+	SilenceErrors: true,
+	Run:           jobRun,
+}
+
+func jobRun(cmd *cobra.Command, args []string) {
+	c := config.DefaultConfig
+
+	sName := cmd.Flag("system").Value.String()
+	sys, err := c.GetSystem(sName)
+	if err != nil {
+		CmdError(cmd, "%s\n请输入export-tool system进行查看", err.Error())
+	}
+
+	cmd.Println("支持的任务：")
+	for _, job := range sys.Jobs {
+		cmd.Printf("%s \n", job.Name)
+	}
+	return
+
+}

cmd/cmd/root.go

@@ -1,94 +1,16 @@
 package cmd
 
 import (
-	"excel_export/biz/config"
-	"fmt"
 	"github.com/spf13/cobra"
-	"strconv"
-	"time"
 )
 
 var rootCmd = &cobra.Command{
-	Use:           "export",
+	Use:           "export-tool",
 	Short:         "导出直充系统订单数据",
 	SilenceUsage:  true,
 	SilenceErrors: true,
-	Run:           rootRun,
 }
 
 func Execute() {
-
-	rootCmd.Flags().IntP("query", "q", 10000, "单次查询数")
-
-	rootCmd.Flags().StringP("system", "s", "", "需要导出的系统")
-	rootCmd.Flags().StringP("job", "j", "", "需要导出的任务")
-	rootCmd.Flags().StringP("begin", "b", "", "开始时间，如：2006-01-02 15:04:05")
-	rootCmd.Flags().StringP("end", "e", "", "结束时间，如：2006-01-02 15:04:05")
-	//	rootCmd.AddCommand(ExportCmd)
-
 	cobra.CheckErr(rootCmd.Execute())
-
-}
-
-func rootRun(cmd *cobra.Command, args []string) {
-	c := config.DefaultConfig
-
-	sName := cmd.Flag("system").Value.String()
-	if sName == "" {
-		cmd.Println("支持的系统")
-		for i, system := range c.Systems {
-			cmd.Printf("\t %d\t%s\n", i+1, system.Name)
-		}
-		return
-	}
-
-	var sys config.System
-	sIndex, err := strconv.Atoi(sName)
-	if err == nil {
-		sys = c.Systems[sIndex-1]
-		sName = sys.Name
-	} else {
-		sys, err = c.GetSystem(sName)
-		if err != nil {
-			CmdError(cmd, "%s", err.Error())
-		}
-	}
-
-	jName := cmd.Flag("job").Value.String()
-	if jName == "" {
-		cmd.Println("支持的任务")
-		for i, job := range sys.Jobs {
-			cmd.Printf("\t %d\t%s \n", i+1, job.Name)
-		}
-		return
-	}
-
-	var job config.Job
-	jIndex, err := strconv.Atoi(jName)
-	if err == nil {
-		job = sys.Jobs[jIndex-1]
-		jName = job.Name
-	} else {
-		job, err = sys.GetJob(jName)
-		if err != nil {
-			CmdError(cmd, "%s", err.Error())
-		}
-	}
-
-	fmt.Printf("[%s]执行【%s】【%s】导出\n", time.Now().Format("2006-01-02 15:04:05"), sName, jName)
-
-	begin := MustFlagsDateTime(cmd, "begin")
-	end := MustFlagsDateTime(cmd, "end")
-
-	query := cmd.Flag("query").Value.String()
-	batch, err := strconv.Atoi(query)
-	if err != nil {
-		CmdError(cmd, "无效的参数：%s", err.Error())
-	}
-	b := time.Now()
-	ee := NewCsv(config.DefaultConfig)
-	ee.Export(sName, jName, begin, end, batch)
-	e := time.Now()
-	fmt.Println("耗时：" + e.Sub(b).String())
-
 }

cmd/cmd/system.go

@@ -0,0 +1,29 @@
+package cmd
+
+import (
+	"excel_export/biz/config"
+	"github.com/spf13/cobra"
+)
+
+func init() {
+	rootCmd.AddCommand(systemCmd)
+}
+
+var systemCmd = &cobra.Command{
+	Use:           "system",
+	Short:         "支持的系统",
+	SilenceUsage:  true,
+	SilenceErrors: true,
+	Run:           systemRun,
+}
+
+func systemRun(cmd *cobra.Command, args []string) {
+	c := config.DefaultConfig
+
+	cmd.Println("支持的系统:")
+	for _, system := range c.Systems {
+		cmd.Printf("%s\n", system.Name)
+	}
+	return
+
+}