Go语言自定义类型与数据库交互：使用Scan接口实现字符串自动大小写转换

type Scanner interface {
    Scan(src interface{}) error
}

type EmailAddress string

func (e *EmailAddress) Scan(src interface{}) error {
    // 实现逻辑
    return nil
}

func (e *EmailAddress) Scan(src interface{}) error {
    // 错误示例：直接断言为string
    *e = EmailAddress(strings.ToLower(src.(string)))
    return nil
}

import (
    "fmt"
    "strings"
)

type EmailAddress string

func (e *EmailAddress) Scan(src interface{}) error {
    if src == nil {
        *e = "" // 处理NULL值，将其视为空字符串
        return nil
    }

    // 尝试断言为[]byte
    b, ok := src.([]byte)
    if !ok {
        // 如果不是[]byte，尝试断言为string
        s, ok := src.(string)
        if !ok {
            // 如果既不是[]byte也不是string，则返回错误
            return fmt.Errorf("EmailAddress.Scan: expected []byte or string, got %T", src)
        }
        b = []byte(s) // 将string转换为[]byte
    }

    // 将字节切片转换为字符串，并转换为小写
    *e = EmailAddress(strings.ToLower(string(b)))
    return nil
}

go get github.com/lib/pq
go get github.com/kisielk/sqlstruct

CREATE TABLE users (
    id SERIAL PRIMARY KEY,
    mobile_phone VARCHAR(20),
    email VARCHAR(100)
);

INSERT INTO users (mobile_phone, email) VALUES
('1234567890', 'TestUser1@EXAMPLE.COM'),
('0987654321', 'AnotherUser@DOMAIN.ORG'),
('1122334455', NULL);

package main

import (
    "database/sql"
    "fmt"
    "log"
    "strings"

    "github.com/kisielk/sqlstruct"
    _ "github.com/lib/pq" // PostgreSQL驱动
)

// EmailAddress 自定义类型，用于存储电子邮件地址
type EmailAddress string

// Scan 实现了sql.Scanner接口，用于从数据库读取时自动转换小写
func (e *EmailAddress) Scan(src interface{}) error {
    if src == nil {
        *e = "" // 处理NULL值，将其视为空字符串
        return nil
    }

    // 尝试断言为[]byte
    b, ok := src.([]byte)
    if !ok {
        // 如果不是[]byte，尝试断言为string
        s, ok := src.(string)
        if !ok {
            // 如果既不是[]byte也不是string，则返回错误
            return fmt.Errorf("EmailAddress.Scan: expected []byte or string, got %T", src)
        }
        b = []byte(s) // 将string转换为[]byte
    }

    // 将字节切片转换为字符串，并转换为小写
    *e = EmailAddress(strings.ToLower(string(b)))
    return nil
}

// User 结构体，映射数据库中的users表
type User struct {
    Id          int
    MobilePhone string `sql:"mobile_phone"`
    Email       EmailAddress
}

func main() {
    // 替换为你的PostgreSQL连接字符串
    // 例如: "host=localhost port=5432 user=youruser password=yourpassword dbname=yourdb sslmode=disable"
    connStr := "host=localhost dbname=test sslmode=disable"
    db, err := sql.Open("postgres", connStr)
    if err != nil {
        log.Fatalf("无法连接到数据库: %v", err)
    }
    defer db.Close()

    // 确保数据库连接正常
    err = db.Ping()
    if err != nil {
        log.Fatalf("数据库连接失败: %v", err)
    }
    log.Println("成功连接到数据库！")

    // 查询数据，COALESCE(email, '') 用于将NULL email转换为''，避免Scan方法接收到nil
    rows, err := db.Query("SELECT id, mobile_phone, COALESCE(email, '') as email FROM users limit 5")
    if err != nil {
        log.Fatalf("查询失败: %v", err)
    }
    defer rows.Close()

    for rows.Next() {
        var user User
        // 使用sqlstruct.Scan将行数据映射到User结构体
        // sqlstruct会自动查找实现sql.Scanner接口的字段并调用其Scan方法
        err := sqlstruct.Scan(&user, rows)
        if err != nil {
            log.Printf("扫描行数据失败: %v", err)
            continue
        }
        log.Printf("%+v\n", user)
    }

    if err = rows.Err(); err != nil {
        log.Fatalf("遍历行时发生错误: %v", err)
    }
}

2023/10/27 10:00:00 成功连接到数据库！
2023/10/27 10:00:00 {Id:1 MobilePhone:1234567890 Email:testuser1@example.com}
2023/10/27 10:00:00 {Id:2 MobilePhone:0987654321 Email:anotheruser@domain.org}
2023/10/27 10:00:00 {Id:3 MobilePhone:1122334455 Email:}