nxshock/csv2db
1
0
Fork 0
mirror of https://github.com/nxshock/csv2db.git synced 2024-11-27 03:31:00 +05:00
Code Issues Projects Releases Wiki Activity

Add WithoutHeader and Silent options

Browse Source
This commit is contained in:
nxshock 2023-02-11 22:07:58 +05:00
parent 3606f38890
commit 3f2a2e2b44
1 changed files with 30 additions and 17 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

47
main.go
View File

@ -73,6 +73,12 @@ var app = &cli.App{
Name: "unknowncolumnnames",
Usage: "insert to table with unknown column names",
},
&cli.BoolFlag{
Name: "withoutheader",
Usage: "file does not contains a header"},
&cli.BoolFlag{
Name: "silent",
Usage: "silens progress output"},
},
Action: func(c *cli.Context) error {
if len(strings.Split(c.String("table"), ".")) != 2 {
@ -146,23 +152,26 @@ func processReader(c *cli.Context, r io.Reader) error {
reader.Comma = []rune(c.String("comma"))[0]
}
header, err := reader.Read()
if err != nil {
return fmt.Errorf("read header: %v", err)
}
headerList := `"`
for i, v := range header {
if c.String("fields")[i] == ' ' {
continue
var header []string
if !c.Bool("withoutheader") {
header, err = reader.Read()
if err != nil {
return fmt.Errorf("read header: %v", err)
}
headerList += v
headerList := `"`
for i, v := range header {
if c.String("fields")[i] == ' ' {
continue
}
if i+1 < len(header) {
headerList += `", "`
} else {
headerList += `"`
headerList += v
if i+1 < len(header) {
headerList += `", "`
} else {
headerList += `"`
}
}
}
@ -207,7 +216,7 @@ func processReader(c *cli.Context, r io.Reader) error {
}
if c.Bool("create") {
err = createTable(tx, c.String("table"), header, c.String("fields"), c.Bool("overwrite"))
err = createTable(tx, c.String("table"), neededHeader, c.String("fields"), c.Bool("overwrite"))
if err != nil {
_ = tx.Rollback()
return fmt.Errorf("create table: %v", err)
@ -225,7 +234,9 @@ func processReader(c *cli.Context, r io.Reader) error {
n := 0
for {
if n%100000 == 0 {
fmt.Fprintf(os.Stderr, "Processed %d records...\r", n)
if !c.Bool("silent") {
fmt.Fprintf(os.Stderr, "Processed %d records...\r", n)
}
}
record, err := reader.Read()
@ -274,7 +285,9 @@ func processReader(c *cli.Context, r io.Reader) error {
_ = tx.Rollback()
return fmt.Errorf("calc rows affected: %v", err)
}
fmt.Fprintf(os.Stderr, "Processed %d records. \n", rowsAffected)
if !c.Bool("silent") {
fmt.Fprintf(os.Stderr, "Processed %d records. \n", rowsAffected)
}
err = stmt.Close()
if err != nil {