/
bndTimeSlice.go
143 lines (133 loc) · 4.86 KB
/
bndTimeSlice.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
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
132
133
134
135
136
137
138
139
140
141
142
143
// Copyright 2014 Rana Ian. All rights reserved.
// Use of this source code is governed by The MIT License
// found in the accompanying LICENSE file.
package ora
/*
#include <oci.h>
#include <stdlib.h>
#include "version.h"
*/
import "C"
import (
"bytes"
"time"
"unsafe"
)
type bndTimeSlice struct {
stmt *Stmt
ocibnd *C.OCIBind
ociDateTimes []*C.OCIDateTime
zoneBuf bytes.Buffer
}
func (bnd *bndTimeSlice) bindOra(values []Time, position int, stmt *Stmt) error {
timeValues := make([]time.Time, len(values))
nullInds := make([]C.sb2, len(values))
for n, _ := range values {
if values[n].IsNull {
nullInds[n] = C.sb2(-1)
} else {
timeValues[n] = values[n].Value
}
}
return bnd.bind(timeValues, nullInds, position, stmt)
}
func (bnd *bndTimeSlice) bind(values []time.Time, nullInds []C.sb2, position int, stmt *Stmt) error {
bnd.stmt = stmt
bnd.ociDateTimes = make([]*C.OCIDateTime, len(values))
if nullInds == nil {
nullInds = make([]C.sb2, len(values))
}
alenp := make([]C.ACTUAL_LENGTH_TYPE, len(values))
rcodep := make([]C.ub2, len(values))
for n, timeValue := range values {
timezoneStr := zoneOffset(timeValue, &bnd.zoneBuf)
cTimezoneStr := C.CString(timezoneStr)
defer func() {
C.free(unsafe.Pointer(cTimezoneStr))
}()
r := C.OCIDescriptorAlloc(
unsafe.Pointer(bnd.stmt.ses.srv.env.ocienv), //CONST dvoid *parenth,
(*unsafe.Pointer)(unsafe.Pointer(&bnd.ociDateTimes[n])), //dvoid **descpp,
C.OCI_DTYPE_TIMESTAMP_TZ, //ub4 type,
0, //size_t xtramem_sz,
nil) //dvoid **usrmempp);
if r == C.OCI_ERROR {
return bnd.stmt.ses.srv.env.ociError()
} else if r == C.OCI_INVALID_HANDLE {
return errNew("unable to allocate oci timestamp handle during bind")
}
r = C.OCIDateTimeConstruct(
unsafe.Pointer(bnd.stmt.ses.srv.env.ocienv), //dvoid *hndl,
bnd.stmt.ses.srv.env.ocierr, //OCIError *err,
bnd.ociDateTimes[n], //OCIDateTime *datetime,
C.sb2(timeValue.Year()), //sb2 year,
C.ub1(int32(timeValue.Month())), //ub1 month,
C.ub1(timeValue.Day()), //ub1 day,
C.ub1(timeValue.Hour()), //ub1 hour,
C.ub1(timeValue.Minute()), //ub1 min,
C.ub1(timeValue.Second()), //ub1 sec,
C.ub4(timeValue.Nanosecond()), //ub4 fsec,
(*C.OraText)(unsafe.Pointer(cTimezoneStr)), //OraText *timezone,
C.size_t(len(timezoneStr))) //size_t timezone_length );
if r == C.OCI_ERROR {
return bnd.stmt.ses.srv.env.ociError()
}
alenp[n] = C.ACTUAL_LENGTH_TYPE(unsafe.Sizeof(bnd.ociDateTimes[n]))
}
r := C.OCIBINDBYPOS(
bnd.stmt.ocistmt, //OCIStmt *stmtp,
(**C.OCIBind)(&bnd.ocibnd), //OCIBind **bindpp,
bnd.stmt.ses.srv.env.ocierr, //OCIError *errhp,
C.ub4(position), //ub4 position,
unsafe.Pointer(&bnd.ociDateTimes[0]), //void *valuep,
C.LENGTH_TYPE(unsafe.Sizeof(bnd.ociDateTimes[0])), //sb8 value_sz,
C.SQLT_TIMESTAMP_TZ, //ub2 dty,
unsafe.Pointer(&nullInds[0]), //void *indp,
&alenp[0], //ub2 *alenp,
&rcodep[0], //ub2 *rcodep,
0, //ub4 maxarr_len,
nil, //ub4 *curelep,
C.OCI_DEFAULT) //ub4 mode );
if r == C.OCI_ERROR {
return bnd.stmt.ses.srv.env.ociError()
}
r = C.OCIBindArrayOfStruct(
bnd.ocibnd,
bnd.stmt.ses.srv.env.ocierr,
C.ub4(unsafe.Sizeof(bnd.ociDateTimes[0])), //ub4 pvskip,
C.ub4(C.sizeof_sb2), //ub4 indskip,
C.ub4(C.sizeof_ub4), //ub4 alskip,
C.ub4(C.sizeof_ub2)) //ub4 rcskip
if r == C.OCI_ERROR {
return bnd.stmt.ses.srv.env.ociError()
}
return nil
}
func (bnd *bndTimeSlice) setPtr() error {
return nil
}
func (bnd *bndTimeSlice) free(n int) {
defer func() {
recover()
}()
C.OCIDescriptorFree(
unsafe.Pointer(bnd.ociDateTimes[n]), //void *descp,
C.OCI_DTYPE_TIMESTAMP_TZ) //ub4 type );
}
func (bnd *bndTimeSlice) close() (err error) {
defer func() {
if value := recover(); value != nil {
err = errR(value)
}
}()
for n := range bnd.ociDateTimes {
bnd.free(n)
}
stmt := bnd.stmt
bnd.stmt = nil
bnd.ocibnd = nil
bnd.ociDateTimes = nil
bnd.zoneBuf.Reset()
stmt.putBnd(bndIdxTimeSlice, bnd)
return nil
}