rogee/quyun-v2
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

feat add tenant users

Browse Source
This commit is contained in:
Rogee
2025-12-16 11:20:40 +08:00
parent 28ab17324d
commit d058b7ffda
17 changed files with 1780 additions and 32 deletions
Download Patch File Download Diff File Expand all files Collapse all files

38
backend/database/models/query.gen.go
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),
}
}

63
backend/database/models/tenant_users.gen.go Normal file
View File

@@ -0,0 +1,63 @@
// Code generated by go.ipao.vip/gen. DO NOT EDIT.
// Code generated by go.ipao.vip/gen. DO NOT EDIT.
// Code generated by go.ipao.vip/gen. DO NOT EDIT.
package models
import (
"context"
"time"
"quyun/v2/pkg/consts"
"go.ipao.vip/gen"
"go.ipao.vip/gen/types"
)
const TableNameTenantUser = "tenant_users"
// TenantUser mapped from table <tenant_users>
type TenantUser struct {
ID int64 `gorm:"column:id;type:bigint;primaryKey;autoIncrement:true" json:"id"`
TenantID int64 `gorm:"column:tenant_id;type:bigint;not null" json:"tenant_id"`
UserID int64 `gorm:"column:user_id;type:bigint;not null" json:"user_id"`
Role types.Array[consts.TenantUserRole] `gorm:"column:role;type:text[];not null;default:ARRAY['member" json:"role"`
Balance float64 `gorm:"column:balance;type:numeric(20,8);not null" json:"balance"`
Status consts.UserStatus `gorm:"column:status;type:character varying(50);not null;default:active" json:"status"`
CreatedAt time.Time `gorm:"column:created_at;type:timestamp with time zone;not null;default:now()" json:"created_at"`
UpdatedAt time.Time `gorm:"column:updated_at;type:timestamp with time zone;not null;default:now()" json:"updated_at"`
}
// Quick operations without importing query package
// Update applies changed fields to the database using the default DB.
func (m *TenantUser) Update(ctx context.Context) (gen.ResultInfo, error) {
return Q.TenantUser.WithContext(ctx).Updates(m)
}
// Save upserts the model using the default DB.
func (m *TenantUser) Save(ctx context.Context) error { return Q.TenantUser.WithContext(ctx).Save(m) }
// Create inserts the model using the default DB.
func (m *TenantUser) Create(ctx context.Context) error {
return Q.TenantUser.WithContext(ctx).Create(m)
}
// Delete removes the row represented by the model using the default DB.
func (m *TenantUser) Delete(ctx context.Context) (gen.ResultInfo, error) {
return Q.TenantUser.WithContext(ctx).Delete(m)
}
// ForceDelete permanently deletes the row (ignores soft delete) using the default DB.
func (m *TenantUser) ForceDelete(ctx context.Context) (gen.ResultInfo, error) {
return Q.TenantUser.WithContext(ctx).Unscoped().Delete(m)
}
// Reload reloads the model from database by its primary key and overwrites current fields.
func (m *TenantUser) Reload(ctx context.Context) error {
fresh, err := Q.TenantUser.WithContext(ctx).GetByID(m.ID)
if err != nil {
return err
}
*m = *fresh
return nil
}

489
backend/database/models/tenant_users.query.gen.go Normal file
View File

