Question

我有一些像这样的结构：

type Datacenter struct {
  Name  string
  Key   string
  Hosts []Host
}

type Host struct {
  Name string
  Port int
}

当填充这些结构时，它们看起来像这样：

[]config.Datacenter (len=2) {
  Datacenter {
    Name: "dc1",
    Key: "string",
    Hosts: []config.Host (len=1) {
      Host {
        Name: "dc1-server-1",
        Port: 8200 (int)
      }
    }
  },
  Datacenter {
    Name: "dc2",
    Key: "Another string",
    Hosts: []config.Host (len=2) {
      Host {
        Name: "dc2-server-1",
        Port: 8200 (int)
      },
      Host {
        Name: "dc2-server-2",
        Port: 8200 (int)
      }
    }
  }
}

我正在创建一个CLI工具，几乎每个命令都以这种方式覆盖这些结构：

for _, d := range datacenters {
  if datacenter == d.Name || datacenter == "" {
    for _, h := range d.Hosts {
      // do something on each host in the range here
    }
  }
}

因为我是一个可怕的，糟糕的开发者，我知道有更好的方法来做到这一点。目前代码不是很干，我实际上正在处理并将范围粘贴到每个新的子命令中，这看起来很糟糕，通常会让我感到愚蠢。

我认为我需要创建一个函数并在其中进行范围调整，然后决定返回值。我无法弄清楚它应该是什么，概念不存在。任何帮助将不胜感激。

Answer 1

听起来你想做类似的事情：

func EachHost(datacenter string, f func(Host)) {
    for _, d := range datacenters {
        if datacenter == d.Name || datacenter == "" {
            for _, h := range d.Hosts {
                // do something on each host in the range here
                f(h)
            }
        }
     }
}

然后你可以这样做：

listHostNames := func(datacenter string) {
    EachHost(datacenter, func(h Host) {
        fmt.Println(h.Name)
    });
}
listHostNames("dc1")

从重复范围创建函数

1 个答案: