forked from si3nloong/sqlike
/
mapper.go
59 lines (49 loc) · 1.05 KB
/
mapper.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
package sqldump
import (
"bytes"
"strconv"
"strings"
"time"
"github.com/RevenueMonster/sqlike/util"
)
var null = []byte(`null`)
// Parser :
type Parser func([]byte) string
func byteToString(data []byte) string {
if len(data) == 0 {
return `""`
}
return strconv.Quote(string(data))
}
func numToString(data []byte) string {
return string(data)
}
func tsToString(data []byte) string {
t, _ := time.Parse(time.RFC3339, string(data))
return t.UTC().Format(`"2006-01-02 15:04:05.999999999"`)
}
func dateToString(data []byte) string {
t, _ := time.Parse(time.RFC3339, string(data))
return t.UTC().Format(`"2006-01-02"`)
}
func jsonToString(data []byte) string {
if bytes.Equal(data, null) {
return `"null"`
}
return strconv.Quote(string(data))
}
func setToString(data []byte) string {
str := string(data)
vals := strings.Split(str, ",")
blr := util.AcquireString()
defer util.ReleaseString(blr)
blr.WriteString(`'`)
for i, v := range vals {
if i > 0 {
blr.WriteByte(',')
}
blr.WriteString(v)
}
blr.WriteString(`'`)
return blr.String()
}