Question

我想将地图的所有键加入到[k1，k2，...]形式的单个字符串中。我并不过分关心订单，只是我可以制作字符串。我知道有strings.Join()函数，但它需要[]string而不是map[string]bool。

我想以最有效/最快的方式做到这一点（即我不想创建密钥的完整副本，因此我可以将其切片）。我无法找到一种方法来获取地图键的一部分，所以我提出了以下功能。显然它不是最好的，因为它做了不必要的写和截断。

有没有办法只切片地图键？

func CreateStringArray(myMap map[string]bool) string {
    if myMap == nil || len(myMap) == 0 {
        return "[ ]"
    }

    buf := bytes.NewBufferString("[")

    for k, _ := range myMap {
        buf.WriteString(k)
        buf.WriteString(", ")
    }

    buf.Truncate(buf.Len() - 2)
    buf.WriteString("]")

    return buf.String()
}

Answer 1

大多数时候，我只是写下明显的代码：

func KeysString(m map[string]bool) string {
    keys := make([]string, 0, len(m))
    for k := range m {
        keys = append(keys, k)
    }
    return "[" + strings.Join(keys, ", ") + "]"
}

如果您需要的效率高于可读性，您可以查看strings.Join的实现，了解如何编写此最小化副本。这与你的代码之间的主要区别在于构造了一个完全正确长度的[]byte，这可以防止在缓冲区必须调整大小时复制数据，因为结果已经建立起来。

func KeysString(m map[string]bool) string {
    if len(m) == 0 {
        return "[]"
    }
    n := 2 * len(m)  // (len-1) commas (", "), and one each of "[" and "]".
    for k := range m {
        n += len(k)
    }
    b := make([]byte, n)
    bp := copy(b, "[")
    first := true
    for k := range m {
        if !first {
            bp += copy(b[bp:], ", ")
        }
        bp += copy(b[bp:], k)
        first = false
    }
    bp += copy(b[bp:], "]")
    return string(b)
}

当然，请务必在您使用此功能的代码环境中进行分析和优化，以确保可读性权衡实际上是值得的。

在golang中加入地图键

1 个答案: