Go语言中处理数据库NULL值的标准方案

CREATE TABLE users (
    id INT PRIMARY KEY,
    name VARCHAR(100) NOT NULL,
    age INT NULL -- 允许为 NULL
);

var age int
err := row.Scan(&age) // 如果 age 为 NULL，这里会报错！

// 源码节选
type NullString struct {
    String string
    Valid  bool // Valid 为 true 时，String 才包含有效数据
}

package main

import (
    "database/sql"
    "fmt"
    "log"
    _ "github.com/go-sql-driver/mysql"
)

func main() {
    db, err := sql.Open("mysql", "user:password@/dbname")
    if err != nil {
        log.Fatal(err)
    }
    defer db.Close()

    var (
        id   int
        name string
        age  sql.NullInt64 // 使用 NullInt64 接收可能为 NULL 的 age
    )

    row := db.QueryRow("SELECT id, name, age FROM users WHERE id = ?", 1)
    err = row.Scan(&id, &name, &age)
    if err != nil {
        log.Fatal(err)
    }

    // 使用前必须检查 Valid
    if age.Valid {
        fmt.Printf("用户年龄: %d\n", age.Int64)
    } else {
        fmt.Println("用户年龄: (未设置)")
    }
}

// 插入一个年龄未知（NULL）的用户
newAge := sql.NullInt64{Valid: false} // Valid 为 false 表示 NULL
// 或者使用 Int64 的零值，但 Valid 为 false
// newAge := sql.NullInt64{}

result, err := db.Exec(
    "INSERT INTO users (name, age) VALUES (?, ?)",
    "张三",
    newAge, // 这里传递 sql.NullInt64
)
if err != nil {
    log.Fatal(err)
}

// 更新，将某个用户的年龄设置为 NULL
_, err = db.Exec(
    "UPDATE users SET age = ? WHERE id = ?",
    sql.NullInt64{}, // 等价于 sql.NullInt64{Valid: false}
    2,
)

func NewNullString(s string) sql.NullString {
    return sql.NullString{
        String: s,
        Valid:  s != "", // 根据业务逻辑定义“有效”条件
    }
}

func NewNullInt64(i int64) sql.NullInt64 {
    return sql.NullInt64{
        Int64: i,
        Valid: true,
    }
}

// 使用
age := NewNullInt64(25)
nullableName := NewNullString("") // Valid 将为 false

type User struct {
    ID   int             `json:"id"`
    Name string          `json:"name"`
    Age  *int64          `json:"age,omitempty"` // 使用指针，nil 对应 JSON null
}

// 从数据库扫描到结构体
row := db.QueryRow("SELECT id, name, age FROM users WHERE id = ?", 1)
var (
    id   int
    name string
    age  sql.NullInt64
)
row.Scan(&id, &name, &age)

user := User{
    ID:   id,
    Name: name,
}
if age.Valid {
    user.Age = &age.Int64 // 只有有效时才赋值指针
}
// user.Age 为 nil 时，JSON 输出中 age 字段会被忽略（omitempty）或为 null

// 业务逻辑
func formatAge(age sql.NullInt64) string {
    if !age.Valid {
        return "保密"
    }
    return fmt.Sprintf("%d岁", age.Int64)
}

// 模板中使用 (例如 html/template)
// {{if .Age.Valid}}{{.Age.Int64}}{{else}}未设置{{end}}

// 错误示例
avgAge := totalAge / userCount // 如果 totalAge 来自某个 SUM(age)，而 age 有 NULL，结果可能不对

var age *int64
err := row.Scan(&age)
if err != nil {
    log.Fatal(err)
}
if age != nil {
    fmt.Println(*age)
} else {
    fmt.Println("NULL")
}