platform/cmd/gen/main.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.Proxy{},
		m.Refund{},
		m.Resource{},
		m.ResourceLong{},
		m.ResourceShort{},
		m.Session{},
		m.Trade{},
		m.User{},
		m.UserRole{},
		m.Whitelist{},
		m.Inquiry{},
	)
	g.Execute()
}