SQL 跳过 select 的第一个值

本篇文章主要是结合我之前面试的各种经历和实战开发中遇到的问题解决经验整理的，希望这篇《SQL 跳过 select 的第一个值》对你有很大帮助！欢迎收藏，分享给更多的需要的朋友学习~

我不确定为什么我的代码省略了 select 语句中收到的第一行。代码是：

func (s *sqlserver) findallproducts() (products []*product, err error) {
    ctx, cancel := getcontext()
    defer cancel()

    rows, err := s.querycontext(ctx,
        "select productid, productname, pricecents, brandname from products")
    if err != nil {
        return
    }   
    defer rows.close()
    
    for rows.next() {
        product := new(product)
        err = rows.scan(
            &product.productid,
            &product.productname,
            &product.pricecents,
            &product.brandname,
            )
        if err != nil {
            return
        }
        products = append(products, product)
    }
    return
}

还有一个单元测试，它使用 go-sqlmock 来提供测试存储库：

func TestProduct(t *testing.T) {
    db, mock, err := sqlmock.New()
    if err != nil {
        require.NoError(t, err)
    }
    s := sqlserver{db}
    defer s.Close()

    product1 := &Product{
        1,
        "soda",
        100,
        "prites",
    }
    product2 := &Product{
        60,
        "slurpee",
        400,
        "koce",
    }
    product3 := &Product{
        21,
        "borg",
        210,
        "ham",
    }

    query := "^insert into *"
    mock.ExpectExec(query).
        WithArgs(product1.ProductID, product1.ProductName, product1.PriceCents, product1.BrandName).
        WillReturnResult(sqlmock.NewResult(0, 1))

    query = "^insert into *"
    mock.ExpectExec(query).
        WithArgs(product2.ProductID, product2.ProductName, product2.PriceCents, product2.BrandName).
        WillReturnResult(sqlmock.NewResult(1, 1))

    query = "^insert into *"
    mock.ExpectExec(query).
        WithArgs(product3.ProductID, product3.ProductName, product3.PriceCents, product3.BrandName).
        WillReturnResult(sqlmock.NewResult(2, 1))

    query = "^select .+ from *"
    rows := sqlmock.NewRows([]string{"productid", "productname", "pricecents", "brandname"}).
        AddRow(product1.ProductID, product1.ProductName, product1.PriceCents, product1.BrandName)
    mock.ExpectQuery(query).
        WithArgs(product1.ProductID).
        WillReturnRows(rows)

    query = "^select .+ from *"
    rows.AddRow(product2.ProductID, product2.ProductName, product2.PriceCents, product2.BrandName)
    rows.AddRow(product3.ProductID, product3.ProductName, product3.PriceCents, product3.BrandName)
    mock.ExpectQuery(query).
        WillReturnRows(rows)

    err = s.InsertProduct(product1)
    assert.NoError(t, err)

    err = s.InsertProduct(product2)
    assert.NoError(t, err)

    err = s.InsertProduct(product3)
    assert.NoError(t, err)

    product, err := s.FindProductFromID(product1.ProductID)
    assert.NoError(t, err)
    assert.Equal(t, product1, product)

    products, err := s.FindAllProducts()
    assert.NoError(t, err)
    assert.Len(t, products, 3)
    assert.Equal(t, *product1, *products[0])
    assert.Equal(t, *product2, *products[1])
    assert.Equal(t, *product3, *products[2])

    err = mock.ExpectationsWereMet()
    assert.NoError(t, err)
}

将产品插入表中工作正常，但是当我尝试使用 s.findallproducts 检索产品时，返回的产品中始终缺少第一个产品（导致 assert.len(t, products, 3) 失败）。非常感谢任何帮助！

编辑：关于在将值扫描到产品中之前调用 rows.next() ，这是文档所说的操作，因为指针应该在第一行之前开始一个索引。但是，我已经调试了代码，问题似乎是指针开始向前一个位置（指向第一行）并调用 rows.next() 将其移动到下一行。我尝试在初始化产品后移动此语句，但代码随后抛出错误： 收到意外错误：sql：扫描调用但未调用 next

编辑 2：这似乎是 go-sqlmock 的直接问题，因为在真实的 sql 数据库上测试后，数据库正确返回所有 3 行。

解决方案

发现错误，因为 rows 是一个指针，由于某种原因 go-sqlmock 修改了行集，导致其中的原始行被删除，并且每次调用 willreturnrows 时都需要重置。以下部分代码：

query = "^select .+ from *"
    rows.addrow(product2.productid, product2.productname, product2.pricecents, product2.brandname)
    rows.addrow(product3.productid, product3.productname, product3.pricecents, product3.brandname)
    mock.expectquery(query).
        willreturnrows(rows)

应该修改以添加原始行，如下所示：

query = "^select .+ from *"
    rows = sqlmock.NewRows([]string{"productid", "productname", "pricecents", "brandname"})
    rows.AddRow(product1.ProductID, product1.ProductName, product1.PriceCents, product2.BrandName)
    rows.AddRow(product2.ProductID, product2.ProductName, product2.PriceCents, product2.BrandName)
    rows.AddRow(product3.ProductID, product3.ProductName, product3.PriceCents, product3.BrandName)
    mock.ExpectQuery(query).
        WillReturnRows(rows)

测试现在通过了！

好了，本文到此结束，带大家了解了《SQL 跳过 select 的第一个值》，希望本文对你有所帮助！关注golang学习网公众号，给大家分享更多Golang知识！