forked from sqls-server/sqls
/
scan_row.go
87 lines (77 loc) · 1.58 KB
/
scan_row.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
package database
import (
"database/sql"
"encoding/json"
"fmt"
"strings"
"time"
)
func Columns(rows *sql.Rows) ([]string, error) {
var cols []string
var err error
cols, err = rows.Columns()
if err != nil {
return nil, fmt.Errorf("cannot get query columns, %w", err)
}
for i, c := range cols {
if strings.TrimSpace(c) == "" {
cols[i] = fmt.Sprintf("col%d", i)
}
}
return cols, nil
}
func ScanRows(rows *sql.Rows, columnLength int) ([][]string, error) {
stringRows := [][]string{}
for rows.Next() {
// scan to []interface{}
rowBuffer := make([]interface{}, columnLength)
for i := range rowBuffer {
rowBuffer[i] = new(interface{})
}
if err := rows.Scan(rowBuffer...); err != nil {
return nil, err
}
stringRow := make([]string, columnLength)
for i, buf := range rowBuffer {
val, err := sqlValToString(buf)
if err != nil {
return nil, err
}
stringRow[i] = val
}
stringRows = append(stringRows, stringRow)
}
return stringRows, nil
}
func sqlValToString(pointer interface{}) (string, error) {
res := ""
if pointer == nil {
return res, nil
}
val := *pointer.(*interface{})
switch v := (val).(type) {
case []byte:
res = string(v)
case string:
res = v
case time.Time:
res = v.Format(time.RFC3339Nano)
case fmt.Stringer:
res = v.String()
case map[string]interface{}:
buf, err := json.Marshal(v)
if err != nil {
return "", err
}
res = string(buf)
case []interface{}:
buf, err := json.Marshal(v)
if err != nil {
return "", err
}
res = string(buf)
default:
res = fmt.Sprintf("%v", v)
}
return res, nil
}