在 Golang 中，我可以使用 make 或 new 来创建预填充的字符串吗？

在 Go 中优化字符串填充库时，使用 make 或 new 创建预填充字符串一直是个难题。传统的解决方案涉及使用 for 循环，这会产生不必要的内存副本。受 strings.Repeat() 函数的启发，可以采用更优雅的方法。通过使用 make 分配工作缓冲区，然后使用 copy() 函数增量复制可填充字符串，可以减少内存副本的数量，从而提高性能。

我正在尝试优化 go 中的 stringpad 库。到目前为止，我发现用已知字符值（例如 0 或“”）填充字符串（实际上是 bytes.buffer）的唯一方法是使用 for 循环。

代码片段是：

// PadLeft pads string on left side with p, c times
func PadLeft(s string, p string, c int) string {
    var t bytes.Buffer
    if c <= 0 {
        return s
    }
    if len(p) < 1 {
        return s
    }
    for i := 0; i < c; i++ {
        t.WriteString(p)
    }
    t.WriteString(s)
    return t.String()
}

字符串 pad 越大，我相信 t 缓冲区的内存副本就越多。是否有更优雅的方法来在初始化时使用已知值创建已知大小的缓冲区？

解决方案

只能使用 make() 和 new() 来分配归零的缓冲区（字节片或数组）。您可以使用 composite literals 来获取最初包含非零值的切片或数组，但您无法动态描述初始值（索引必须是常量）。

从类似但非常高效的 strings.Repeat() 函数中获取灵感。它以给定的次数重复给定的字符串：

func repeat(s string, count int) string {
    // since we cannot return an error on overflow,
    // we should panic if the repeat will generate
    // an overflow.
    // see issue golang.org/issue/16237
    if count < 0 {
        panic("strings: negative repeat count")
    } else if count > 0 && len(s)*count/count != len(s) {
        panic("strings: repeat count causes overflow")
    }

    b := make([]byte, len(s)*count)
    bp := copy(b, s)
    for bp < len(b) {
        copy(b[bp:], b[:bp])
        bp *= 2
    }
    return string(b)
}

strings.repeat() 执行一次分配来获取工作缓冲区（这将是一个字节切片 []byte），并使用内置的 copy() 函数复制可重复的字符串。值得注意的一件事是，它使用工作副本并尝试增量复制整个工作副本，这意味着例如如果字符串已被复制 4 次，则复制此缓冲区将使其复制 8 次，依此类推。这将最大限度地减少对 copy() 的调用。该解决方案还利用了 copy() 可以从 string 复制字节，而无需将其转换为字节切片。

我们想要的是类似的东西，但我们希望将结果添加到字符串前面。

我们可以解决这个问题，只需分配一个在 repeat() 内部使用的缓冲区加上我们要填充的字符串的长度。

结果（不检查 count 参数）：

func padleft(s, p string, count int) string {
    ret := make([]byte, len(p)*count+len(s))

    b := ret[:len(p)*count]
    bp := copy(b, p)
    for bp < len(b) {
        copy(b[bp:], b[:bp])
        bp *= 2
    }
    copy(ret[len(b):], s)
    return string(ret)
}

测试它：

fmt.println(padleft("aa", "x", 1))
fmt.println(padleft("aa", "x", 2))
fmt.println(padleft("abc", "xy", 3))

输出（在 Go Playground 上尝试）：

xaa
xxaa
xyxyxyabc

查看类似/相关问题：Is there analog of memset in go?

文中关于的知识介绍，希望对你的学习有所帮助！若是受益匪浅，那就动动鼠标收藏这篇《在 Golang 中，我可以使用 make 或 new 来创建预填充的字符串吗？》文章吧，也可关注golang学习网公众号了解相关技术文章。