mirror of
https://github.com/nxshock/csv2db.git
synced 2025-04-20 01:21:51 +05:00
Compare commits
2 Commits
c4f18766cb
...
3606f38890
Author | SHA1 | Date | |
---|---|---|---|
3606f38890 | |||
06139693c6 |
7
main.go
7
main.go
@ -4,6 +4,7 @@ import (
|
|||||||
"bufio"
|
"bufio"
|
||||||
"database/sql"
|
"database/sql"
|
||||||
"encoding/csv"
|
"encoding/csv"
|
||||||
|
"errors"
|
||||||
"fmt"
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"log"
|
"log"
|
||||||
@ -74,6 +75,10 @@ var app = &cli.App{
|
|||||||
},
|
},
|
||||||
},
|
},
|
||||||
Action: func(c *cli.Context) error {
|
Action: func(c *cli.Context) error {
|
||||||
|
if len(strings.Split(c.String("table"), ".")) != 2 {
|
||||||
|
return errors.New("table name must be in schema.name format")
|
||||||
|
}
|
||||||
|
|
||||||
var err error
|
var err error
|
||||||
|
|
||||||
db, err = sql.Open("sqlserver", fmt.Sprintf("sqlserver://%s?database=%s", c.String("server"), c.String("database")))
|
db, err = sql.Open("sqlserver", fmt.Sprintf("sqlserver://%s?database=%s", c.String("server"), c.String("database")))
|
||||||
@ -135,7 +140,7 @@ func processReader(c *cli.Context, r io.Reader) error {
|
|||||||
reader.TrimLeadingSpace = false
|
reader.TrimLeadingSpace = false
|
||||||
reader.FieldsPerRecord = len(c.String("fields"))
|
reader.FieldsPerRecord = len(c.String("fields"))
|
||||||
|
|
||||||
if []rune(c.String("comma"))[0] == 't' {
|
if runes := []rune(c.String("comma")); len(runes) > 0 && runes[0] == 't' {
|
||||||
reader.Comma = '\t'
|
reader.Comma = '\t'
|
||||||
} else {
|
} else {
|
||||||
reader.Comma = []rune(c.String("comma"))[0]
|
reader.Comma = []rune(c.String("comma"))[0]
|
||||||
|
Loading…
x
Reference in New Issue
Block a user