Gorm-高级特性-函数示例

package main

import (
    "fmt"
    "time"

    "gorm.io/driver/mysql"
    "gorm.io/gorm"
)

type User struct {
    ID   uint
    Name string
    Age  uint
}

func (u *User) CalculateAge() uint {
    return time.Now().Year() - u.Age
}

func main() {
    dsn := "root:password@tcp(127.0.0.1:3306)/test?charset=utf8mb4&parseTime=True&loc=Local"
    db, err := gorm.Open(mysql.Open(dsn), &gorm.Config{})
    if err != nil {
        panic(err)
    }

    // 自动迁移模型
    db.AutoMigrate(&User{})

    // 创建新用户
    user := User{Name: "Tom", Age: 2000}
    db.Create(&user)

    // 查询所有用户的年龄
    var users []User
    db.Find(&users).Select("id, name, age, calculate_age() as age_now")
    fmt.Println(users)

    // 查询年龄大于等于18岁的用户
    db.Where("calculate_age() >= ?", 18).Find(&users)
    fmt.Println(users)

    // 使用原始SQL语句查询年龄
    var age uint
    db.Raw("SELECT calculate_age() FROM users WHERE id = ?", user.ID).Scan(&age)
    fmt.Println(age)
}

[{1 Tom 22}]
[{1 Tom 22}]
22