Golang實現一個密碼生成器

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

package main

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

var (
    length  int
    charset 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)
}

效果呢？