diff options
Diffstat (limited to 'sqlite3.go')
| -rw-r--r-- | sqlite3.go | 61 |
1 files changed, 57 insertions, 4 deletions
@@ -121,6 +121,7 @@ type SQLiteTx struct { type SQLiteStmt struct { c *SQLiteConn s *C.sqlite3_stmt + nv int t string closed bool cls bool @@ -368,7 +369,19 @@ func (c *SQLiteConn) Prepare(query string) (driver.Stmt, error) { if tail != nil && C.strlen(tail) > 0 { t = strings.TrimSpace(C.GoString(tail)) } - ss := &SQLiteStmt{c: c, s: s, t: t} + nv := int(C.sqlite3_bind_parameter_count(s)) + if nv > 0 { + pn := C.GoString(C.sqlite3_bind_parameter_name(s, 1)) + /* TODO: map argument for named parameters + if len(pn) > 0 && pn[0] == '$' && pn[1] != '1' { + nv = -1 + } + */ + if len(pn) > 0 && pn[0] != '?' { + nv = -1 + } + } + ss := &SQLiteStmt{c: c, s: s, nv: nv, t: t} runtime.SetFinalizer(ss, (*SQLiteStmt).Close) return ss, nil } @@ -392,7 +405,12 @@ func (s *SQLiteStmt) Close() error { // Return a number of parameters. func (s *SQLiteStmt) NumInput() int { - return int(C.sqlite3_bind_parameter_count(s.s)) + return s.nv +} + +type bindArg struct { + n int + v driver.Value } func (s *SQLiteStmt) bind(args []driver.Value) error { @@ -401,8 +419,43 @@ func (s *SQLiteStmt) bind(args []driver.Value) error { return s.c.lastError() } - for i, v := range args { - n := C.int(i + 1) + var vargs []bindArg + narg := len(args) + if s.nv == -1 { + /* TODO: map argument for named parameters + if narg == 1 { + if m, ok := args[0].(map[string]driver.Value); ok { + for k, v := range m { + pn := C.CString(k) + if pi := int(C.sqlite3_bind_parameter_index(s.s, pn)); pi > 0 { + println(pi) + vargs = append(vargs, bindArg{pi, v}) + } + C.free(unsafe.Pointer(pn)) + } + } + narg = 0 + } + */ + if narg > 0 { + for i := 0; i < narg; i++ { + pn := C.CString(fmt.Sprint(i + 1)) + if pi := int(C.sqlite3_bind_parameter_index(s.s, pn)); pi > 0 { + vargs = append(vargs, bindArg{pi, args[i]}) + } + C.free(unsafe.Pointer(pn)) + } + } + } else { + vargs = make([]bindArg, narg) + for i, v := range args { + vargs[i] = bindArg{i + 1, v} + } + } + + for _, varg := range vargs { + n := C.int(varg.n) + v := varg.v switch v := v.(type) { case nil: rv = C.sqlite3_bind_null(s.s, n) |