feat add tenant users

This commit is contained in:
2025-12-16 11:20:40 +08:00
parent 28ab17324d
commit d058b7ffda
17 changed files with 1780 additions and 32 deletions

View File

@@ -16,34 +16,44 @@ import (
)
var (
Q = new(Query)
UserQuery *userQuery
Q = new(Query)
TenantQuery *tenantQuery
TenantUserQuery *tenantUserQuery
UserQuery *userQuery
)
func SetDefault(db *gorm.DB, opts ...gen.DOOption) {
*Q = *Use(db, opts...)
TenantQuery = &Q.Tenant
TenantUserQuery = &Q.TenantUser
UserQuery = &Q.User
}
func Use(db *gorm.DB, opts ...gen.DOOption) *Query {
return &Query{
db: db,
User: newUser(db, opts...),
db: db,
Tenant: newTenant(db, opts...),
TenantUser: newTenantUser(db, opts...),
User: newUser(db, opts...),
}
}
type Query struct {
db *gorm.DB
User userQuery
Tenant tenantQuery
TenantUser tenantUserQuery
User userQuery
}
func (q *Query) Available() bool { return q.db != nil }
func (q *Query) clone(db *gorm.DB) *Query {
return &Query{
db: db,
User: q.User.clone(db),
db: db,
Tenant: q.Tenant.clone(db),
TenantUser: q.TenantUser.clone(db),
User: q.User.clone(db),
}
}
@@ -57,18 +67,24 @@ func (q *Query) WriteDB() *Query {
func (q *Query) ReplaceDB(db *gorm.DB) *Query {
return &Query{
db: db,
User: q.User.replaceDB(db),
db: db,
Tenant: q.Tenant.replaceDB(db),
TenantUser: q.TenantUser.replaceDB(db),
User: q.User.replaceDB(db),
}
}
type queryCtx struct {
User *userQueryDo
Tenant *tenantQueryDo
TenantUser *tenantUserQueryDo
User *userQueryDo
}
func (q *Query) WithContext(ctx context.Context) *queryCtx {
return &queryCtx{
User: q.User.WithContext(ctx),
Tenant: q.Tenant.WithContext(ctx),
TenantUser: q.TenantUser.WithContext(ctx),
User: q.User.WithContext(ctx),
}
}