Golang生成随机字符串|极客教程

Golang生成随机字符串

在开发过程中，经常会遇到需要生成随机字符串的情况，比如生成随机密码、token等。在Golang中，我们可以利用内置的随机数生成器来生成随机字符串。本文将详细介绍如何在Golang中生成随机字符串。

1. 使用math库生成随机字符串

在Golang中，我们可以使用math库中的Rand函数生成随机数。结合字符串转换函数，我们可以生成随机的ASCII字符。

package main

import (
    "fmt"
    "math/rand"
    "time"
)

func randString(n int) string {
    var letterRunes = []rune("abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ")

    rand.Seed(time.Now().UnixNano())
    b := make([]rune, n)
    for i := range b {
        b[i] = letterRunes[rand.Intn(len(letterRunes))]
    }
    return string(b)
}

func main() {
    fmt.Println(randString(10))
}

运行上述代码，将生成一个包含10个随机字符的字符串，如下所示：

JTIMhfLDvA

2. 使用crypto/rand库生成随机字符串

除了使用math库生成随机数外，还可以使用crypto/rand库生成更安全的随机数。crypto/rand库提供了更安全的随机数生成器，适用于生成密码、token等安全敏感的随机字符串。

package main

import (
    "crypto/rand"
    "fmt"
    "math/big"
)

func randString(n int) string {
    const letterBytes = "abcdefghijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
    const letterIdxBits = 6
    const letterIdxMask = 1<<letterIdxBits - 1
    const letterIdxMax = 63 / letterIdxBits

    randStr := make([]byte, n)
    for i, cache, remain := n-1, 0, letterIdxMax; i >= 0; {
        if remain == 0 {
            cache, remain = int(rand.Int63()), letterIdxMax
        }
        if idx := int(cache & letterIdxMask); idx < len(letterBytes) {
            randStr[i] = letterBytes[idx]
            i--
        }
        cache >>= letterIdxBits
        remain--
    }

    return string(randStr)
}

func main() {
    fmt.Println(randString(10))
}

运行上述代码，同样将生成一个包含10个随机字符的字符串，如下所示：