Merge pull request #1440 from mattn/fix-limit-example-stmt-leak
Close prepared statement and fail if limit is not enforced in limit example
This commit is contained in:
@@ -32,6 +32,7 @@ func bulkInsert(db *sql.DB, query string, args []any) (err error) {
|
||||
if err != nil {
|
||||
return
|
||||
}
|
||||
defer stmt.Close()
|
||||
|
||||
_, err = stmt.Exec(args...)
|
||||
if err != nil {
|
||||
@@ -89,11 +90,13 @@ func main() {
|
||||
|
||||
query, args = createBulkInsertQuery(num, num)
|
||||
err = bulkInsert(db, query, args)
|
||||
if err != nil {
|
||||
if err != nil {
|
||||
log.Printf("expect failed since SQLITE_LIMIT_VARIABLE_NUMBER is too small: %v", err)
|
||||
}
|
||||
if err == nil {
|
||||
log.Fatal("expected failure since SQLITE_LIMIT_VARIABLE_NUMBER is too small, but insert succeeded")
|
||||
}
|
||||
if !strings.Contains(err.Error(), "too many SQL variables") {
|
||||
log.Fatalf("expected too many SQL variables error, got: %v", err)
|
||||
}
|
||||
log.Printf("expect failed since SQLITE_LIMIT_VARIABLE_NUMBER is too small: %v", err)
|
||||
|
||||
bigLimitVariableNumber := 999999
|
||||
sqlite3conn.SetLimit(sqlite3.SQLITE_LIMIT_VARIABLE_NUMBER, bigLimitVariableNumber)
|
||||
@@ -104,9 +107,7 @@ func main() {
|
||||
query, args = createBulkInsertQuery(500, num+num)
|
||||
err = bulkInsert(db, query, args)
|
||||
if err != nil {
|
||||
if err != nil {
|
||||
log.Fatal(err)
|
||||
}
|
||||
log.Fatal(err)
|
||||
}
|
||||
|
||||
log.Println("no error if SQLITE_LIMIT_VARIABLE_NUMBER > 999")
|
||||
|
||||
Reference in New Issue
Block a user