feat:完善项目
This commit is contained in:
@ -1,10 +1,7 @@
|
||||
package model
|
||||
|
||||
import (
|
||||
"TenantServe/app/common/model"
|
||||
"TenantServe/serve/orm"
|
||||
"fmt"
|
||||
"time"
|
||||
"ArmedPolice/app/common/model"
|
||||
)
|
||||
|
||||
type SysTenant struct {
|
||||
@ -21,65 +18,8 @@ type (
|
||||
DeviceCount int `json:"device_count"`
|
||||
CustomerDeviceCount int `json:"customer_device_count"`
|
||||
}
|
||||
// SysTenantSubsetInfo 租户子集信息
|
||||
SysTenantSubsetInfo struct {
|
||||
ID uint64 `json:"id"`
|
||||
ParentID uint64 `json:"parent_id"`
|
||||
Name string `json:"name"`
|
||||
DeviceCount int `json:"device_count"`
|
||||
}
|
||||
// SysTenantDeviceCount 设备数量
|
||||
SysTenantDeviceCount struct {
|
||||
TenantID uint64 `json:"tenant_id"`
|
||||
Count int `json:"count"`
|
||||
CustomerCount int `json:"customer_count"`
|
||||
}
|
||||
)
|
||||
|
||||
// IsInvalid 判断是否有效
|
||||
func (m *SysTenant) IsInvalid() bool {
|
||||
return m.Status == model.SysTenantStatusForDisable || m.Status == model.SysTenantStatusForExpired || m.Deadline.Before(time.Now())
|
||||
}
|
||||
|
||||
// Tenants 租户信息
|
||||
func (m *SysTenant) Tenants(page, pageSize int, count *int64, where ...*model.ModelWhere) ([]*SysTenantInfo, error) {
|
||||
|
||||
db := orm.GetDB().Table(m.TableName()).Where("is_deleted = ?", model.DeleteStatusForNot)
|
||||
|
||||
if len(where) > 0 {
|
||||
for _, wo := range where {
|
||||
db = db.Where(wo.Condition, wo.Value)
|
||||
}
|
||||
}
|
||||
out := make([]*SysTenantInfo, 0)
|
||||
|
||||
if err := db.Count(count).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
if err := db.Order("id " + model.OrderModeToDesc).Offset((page - 1) * pageSize).Limit(pageSize).Scan(&out).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
// TenantSubset 租户子集信息
|
||||
func (m *SysTenant) TenantSubset(tenantID uint64) ([]*SysTenantSubsetInfo, error) {
|
||||
|
||||
sql := fmt.Sprintf(`SELECT t3.id, t3.parent_id, t3.name, d.count AS device_count FROM
|
||||
(SELECT t1.id, t1.parent_id, t1.name, IF(FIND_IN_SET( parent_id, @pids ) > 0, @pids := concat( @pids, ',', id ), 0 ) AS is_child
|
||||
FROM (SELECT id, parent_id, name FROM %s WHERE is_deleted = 0 ORDER BY id DESC) AS t1,
|
||||
(SELECT @pids := %d) AS t2) AS t3
|
||||
LEFT JOIN (SELECT tenant_id, COUNT(id) AS count FROM %s WHERE is_deleted = 0 GROUP BY tenant_id) AS d ON t3.id = d.tenant_id
|
||||
WHERE is_child != 0`, m.TableName(), tenantID, "数据表")
|
||||
|
||||
out := make([]*SysTenantSubsetInfo, 0)
|
||||
|
||||
if err := orm.GetDB().Raw(sql).Scan(&out).Error; err != nil {
|
||||
return nil, err
|
||||
}
|
||||
return out, nil
|
||||
}
|
||||
|
||||
func NewSysTenant() *SysTenant {
|
||||
return &SysTenant{SysTenant: model.NewSysTenant()}
|
||||
}
|
||||
|
Reference in New Issue
Block a user