1
type T struct {
2
  model.DictGenre
3
  Children *[]*T `json:"children"`
4
}
5

6
treeList := func() []T {
7
  var treeList []T
8
  var flatPtr []T
9
  for _, i := range flat {
10
    t := T{}
11
    t.Pid = i.Pid
12
    t.ID = i.ID
13
    t.NameZh = i.NameZh
14
    t.NameEn = i.NameEn
15
    t.AliasZh = i.AliasZh
16
    t.AliasEn = i.AliasEn
17
    t.CreateTime = i.CreateTime
18
    t.UpdateTime = i.UpdateTime
19
    t.Children = nil
20
    flatPtr = append(flatPtr, t)
21
  }
22
  for m := range flatPtr {
23
    for n := range flatPtr {
24
      if flatPtr[m].ID == flatPtr[n].Pid {
25
        if flatPtr[m].Children == nil {
26
          flatPtr[m].Children = &[]*T{}
27
        }
28
        *(flatPtr[m].Children) = append(*(flatPtr[m].Children), &flatPtr[n])
29
      }
30
    }
31
  }
32
  for _, j := range flatPtr {
33
    if j.Pid == 0 {
34
      treeList = append(treeList, j)
35
    }
36
  }
37
  return treeList
38
}()

1
flat := func() []model.DictGenre {
2
  var flat []model.DictGenre
3
  for _, v := range genres {
4
    u := model.DictGenre{}
5
    u.Pid = v.Pid
6
    u.ID = v.ID
7
    u.NameZh = v.NameZh
8
    u.NameEn = v.NameEn
9
    u.AliasZh = v.AliasZh
10
    u.AliasEn = v.AliasEn
11
    u.CreateTime = v.CreateTime
12
    u.UpdateTime = v.UpdateTime
13
    flat = append(flat, u)
14
  }
15
  return flat
16
}()