rts-sim-testing-service/db/generator/main.go

38 lines
985 B
Go

package main
import (
"gorm.io/driver/mysql"
"gorm.io/gen"
"gorm.io/gorm"
)
// Dynamic SQL
type Querier interface {
// SELECT * FROM @@table WHERE name = @name{{if role !=""}} AND role = @role{{end}}
FilterWithNameAndRole(name, role string) ([]gen.T, error)
}
func main() {
g := gen.NewGenerator(gen.Config{
OutPath: "../dbquery",
Mode: gen.WithoutContext | gen.WithDefaultQuery | gen.WithQueryInterface, // generate mode
})
dsn := "root:joylink0503@tcp(192.168.3.233:3306)/bj-rtss?charset=utf8mb4&parseTime=true&loc=Local"
gormdb, err := gorm.Open(mysql.Open(dsn))
if err != nil {
panic(err)
}
g.UseDB(gormdb) // reuse your gorm db
// Generate basic type-safe DAO API for struct `model.User` following conventions
g.ApplyBasic(g.GenerateAllTable()...)
// Generate Type Safe API with Dynamic SQL defined on Querier interface for `model.User` and `model.Company`
// g.ApplyInterface(func(Querier) {}, entity.User{})
// Generate the code
g.Execute()
}