数据关联手机号查询

This commit is contained in:
2026-03-20 14:37:41 +08:00
parent bb895eccdf
commit 71f1c6f141
10 changed files with 417 additions and 8 deletions

View File

@@ -47,6 +47,76 @@ func newTrade(db *gorm.DB, opts ...gen.DOOption) trade {
_trade.PaymentURL = field.NewString(tableName, "payment_url")
_trade.CompletedAt = field.NewTime(tableName, "completed_at")
_trade.CanceledAt = field.NewTime(tableName, "canceled_at")
_trade.User = tradeBelongsToUser{
db: db.Session(&gorm.Session{}),
RelationField: field.NewRelation("User", "models.User"),
Admin: struct {
field.RelationField
Roles struct {
field.RelationField
Permissions struct {
field.RelationField
Parent struct {
field.RelationField
}
Children struct {
field.RelationField
}
}
}
}{
RelationField: field.NewRelation("User.Admin", "models.Admin"),
Roles: struct {
field.RelationField
Permissions struct {
field.RelationField
Parent struct {
field.RelationField
}
Children struct {
field.RelationField
}
}
}{
RelationField: field.NewRelation("User.Admin.Roles", "models.AdminRole"),
Permissions: struct {
field.RelationField
Parent struct {
field.RelationField
}
Children struct {
field.RelationField
}
}{
RelationField: field.NewRelation("User.Admin.Roles.Permissions", "models.Permission"),
Parent: struct {
field.RelationField
}{
RelationField: field.NewRelation("User.Admin.Roles.Permissions.Parent", "models.Permission"),
},
Children: struct {
field.RelationField
}{
RelationField: field.NewRelation("User.Admin.Roles.Permissions.Children", "models.Permission"),
},
},
},
},
Roles: struct {
field.RelationField
Permissions struct {
field.RelationField
}
}{
RelationField: field.NewRelation("User.Roles", "models.UserRole"),
Permissions: struct {
field.RelationField
}{
RelationField: field.NewRelation("User.Roles.Permissions", "models.Permission"),
},
},
}
_trade.fillFieldMap()
@@ -77,6 +147,7 @@ type trade struct {
PaymentURL field.String
CompletedAt field.Time
CanceledAt field.Time
User tradeBelongsToUser
fieldMap map[string]field.Expr
}
@@ -129,7 +200,7 @@ func (t *trade) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
}
func (t *trade) fillFieldMap() {
t.fieldMap = make(map[string]field.Expr, 20)
t.fieldMap = make(map[string]field.Expr, 21)
t.fieldMap["id"] = t.ID
t.fieldMap["created_at"] = t.CreatedAt
t.fieldMap["updated_at"] = t.UpdatedAt
@@ -150,18 +221,125 @@ func (t *trade) fillFieldMap() {
t.fieldMap["payment_url"] = t.PaymentURL
t.fieldMap["completed_at"] = t.CompletedAt
t.fieldMap["canceled_at"] = t.CanceledAt
}
func (t trade) clone(db *gorm.DB) trade {
t.tradeDo.ReplaceConnPool(db.Statement.ConnPool)
t.User.db = db.Session(&gorm.Session{Initialized: true})
t.User.db.Statement.ConnPool = db.Statement.ConnPool
return t
}
func (t trade) replaceDB(db *gorm.DB) trade {
t.tradeDo.ReplaceDB(db)
t.User.db = db.Session(&gorm.Session{})
return t
}
type tradeBelongsToUser struct {
db *gorm.DB
field.RelationField
Admin struct {
field.RelationField
Roles struct {
field.RelationField
Permissions struct {
field.RelationField
Parent struct {
field.RelationField
}
Children struct {
field.RelationField
}
}
}
}
Roles struct {
field.RelationField
Permissions struct {
field.RelationField
}
}
}
func (a tradeBelongsToUser) Where(conds ...field.Expr) *tradeBelongsToUser {
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 tradeBelongsToUser) WithContext(ctx context.Context) *tradeBelongsToUser {
a.db = a.db.WithContext(ctx)
return &a
}
func (a tradeBelongsToUser) Session(session *gorm.Session) *tradeBelongsToUser {
a.db = a.db.Session(session)
return &a
}
func (a tradeBelongsToUser) Model(m *models.Trade) *tradeBelongsToUserTx {
return &tradeBelongsToUserTx{a.db.Model(m).Association(a.Name())}
}
func (a tradeBelongsToUser) Unscoped() *tradeBelongsToUser {
a.db = a.db.Unscoped()
return &a
}
type tradeBelongsToUserTx struct{ tx *gorm.Association }
func (a tradeBelongsToUserTx) Find() (result *models.User, err error) {
return result, a.tx.Find(&result)
}
func (a tradeBelongsToUserTx) Append(values ...*models.User) (err error) {
targetValues := make([]interface{}, len(values))
for i, v := range values {
targetValues[i] = v
}
return a.tx.Append(targetValues...)
}
func (a tradeBelongsToUserTx) Replace(values ...*models.User) (err error) {
targetValues := make([]interface{}, len(values))
for i, v := range values {
targetValues[i] = v
}
return a.tx.Replace(targetValues...)
}
func (a tradeBelongsToUserTx) Delete(values ...*models.User) (err error) {
targetValues := make([]interface{}, len(values))
for i, v := range values {
targetValues[i] = v
}
return a.tx.Delete(targetValues...)
}
func (a tradeBelongsToUserTx) Clear() error {
return a.tx.Clear()
}
func (a tradeBelongsToUserTx) Count() int64 {
return a.tx.Count()
}
func (a tradeBelongsToUserTx) Unscoped() *tradeBelongsToUserTx {
a.tx = a.tx.Unscoped()
return &a
}
type tradeDo struct{ gen.DO }
func (t tradeDo) Debug() *tradeDo {