go - 如何使用将多个结构连接到同一个结构中

Question

我正在尝试使用递归结构，当我有多个递归结构时，我可以将它们加在一起，创建一个嵌入这些结构的新结构。但是，我不确定处理此问题的正确方法是什么。

我在下面包含了一个代码片段以进一步说明我的意思。

package main

import "fmt"

type Container struct {
    F          int
    Collection []SubContainer
}

type SubContainer struct {
    Key    string
    Value  int
}

func main() {
    commits := map[string]int{
        "a": 1,
        "b": 2,
        "c": 3,
        "d": 4,
    }

    sc := []SubContainer{}
    c := []Container{}
    count := 0

    for k, v := range commits {
        sc = append(sc, SubContainer{Key: k, Value: v})
        count++

        if len(sc) == 2 {
            c = append(c, Container{Collection: sc, F: count})
            sc = nil
        }
    }

    for _, r := range c {
        fmt.Println(r)
    }
}

结果:

{2 [{a 1} {b 2}]}
{4 [{c 3} {d 4}]}

期望的结果:

{6 {2 [{a 1} {b 2}]} {4 [{c 3} {d 4}]}}

Playground 链接:https://play.golang.org/p/j6rbhgcOoT

我无法理解的一个警告是 commits 长度可能会改变(我最初认为我可以创建一个不同的父结构)。任何建议将不胜感激......以某种方式使用递归结构来完成此操作是否是正确的方法？谢谢!

Answer 1

我试图在不确定确切目标的情况下接近所需的输出，您将在下面找到您提供的代码段的修改版本。

您可以在集合上使用 String() string 方法来自定义格式。

package main

import "fmt"

type ContainerCollection struct {
    Count int
    List  []Container
}

func (cc ContainerCollection) String() string {
    total := 0
    for _, c := range cc.List {
        total += c.F
    }
    return fmt.Sprintf("{%d %v}", total, cc.List)
}

type Container struct {
    F          int
    Collection []SubContainer
}

type SubContainer struct {
    Key   string
    Value int
}

func main() {
    commits := map[string]int{
        "a": 1,
        "b": 2,
        "c": 3,
        "d": 4,
    }

    c := ContainerCollection{Count: 0}
    sc := []SubContainer{}

    for k, v := range commits {
        sc = append(sc, SubContainer{Key: k, Value: v})
        c.Count++
        if len(sc) == 2 {
            c.List = append(c.List, Container{Collection: sc, F: c.Count})
            sc = []SubContainer{}
        }
    }
    // Should also cover odd number of commits
    if len(sc) != 0 {
        c.Count++
        c.List = append(c.List, Container{Collection: sc, F: c.Count})
    }

    // for _, r := range c.List { fmt.Println(r) }

    fmt.Println(c)
}

结果:

{6 [{2 [{a 1} {b 2}]} {4 [{c 3} {d 4}]}]}

Playground