如何预处理使用 sqlx 获取的行?
How do I preprocess rows fetched with 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
中执行此操作的规范方法是什么?
虽然不支持开箱即用,但您可以创建实现 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
}
考虑以下示例:
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
中执行此操作的规范方法是什么?
虽然不支持开箱即用,但您可以创建实现 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
}