golang命令行

flag

案例

使用阿里云oss实现文件中转站

package main

import (
	"flag"
	"fmt"
	"os"

	"github.com/aliyun/aliyun-oss-go-sdk/oss"
)

var (
	endpint        = "oss-cn-beijing.aliyuncs.com"
	ak             = "xx"
	sk             = "xx"
	bucketName     = "clouds-station"
	uploadFilePath = ""
	help           bool
)

func usage() {
	fmt.Fprintf(os.Stderr, `Usage: cloud-station -f <uplaod_file_path>
`)
	flag.PrintDefaults()
	os.Exit(0)
}

func loadParam() error {
	flag.BoolVar(&help, "h", false, "help usage")
	flag.StringVar(&uploadFilePath, "f", "", "upload file path")
	flag.Parse()

	if uploadFilePath == "" {
		usage()
	}

	if help {
		usage()
	}

	return nil
}

func upload(filePath string) error {
	client, err := oss.New(endpint, ak, sk)
	if err != nil {
		return err
	}

	bucket, err := client.Bucket(bucketName)
	if err != nil {
		return err
	}

	// 第一个参数: 上传到oss里面的文件的key(路径)
	// 第二个参数: 需要上传的文件的路径
	err = bucket.PutObjectFromFile(filePath, filePath)
	if err != nil {
		return err
	}

	// 打印下载URL
	// sts, 临时授权token(有效期1天)
	signedURL, err := bucket.SignURL(filePath, oss.HTTPGet, 60*60*24)
	if err != nil {
		return fmt.Errorf("sign file download url error, %s", err)
	}
	fmt.Printf("upload file: %s success\n", uploadFilePath)
	fmt.Printf("下载链接: %s\n", signedURL)
	fmt.Println("\n注意: 文件下载有效期为1天, 中转站保存时间为3天, 请及时下载")
	return nil
}

func main() {
	// 读取用户输入的参数
	loadParam()

	// 执行文件上传
	if err := upload(uploadFilePath); err != nil {
		fmt.Printf("upload file error, %s\n", err)
		os.Exit(1)
	}

}

cobra

第三方库：github.com/spf13/cobra

package main

import (
	"fmt"

	"github.com/spf13/cobra"
)

var RootCmd = &cobra.Command{
	Use:     "cloud-station-cli",
	Long:    "欢迎使用云存储文件中转站",
	Short:   "欢迎使用云存储文件中转站",
	Example: "cloud-station-cli cmds",
	RunE: func(cmd *cobra.Command, args []string) error {
		if version {
			fmt.Println("cloud-station cli v0.0.1")
		}
		return nil
	},
}

var (
	version bool
)

func init() {
	RootCmd.PersistentFlags().BoolVarP(&version, "version", "v", false, "show version")
    UploadCmd.PersistentFlags().StringVarP(&accessKey, "access_id", "k", "", "the ali oss access id")
	UploadCmd.PersistentFlags().StringVarP(&secretKey, "secret_key", "s", "", "the ali oss access key")
	UploadCmd.PersistentFlags().StringVarP(&ossProvider, "oss_provider", "p", "aliyun", "the oss provider [aliyun/qcloud]")
	UploadCmd.PersistentFlags().StringVarP(&ossEndpoint, "oss_endpoint", "e", "oss-cn-beijing.aliyuncs.com", "oss store endpointer")
	UploadCmd.PersistentFlags().StringVarP(&bucketName, "bucket_name", "b", "devcloud-station", "bucket name")
	UploadCmd.PersistentFlags().StringVarP(&uploadFile, "upload_file", "f", "", "upload file")
	RootCmd.AddCommand(UploadCmd)
}

var UploadCmd = &cobra.Command{
	Use:     "upload",
	Long:    "upload 文件上传",
	Short:   "upload 文件上传",
	Example: "upload -f filename",
	RunE: func(cmd *cobra.Command, args []string) error {
		// 设置ak默认值
		setAk()

		var (
			uploader store.Uploader
			err      error
		)
		switch ossProvider {
		case "aliyun":
			uploader, err = aliyun.NewAliOssStore(ossEndpoint, accessKey, secretKey)
		case "qcloud":
		default:
			return fmt.Errorf("not support oss storage provider")
		}

		if err != nil {
			return err
		}

		// 使用uploader上传
		return uploader.Upload(bucketName, uploadFile, uploadFile)
	},
}


func main() {
	if err := RootCmd.Execute(); err != nil {
		log.Fatal(err)
	}
}

加密输入

第三方库：https://github.com/go-survey/survey

package main

import (
	"fmt"

	"github.com/AlecAivazis/survey/v2"
)

var aliSecretKey string

func main() {
	getSecretKeyFromInputV2()
}

func getSecretKeyFromInputV2() {
	prompt := &survey.Password{
		Message: "请输入access key: ",
	}
	survey.AskOne(prompt, &aliSecretKey)
	fmt.Printf("aliSecretKey: %v\n", aliSecretKey)
}

运行
PS C:\Users\ren\Desktop\新建文件夹> go run .\main.go
? 请输入access key: *******

进度条

UI库：github.com/schollz/progressbar

阿里云oss sdk上传进度条实现

package aliyun

import (
	"fmt"

	"github.com/aliyun/aliyun-oss-go-sdk/oss"
	"github.com/schollz/progressbar/v3"
)

type ProgressListener struct {
	bar *progressbar.ProgressBar
}

func (p *ProgressListener) ProgressChanged(event *oss.ProgressEvent) {
	switch event.EventType {
	case oss.TransferStartedEvent:
		p.bar = progressbar.DefaultBytes(
			event.TotalBytes,
			"文件上传中",
		)
	case oss.TransferDataEvent:
		p.bar.Add64(event.RwBytes)
	case oss.TransferCompletedEvent:
		fmt.Printf("\n上传完成\n")
	case oss.TransferFailedEvent:
		fmt.Printf("\n上传失败\n")
	default:
	}
}

func NewDefaultProgressListener() *ProgressListener {
	return &ProgressListener{}
}

文件上传中 100% |███████████████████████████| (1.9/1.9 kB, 86.513 kB/s)