forked from guypeled76/go-bigquery-driver
/
columns.go
97 lines (76 loc) · 2.01 KB
/
columns.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
91
92
93
94
95
96
97
package driver
import (
"database/sql/driver"
"encoding/json"
"cloud.google.com/go/bigquery"
"gorm.io/driver/bigquery/adaptor"
)
type bigQuerySchema interface {
ColumnNames() []string
ConvertColumnValue(index int, value bigquery.Value) (driver.Value, error)
}
type bigQueryColumns struct {
names []string
columns []bigQueryColumn
}
func (columns bigQueryColumns) ConvertColumnValue(index int, value bigquery.Value) (driver.Value, error) {
if index > -1 && len(columns.columns) > index {
column := columns.columns[index]
return column.ConvertValue(value)
}
return value, nil
}
func (columns bigQueryColumns) ColumnNames() []string {
return columns.names
}
type bigQueryReroutedColumn struct {
values []bigquery.Value
schema bigquery.Schema
}
func (c bigQueryReroutedColumn) MarshalJSON() ([]byte, error) {
return json.Marshal(c.values)
}
type bigQueryColumn struct {
Name string
Schema bigquery.Schema
Adaptor adaptor.SchemaColumnAdaptor
}
func (column bigQueryColumn) ConvertValue(value bigquery.Value) (driver.Value, error) {
if len(column.Schema) == 0 {
return value, nil
}
values, ok := value.([]bigquery.Value)
if ok {
if len(values) > 0 {
if _, isRows := values[0].([]bigquery.Value); !isRows {
values = []bigquery.Value{values}
}
}
value = bigQueryReroutedColumn{values: values, schema: column.Schema}
}
if columnAdaptor := column.Adaptor; columnAdaptor != nil {
return columnAdaptor.AdaptValue(value)
}
return value, nil
}
func createBigQuerySchema(schema bigquery.Schema, schemaAdaptor adaptor.SchemaAdaptor) bigQuerySchema {
var names []string
var columns []bigQueryColumn
for _, column := range schema {
name := column.Name
var columnAdaptor adaptor.SchemaColumnAdaptor
if schemaAdaptor != nil {
columnAdaptor = schemaAdaptor.GetColumnAdaptor(name)
}
names = append(names, name)
columns = append(columns, bigQueryColumn{
Name: name,
Schema: column.Schema,
Adaptor: columnAdaptor,
})
}
return &bigQueryColumns{
names,
columns,
}
}