Golang實現一個密碼生成器

小地鼠防止有人偷他的果實，在家里上了一把鎖。這個鎖怎么來的呢？請往下看。。

package mainimport ("flag""fmt""math/rand""time"
)var (length  intcharset string
)const (NUmStr  = "0123456789"CharStr = "ABCDEFGHIJKLMNOPQRSTUVWXYZabcdefghijklmnopqrstuvwxyz"SpecStr = "+=-@#~,.[]()!%^*$"
)//解析參數
func parseArgs() {//需要接受指針，就傳遞地址,&flag.IntVar(&length, "l", 16, "-l 生成密碼的長度")flag.StringVar(&charset, "t", "num",//反引號以原樣輸出`-t 制定密碼生成的字符集,num:只使用數字[0-9],char:只使用英文字母[a-zA-Z],mix:使用數字和字母，advance:使用數字、字母以及特殊字符`)flag.Parse()
}//檢測字符串中的空格
func test1() {for i := 0; i < len(CharStr); i++ {if CharStr[i] != ' ' {fmt.Printf("%c", CharStr[i])}}
}func generatePasswd() string {//初始化密碼切片var passwd []byte = make([]byte, length, length)//源字符串var sourceStr string//判斷字符類型,如果是數字if charset == "num" {sourceStr = NUmStr//如果選的是字符} else if charset == "char" {sourceStr = charset//如果選的是混合模式} else if charset == "mix" {sourceStr = fmt.Sprintf("%s%s", NUmStr, CharStr)//如果選的是高級模式} else if charset == "advance" {sourceStr = fmt.Sprintf("%s%s%s", NUmStr, CharStr, SpecStr)} else {sourceStr = NUmStr}fmt.Println("source:", sourceStr)//遍歷，生成一個隨機index索引,for i := 0; i < length; i++ {index := rand.Intn(len(sourceStr))passwd[i] = sourceStr[index]}return string(passwd)
}func main() {//隨機種子
    rand.Seed(time.Now().UnixNano())parseArgs()fmt.Printf("length:%d charset:%s\n", length, charset)//test1()passwd := generatePasswd()fmt.Println(passwd)fmt.Printf("length:%d charset:%s\n", length, charset)
}

效果呢？

?