如何预处理用sqlx提取的行?

时间:2019-03-28 11:56:56

标签: mysql go sqlx

考虑以下示例:

package main

import (
    "fmt"
    "github.com/jmoiron/sqlx"
    _ "github.com/go-sql-driver/mysql"
)

type Data struct {
    Stuff string
}

func main() {
    db, _ := sqlx.Connect("mysql", "root:root@(localhost:3306)/data")
    var datas []Data
    db.Select(&datas, "select 'a,b' stuff from data limit 10")
    fmt.Println(datas)
}

我想做的是将Stuff设为[]string,在其中将,拆分。我想我可以添加一个额外的[]string字段并遍历填充该字段并删除源数据的结果,但这似乎效率很低。在sqlx中执行此操作的规范方法是什么?

1 个答案:

答案 0 :(得分:1)

虽然不支持此功能,但是您可以创建实现扫描程序界面(https://golang.org/pkg/database/sql/#Scanner)的自定义结构:

type StringList []string

// Scan implements Scanner interface and parses 
// database result to StringList struct.
func (s *StringList) Scan(src interface{}) error {
    var source string

    switch src.(type) {
    case string:
        source = src.(string)
    case []byte:
        source = string(src.([]byte))
    default:
        return errors.New("Incompatible type for StringList")
    }

    *s = strings.Split(source, ",")
    for i := range *s {
        (*s)[i] = strings.TrimSpace((*s)[i])
    }

    return nil
}

并使用此结构代替string

type Data struct {
    Stuff StringList
}