完善输出

Author: cyy0523xc

csv2es/cmd/import.go

@@ -92,13 +92,12 @@ csv2es import --host=locahost --port=9200 --mapping=mapping_filename.json --inde
 		for {
 			row, err := reader.Read()
 			if err == io.EOF {
-				fmt.Printf("the file is over")
+				fmt.Println("the file is over")
 				break
 			}
 			if err != nil {
 				panic(err)
 			}
-			fmt.Printf("Row 1: %+v\n", row)
 
 			req := elastic.NewBulkIndexRequest().Index(cParams.IndexName).Type(cParams.DocType).Doc(row)
 			bulk.Add(req)

csv2es/cmd/root.go

@@ -24,6 +24,8 @@ import (
 )
 
 type CommonParams struct {
+	Debug bool
+
 	// es config
 	Host      string
 	Port      int
@@ -69,6 +71,7 @@ func init() {
 	// Here you will define your flags and configuration settings.
 	// Cobra supports persistent flags, which, if defined here,
 	// will be global for your application.
+	rootCmd.PersistentFlags().BoolVar(&cParams.Debug, "debug", false, "debug mode")
 	rootCmd.PersistentFlags().StringVar(&cParams.Host, "host", "localhost", "es host")
 	rootCmd.PersistentFlags().IntVar(&cParams.Port, "port", 9200, "es port")
 	rootCmd.PersistentFlags().StringVar(&cParams.IndexName, "index", "", "es index name")
@@ -77,7 +80,7 @@ func init() {
 
 	// Cobra also supports local flags, which will only run
 	// when this action is called directly.
-	rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
+	//rootCmd.Flags().BoolP("toggle", "t", false, "Help message for toggle")
 }
 
 // initConfig reads in config file and ENV variables if set.

examples/import.sh

@@ -9,5 +9,5 @@ if [ -f csv2es ]; then
     cmd="./csv2es"
 fi
 
-$cmd --host=100.115.147.50 --port=9200 --index=test --type=test --mapping=./eyenlp_area2016.json --csv=./eyenlp_area2016.csv
+$cmd import --host=100.115.147.50 --port=9200 --index=test --type=test --mapping=./eyenlp_area2016.json --csv=./eyenlp_area2016.csv --delete-index=true