Gorm-自定义查询（二）

package main

import (
    "fmt"

    "gorm.io/driver/sqlite"
    "gorm.io/gorm"
)

type User struct {
    ID   uint
    Name string
}

func (u User) FindByName(db *gorm.DB, name string) *gorm.DB {
    return db.Where("name = ?", name)
}

func main() {
    dsn := "test.db"
    db, err := gorm.Open(sqlite.Open(dsn), &gorm.Config{})
    if err != nil {
        panic("failed to connect database")
    }
    db.AutoMigrate(&User{})

    // Create some test data
    db.Create(&User{Name: "Tom"})
    db.Create(&User{Name: "Jerry"})
    db.Create(&User{Name: "Mike"})
    db.Create(&User{Name: "Tom"})

    // Use custom query method
    var users []User
    db.Model(User{}).FindByName("Tom").Find(&users)

    fmt.Println(users)
}