daihengshun/ArmedPolice
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

feat：完善项目

Browse Source
This commit is contained in:
henry
2021-11-11 14:05:52 +08:00
parent d331a682b7
commit 3b09ec1efc
12 changed files with 341 additions and 49 deletions
Download Patch File Download Diff File Expand all files Collapse all files

13
app/model/sys_tenant.go
View File

@ -3,6 +3,7 @@ package model
import (
"ArmedPolice/app/common/model"
"ArmedPolice/serve/orm"
"fmt"
)
type SysTenant struct {
@ -25,16 +26,16 @@ type (
// Parent 父集元素，包含自己
func (m *SysTenant) Parent(tenantID uint64) ([]*SysTenantBasic, error) {
sql := `SELECT T2.id, T2.name FROM
(SELECT @r AS _id, (SELECT @r := parent_id FROM ? WHERE id = _id) AS parent_id,
@l := @l + 1 AS lvl
FROM (SELECT @r := ?, @l := 0 ) vars, ? h WHERE @r <> 0) T1
JOIN ? T2 ON T1._id = T2.id
sql := `SELECT T2.id, T2.name FROM
(SELECT @r AS _id, (SELECT @r := parent_id FROM %s WHERE id = _id) AS parent_id,
@l := @l + 1 AS lvl
FROM (SELECT @r := %d, @l := 0 ) vars, %s h WHERE @r <> 0) T1
JOIN %s T2 ON T1._id = T2.id
ORDER BY T1.lvl DESC`
out := make([]*SysTenantBasic, 0)
err := orm.GetDB().Raw(sql, m.TableName(), tenantID, m.TableName(), m.TableName()).Scan(&out).Error
err := orm.GetDB().Raw(fmt.Sprintf(sql, m.TableName(), tenantID, m.TableName(), m.TableName())).Scan(&out).Error
return out, err
}