@@ -0,0 +1,489 @@
// Code generated by go.ipao.vip/gen. DO NOT EDIT.
// Code generated by go.ipao.vip/gen. DO NOT EDIT.
// Code generated by go.ipao.vip/gen. DO NOT EDIT.
package models
import (
"context"
"gorm.io/gorm"
"gorm.io/gorm/clause"
"gorm.io/gorm/schema"
"go.ipao.vip/gen"
"go.ipao.vip/gen/field"
"gorm.io/plugin/dbresolver"
)
func newTenantUser(db *gorm.DB, opts ...gen.DOOption) tenantUserQuery {
_tenantUserQuery := tenantUserQuery{}
_tenantUserQuery.tenantUserQueryDo.UseDB(db, opts...)
_tenantUserQuery.tenantUserQueryDo.UseModel(&TenantUser{})
tableName := _tenantUserQuery.tenantUserQueryDo.TableName()
_tenantUserQuery.ALL = field.NewAsterisk(tableName)
_tenantUserQuery.ID = field.NewInt64(tableName, "id")
_tenantUserQuery.TenantID = field.NewInt64(tableName, "tenant_id")
_tenantUserQuery.UserID = field.NewInt64(tableName, "user_id")
_tenantUserQuery.Role = field.NewArray(tableName, "role")
_tenantUserQuery.Balance = field.NewFloat64(tableName, "balance")
_tenantUserQuery.Status = field.NewField(tableName, "status")
_tenantUserQuery.CreatedAt = field.NewTime(tableName, "created_at")
_tenantUserQuery.UpdatedAt = field.NewTime(tableName, "updated_at")
_tenantUserQuery.fillFieldMap()
return _tenantUserQuery
}
type tenantUserQuery struct {
tenantUserQueryDo tenantUserQueryDo
ALL field.Asterisk
ID field.Int64
TenantID field.Int64
UserID field.Int64
Role field.Array
Balance field.Float64
Status field.Field
CreatedAt field.Time
UpdatedAt field.Time
fieldMap map[string]field.Expr
}
func (t tenantUserQuery) Table(newTableName string) *tenantUserQuery {
t.tenantUserQueryDo.UseTable(newTableName)
return t.updateTableName(newTableName)
}
func (t tenantUserQuery) As(alias string) *tenantUserQuery {
t.tenantUserQueryDo.DO = *(t.tenantUserQueryDo.As(alias).(*gen.DO))
return t.updateTableName(alias)
}
func (t *tenantUserQuery) updateTableName(table string) *tenantUserQuery {
t.ALL = field.NewAsterisk(table)
t.ID = field.NewInt64(table, "id")
t.TenantID = field.NewInt64(table, "tenant_id")
t.UserID = field.NewInt64(table, "user_id")
t.Role = field.NewArray(table, "role")
t.Balance = field.NewFloat64(table, "balance")
t.Status = field.NewField(table, "status")
t.CreatedAt = field.NewTime(table, "created_at")
t.UpdatedAt = field.NewTime(table, "updated_at")
t.fillFieldMap()
return t
}
func (t *tenantUserQuery) QueryContext(ctx context.Context) (*tenantUserQuery, *tenantUserQueryDo) {
return t, t.tenantUserQueryDo.WithContext(ctx)
}
func (t *tenantUserQuery) WithContext(ctx context.Context) *tenantUserQueryDo {
return t.tenantUserQueryDo.WithContext(ctx)
}
func (t tenantUserQuery) TableName() string { return t.tenantUserQueryDo.TableName() }
func (t tenantUserQuery) Alias() string { return t.tenantUserQueryDo.Alias() }
func (t tenantUserQuery) Columns(cols ...field.Expr) gen.Columns {
return t.tenantUserQueryDo.Columns(cols...)
}
func (t *tenantUserQuery) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
_f, ok := t.fieldMap[fieldName]
if !ok || _f == nil {
return nil, false
}
_oe, ok := _f.(field.OrderExpr)
return _oe, ok
}
func (t *tenantUserQuery) fillFieldMap() {
t.fieldMap = make(map[string]field.Expr, 8)
t.fieldMap["id"] = t.ID
t.fieldMap["tenant_id"] = t.TenantID
t.fieldMap["user_id"] = t.UserID
t.fieldMap["role"] = t.Role
t.fieldMap["balance"] = t.Balance
t.fieldMap["status"] = t.Status
t.fieldMap["created_at"] = t.CreatedAt
t.fieldMap["updated_at"] = t.UpdatedAt
}
func (t tenantUserQuery) clone(db *gorm.DB) tenantUserQuery {
t.tenantUserQueryDo.ReplaceConnPool(db.Statement.ConnPool)
return t
}
func (t tenantUserQuery) replaceDB(db *gorm.DB) tenantUserQuery {
t.tenantUserQueryDo.ReplaceDB(db)
return t
}
type tenantUserQueryDo struct{ gen.DO }
func (t tenantUserQueryDo) Debug() *tenantUserQueryDo {
return t.withDO(t.DO.Debug())
}
func (t tenantUserQueryDo) WithContext(ctx context.Context) *tenantUserQueryDo {
return t.withDO(t.DO.WithContext(ctx))
}
func (t tenantUserQueryDo) ReadDB() *tenantUserQueryDo {
return t.Clauses(dbresolver.Read)
}
func (t tenantUserQueryDo) WriteDB() *tenantUserQueryDo {
return t.Clauses(dbresolver.Write)
}
func (t tenantUserQueryDo) Session(config *gorm.Session) *tenantUserQueryDo {
return t.withDO(t.DO.Session(config))
}
func (t tenantUserQueryDo) Clauses(conds ...clause.Expression) *tenantUserQueryDo {
return t.withDO(t.DO.Clauses(conds...))
}
func (t tenantUserQueryDo) Returning(value interface{}, columns ...string) *tenantUserQueryDo {
return t.withDO(t.DO.Returning(value, columns...))
}
func (t tenantUserQueryDo) Not(conds ...gen.Condition) *tenantUserQueryDo {
return t.withDO(t.DO.Not(conds...))
}
func (t tenantUserQueryDo) Or(conds ...gen.Condition) *tenantUserQueryDo {
return t.withDO(t.DO.Or(conds...))
}
func (t tenantUserQueryDo) Select(conds ...field.Expr) *tenantUserQueryDo {
return t.withDO(t.DO.Select(conds...))
}
func (t tenantUserQueryDo) Where(conds ...gen.Condition) *tenantUserQueryDo {
return t.withDO(t.DO.Where(conds...))
}
func (t tenantUserQueryDo) Order(conds ...field.Expr) *tenantUserQueryDo {
return t.withDO(t.DO.Order(conds...))
}
func (t tenantUserQueryDo) Distinct(cols ...field.Expr) *tenantUserQueryDo {
return t.withDO(t.DO.Distinct(cols...))
}
func (t tenantUserQueryDo) Omit(cols ...field.Expr) *tenantUserQueryDo {
return t.withDO(t.DO.Omit(cols...))
}
func (t tenantUserQueryDo) Join(table schema.Tabler, on ...field.Expr) *tenantUserQueryDo {
return t.withDO(t.DO.Join(table, on...))
}
func (t tenantUserQueryDo) LeftJoin(table schema.Tabler, on ...field.Expr) *tenantUserQueryDo {
return t.withDO(t.DO.LeftJoin(table, on...))
}
func (t tenantUserQueryDo) RightJoin(table schema.Tabler, on ...field.Expr) *tenantUserQueryDo {
return t.withDO(t.DO.RightJoin(table, on...))
}
func (t tenantUserQueryDo) Group(cols ...field.Expr) *tenantUserQueryDo {
return t.withDO(t.DO.Group(cols...))
}
func (t tenantUserQueryDo) Having(conds ...gen.Condition) *tenantUserQueryDo {
return t.withDO(t.DO.Having(conds...))
}
func (t tenantUserQueryDo) Limit(limit int) *tenantUserQueryDo {
return t.withDO(t.DO.Limit(limit))
}
func (t tenantUserQueryDo) Offset(offset int) *tenantUserQueryDo {
return t.withDO(t.DO.Offset(offset))
}
func (t tenantUserQueryDo) Scopes(funcs ...func(gen.Dao) gen.Dao) *tenantUserQueryDo {
return t.withDO(t.DO.Scopes(funcs...))
}
func (t tenantUserQueryDo) Unscoped() *tenantUserQueryDo {
return t.withDO(t.DO.Unscoped())
}
func (t tenantUserQueryDo) Create(values ...*TenantUser) error {
if len(values) == 0 {
return nil
}
return t.DO.Create(values)
}
func (t tenantUserQueryDo) CreateInBatches(values []*TenantUser, batchSize int) error {
return t.DO.CreateInBatches(values, batchSize)
}
// Save : !!! underlying implementation is different with GORM
// The method is equivalent to executing the statement: db.Clauses(clause.OnConflict{UpdateAll: true}).Create(values)
func (t tenantUserQueryDo) Save(values ...*TenantUser) error {
if len(values) == 0 {
return nil
}
return t.DO.Save(values)
}
func (t tenantUserQueryDo) First() (*TenantUser, error) {
if result, err := t.DO.First(); err != nil {
return nil, err
} else {
return result.(*TenantUser), nil
}
}
func (t tenantUserQueryDo) Take() (*TenantUser, error) {
if result, err := t.DO.Take(); err != nil {
return nil, err
} else {
return result.(*TenantUser), nil
}
}
func (t tenantUserQueryDo) Last() (*TenantUser, error) {
if result, err := t.DO.Last(); err != nil {
return nil, err
} else {
return result.(*TenantUser), nil
}
}
func (t tenantUserQueryDo) Find() ([]*TenantUser, error) {
result, err := t.DO.Find()
return result.([]*TenantUser), err
}
func (t tenantUserQueryDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*TenantUser, err error) {
buf := make([]*TenantUser, 0, batchSize)
err = t.DO.FindInBatches(&buf, batchSize, func(tx gen.Dao, batch int) error {
defer func() { results = append(results, buf...) }()
return fc(tx, batch)
})
return results, err
}
func (t tenantUserQueryDo) FindInBatches(result *[]*TenantUser, batchSize int, fc func(tx gen.Dao, batch int) error) error {
return t.DO.FindInBatches(result, batchSize, fc)
}
func (t tenantUserQueryDo) Attrs(attrs ...field.AssignExpr) *tenantUserQueryDo {
return t.withDO(t.DO.Attrs(attrs...))
}
func (t tenantUserQueryDo) Assign(attrs ...field.AssignExpr) *tenantUserQueryDo {
return t.withDO(t.DO.Assign(attrs...))
}
func (t tenantUserQueryDo) Joins(fields ...field.RelationField) *tenantUserQueryDo {
for _, _f := range fields {
t = *t.withDO(t.DO.Joins(_f))
}
return &t
}
func (t tenantUserQueryDo) Preload(fields ...field.RelationField) *tenantUserQueryDo {
for _, _f := range fields {
t = *t.withDO(t.DO.Preload(_f))
}
return &t
}
func (t tenantUserQueryDo) FirstOrInit() (*TenantUser, error) {
if result, err := t.DO.FirstOrInit(); err != nil {
return nil, err
} else {
return result.(*TenantUser), nil
}
}
func (t tenantUserQueryDo) FirstOrCreate() (*TenantUser, error) {
if result, err := t.DO.FirstOrCreate(); err != nil {
return nil, err
} else {
return result.(*TenantUser), nil
}
}
func (t tenantUserQueryDo) FindByPage(offset int, limit int) (result []*TenantUser, count int64, err error) {
result, err = t.Offset(offset).Limit(limit).Find()
if err != nil {
return
}
if size := len(result); 0 < limit && 0 < size && size < limit {
count = int64(size + offset)
return
}
count, err = t.Offset(-1).Limit(-1).Count()
return
}
func (t tenantUserQueryDo) ScanByPage(result interface{}, offset int, limit int) (count int64, err error) {
count, err = t.Count()
if err != nil {
return
}
err = t.Offset(offset).Limit(limit).Scan(result)
return
}
func (t tenantUserQueryDo) Scan(result interface{}) (err error) {
return t.DO.Scan(result)
}
func (t tenantUserQueryDo) Delete(models ...*TenantUser) (result gen.ResultInfo, err error) {
return t.DO.Delete(models)
}
// ForceDelete performs a permanent delete (ignores soft-delete) for current scope.
func (t tenantUserQueryDo) ForceDelete() (gen.ResultInfo, error) {
return t.Unscoped().Delete()
}
// Inc increases the given column by step for current scope.
func (t tenantUserQueryDo) Inc(column field.Expr, step int64) (gen.ResultInfo, error) {
// column = column + step
e := field.NewUnsafeFieldRaw("?+?", column.RawExpr(), step)
return t.DO.UpdateColumn(column, e)
}
// Dec decreases the given column by step for current scope.
func (t tenantUserQueryDo) Dec(column field.Expr, step int64) (gen.ResultInfo, error) {
// column = column - step
e := field.NewUnsafeFieldRaw("?-?", column.RawExpr(), step)
return t.DO.UpdateColumn(column, e)
}
// Sum returns SUM(column) for current scope.
func (t tenantUserQueryDo) Sum(column field.Expr) (float64, error) {
var _v float64
agg := field.NewUnsafeFieldRaw("SUM(?)", column.RawExpr())
if err := t.Select(agg).Scan(&_v); err != nil {
return 0, err
}
return _v, nil
}
// Avg returns AVG(column) for current scope.
func (t tenantUserQueryDo) Avg(column field.Expr) (float64, error) {
var _v float64
agg := field.NewUnsafeFieldRaw("AVG(?)", column.RawExpr())
if err := t.Select(agg).Scan(&_v); err != nil {
return 0, err
}
return _v, nil
}
// Min returns MIN(column) for current scope.
func (t tenantUserQueryDo) Min(column field.Expr) (float64, error) {
var _v float64
agg := field.NewUnsafeFieldRaw("MIN(?)", column.RawExpr())
if err := t.Select(agg).Scan(&_v); err != nil {
return 0, err
}
return _v, nil
}
// Max returns MAX(column) for current scope.
func (t tenantUserQueryDo) Max(column field.Expr) (float64, error) {
var _v float64
agg := field.NewUnsafeFieldRaw("MAX(?)", column.RawExpr())
if err := t.Select(agg).Scan(&_v); err != nil {
return 0, err
}
return _v, nil
}
// PluckMap returns a map[key]value for selected key/value expressions within current scope.
func (t tenantUserQueryDo) PluckMap(key, val field.Expr) (map[interface{}]interface{}, error) {
do := t.Select(key, val)
rows, err := do.DO.Rows()
if err != nil {
return nil, err
}
defer rows.Close()
mm := make(map[interface{}]interface{})
for rows.Next() {
var k interface{}
var v interface{}
if err := rows.Scan(&k, &v); err != nil {
return nil, err
}
mm[k] = v
}
return mm, rows.Err()
}
// Exists returns true if any record matches the given conditions.
func (t tenantUserQueryDo) Exists(conds ...gen.Condition) (bool, error) {
cnt, err := t.Where(conds...).Count()
if err != nil {
return false, err
}
return cnt > 0, nil
}
// PluckIDs returns all primary key values under current scope.
func (t tenantUserQueryDo) PluckIDs() ([]int64, error) {
ids := make([]int64, 0, 16)
pk := field.NewInt64(t.TableName(), "id")
if err := t.DO.Pluck(pk, &ids); err != nil {
return nil, err
}
return ids, nil
}
// GetByID finds a single record by primary key.
func (t tenantUserQueryDo) GetByID(id int64) (*TenantUser, error) {
pk := field.NewInt64(t.TableName(), "id")
return t.Where(pk.Eq(id)).First()
}
// GetByIDs finds records by primary key list.
func (t tenantUserQueryDo) GetByIDs(ids ...int64) ([]*TenantUser, error) {
if len(ids) == 0 {
return []*TenantUser{}, nil
}
pk := field.NewInt64(t.TableName(), "id")
return t.Where(pk.In(ids...)).Find()
}
// DeleteByID deletes records by primary key.
func (t tenantUserQueryDo) DeleteByID(id int64) (gen.ResultInfo, error) {
pk := field.NewInt64(t.TableName(), "id")
return t.Where(pk.Eq(id)).Delete()
}
// DeleteByIDs deletes records by a list of primary keys.
func (t tenantUserQueryDo) DeleteByIDs(ids ...int64) (gen.ResultInfo, error) {
if len(ids) == 0 {
return gen.ResultInfo{RowsAffected: 0, Error: nil}, nil
}
pk := field.NewInt64(t.TableName(), "id")
return t.Where(pk.In(ids...)).Delete()
}
func (t *tenantUserQueryDo) withDO(do gen.Dao) *tenantUserQueryDo {
t.DO = *do.(*gen.DO)
return t
}

