计算golang中的行数

Question

我想使用Go显示数据库中的行数。如何显示行数？

count, err := db.Query("SELECT COUNT(*) FROM main_table")

Answer 1

查询将在变量计数中返回一行。因此，下一个要做的就是读取该行，并使用函数将结果分配到新变量中Scan()。这就是它的工作方式。

rows, err := db.Query("SELECT COUNT(*) FROM main_table")
if err != nil {
    log.Fatal(err)
}
defer rows.Close()

var count int

for rows.Next() {   
    if err := rows.Scan(&count); err != nil {
        log.Fatal(err)
    }
}

fmt.Printf("Number of rows are %s\n", count)

最好的选择是使用，QueryRow()因为您希望只读取一行。代码将是。

var count int

err := db.QueryRow("SELECT COUNT(*) FROM main_table").Scan(&count)
switch {    
case err != nil:
    log.Fatal(err)
default:
    fmt.Printf("Number of rows are %s\n", count)
}