如何一次使用GORM读取一行SQLite数据库

时间:2018-08-24 07:27:07

标签: go go-gorm

我有以下代码

package main

import (
    "database/sql"
    "fmt"
    "log"

    _ "github.com/mattn/go-sqlite3"
)

func getDatabaseHandle(dbpath string) (*sql.DB, error) {
    database, err := sql.Open("sqlite3", dbpath)
    if err != nil {
        log.Printf("Failed to create the handle")
        return nil, err
    }
    if err = database.Ping(); err != nil {
        fmt.Printf("Failed to keep connection alive")
        return nil, err
    }
    return database, nil
}

func getAllRows(database *sql.DB, table string) {
    query := fmt.Sprintf("SELECT User, AppName FROM %s LIMIT 10", table)
    rows, err := database.Query(query)
    if err != nil {
        panic(err)
    }
    defer rows.Close()
    for rows.Next() {
        var id int
        var app string
        rows.Scan(&id, &app)
        fmt.Println(id, app)
    }
}

func main() {
    db, err := getDatabaseHandle("./gozo.db")
    if err != nil {
        panic(err)
    }
    defer db.Close()
    getAllRows(db, "timesheet")
}

这是针对具有以下列的SQLite数据库

id, User, Matter, AppName, AppDesc, Duration, Type, Timestamp

以上代码字完美。但是有两个问题

  1. 我需要为打算在for rows.Next()的for循环内使用的每一列声明变量,这很烦人并且无法轻松地模块化。

  2. 我需要使用ORM工具来实现数据库的可移植性。

因此我系上GORM,这是我的代码

type Timesheet struct {
    id        int
    User      int
    Matter    int
    AppName   string
    AppDesc   string
    Duration  int64
    Type      string
    Timestamp string
}

// TableName -- Sets the table name
func (ts Timesheet) TableName() string {
    return "timesheet"
}

func main() {
    db, err := gorm.Open("sqlite3", "./gozo.db")
    if err != nil {
        panic(err)
    }
    defer db.Close()
    var ts []Timesheet
    db.Find(&ts).Limit(5)
    fmt.Println(ts)
}

但这不能正确地给我数据,而是给我所有值0。而且,这没有使用每行扫描迭代,因此我可以在执行其他一些相关操作的同时包装go并发。这似乎正在提取所有数据,这也是错误的。请让我知道如何循环扫描每一行并使用GORM获取正确的数据。

1 个答案:

答案 0 :(得分:2)

我找到了答案。 GORM文档不是那么冗长,但是可能是答案。

package main

import (
    "fmt"

    "github.com/jinzhu/gorm"
    _ "github.com/jinzhu/gorm/dialects/sqlite"
)

// Product -- Represents a product
type Product struct {
    gorm.Model
    Code  string
    Price uint
}

// TableName setting the table name
func (Product) TableName() string {
    return "allProducts"
}

func main() {
    db, err := gorm.Open("sqlite3", "test.db")
    if err != nil {
        panic("failed to connect database")
    }
    defer db.Close()
    var product Product
    rows, err := db.Model(&Product{}).Rows()
    defer rows.Close()
    if err != nil {
        panic(err)
    }
    for rows.Next() {
        db.ScanRows(rows, &product)
        fmt.Println(product)
    }
}