63
backend/database/models/tenants.gen.go Normal file
View File

@@ -0,0 +1,63 @@
// Code generated by go.ipao.vip/gen. DO NOT EDIT.
// Code generated by go.ipao.vip/gen. DO NOT EDIT.
// Code generated by go.ipao.vip/gen. DO NOT EDIT.
package models
import (
"context"
"time"
"quyun/v2/pkg/consts"
"go.ipao.vip/gen"
"go.ipao.vip/gen/types"
)
const TableNameTenant = "tenants"
// Tenant mapped from table <tenants>
type Tenant struct {
ID int64 `gorm:"column:id;type:bigint;primaryKey;autoIncrement:true" json:"id"`
UserID int64 `gorm:"column:user_id;type:bigint;not null" json:"user_id"`
Code string `gorm:"column:code;type:character varying(64);not null" json:"code"`
UUID types.UUID `gorm:"column:uuid;type:uuid;not null" json:"uuid"`
Name string `gorm:"column:name;type:character varying(128);not null" json:"name"`
Status consts.TenantStatus `gorm:"column:status;type:character varying(64);not null" json:"status"`
Config types.JSON `gorm:"column:config;type:jsonb;not null;default:{}" json:"config"`
ExpiredAt time.Time `gorm:"column:expired_at;type:timestamp with time zone" json:"expired_at"`
CreatedAt time.Time `gorm:"column:created_at;type:timestamp with time zone;not null;default:now()" json:"created_at"`
UpdatedAt time.Time `gorm:"column:updated_at;type:timestamp with time zone;not null;default:now()" json:"updated_at"`
}
// Quick operations without importing query package
// Update applies changed fields to the database using the default DB.
func (m *Tenant) Update(ctx context.Context) (gen.ResultInfo, error) {
return Q.Tenant.WithContext(ctx).Updates(m)
}
// Save upserts the model using the default DB.
func (m *Tenant) Save(ctx context.Context) error { return Q.Tenant.WithContext(ctx).Save(m) }
// Create inserts the model using the default DB.
func (m *Tenant) Create(ctx context.Context) error { return Q.Tenant.WithContext(ctx).Create(m) }
// Delete removes the row represented by the model using the default DB.
func (m *Tenant) Delete(ctx context.Context) (gen.ResultInfo, error) {
return Q.Tenant.WithContext(ctx).Delete(m)
}
// ForceDelete permanently deletes the row (ignores soft delete) using the default DB.
func (m *Tenant) ForceDelete(ctx context.Context) (gen.ResultInfo, error) {
return Q.Tenant.WithContext(ctx).Unscoped().Delete(m)
}
// Reload reloads the model from database by its primary key and overwrites current fields.
func (m *Tenant) Reload(ctx context.Context) error {
fresh, err := Q.Tenant.WithContext(ctx).GetByID(m.ID)
if err != nil {
return err
}
*m = *fresh
return nil
}

