在进行数据库查询时，我们经常会遇到分页查询的情况。然而，当取消分页查询时，有时会导致计时不准确的问题。这个问题对于需要精确计时的应用程序来说是非常关键的。在本文中，php小编百草将为您介绍如何处理这个问题，从而确保计时的准确性和精确性。我们将探讨一些可能的原因和解决方案，帮助您更好地理解和应对这个问题。

问题内容

我有一个对象用于进行分页 sql 查询，该查询允许异步运行查询：

type pagedquery[t any] struct {
    results   chan []*t
    errors    chan error
    done      chan error
    quit      chan error
    client    *sql.db
}

func newpagedquery[t any](client *sql.db) *pagedquery[t] {
    return &pagedquery[t]{
        results:   make(chan []*t, 1),
        errors:    make(chan error, 1),
        done:      make(chan error, 1),
        quit:      make(chan error, 1),
        client:    client,
    }
}

func (paged *pagedquery[t]) requestasync(ctx context.context, queries ...*query) {

    conn, err := client.conn(ctx)
    if err != nil {
        paged.errors <- err
        return
    }

    defer func() {
        conn.close()
        paged.done <- nil
    }()

    for i, query := range queries {
        select {
        case <-ctx.done():
            return
        case <-paged.quit:
            return
        default:
        }

        rows, err := conn.querycontext(ctx, query.string, query.arguments...)
        if err != nil {
            paged.errors <- err
            return
        }

        data, err := sql.readrows[t](rows)
        if err != nil {
            paged.errors <- err
            return
        }

        paged.results <- data
    }
}

我正在尝试测试此代码，特别是取消部分。我的测试代码如下所示：

svc, mock := createServiceMock("TEST_DATABASE", "TEST_SCHEMA")

mock.ExpectQuery(regexp.QuoteMeta("TEST QUERY")).
    WithArgs(...).
    WillReturnRows(mock.NewRows([]string{"t", "v", "o", "c", "h", "l", "vw", "n"}))

ctx, cancel := context.WithCancel(context.Background())
go svc.requestAsync(ctx, query1, query2, query3, query4)

time.Sleep(50 * time.Millisecond)
cancel()

results := make([]data, 0)
loop:
for {
    select {
    case <-query.Done:
        break loop
    case err := <-query.Errors:
        Expect(err).ShouldNot(HaveOccurred())
    case r := <-query.Results:
        results = append(results, r...)
    }
}

Expect(results).Should(BeEmpty())
Expect(mock.ExpectationsWereMet()).ShouldNot(HaveOccurred()) // fails here

我遇到的问题是，此测试偶尔会在我的评论所指示的行处失败，因为当调用 cancel() 时，不能保证在我检查 < 的 switch 语句处执行 - ctx.done 或 <-退出。执行可以在循环中的任何位置进行，直到我将结果发送到 results 通道。但这没有意义，因为执行应该阻塞，直到我从 results 通道接收到数据，直到我调用 cancel() 后我才会这样做。此外，我依靠 sqlmock 包进行 sql 测试，它不允许对 sql 查询进行任何类型的模糊检查。为什么我会遇到此故障以及如何修复它？

解决方法

我的问题是由于我自己对 go 通道缺乏了解而导致的。我认为，通过创建 chan([]*t, 1) 意味着通道在满时（即当它包含单个项目时）会阻塞，但事实并非如此。相反，当我尝试在缓冲区已满时发送到通道时，会发生阻塞。因此，通过像这样修改 results ：

func NewPagedQuery[T any](client *sql.DB) *PagedQuery[T] {
    return &PagedQuery[T]{
        Results:   make(chan []*T),    // Remove buffer here
        Errors:    make(chan error, 1),
        Done:      make(chan error, 1),
        Quit:      make(chan error, 1),
        client:    client,
    }
}

我可以确保通道阻塞，直到接收到它包含的数据。这一更改解决了测试的所有问题。