使用rows.Next()遍历SQLite行时，无法通过exitChannel退出循环的问题

我在Go语言里用rows.Next()循环遍历SQLite数据库的行，需要通过exitChannel传入的信号停止迭代并退出循环，写了下面的代码：

for rows.Next() {
    select {
    case <-exitChannel:
        fmt.Println("Exiting")
        break
    default:
        fmt.Println("Entering")
        doSomething()
    }
}

即使exitChannel已经触发，代码始终不会进入对应的case，也不会打印“Exiting”。

为了排查是Go语言还是SQLite的问题，我跑了一个不含rows.Next()的连续循环测试：

for {
    select {
    case <-exitChannel:
        fmt.Println("Exiting")
        break
    default:
        fmt.Println("Entering")
    }
}

这段代码能正常打印“Exiting”。

问题原因 #

核心问题在于rows.Next()的阻塞特性：当调用rows.Next()时，它会阻塞直到从SQLite数据库获取到下一行数据、查询完成或出现错误。在这个阻塞期间，你的select语句根本没有执行的机会，自然无法响应exitChannel的信号。

解决方案 #

方案1：使用上下文(Context)中断查询

主流的SQLite Go驱动（如github.com/mattn/go-sqlite3）支持通过上下文(Context)来中断正在执行的查询。你可以在执行查询时传入带取消功能的上下文，当exitChannel触发时，调用取消函数让rows.Next()提前返回错误，从而退出循环。

示例代码：

import (
    "context"
    "database/sql"
    "fmt"
    _ "github.com/mattn/go-sqlite3"
)

func main() {
    db, err := sql.Open("sqlite3", "./test.db")
    if err != nil {
        panic(err)
    }
    defer db.Close()

    exitChannel := make(chan struct{})

    ctx, cancel := context.WithCancel(context.Background())
    defer cancel()

    // 启动协程监听退出信号
    go func() {
        <-exitChannel
        fmt.Println("Exiting")
        cancel()
    }()

    // 使用QueryContext传入上下文
    rows, err := db.QueryContext(ctx, "SELECT * FROM your_table")
    if err != nil {
        panic(err)
    }
    defer rows.Close()

    for rows.Next() {
        fmt.Println("Entering")
        doSomething()
    }

    // 检查遍历后的错误，上下文取消会返回context.Canceled
    if err := rows.Err(); err != nil && err != context.Canceled {
        fmt.Printf("Query error: %v\n", err)
    }
}

func doSomething() {
    // 你的业务逻辑
}

方案2：将数据库遍历放入协程，通过通道传递结果

把遍历rows的逻辑放到独立协程中，通过结果通道传递每行数据，主循环同时监听结果通道和exitChannel，这样就能及时响应退出信号。

示例代码：

import (
    "database/sql"
    "fmt"
    _ "github.com/mattn/go-sqlite3"
)

type YourRowType struct {
    Field1 string
    Field2 int
}

func main() {
    db, err := sql.Open("sqlite3", "./test.db")
    if err != nil {
        panic(err)
    }
    defer db.Close()

    exitChannel := make(chan struct{})
    resultChan := make(chan YourRowType)
    errChan := make(chan error)

    // 协程中执行数据库遍历
    go func() {
        rows, err := db.Query("SELECT field1, field2 FROM your_table")
        if err != nil {
            errChan <- err
            close(resultChan)
            return
        }
        defer rows.Close()

        for rows.Next() {
            var row YourRowType
            if err := rows.Scan(&row.Field1, &row.Field2); err != nil {
                errChan <- err
                close(resultChan)
                return
            }
            resultChan <- row
        }

        if err := rows.Err(); err != nil {
            errChan <- err
        }
        close(resultChan)
    }()

    // 主循环监听退出信号和结果
    for {
        select {
        case <-exitChannel:
            fmt.Println("Exiting")
            return
        case row, ok := <-resultChan:
            if !ok {
                fmt.Println("遍历完成")
                return
            }
            fmt.Println("Entering")
            doSomething(row)
        case err := <-errChan:
            fmt.Printf("查询错误: %v\n", err)
            return
        }
    }
}

func doSomething(row YourRowType) {
    // 你的业务逻辑
}

内容的提问来源于stack exchange，提问作者Kuntal Pithwa