forked from mikespook/mymysql
-
Notifications
You must be signed in to change notification settings - Fork 0
/
unsafe.go
116 lines (95 loc) · 2.31 KB
/
unsafe.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
110
111
112
113
114
115
116
package native
import (
"github.com/ziutek/mymysql/mysql"
"io"
"time"
"unsafe"
)
type paramValue struct {
typ uint16
addr unsafe.Pointer
raw bool
length int // >=0 - length of value, <0 - unknown length
}
func (pv *paramValue) SetAddr(addr uintptr) {
pv.addr = unsafe.Pointer(addr)
}
func (val *paramValue) Len() int {
if val.addr == nil {
// Invalid Value was binded
return 0
}
// val.addr always points to the pointer - lets dereference it
ptr := *(*unsafe.Pointer)(val.addr)
if ptr == nil {
// Binded Ptr Value is nil
return 0
}
if val.length >= 0 {
return val.length
}
switch val.typ {
case MYSQL_TYPE_STRING:
return lenStr(*(*string)(ptr))
case MYSQL_TYPE_DATE:
return lenDate(*(*mysql.Date)(ptr))
case MYSQL_TYPE_TIMESTAMP, MYSQL_TYPE_DATETIME:
return lenTime(*(*time.Time)(ptr))
case MYSQL_TYPE_TIME:
return lenDuration(*(*time.Duration)(ptr))
case MYSQL_TYPE_TINY: // val.length < 0 so this is bool
return 1
}
// MYSQL_TYPE_VAR_STRING, MYSQL_TYPE_BLOB and type of Raw value
return lenBin(*(*[]byte)(ptr))
}
func writeValue(wr io.Writer, val *paramValue) {
if val.addr == nil {
// Invalid Value was binded
return
}
// val.addr always points to the pointer - lets dereference it
ptr := *(*unsafe.Pointer)(val.addr)
if ptr == nil {
// Binded Ptr Value is nil
return
}
if val.raw || val.typ == MYSQL_TYPE_VAR_STRING ||
val.typ == MYSQL_TYPE_BLOB {
writeBin(wr, *(*[]byte)(ptr))
return
}
// We don't need unsigned bit to check type
switch val.typ & ^MYSQL_UNSIGNED_MASK {
case MYSQL_TYPE_NULL:
// Don't write null values
case MYSQL_TYPE_STRING:
writeStr(wr, *(*string)(ptr))
case MYSQL_TYPE_LONG, MYSQL_TYPE_FLOAT:
writeU32(wr, *(*uint32)(ptr))
case MYSQL_TYPE_SHORT:
writeU16(wr, *(*uint16)(ptr))
case MYSQL_TYPE_TINY:
if val.length == -1 {
// Translate bool value to MySQL tiny
if *(*bool)(ptr) {
writeByte(wr, 1)
} else {
writeByte(wr, 0)
}
} else {
writeByte(wr, *(*byte)(ptr))
}
case MYSQL_TYPE_LONGLONG, MYSQL_TYPE_DOUBLE:
writeU64(wr, *(*uint64)(ptr))
case MYSQL_TYPE_DATE:
writeDate(wr, *(*mysql.Date)(ptr))
case MYSQL_TYPE_TIMESTAMP, MYSQL_TYPE_DATETIME:
writeTime(wr, *(*time.Time)(ptr))
case MYSQL_TYPE_TIME:
writeDuration(wr, *(*time.Duration)(ptr))
default:
panic(BIND_UNK_TYPE)
}
return
}