Fix error on Execer. Close #124

sqlite3.go

@@ -144,52 +144,55 @@ func (c *SQLiteConn) lastError() Error {
 
 // Implements Execer
 func (c *SQLiteConn) Exec(query string, args []driver.Value) (driver.Result, error) {
-       if len(args) == 0 {
+	if len(args) == 0 {
-               return c.exec(query)
+		return c.exec(query)
-       }
+	}
 
-       for {
+	for {
-               s, err := c.Prepare(query)
+		s, err := c.Prepare(query)
-               if err != nil {
+		if err != nil {
-                       return nil, err
+			return nil, err
-               }
+		}
-               na := s.NumInput()
+		var res driver.Result
-               res, err := s.Exec(args[:na])
+		if s.(*SQLiteStmt).s != nil {
-               if err != nil && err != driver.ErrSkip {
+			na := s.NumInput()
-                       s.Close()
+			res, err = s.Exec(args[:na])
-                       return nil, err
+			if err != nil && err != driver.ErrSkip {
-               }
+				s.Close()
-               args = args[na:]
+				return nil, err
-               tail := s.(*SQLiteStmt).t
+			}
-               if tail == "" {
+			args = args[na:]
-                       return res, nil
+		}
-               }
+		tail := s.(*SQLiteStmt).t
-               s.Close()
+		if tail == "" {
-               query = tail
+			return res, nil
-       }
+		}
+		s.Close()
+		query = tail
+	}
 }
 
 // Implements Queryer
 func (c *SQLiteConn) Query(query string, args []driver.Value) (driver.Rows, error) {
-       for {
+	for {
-               s, err := c.Prepare(query)
+		s, err := c.Prepare(query)
-               if err != nil {
+		if err != nil {
-                       return nil, err
+			return nil, err
-               }
+		}
-               na := s.NumInput()
+		na := s.NumInput()
-               rows, err := s.Query(args[:na])
+		rows, err := s.Query(args[:na])
-               if err != nil && err != driver.ErrSkip {
+		if err != nil && err != driver.ErrSkip {
-                       s.Close()
+			s.Close()
-                       return nil, err
+			return nil, err
-               }
+		}
-               args = args[na:]
+		args = args[na:]
-               tail := s.(*SQLiteStmt).t
+		tail := s.(*SQLiteStmt).t
-               if tail == "" {
+		if tail == "" {
-                       return rows, nil
+			return rows, nil
-               }
+		}
-               s.Close()
+		s.Close()
-               query = tail
+		query = tail
-       }
+	}
 }
 
 func (c *SQLiteConn) exec(cmd string) (driver.Result, error) {

sqlite3_test.go

@@ -1,7 +1,6 @@
 package sqlite3
 
 import (
-	"./sqltest"
 	"crypto/rand"
 	"database/sql"
 	"encoding/hex"
@@ -9,6 +8,8 @@ import (
 	"path/filepath"
 	"testing"
 	"time"
+
+	"./sqltest"
 )
 
 func TempFilename() string {
@@ -639,29 +640,26 @@ func TestSuite(t *testing.T) {
 
 // TODO: Execer & Queryer currently disabled
 // https://github.com/mattn/go-sqlite3/issues/82
-//func TestExecer(t *testing.T) {
+func TestExecer(t *testing.T) {
-//	tempFilename := TempFilename()
+	tempFilename := TempFilename()
-//	db, err := sql.Open("sqlite3", tempFilename)
+	db, err := sql.Open("sqlite3", tempFilename)
-//	if err != nil {
+	if err != nil {
-//		t.Fatal("Failed to open database:", err)
+		t.Fatal("Failed to open database:", err)
-//	}
+	}
-//	defer os.Remove(tempFilename)
+	defer os.Remove(tempFilename)
-//	defer db.Close()
+	defer db.Close()
-//
+
-//	_, err = db.Exec(`
+	_, err = db.Exec(`
-//	create table foo (id integer);
+       create table foo (id integer); -- one comment
-//	insert into foo(id) values(?);
+       insert into foo(id) values(?);
-//	insert into foo(id) values(?);
+       insert into foo(id) values(?);
-//	insert into foo(id) values(?);
+       insert into foo(id) values(?); -- another comment
-//	`, 1, 2, 3)
+       `, 1, 2, 3)
-//	if err != nil {
+	if err != nil {
-//		t.Error("Failed to call db.Exec:", err)
+		t.Error("Failed to call db.Exec:", err)
-//	}
+	}
-//	if err != nil {
+}
-//		t.Error("Failed to call res.RowsAffected:", err)
+
-//	}
-//}
-//
 //func TestQueryer(t *testing.T) {
 //	tempFilename := TempFilename()
 //	db, err := sql.Open("sqlite3", tempFilename)