重构 orm 代码生成逻辑,实现 bill 接口,优化请求字段检查与 list total 查询逻辑

This commit is contained in:
2025-04-11 17:36:34 +08:00
parent c1b4f8d605
commit 636bdcdb87
36 changed files with 794 additions and 193 deletions

View File

@@ -33,6 +33,12 @@ func newResource(db *gorm.DB, opts ...gen.DOOption) resource {
_resource.CreatedAt = field.NewField(tableName, "created_at")
_resource.UpdatedAt = field.NewField(tableName, "updated_at")
_resource.DeletedAt = field.NewField(tableName, "deleted_at")
_resource.ResourceNo = field.NewString(tableName, "resource_no")
_resource.Pss = resourceHasOnePss{
db: db.Session(&gorm.Session{}),
RelationField: field.NewRelation("Pss", "models.ResourcePss"),
}
_resource.fillFieldMap()
@@ -42,13 +48,15 @@ func newResource(db *gorm.DB, opts ...gen.DOOption) resource {
type resource struct {
resourceDo
ALL field.Asterisk
ID field.Int32 // 套餐ID
UserID field.Int32 // 用户ID
Active field.Bool // 套餐状态
CreatedAt field.Field // 创建时间
UpdatedAt field.Field // 更新时间
DeletedAt field.Field // 删除时间
ALL field.Asterisk
ID field.Int32 // 套餐ID
UserID field.Int32 // 用户ID
Active field.Bool // 套餐状态
CreatedAt field.Field // 创建时间
UpdatedAt field.Field // 更新时间
DeletedAt field.Field // 删除时间
ResourceNo field.String
Pss resourceHasOnePss
fieldMap map[string]field.Expr
}
@@ -71,6 +79,7 @@ func (r *resource) updateTableName(table string) *resource {
r.CreatedAt = field.NewField(table, "created_at")
r.UpdatedAt = field.NewField(table, "updated_at")
r.DeletedAt = field.NewField(table, "deleted_at")
r.ResourceNo = field.NewString(table, "resource_no")
r.fillFieldMap()
@@ -87,13 +96,15 @@ func (r *resource) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
}
func (r *resource) fillFieldMap() {
r.fieldMap = make(map[string]field.Expr, 6)
r.fieldMap = make(map[string]field.Expr, 8)
r.fieldMap["id"] = r.ID
r.fieldMap["user_id"] = r.UserID
r.fieldMap["active"] = r.Active
r.fieldMap["created_at"] = r.CreatedAt
r.fieldMap["updated_at"] = r.UpdatedAt
r.fieldMap["deleted_at"] = r.DeletedAt
r.fieldMap["resource_no"] = r.ResourceNo
}
func (r resource) clone(db *gorm.DB) resource {
@@ -106,6 +117,77 @@ func (r resource) replaceDB(db *gorm.DB) resource {
return r
}
type resourceHasOnePss struct {
db *gorm.DB
field.RelationField
}
func (a resourceHasOnePss) Where(conds ...field.Expr) *resourceHasOnePss {
if len(conds) == 0 {
return &a
}
exprs := make([]clause.Expression, 0, len(conds))
for _, cond := range conds {
exprs = append(exprs, cond.BeCond().(clause.Expression))
}
a.db = a.db.Clauses(clause.Where{Exprs: exprs})
return &a
}
func (a resourceHasOnePss) WithContext(ctx context.Context) *resourceHasOnePss {
a.db = a.db.WithContext(ctx)
return &a
}
func (a resourceHasOnePss) Session(session *gorm.Session) *resourceHasOnePss {
a.db = a.db.Session(session)
return &a
}
func (a resourceHasOnePss) Model(m *models.Resource) *resourceHasOnePssTx {
return &resourceHasOnePssTx{a.db.Model(m).Association(a.Name())}
}
type resourceHasOnePssTx struct{ tx *gorm.Association }
func (a resourceHasOnePssTx) Find() (result *models.ResourcePss, err error) {
return result, a.tx.Find(&result)
}
func (a resourceHasOnePssTx) Append(values ...*models.ResourcePss) (err error) {
targetValues := make([]interface{}, len(values))
for i, v := range values {
targetValues[i] = v
}
return a.tx.Append(targetValues...)
}
func (a resourceHasOnePssTx) Replace(values ...*models.ResourcePss) (err error) {
targetValues := make([]interface{}, len(values))
for i, v := range values {
targetValues[i] = v
}
return a.tx.Replace(targetValues...)
}
func (a resourceHasOnePssTx) Delete(values ...*models.ResourcePss) (err error) {
targetValues := make([]interface{}, len(values))
for i, v := range values {
targetValues[i] = v
}
return a.tx.Delete(targetValues...)
}
func (a resourceHasOnePssTx) Clear() error {
return a.tx.Clear()
}
func (a resourceHasOnePssTx) Count() int64 {
return a.tx.Count()
}
type resourceDo struct{ gen.DO }
func (r resourceDo) Debug() *resourceDo {