/
bndBinSlice.go
107 lines (100 loc) · 2.74 KB
/
bndBinSlice.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
// 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 "version.h"
*/
import "C"
import (
"unsafe"
)
type bndBinSlice struct {
stmt *Stmt
ocibnd *C.OCIBind
buf []byte
}
func (bnd *bndBinSlice) bindOra(values []Raw, position int, lobBufferSize int, stmt *Stmt) error {
binValues := make([][]byte, len(values))
nullInds := make([]C.sb2, len(values))
for i := range values {
if values[i].IsNull {
nullInds[i] = C.sb2(-1)
} else {
binValues[i] = values[i].Value
}
}
return bnd.bind(binValues, nullInds, position, lobBufferSize, stmt)
}
func (bnd *bndBinSlice) bind(values [][]byte, nullInds []C.sb2, position int, lobBufferSize int, stmt *Stmt) error {
bnd.stmt = stmt
if nullInds == nil {
nullInds = make([]C.sb2, len(values))
}
alenp := make([]C.ACTUAL_LENGTH_TYPE, len(values))
rcodep := make([]C.ub2, len(values))
var maxLen int
for _, b := range values {
if len(b) > maxLen {
maxLen = len(b)
}
}
n := maxLen * len(values)
if cap(bnd.buf) < n {
bnd.buf = make([]byte, n)
} else {
bnd.buf = bnd.buf[:n]
// reset buffer
for i := range bnd.buf {
bnd.buf[i] = 0
}
}
for i, b := range values {
copy(bnd.buf[i*maxLen:], b)
alenp[i] = C.ACTUAL_LENGTH_TYPE(len(b))
}
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.buf[0]), //void *valuep,
C.LENGTH_TYPE(maxLen), //sb8 value_sz,
C.SQLT_LBI, //ub2 dty,
unsafe.Pointer(&nullInds[0]), //void *indp,
&alenp[0], //ub4 *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(maxLen), //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 *bndBinSlice) setPtr() error {
return nil
}
func (bnd *bndBinSlice) close() (err error) {
defer func() {
if value := recover(); value != nil {
err = errR(value)
}
}()
stmt := bnd.stmt
bnd.stmt = nil
bnd.ocibnd = nil
stmt.putBnd(bndIdxBinSlice, bnd)
return nil
}