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

17
main.go
View File

@ -73,6 +73,12 @@ var app = &cli.App{
Name: "unknowncolumnnames", Name: "unknowncolumnnames",
Usage: "insert to table with unknown column names", 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 { Action: func(c *cli.Context) error {
if len(strings.Split(c.String("table"), ".")) != 2 { if len(strings.Split(c.String("table"), ".")) != 2 {
@ -146,7 +152,9 @@ func processReader(c *cli.Context, r io.Reader) error {
reader.Comma = []rune(c.String("comma"))[0] reader.Comma = []rune(c.String("comma"))[0]
} }
header, err := reader.Read() var header []string
if !c.Bool("withoutheader") {
header, err = reader.Read()
if err != nil { if err != nil {
return fmt.Errorf("read header: %v", err) return fmt.Errorf("read header: %v", err)
} }
@ -165,6 +173,7 @@ func processReader(c *cli.Context, r io.Reader) error {
headerList += `"` headerList += `"`
} }
} }
}
var neededHeader []string var neededHeader []string
@ -207,7 +216,7 @@ func processReader(c *cli.Context, r io.Reader) error {
} }
if c.Bool("create") { 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 { if err != nil {
_ = tx.Rollback() _ = tx.Rollback()
return fmt.Errorf("create table: %v", err) return fmt.Errorf("create table: %v", err)
@ -225,8 +234,10 @@ func processReader(c *cli.Context, r io.Reader) error {
n := 0 n := 0
for { for {
if n%100000 == 0 { if n%100000 == 0 {
if !c.Bool("silent") {
fmt.Fprintf(os.Stderr, "Processed %d records...\r", n) fmt.Fprintf(os.Stderr, "Processed %d records...\r", n)
} }
}
record, err := reader.Read() record, err := reader.Read()
if err == io.EOF { if err == io.EOF {
@ -274,7 +285,9 @@ func processReader(c *cli.Context, r io.Reader) error {
_ = tx.Rollback() _ = tx.Rollback()
return fmt.Errorf("calc rows affected: %v", err) return fmt.Errorf("calc rows affected: %v", err)
} }
if !c.Bool("silent") {
fmt.Fprintf(os.Stderr, "Processed %d records. \n", rowsAffected) fmt.Fprintf(os.Stderr, "Processed %d records. \n", rowsAffected)
}
err = stmt.Close() err = stmt.Close()
if err != nil { if err != nil {