首页 > 后端开发 > Golang > Gorm 在自定义字符串类型上返回扫描仪错误

Gorm 在自定义字符串类型上返回扫描仪错误

王林
发布: 2024-02-11 18:21:21
转载
1163 人浏览过

Gorm 在自定义字符串类型上返回扫描仪错误

php小编香蕉介绍:在Gorm中,当你尝试在自定义的字符串类型上进行扫描操作时,可能会遇到错误。这个问题可能会导致扫描仪无法正确解析字符串,从而导致程序出错。这是因为Gorm默认使用`Scan`方法来扫描字符串类型的字段,但对于自定义的字符串类型,`Scan`方法可能无法正确处理。解决这个问题的方法是使用`Value`方法来手动解析字符串,确保程序能够正确执行。通过这种方式,你可以避免在使用Gorm时遇到关于扫描仪错误的问题。

问题内容

我编写了以下实体:

type datacategory string

const (
    datacategory1 datacategory = "category1"
    datacategory2 datacategory = "category2"
)

type data struct {
    name            string         `json:"name"`
    categories      []datacategory `json:"category" gorm:"type:text[]"`
}
登录后复制

我在数据库中手动创建了一行,并用 category1 和 category2 填充了类别数组类型。

但是读取数据时会出现以下错误:

sql: scan error on column index 19, name "category": unsupported scan, storing driver.value type string into type *[]datacategory
登录后复制

示例值方法:

func (s DataCategorySlice) Value() (driver.Value, error) {
    if s == nil {
        return nil, nil
    }
    if len(s) == 0 {
        return "{}", nil
    }

    v := []string{}
    for _, dc := range s {
        v = append(v, string(dc))
    }
    result := fmt.Sprintf("{%s}", strings.Join(v, ","))
    return result, nil
}
登录后复制

解决方法

以下示例假设您使用 postgresql 作为 rdbms,并且 datacategory 值不包含逗号或未转义的单引号。

// declare the custom type
type datacategoryslice []datacategory
登录后复制
// implement driver.valuer to encode the value into the
// correct format that is accepted by the target rdbms
func (s datacategoryslice) value() (driver.value, error) {
    if s == nil {
        return nil, nil
    }
    if len(s) == 0 {
        return []byte(`{}`), nil
    }

    v := []byte(`{`)
    for i := range s {
        v = append(v, s[i]...)
        v = append(v, ',')
    }
    v[len(v)-1] = '}' // replace last comma with closing brace
    return v, nil
}
登录后复制
// implement scanner to decode the raw source
// value as retrieved from the database
func (s *datacategoryslice) scan(src any) error {
    var data []byte
    switch v := src.(type) {
    case []byte:
        data = v
    case string:
        data = []byte(v)
    case nil:
        return nil
    default:
        return fmt.errorf("unsupported type: %t", src)
    }
    if len(data) == 0 {
        return nil
    }
    data = data[1:len(data)-1] // remove surrounding braces
    for _, v := range bytes.split(data, []byte{','}) {
        *s = append(*s, datacategory(v))
    }
    return nil
}
登录后复制

或者,如果从 value() (driver.value, error) 返回 <code>[]byte 不起作用,例如它会导致 “错误:格式错误的数组文字:<hex 字段值的表示形式> (sqlstate 22p02)”,那么您可以尝试使用 string 作为返回类型。

来自@kozhioyrin的示例

// implement driver.Valuer to encode the value into the
// correct format that is accepted by the target RDBMS
func (s DataCategorySlice) Value() (driver.Value, error) {
    if s == nil {
        return nil, nil
    }
    if len(s) == 0 {
        return "{}", nil
    }

    v := []string{}
    for _, dc := range s {
        v = append(v, string(dc))
    }
    result := fmt.Sprintf("{%s}", strings.Join(v, ","))
    return result, nil
}
登录后复制

以上是Gorm 在自定义字符串类型上返回扫描仪错误的详细内容。更多信息请关注PHP中文网其他相关文章!

相关标签:
来源:stackoverflow.com
本站声明
本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn
最新问题
热门教程
更多>
最新下载
更多>
网站特效
网站源码
网站素材
前端模板