/
inserts.go
90 lines (81 loc) · 2.01 KB
/
inserts.go
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
// Implements INSERTS output for exported rows
// e.g.
// INSERT INTO <table> (name, last_name, something);
package formats
import (
"bufio"
"bytes"
"fmt"
"io"
"strconv"
"strings"
"time"
)
const (
insertQuery = "INSERT INTO %s (%s) "
)
// very simple quoting for sql values
func quote(val string) string {
buf := bytes.NewBufferString("'")
buf.WriteString(strings.Replace(val, "'", "''", -1))
buf.WriteString("'")
return buf.String()
}
type InsertsFormat struct {
DataFormat
tableName string
columns []string
multiInsert bool
writer *bufio.Writer
}
func NewInsertsFormat(w io.Writer, fileName string, tableName string) (*InsertsFormat, error) {
return &InsertsFormat{
writer: bufio.NewWriter(w),
tableName: tableName,
columns: make([]string, 0),
multiInsert: false,
}, nil
}
func (f *InsertsFormat) WriteHeader(columns []string) error {
f.columns = columns
return nil
}
func (f *InsertsFormat) Flush() error { return nil }
func (f *InsertsFormat) WriteRow(values map[string]interface{}) error {
columnsVal := strings.Join(f.columns, ",")
queryStr := fmt.Sprintf(insertQuery, f.tableName, columnsVal)
buf := bytes.NewBufferString(queryStr)
record := []string{}
for _, col := range f.columns {
switch value := (values[col]).(type) {
case string:
record = append(record, quote(value))
case []byte:
record = append(record, quote(string(value)))
case int64:
record = append(record, fmt.Sprintf("%d", value))
case float64:
record = append(record, strconv.FormatFloat(value, 'f', -1, 64))
case time.Time:
record = append(record, quote(value.Format(time.RFC3339)))
case bool:
if value == true {
record = append(record, "true")
} else {
record = append(record, "false")
}
case nil:
record = append(record, "null")
}
}
buf.WriteString("VALUES (")
buf.WriteString(strings.Join(record, ","))
buf.WriteString(");\n")
_, err := f.writer.Write(buf.Bytes())
if err != nil {
return err
}
err = f.writer.Flush()
// defer f.writer.Close()
return err
}