假设我有一个结构:
type User struct {
Name string
Id int
Score int
}
具有相同架构的数据库表。将数据库行解析为结构的最简单方法是什么?我在下面添加了一个答案,但我不确定它是最好的答案。
答案 0 :(得分:66)
Go package测试经常提供关于做事方式的线索。例如,来自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)
我知道这与您的解决方案类似,但重要的是展示如何找到解决方案。
答案 1 :(得分:32)
来自自述文件:
sqlx是一个库,它提供了一套符合标准的扩展
database/sql
库。sql.DB
,sql.TX
的sqlx版本,sql.Stmt
,等。所有都保持底层接口不变,所以 他们的接口是标准接口的超集。这使得 使用集成现有代码库相对轻松 带有sqlx的数据库/ sql。主要的附加概念是:
- Marshal行进入结构体(具有嵌入式结构支持),地图和切片
- 命名参数支持,包括准备好的陈述
Get
和Select
可以快速从查询到结构/切片
自述文件还包括一个代码片段,演示如何将行扫描到结构中:
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
字段)。您可以在the documentation中了解更多相关信息。
答案 2 :(得分:28)
这是一种方法 - 只需在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
}
}
答案 3 :(得分:3)
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)
}
答案 4 :(得分:2)
答案 5 :(得分:0)
您可以使用github.com/gocraft/dbr
(godoc)将行映射到结构中。
import (
"github.com/gocraft/dbr"
)
func GetUser(name string) (*User, error) {
var u User
rows, err := db.Query("SELECT * FROM users WHERE name = $1 LIMIT 1", name)
if err != nil {
return nil, err
}
// Load uses reflection to map values into a struct.
n, err := dbr.Load(rows, &u)
if err != nil {
return nil, err
}
if n != 1 {
return nil, NotFound
}
return u, nil
}
答案 6 :(得分:0)
这里是一个图书馆,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`)
有据可查且易于使用。
免责声明:我是这个图书馆的作者。
答案 7 :(得分:-1)
使用: go-models-mysql sqlbuilder
val, err = m.ScanRowType(row, (*UserTb)(nil))
或完整代码
import (
"database/sql"
"fmt"
lib "github.com/eehsiao/go-models-lib"
mysql "github.com/eehsiao/go-models-mysql"
)
// MyUserDao : extend from mysql.Dao
type MyUserDao struct {
*mysql.Dao
}
// UserTb : sql table struct that to store into mysql
type UserTb struct {
Name sql.NullString `TbField:"Name"`
Id int `TbField:"Id"`
Score int `TbField:"Score"`
}
// GetFirstUser : this is a data logical function, you can write more logical in there
// sample data logical function to get the first user
func (m *MyUserDao) GetFirstUser() (user *User, err error) {
m.Select("Name", "Id", "Score").From("user").Limit(1)
fmt.Println("GetFirstUser", m.BuildSelectSQL().BuildedSQL())
var (
val interface{}
row *sql.Row
)
if row, err = m.GetRow(); err == nil {
if val, err = m.ScanRowType(row, (*UserTb)(nil)); err == nil {
u, _ := val.(*UserTb)
user = &User{
Name: lib.Iif(u.Name.Valid, u.Nae.String, "").(string),
Id: u.Id,
Score: u.Score,
}
}
}
row, val = nil, nil
return
}