如何在Gorp中使用TypeConverter？

Question

我想使用Gorp从包含专门类型的DB中加载和保存结构。除其他外，这对于枚举字符串(如角色)非常有用：

type Role string

type Account struct {
    User string
    Role Role
}

这“开箱而出”是行不通的。引发错误消息，如

panic: sql: converting Exec argument #0's type: unsupported type user.Role, a string

我怀疑我需要使用gorp.TypeConverter来解决这个问题，但是没有关于如何这样做的文档。

你能帮上忙吗？

Answer 1

估价师和扫描仪接口可以做您想做的事情。下面是一个有用的例子：

package roleGorp

import (
    "gopkg.in/gorp.v1"
    "github.com/DATA-DOG/go-sqlmock"
    "fmt"
    "testing"
    "database/sql/driver"
)

type Role string

func (r *Role) Scan(value interface{}) error { *r = Role(value.(string)); return nil }
func (r Role) Value() (driver.Value, error)  { return string(r), nil }

type Account struct {
    User string `db:"user"`
    Role Role `db:"role"`
}

func TestRoleGorp(t *testing.T) {
    db, err := sqlmock.New()
    if err != nil {
        panic(err)
    }
    dbMap := gorp.DbMap{
        Db: db,
        Dialect: gorp.MySQLDialect{
            Engine: "InnoDB",
        },
    }

    rows := sqlmock.NewRows([]string{"user", "role"}).AddRow("user1", "admin")

    sqlmock.ExpectQuery(`SELECT \* FROM account LIMIT 1`).WillReturnRows(rows)

    dbMap.AddTableWithName(Account{}, "account")

    result := &Account{}
    err = dbMap.SelectOne(result, "SELECT * FROM account LIMIT 1")
    if err != nil {
        panic(err)
    }

    fmt.Printf("%+v\n", *result)

    result2 := &Account{
        User: "user2",
        Role: Role("moderator"),
    }

    sqlmock.ExpectExec("insert into `account` \\(`user`,`role`\\) values \\(\\?,\\?\\);").WithArgs("user2", "moderator").WillReturnResult(sqlmock.NewResult(1, 1))

    err = dbMap.Insert(result2)
    if err != nil {
        panic(err)
    }
}