mirror of https://github.com/mattn/go-sqlite3.git
merged bradfitz's patch. thanks.
This commit is contained in:
parent
a620c1a086
commit
1fc69c55a3
20
sqlite3.go
20
sqlite3.go
|
@ -30,7 +30,6 @@ func init() {
|
|||
}
|
||||
|
||||
type SQLiteDriver struct {
|
||||
|
||||
}
|
||||
|
||||
type SQLiteConn struct {
|
||||
|
@ -112,6 +111,7 @@ type SQLiteStmt struct {
|
|||
c *SQLiteConn
|
||||
s *C.sqlite3_stmt
|
||||
t string
|
||||
closed bool
|
||||
}
|
||||
|
||||
func (c *SQLiteConn) Prepare(query string) (driver.Stmt, error) {
|
||||
|
@ -127,10 +127,14 @@ func (c *SQLiteConn) Prepare(query string) (driver.Stmt, error) {
|
|||
if perror != nil && C.strlen(perror) > 0 {
|
||||
t = C.GoString(perror)
|
||||
}
|
||||
return &SQLiteStmt{c, s, t}, nil
|
||||
return &SQLiteStmt{c: c, s: s, t: t}, nil
|
||||
}
|
||||
|
||||
func (s *SQLiteStmt) Close() error {
|
||||
if s.closed {
|
||||
return nil
|
||||
}
|
||||
s.closed = true
|
||||
rv := C.sqlite3_finalize(s.s)
|
||||
if rv != C.SQLITE_OK {
|
||||
return errors.New(C.GoString(C.sqlite3_errmsg(s.c.db)))
|
||||
|
@ -142,7 +146,7 @@ func (s *SQLiteStmt) NumInput() int {
|
|||
return int(C.sqlite3_bind_parameter_count(s.s))
|
||||
}
|
||||
|
||||
func (s *SQLiteStmt) bind(args []interface{}) error {
|
||||
func (s *SQLiteStmt) bind(args []driver.Value) error {
|
||||
rv := C.sqlite3_reset(s.s)
|
||||
if rv != C.SQLITE_ROW && rv != C.SQLITE_OK && rv != C.SQLITE_DONE {
|
||||
return errors.New(C.GoString(C.sqlite3_errmsg(s.c.db)))
|
||||
|
@ -169,7 +173,7 @@ func (s *SQLiteStmt) bind(args []interface{}) error {
|
|||
rv = C.sqlite3_bind_int(s.s, n, C.int(v))
|
||||
case bool:
|
||||
if bool(v) {
|
||||
rv = C.sqlite3_bind_int(s.s, n, 1)
|
||||
rv = C.sqlite3_bind_int(s.s, n, -1)
|
||||
} else {
|
||||
rv = C.sqlite3_bind_int(s.s, n, 0)
|
||||
}
|
||||
|
@ -191,7 +195,7 @@ func (s *SQLiteStmt) bind(args []interface{}) error {
|
|||
return nil
|
||||
}
|
||||
|
||||
func (s *SQLiteStmt) Query(args []interface{}) (driver.Rows, error) {
|
||||
func (s *SQLiteStmt) Query(args []driver.Value) (driver.Rows, error) {
|
||||
if err := s.bind(args); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
@ -210,7 +214,7 @@ func (r *SQLiteResult) RowsAffected() (int64, error) {
|
|||
return int64(C.sqlite3_changes(r.s.c.db)), nil
|
||||
}
|
||||
|
||||
func (s *SQLiteStmt) Exec(args []interface{}) (driver.Result, error) {
|
||||
func (s *SQLiteStmt) Exec(args []driver.Value) (driver.Result, error) {
|
||||
if err := s.bind(args); err != nil {
|
||||
return nil, err
|
||||
}
|
||||
|
@ -228,7 +232,7 @@ type SQLiteRows struct {
|
|||
}
|
||||
|
||||
func (rc *SQLiteRows) Close() error {
|
||||
return nil
|
||||
return rc.s.Close()
|
||||
}
|
||||
|
||||
func (rc *SQLiteRows) Columns() []string {
|
||||
|
@ -241,7 +245,7 @@ func (rc *SQLiteRows) Columns() []string {
|
|||
return rc.cols
|
||||
}
|
||||
|
||||
func (rc *SQLiteRows) Next(dest []interface{}) error {
|
||||
func (rc *SQLiteRows) Next(dest []driver.Value) error {
|
||||
rv := C.sqlite3_step(rc.s.s)
|
||||
if rv != C.SQLITE_ROW {
|
||||
return errors.New(C.GoString(C.sqlite3_errmsg(rc.s.c.db)))
|
||||
|
|
Loading…
Reference in New Issue