Go语言学习笔记十三--Go语言练习 go语言实现一个密码生成工具

go语言实现一个密码生成工具，支持以下功能

代码讲解地址：https://www.bilibili.com/video/BV1v5411W7t3?p=52

package main

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

var (
    length int
    charset string
)
//字符集常量
const (
    NumStr = "0123456789"
    CharStr = "abcdefglijklmnopqrstuvwxyzABCDEFGHIJKLMNOPQRSTUVWXYZ"
    SpecStr = "!@#$%^&*()[]_+-=,."
)

func parseArgs(){
    //参数分别表示 读取到达参数存到length 选项名称l 长度16 解释如果输入错误会弹出列表
    flag.IntVar(&length,"l",16,"-l 生成密码的长度")
    flag.StringVar(&charset,"t","num",
        `-t 制定密码生成的字符集，
                num:只使用数字【1-9】，
                char:只使用英文字母【A-Z，a-z】，
                mix:使用数字和字母，
                advance:使用数字字母和特殊字符
                `)
    flag.Parse()
}

func generatePassword() string{
    var passwd []byte = make([]byte, length, length)
    var sourceStr string = ""
    if charset == "num" {
        sourceStr = NumStr
    } else if charset == "char" {
        sourceStr = CharStr
    } 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
    }
    //随机数生成密码
    for i := 0; i < length; i++ {
        index := rand.Intn(len(sourceStr))
        passwd[i] = sourceStr[index]
    }
    //fmt.Println(sourceStr)
    return string(passwd)
}
func main() {
    rand.Seed(time.Now().Unix())
    parseArgs()
    //fmt.Println("length：", length)
    //fmt.Println("charset:", charset)
    passwd := generatePassword()
    fmt.Println(passwd)
}

代码运行命令 

//-l 表示密码长度为32位 
//-t 表示密码方式为num

go run args.go -l 32 -t num

go run args.go -l 64 -t mix