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

Question

我想创建一个给定长度的虚拟字符串来进行性能测试。例如，我想先用 1 KB 的字符串进行测试，然后可能是 10 KB 的字符串等。我不关心它填充的是哪个字符(或 rune ？)。我知道 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() 创建一个具有给定长度的字节 slice ，默认情况下用零填充 (zero value)。然后将 len 数量的 runes 附加到此 slice ，因此结果将是您打算创建的长度的两倍(假设 rune 更少比 127)，这就是为什么在打印时您会看到零，然后是 'A' rune 。

如果将其更改为:

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

那么结果就是你想要的。

但更简单的方法是简单地使用 strings.Repeat()它重复给定的 string 值 n 次。重复一个 rune (或更具体地说是一个字节)string 值 n 次，然后你得到你想要的:

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

这将输出(在 Go Playground 上尝试):

10 AAAAAAAAAA

如果您不关心 string 的内容，只关心它的长度，那么只需像这样转换一个 byte slice :

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

或者像这样:

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

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

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