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

问题描述 投票:0回答:1

请考虑以下示例:

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这样做的规范方法是什么?

mysql go sqlx
1个回答
1
投票

虽然不支持开箱即用,但您可以创建实现Scanner界面的自定义结构(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
}
© www.soinside.com 2019 - 2024. All rights reserved.