zishuo/oneterm
1
0
Fork 0
mirror of https://github.com/veops/oneterm.git synced 2025-10-09 01:00:09 +08:00
Code Issues Packages Projects Releases Wiki Activity

fix(api): monitor, replay

Browse Source
This commit is contained in:
ttk
2024-09-06 19:45:01 +08:00
parent 5a63cc375a
commit a30bd94541
6 changed files with 75 additions and 96 deletions
Download Patch File Download Diff File Expand all files Collapse all files

16
backend/api/controller/stat.go
View File

@@ -277,25 +277,29 @@ func toListData[T any](data []T) *ListData {
}
}
func nodeCountAsset() (res map[int]int64, err error) {
func nodeCountAsset() (m map[int]int64, err error) {
assets := make([]*model.AssetIdPid, 0)
if err = mysql.DB.Model(&model.Asset{}).Find(&assets).Error; err != nil {
return
}
res = make(map[int]int64)
nodes := make([]*model.NodeIdPid, 0)
if err = mysql.DB.Model(&model.Node{}).Find(&nodes).Error; err != nil {
return
}
m = make(map[int]int64)
for _, a := range assets {
res[a.ParentId] += 1
m[a.ParentId] += 1
}
g := make(map[int][]int)
for _, n := range assets {
for _, n := range nodes {
g[n.ParentId] = append(g[n.ParentId], n.Id)
}
var dfs func(int) int64
dfs = func(x int) int64 {
for _, y := range g[x] {
res[x] += dfs(y)
m[x] += dfs(y)
}
return res[x]
return m[x]
}
dfs(0)