Как мне предварительно обработать строки, извлеченные с помощью sqlx? - PullRequest
Новая
       23

Как мне предварительно обработать строки, извлеченные с помощью sqlx?

0 голосов
/ 28 марта 2019

Рассмотрим следующий пример: 

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 Ответ

1 голос
/ 28 марта 2019

Хотя это не поддерживается "из коробки", вы можете создать собственную структуру, реализующую интерфейс сканера (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
}
...