Files
platform/cmd/gen/main.go

73 lines
1.3 KiB
Go

package main
import (
m "platform/web/models"
"gorm.io/driver/postgres"
"gorm.io/gen"
"gorm.io/gorm"
"gorm.io/gorm/schema"
)
var g *gen.Generator
func main() {
// 初始化
db, _ := gorm.Open(
postgres.Open("host=localhost user=dev password=dev dbname=app port=5432 sslmode=disable TimeZone=Asia/Shanghai"),
&gorm.Config{
NamingStrategy: schema.NamingStrategy{
SingularTable: true,
},
},
)
g = gen.NewGenerator(gen.Config{
OutPath: "web/queries",
FieldNullable: true,
FieldSignable: true,
FieldWithTypeTag: true,
Mode: gen.WithDefaultQuery | gen.WithoutContext,
})
g.UseDB(db)
g.ApplyBasic(
m.Admin{},
m.AdminRole{},
m.Announcement{},
m.Bill{},
m.Channel{},
m.Client{},
m.Coupon{},
m.Edge{},
m.LinkAdminRole{},
m.LinkAdminRolePermission{},
m.LinkClientPermission{},
m.LinkUserRole{},
m.LinkUserRolePermission{},
m.LogsLogin{},
m.LogsRequest{},
m.LogsUserBandwidth{},
m.LogsUserUsage{},
m.Permission{},
m.Product{},
m.ProductSku{},
m.ProductSkuUser{},
m.Proxy{},
m.Refund{},
m.Resource{},
m.ResourceLong{},
m.ResourceShort{},
m.Session{},
m.Trade{},
m.User{},
m.UserRole{},
m.Whitelist{},
m.Inquiry{},
m.ProductDiscount{},
m.BalanceActivity{},
m.CouponUser{},
)
g.Execute()
}