logger

cmd/extract.go

@@ -1,20 +1,26 @@
 package cmd
 
 import (
+	"fmt"
+
 	"github.com/spf13/cobra"
 	"starwiz.cn/sjy01/preprocessing/extract"
 )
 
+var (
+	dataId string
+)
+
 var extractCmd = &cobra.Command{
 	Use:   "extract",
 	Short: "Extract data from raw data files",
 	Long:  `Extract data from raw data files`,
 	Run: func(cmd *cobra.Command, args []string) {
 		params := extract.Params{
-			InputData:  "demo/051513.dat",
+			InputData:  fmt.Sprintf("demo/%s.dat", dataId),
 			OutputPath: "demo/output",
 			TempPath:   "demo/temp",
-			DataId:     "051513",
+			DataId:     dataId,
 		}
 		p := extract.NewExtractor(&params)
 		p.ExtractAosData()
@@ -27,4 +33,6 @@ var extractCmd = &cobra.Command{
 
 func init() {
 	rootCmd.AddCommand(extractCmd)
+
+	extractCmd.Flags().StringVarP(&dataId, "data id", "d", "051513", "051513")
 }