495
backend/database/models/tenants.query.gen.go Normal file
View File

@@ -0,0 +1,495 @@
// Code generated by go.ipao.vip/gen. DO NOT EDIT.
// Code generated by go.ipao.vip/gen. DO NOT EDIT.
// Code generated by go.ipao.vip/gen. DO NOT EDIT.
package models
import (
"context"
"gorm.io/gorm"
"gorm.io/gorm/clause"
"gorm.io/gorm/schema"
"go.ipao.vip/gen"
"go.ipao.vip/gen/field"
"gorm.io/plugin/dbresolver"
)
func newTenant(db *gorm.DB, opts ...gen.DOOption) tenantQuery {
_tenantQuery := tenantQuery{}
_tenantQuery.tenantQueryDo.UseDB(db, opts...)
_tenantQuery.tenantQueryDo.UseModel(&Tenant{})
tableName := _tenantQuery.tenantQueryDo.TableName()
_tenantQuery.ALL = field.NewAsterisk(tableName)
_tenantQuery.ID = field.NewInt64(tableName, "id")
_tenantQuery.UserID = field.NewInt64(tableName, "user_id")
_tenantQuery.Code = field.NewString(tableName, "code")
_tenantQuery.UUID = field.NewField(tableName, "uuid")
_tenantQuery.Name = field.NewString(tableName, "name")
_tenantQuery.Status = field.NewField(tableName, "status")
_tenantQuery.Config = field.NewJSONB(tableName, "config")
_tenantQuery.ExpiredAt = field.NewTime(tableName, "expired_at")
_tenantQuery.CreatedAt = field.NewTime(tableName, "created_at")
_tenantQuery.UpdatedAt = field.NewTime(tableName, "updated_at")
_tenantQuery.fillFieldMap()
return _tenantQuery
}
type tenantQuery struct {
tenantQueryDo tenantQueryDo
ALL field.Asterisk
ID field.Int64
UserID field.Int64
Code field.String
UUID field.Field
Name field.String
Status field.Field
Config field.JSONB
ExpiredAt field.Time
CreatedAt field.Time
UpdatedAt field.Time
fieldMap map[string]field.Expr
}
func (t tenantQuery) Table(newTableName string) *tenantQuery {
t.tenantQueryDo.UseTable(newTableName)
return t.updateTableName(newTableName)
}
func (t tenantQuery) As(alias string) *tenantQuery {
t.tenantQueryDo.DO = *(t.tenantQueryDo.As(alias).(*gen.DO))
return t.updateTableName(alias)
}
func (t *tenantQuery) updateTableName(table string) *tenantQuery {
t.ALL = field.NewAsterisk(table)
t.ID = field.NewInt64(table, "id")
t.UserID = field.NewInt64(table, "user_id")
t.Code = field.NewString(table, "code")
t.UUID = field.NewField(table, "uuid")
t.Name = field.NewString(table, "name")
t.Status = field.NewField(table, "status")
t.Config = field.NewJSONB(table, "config")
t.ExpiredAt = field.NewTime(table, "expired_at")
t.CreatedAt = field.NewTime(table, "created_at")
t.UpdatedAt = field.NewTime(table, "updated_at")
t.fillFieldMap()
return t
}
func (t *tenantQuery) QueryContext(ctx context.Context) (*tenantQuery, *tenantQueryDo) {
return t, t.tenantQueryDo.WithContext(ctx)
}
func (t *tenantQuery) WithContext(ctx context.Context) *tenantQueryDo {
return t.tenantQueryDo.WithContext(ctx)
}
func (t tenantQuery) TableName() string { return t.tenantQueryDo.TableName() }
func (t tenantQuery) Alias() string { return t.tenantQueryDo.Alias() }
func (t tenantQuery) Columns(cols ...field.Expr) gen.Columns { return t.tenantQueryDo.Columns(cols...) }
func (t *tenantQuery) GetFieldByName(fieldName string) (field.OrderExpr, bool) {
_f, ok := t.fieldMap[fieldName]
if !ok || _f == nil {
return nil, false
}
_oe, ok := _f.(field.OrderExpr)
return _oe, ok
}
func (t *tenantQuery) fillFieldMap() {
t.fieldMap = make(map[string]field.Expr, 10)
t.fieldMap["id"] = t.ID
t.fieldMap["user_id"] = t.UserID
t.fieldMap["code"] = t.Code
t.fieldMap["uuid"] = t.UUID
t.fieldMap["name"] = t.Name
t.fieldMap["status"] = t.Status
t.fieldMap["config"] = t.Config
t.fieldMap["expired_at"] = t.ExpiredAt
t.fieldMap["created_at"] = t.CreatedAt
t.fieldMap["updated_at"] = t.UpdatedAt
}
func (t tenantQuery) clone(db *gorm.DB) tenantQuery {
t.tenantQueryDo.ReplaceConnPool(db.Statement.ConnPool)
return t
}
func (t tenantQuery) replaceDB(db *gorm.DB) tenantQuery {
t.tenantQueryDo.ReplaceDB(db)
return t
}
type tenantQueryDo struct{ gen.DO }
func (t tenantQueryDo) Debug() *tenantQueryDo {
return t.withDO(t.DO.Debug())
}
func (t tenantQueryDo) WithContext(ctx context.Context) *tenantQueryDo {
return t.withDO(t.DO.WithContext(ctx))
}
func (t tenantQueryDo) ReadDB() *tenantQueryDo {
return t.Clauses(dbresolver.Read)
}
func (t tenantQueryDo) WriteDB() *tenantQueryDo {
return t.Clauses(dbresolver.Write)
}
func (t tenantQueryDo) Session(config *gorm.Session) *tenantQueryDo {
return t.withDO(t.DO.Session(config))
}
func (t tenantQueryDo) Clauses(conds ...clause.Expression) *tenantQueryDo {
return t.withDO(t.DO.Clauses(conds...))
}
func (t tenantQueryDo) Returning(value interface{}, columns ...string) *tenantQueryDo {
return t.withDO(t.DO.Returning(value, columns...))
}
func (t tenantQueryDo) Not(conds ...gen.Condition) *tenantQueryDo {
return t.withDO(t.DO.Not(conds...))
}
func (t tenantQueryDo) Or(conds ...gen.Condition) *tenantQueryDo {
return t.withDO(t.DO.Or(conds...))
}
func (t tenantQueryDo) Select(conds ...field.Expr) *tenantQueryDo {
return t.withDO(t.DO.Select(conds...))
}
func (t tenantQueryDo) Where(conds ...gen.Condition) *tenantQueryDo {
return t.withDO(t.DO.Where(conds...))
}
func (t tenantQueryDo) Order(conds ...field.Expr) *tenantQueryDo {
return t.withDO(t.DO.Order(conds...))
}
func (t tenantQueryDo) Distinct(cols ...field.Expr) *tenantQueryDo {
return t.withDO(t.DO.Distinct(cols...))
}
func (t tenantQueryDo) Omit(cols ...field.Expr) *tenantQueryDo {
return t.withDO(t.DO.Omit(cols...))
}
func (t tenantQueryDo) Join(table schema.Tabler, on ...field.Expr) *tenantQueryDo {
return t.withDO(t.DO.Join(table, on...))
}
func (t tenantQueryDo) LeftJoin(table schema.Tabler, on ...field.Expr) *tenantQueryDo {
return t.withDO(t.DO.LeftJoin(table, on...))
}
func (t tenantQueryDo) RightJoin(table schema.Tabler, on ...field.Expr) *tenantQueryDo {
return t.withDO(t.DO.RightJoin(table, on...))
}
func (t tenantQueryDo) Group(cols ...field.Expr) *tenantQueryDo {
return t.withDO(t.DO.Group(cols...))
}
func (t tenantQueryDo) Having(conds ...gen.Condition) *tenantQueryDo {
return t.withDO(t.DO.Having(conds...))
}
func (t tenantQueryDo) Limit(limit int) *tenantQueryDo {
return t.withDO(t.DO.Limit(limit))
}
func (t tenantQueryDo) Offset(offset int) *tenantQueryDo {
return t.withDO(t.DO.Offset(offset))
}
func (t tenantQueryDo) Scopes(funcs ...func(gen.Dao) gen.Dao) *tenantQueryDo {
return t.withDO(t.DO.Scopes(funcs...))
}
func (t tenantQueryDo) Unscoped() *tenantQueryDo {
return t.withDO(t.DO.Unscoped())
}
func (t tenantQueryDo) Create(values ...*Tenant) error {
if len(values) == 0 {
return nil
}
return t.DO.Create(values)
}
func (t tenantQueryDo) CreateInBatches(values []*Tenant, batchSize int) error {
return t.DO.CreateInBatches(values, batchSize)
}
// Save : !!! underlying implementation is different with GORM
// The method is equivalent to executing the statement: db.Clauses(clause.OnConflict{UpdateAll: true}).Create(values)
func (t tenantQueryDo) Save(values ...*Tenant) error {
if len(values) == 0 {
return nil
}
return t.DO.Save(values)
}
func (t tenantQueryDo) First() (*Tenant, error) {
if result, err := t.DO.First(); err != nil {
return nil, err
} else {
return result.(*Tenant), nil
}
}
func (t tenantQueryDo) Take() (*Tenant, error) {
if result, err := t.DO.Take(); err != nil {
return nil, err
} else {
return result.(*Tenant), nil
}
}
func (t tenantQueryDo) Last() (*Tenant, error) {
if result, err := t.DO.Last(); err != nil {
return nil, err
} else {
return result.(*Tenant), nil
}
}
func (t tenantQueryDo) Find() ([]*Tenant, error) {
result, err := t.DO.Find()
return result.([]*Tenant), err
}
func (t tenantQueryDo) FindInBatch(batchSize int, fc func(tx gen.Dao, batch int) error) (results []*Tenant, err error) {
buf := make([]*Tenant, 0, batchSize)
err = t.DO.FindInBatches(&buf, batchSize, func(tx gen.Dao, batch int) error {
defer func() { results = append(results, buf...) }()
return fc(tx, batch)
})
return results, err
}
func (t tenantQueryDo) FindInBatches(result *[]*Tenant, batchSize int, fc func(tx gen.Dao, batch int) error) error {
return t.DO.FindInBatches(result, batchSize, fc)
}
func (t tenantQueryDo) Attrs(attrs ...field.AssignExpr) *tenantQueryDo {
return t.withDO(t.DO.Attrs(attrs...))
}
func (t tenantQueryDo) Assign(attrs ...field.AssignExpr) *tenantQueryDo {
return t.withDO(t.DO.Assign(attrs...))
}
func (t tenantQueryDo) Joins(fields ...field.RelationField) *tenantQueryDo {
for _, _f := range fields {
t = *t.withDO(t.DO.Joins(_f))
}
return &t
}
func (t tenantQueryDo) Preload(fields ...field.RelationField) *tenantQueryDo {
for _, _f := range fields {
t = *t.withDO(t.DO.Preload(_f))
}
return &t
}
func (t tenantQueryDo) FirstOrInit() (*Tenant, error) {
if result, err := t.DO.FirstOrInit(); err != nil {
return nil, err
} else {
return result.(*Tenant), nil
}
}
func (t tenantQueryDo) FirstOrCreate() (*Tenant, error) {
if result, err := t.DO.FirstOrCreate(); err != nil {
return nil, err
} else {
return result.(*Tenant), nil
}
}
func (t tenantQueryDo) FindByPage(offset int, limit int) (result []*Tenant, count int64, err error) {
result, err = t.Offset(offset).Limit(limit).Find()
if err != nil {
return
}
if size := len(result); 0 < limit && 0 < size && size < limit {
count = int64(size + offset)
return
}
count, err = t.Offset(-1).Limit(-1).Count()
return
}
func (t tenantQueryDo) ScanByPage(result interface{}, offset int, limit int) (count int64, err error) {
count, err = t.Count()
if err != nil {
return
}
err = t.Offset(offset).Limit(limit).Scan(result)
return
}
func (t tenantQueryDo) Scan(result interface{}) (err error) {
return t.DO.Scan(result)
}
func (t tenantQueryDo) Delete(models ...*Tenant) (result gen.ResultInfo, err error) {
return t.DO.Delete(models)
}
// ForceDelete performs a permanent delete (ignores soft-delete) for current scope.
func (t tenantQueryDo) ForceDelete() (gen.ResultInfo, error) {
return t.Unscoped().Delete()
}
// Inc increases the given column by step for current scope.
func (t tenantQueryDo) Inc(column field.Expr, step int64) (gen.ResultInfo, error) {
// column = column + step
e := field.NewUnsafeFieldRaw("?+?", column.RawExpr(), step)
return t.DO.UpdateColumn(column, e)
}
// Dec decreases the given column by step for current scope.
func (t tenantQueryDo) Dec(column field.Expr, step int64) (gen.ResultInfo, error) {
// column = column - step
e := field.NewUnsafeFieldRaw("?-?", column.RawExpr(), step)
return t.DO.UpdateColumn(column, e)
}
// Sum returns SUM(column) for current scope.
func (t tenantQueryDo) Sum(column field.Expr) (float64, error) {
var _v float64
agg := field.NewUnsafeFieldRaw("SUM(?)", column.RawExpr())
if err := t.Select(agg).Scan(&_v); err != nil {
return 0, err
}
return _v, nil
}
// Avg returns AVG(column) for current scope.
func (t tenantQueryDo) Avg(column field.Expr) (float64, error) {
var _v float64
agg := field.NewUnsafeFieldRaw("AVG(?)", column.RawExpr())
if err := t.Select(agg).Scan(&_v); err != nil {
return 0, err
}
return _v, nil
}
// Min returns MIN(column) for current scope.
func (t tenantQueryDo) Min(column field.Expr) (float64, error) {
var _v float64
agg := field.NewUnsafeFieldRaw("MIN(?)", column.RawExpr())
if err := t.Select(agg).Scan(&_v); err != nil {
return 0, err
}
return _v, nil
}
// Max returns MAX(column) for current scope.
func (t tenantQueryDo) Max(column field.Expr) (float64, error) {
var _v float64
agg := field.NewUnsafeFieldRaw("MAX(?)", column.RawExpr())
if err := t.Select(agg).Scan(&_v); err != nil {
return 0, err
}
return _v, nil
}
// PluckMap returns a map[key]value for selected key/value expressions within current scope.
func (t tenantQueryDo) PluckMap(key, val field.Expr) (map[interface{}]interface{}, error) {
do := t.Select(key, val)
rows, err := do.DO.Rows()
if err != nil {
return nil, err
}
defer rows.Close()
mm := make(map[interface{}]interface{})
for rows.Next() {
var k interface{}
var v interface{}
if err := rows.Scan(&k, &v); err != nil {
return nil, err
}
mm[k] = v
}
return mm, rows.Err()
}
// Exists returns true if any record matches the given conditions.
func (t tenantQueryDo) Exists(conds ...gen.Condition) (bool, error) {
cnt, err := t.Where(conds...).Count()
if err != nil {
return false, err
}
return cnt > 0, nil
}
// PluckIDs returns all primary key values under current scope.
func (t tenantQueryDo) PluckIDs() ([]int64, error) {
ids := make([]int64, 0, 16)
pk := field.NewInt64(t.TableName(), "id")
if err := t.DO.Pluck(pk, &ids); err != nil {
return nil, err
}
return ids, nil
}
// GetByID finds a single record by primary key.
func (t tenantQueryDo) GetByID(id int64) (*Tenant, error) {
pk := field.NewInt64(t.TableName(), "id")
return t.Where(pk.Eq(id)).First()
}
// GetByIDs finds records by primary key list.
func (t tenantQueryDo) GetByIDs(ids ...int64) ([]*Tenant, error) {
if len(ids) == 0 {
return []*Tenant{}, nil
}
pk := field.NewInt64(t.TableName(), "id")
return t.Where(pk.In(ids...)).Find()
}
// DeleteByID deletes records by primary key.
func (t tenantQueryDo) DeleteByID(id int64) (gen.ResultInfo, error) {
pk := field.NewInt64(t.TableName(), "id")
return t.Where(pk.Eq(id)).Delete()
}
// DeleteByIDs deletes records by a list of primary keys.
func (t tenantQueryDo) DeleteByIDs(ids ...int64) (gen.ResultInfo, error) {
if len(ids) == 0 {
return gen.ResultInfo{RowsAffected: 0, Error: nil}, nil
}
pk := field.NewInt64(t.TableName(), "id")
return t.Where(pk.In(ids...)).Delete()
}
func (t *tenantQueryDo) withDO(do gen.Dao) *tenantQueryDo {
t.DO = *do.(*gen.DO)
return t
}