forked from aliyun/aliyun-odps-go-sdk
-
Notifications
You must be signed in to change notification settings - Fork 0
/
type_conversion.go
109 lines (92 loc) · 2.79 KB
/
type_conversion.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
98
99
100
101
102
103
104
105
106
107
108
109
// Licensed to the Apache Software Foundation (ASF) under one
// or more contributor license agreements. See the NOTICE file
// distributed with this work for additional information
// regarding copyright ownership. The ASF licenses this file
// to you under the Apache License, Version 2.0 (the
// "License"); you may not use this file except in compliance
// with the License. You may obtain a copy of the License at
//
// http://www.apache.org/licenses/LICENSE-2.0
//
// Unless required by applicable law or agreed to in writing, software
// distributed under the License is distributed on an "AS IS" BASIS,
// WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
// See the License for the specific language governing permissions and
// limitations under the License.
package datatype
import (
"github.com/pkg/errors"
"reflect"
"strings"
)
var odpsDataType = reflect.TypeOf((*DataType)(nil)).Elem()
func TryConvertGoToOdpsType(i interface{}) (DataType, error) {
t, ok := i.(reflect.Type)
if !ok {
tt := reflect.TypeOf(i)
if tt.Implements(odpsDataType) {
return i.(DataType), nil
}
return nil, errors.Errorf("cannot convert %s to odps type", tt.Name())
}
if t.Implements(odpsDataType) {
switch strings.ToLower(t.Name()) {
case "char", "varchar", "decimal", "array", "map", "struct":
return nil, errors.Errorf("char, varchar, decimal need extra parameter to specific concrete type")
}
}
if t.Kind() == reflect.Slice && t.Elem().Kind() == reflect.Uint8 {
return BinaryType, nil
}
switch t.Kind() {
case reflect.Bool:
return BooleanType, nil
case reflect.Int8:
return TinyIntType, nil
case reflect.Int32:
return IntType, nil
case reflect.Int16:
return SmallIntType, nil
case reflect.Int64:
return BigIntType, nil
case reflect.Float32:
return FloatType, nil
case reflect.Float64:
return DoubleType, nil
case reflect.String:
return StringType, nil
case reflect.Array, reflect.Slice:
elemType, err := TryConvertGoToOdpsType(t.Elem())
if err != nil {
return nil, err
}
return NewArrayType(elemType), nil
case reflect.Map:
keyType, err := TryConvertGoToOdpsType(t.Key())
if err != nil {
return nil, err
}
valueType, err := TryConvertGoToOdpsType(t.Elem())
if err != nil {
return nil, err
}
return NewMapType(keyType, valueType), nil
case reflect.Struct:
n := t.NumField()
fields := make([]StructFieldType, n)
for i := 0; i < n; i++ {
field := t.Field(i)
fieldName := field.Tag.Get("odps")
if fieldName == "" {
fieldName = field.Name
}
fieldType, err := TryConvertGoToOdpsType(field.Type)
if err != nil {
return nil, err
}
fields[i] = NewStructFieldType(fieldName, fieldType)
}
return NewStructType(fields...), nil
}
return nil, errors.Errorf("Cannot convert %s to odps type", t.Name())
}