如何在Go中将数据库行转换为结构？

Question

假设我有一个结构:

type User struct {
    Name  string
    Id    int
    Score int
}

和具有相同模式的数据库表.将数据库行解析为结构的最简单方法是什么？我在下面添加了一个答案,但我不确定它是最好的答案.

Answer 1

包裹测试经常提供关于做事方式的线索.例如,来自database/sql/sql_test.go,

func TestQuery(t *testing.T) {
    /* . . . */
    rows, err := db.Query("SELECT|people|age,name|")
    if err != nil {
            t.Fatalf("Query: %v", err)
    }
    type row struct {
            age  int
            name string
    }
    got := []row{}
    for rows.Next() {
            var r row
            err = rows.Scan(&r.age, &r.name)
            if err != nil {
                    t.Fatalf("Scan: %v", err)
            }
            got = append(got, r)
    }
    /* . . . */
}

func TestQueryRow(t *testing.T) {
    /* . . . */
    var name string
    var age int
    var birthday time.Time
    err := db.QueryRow("SELECT|people|age,name|age=?", 3).Scan(&age)
    /* . . . */
}

对于您的问题,将行查询到结构中,将转换为以下内容:

var row struct {
    age  int
    name string
}
err = db.QueryRow("SELECT|people|age,name|age=?", 3).Scan(&row.age, &row.name)

我知道这看起来与您的解决方案类似,但重要的是展示如何找到解决方案.

Answer 2

我推荐github.com/jmoiron/sqlx.

来自README:

sqlx是一个库,它在go的标准database/sql库上提供了一组扩展 .的SQLX版本sql.DB,sql.TX, sql.Stmt,等.所有都保持底层接口不变,因此它们的接口是标准接口的超集.这使得使用数据库/ sql与sqlx集成现有代码库相对轻松.

主要的附加概念是:

• Marshal行成结构(具有嵌入式结构支持),映射和切片
• 命名参数支持包括预准备语句
• Get并Select快速从查询到结构/切片

README还包括一个代码片段,演示如何将行扫描到结构中:

type Place struct {
    Country       string
    City          sql.NullString
    TelephoneCode int `db:"telcode"`
}
// Loop through rows using only one struct
place := Place{}
rows, err := db.Queryx("SELECT * FROM place")
for rows.Next() {
    err := rows.StructScan(&place)
    if err != nil {
        log.Fatalln(err)
    } 
    fmt.Printf("%#v\n", place)
}

请注意,我们不必手动将每列映射到结构的字段.sqlx有一些结构字段到数据库列的默认映射,以及能够使用标记指定数据库列(请注意上面结构的TelephoneCode字段Place).您可以在文档中阅读更多相关信息.

Answer 3

这是一种方法 - 只需在Scan函数中手动分配所有结构值.

func getUser(name string) (*User, error) {
    var u User
    // this calls sql.Open, etc.
    db := getConnection()
    // note the below syntax only works for postgres
    err := db.QueryRow("SELECT * FROM users WHERE name = $1", name).Scan(&u.Id, &u.Name, &u.Score)
    if err != nil {
        return &User{}, err
    } else {
        return &u, nil
    }
}

Answer 4

rows, err := connection.Query("SELECT `id`, `username`, `email` FROM `users`")

if err != nil {
    panic(err.Error())
}

for rows.Next() {
    var user User

    if err := rows.Scan(&user.Id, &user.Username, &user.Email); err != nil {
        log.Println(err.Error())
    }

    users = append(users, user)
}

完整示例

Answer 5

这是一个专门用于此目的的库：scany。

你可以这样使用它：

type User struct {
    Name  string
    Id    int
    Score int
}

// db is your *sql.DB instance
// ctx is your current context.Context instance

// Use sqlscan.Select to query multiple records.
var users []*User
sqlscan.Select(ctx, db, &users, `SELECT name, id, score FROM users`)

// Use sqlscan.Get to query exactly one record.
var user User
sqlscan.Get(ctx, db, &user, `SELECT name, id, score FROM users WHERE id=123`)

它有详细的文档记录并且易于使用。

免责声明：我是这个库的作者。