如何创建任意长度的字符串

Question

我想创建一个给定长度的虚拟字符串来进行性能测试。例如，我想首先使用 1 KB 的字符串进行测试，然后可能是 10 KB 的字符串等。我不在乎它填充了哪个字符（或符文？）。我知道 Go 中的字符串由字节数组支持。所以，我希望最终的字符串由大小相当于 1 KB 的字节数组支持（如果我给出 1024 作为参数）。

例如，我尝试了下面的蛮力代码：

...
    oneKBPayload := createPayload(1024, 'A')
...
//I don't mind even if the char argument is removed and 'A' is used for example 
func createPayload(len int, char rune) string {
    payload := make([]byte, len)
    for i := 0; i < len; i++ {
        payload = append(payload, byte(char))
    }
    return string(payload[:])
}

它产生了（10个长度）的结果

"\u0000\u0000\u0000\u0000\u0000\u0000\u0000\u0000\u0000\u0000AAAAAAAAAA"

我意识到这与编码有关。但是如何解决这个问题，以便我创建由给定长度的字节数组支持的任何字符串，以便当我通过网络写入它时，我会生成预期的有效负载。

Answer 1

您的createPayload() 创建一个具有给定长度的字节切片，默认情况下用零填充（zero value）。然后将len 的runes 的数量附加到此切片，因此结果将是您打算创建的长度的两倍（假设rune 小于127），这就是为什么您会看到零，然后是打印时的'A' 符文。

如果你把它改成：

payload := make([]byte, 0, len)

那么结果就是你想要的。

但更简单的方法是简单地使用strings.Repeat()，它重复给定的string 值n 次。重复一个符文（或更具体地说是一个字节）string 值 n 次，你就会得到你想要的：

s := strings.Repeat("A", 10)
fmt.Println(len(s), s)

这将输出（在Go Playground 上尝试）：

10 AAAAAAAAAA

如果您不关心string 的内容，只关心它的长度，那么只需像这样转换一个字节切片：

s := string(make([]byte, 1024))
fmt.Println(len(s))

或者像这样：

s2 := string([]byte{1023: 0})
fmt.Println(len(s2))

两者都打印1024。在Go Playground 上试用。

如果您确实关心内容并且您已经分配了一个字节片，那么您可以通过以下方式有效地填充它：Is there analog of memset in go?