在进行数据库查询时，我们经常会遇到分页查询的情况。然而，当取消分页查询时，有时会导致计时不准确的问题。这个问题对于需要精确计时的应用程序来说是非常关键的。在本文中，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,
}
}
我可以确保通道阻塞，直到接收到它包含的数据。这一更改解决了测试的所有问题。
