首页 > 后端开发 > Golang > 实现database/sql.Scanner接口

实现database/sql.Scanner接口

王林
发布: 2024-02-10 13:30:08
转载
628 人浏览过

实现database/sql.Scanner接口

php小编鱼仔在这篇文章中将向大家介绍如何实现database/sql.Scanner接口。在Go语言中,database/sql包是用于操作关系型数据库的核心包。Scanner接口用于将数据库查询结果中的值扫描到Go语言的变量中。通过实现Scanner接口,我们可以自定义将数据库查询结果中的值转换为我们想要的类型。本文将详细讲解如何实现Scanner接口,帮助读者更好地理解和应用Go语言中的数据库操作。

问题内容

如何实现database/sql.Scanner接口?

在此查询中,SELECT 子句中有 3 个字段:

  • idsmallint 无符号
  • is_suspended tinyint 无符号
  • name varchar

database/sql中,3列的数据类型是:

  • int64
  • int64
  • []uint8

这适用于 []接口{},但希望将每个列类型直接实现到 []接口{},但希望将每个列类型直接实现到 database/sql.Scanner 接口

cols    := make([]interface{}, 3)
ptr     := make([]interface{}, 3)

for i, _ := range cols {
    ptr[i] = &cols[i]
}

if err := rows.Scan(ptr...); err != nil {
    fmt.Println("err:", err)
}

// pair column data with column name
res := map[string]any
for i, name := range res_cols {
    res[name] = *ptr[i].(*any)
    
    fmt.Printf("Type: %T %s\n", res[name], name)
}
登录后复制

我尝试做的事情,但无法真正使其发挥作用

类型

type Type_int int

func (t *Type_int) Scan(value interface{}) error {
    switch value := value.(type) {
    case int64:
        *t = Type_int(value)
    default:
        return fmt.Errorf("Invalid database type: %T %v", value, value)
    }
    return nil
}

type Type_string string

func (t *Type_string) Scan(value interface{}) error {
    switch value := value.(type) {
    case []uint8:
        *t = Type_string(value)
    default:
        return fmt.Errorf("Invalid database type: %T %v", value, value)
    }
    return nil
}
登录后复制

代码

ptr     := make([]interface{}, 3)

cols    := []interface{}{
    Type_int,
    Type_int,
    Type_string,
}

for i, _ := range cols {
    ptr[i] = &cols[i]
}

if err := rows.Scan(ptr...); err != nil {
    fmt.Println("err:", err)
}

// pair column data with column name
res := map[string]any
for i, name := range res_cols {
    res[name] = *ptr[i].(*any)
    
    fmt.Printf("Type: %T %s\n", res[name], name)
}
登录后复制

解决方法

使用指向给定类型值的指针初始化 ptr

var c1 Type_int
var c2 Type_int
var c3 Type_string
ptr := []any{&c1, &c2, &c3}
if err := rows.Scan(ptr...); err != nil {
    fmt.Println("err:", err)
}
登录后复制

ptr 切片在上面的代码片段中没有提供任何值。代码可以简化为:

var c1 Type_int
var c2 Type_int
var c3 Type_string
if err := rows.Scan(&c1, &c2, &c3); err != nil {
    fmt.Println("err:", err)
}
登录后复制

以上是实现database/sql.Scanner接口的详细内容。更多信息请关注PHP中文网其他相关文章!

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