ls 命令

要實(shí)現(xiàn)ls，首先先我們復(fù)習(xí)一下ls命令。

• ls不加任何參數(shù)，表示查詢當(dāng)前目錄下的文件/文件夾
• ls后面加上路徑，列出所有，不包含隱藏文件/文件夾
• ls -a列出所有，包括以.開(kāi)頭的隱藏文件

此命令還有很多功能，這一節(jié)的重點(diǎn)是命令行程度的編寫(xiě)，就實(shí)現(xiàn)上面四個(gè)。

os.Args 讀取參數(shù)

定義如下，一個(gè)數(shù)組。該數(shù)組長(zhǎng)度由參數(shù)個(gè)數(shù)決定，第一個(gè)參數(shù)為程序本身，后面為參數(shù)列表。

var Args []string

比如ls -l，Args=["ls","-l"]，我們現(xiàn)在實(shí)現(xiàn)ls+目錄，或者缺省目錄時(shí)默認(rèn)當(dāng)前目錄代碼如下。

func main() {
	targetDirPath := "./"
	if len(os.Args) > 1 {
		targetDirPath = os.Args[1]
	}

	if dirList, err := ioutil.ReadDir(targetDirPath); err == nil {
		for _, dirInfo := range dirList {
			fmt.Print(dirInfo.Name() + " ")
		}
	} else {
		fmt.Println(err.Error())
	}
}

• 使用 os.Agrs 來(lái)獲取傳入程序的參數(shù)
• 使用 ioutil.ReadDir 讀取目錄,將返回值[] FileInfo遍歷,輸出目錄/文件信息
• 將遇到的error輸出到 終端

輸出

$ go build -o ls main.go
$ ./ls 
main.go folder ls
$ ./ls folder
file1 file2

看，我們已經(jīng)簡(jiǎn)單的實(shí)現(xiàn)了。但我們會(huì)發(fā)現(xiàn)，要使用-開(kāi)頭的參數(shù)時(shí)，用os.Args這個(gè)還需要手寫(xiě)更多復(fù)雜邏輯去判斷輸入的到底是路徑，還是其他參數(shù)，于是下一個(gè)包登場(chǎng)了。

pflag

昨天的每日一Go已經(jīng)講了flag包，pflag 包與 flag 包的工作原理甚至是代碼實(shí)現(xiàn)都是類似的，下面是 pflag 相對(duì) flag 的一些優(yōu)勢(shì)：

• 支持更加精細(xì)的參數(shù)類型：例如，flag 只支持 uint 和 uint64，而 pflag 額外支持 uint8、uint16、int32 等類型。
• 支持更多參數(shù)類型：ip、ip mask、ip net、count、以及所有類型的 slice 類型。
• 兼容標(biāo)準(zhǔn) flag 庫(kù)的 Flag 和 FlagSet：pflag 更像是對(duì) flag 的擴(kuò)展。
• 原生支持更豐富的功能：支持 shorthand、deprecated、hidden 等高級(jí)功能。

var a = pflag.BoolP("all", "a", false, "Include directory entries whose names begin with a dot (.).")
	var help = pflag.BoolP("help", "h", false, "Show this help message.")
	pflag.Parse()
	if *help {
		pflag.Usage()
		return
	}
	args := pflag.Args()
	if len(args) == 0 {
		args = append(args, "./")
	}
	if len(args) == 1 {
		ShowPath(args[0], *a)
	} else {
		for _, v := range args {
			fmt.Println(v + ":")
			ShowPath(v, *a)
		}
	}

• 列出文件夾的方式寫(xiě)到了ShowPath函數(shù)里，第二參數(shù)是一個(gè)bool判斷是否列出全部文件（包括隱藏文件）感興趣去看我源代碼，這里不是重點(diǎn)。
• pflag.BoolP 定義一個(gè)bool類型的參數(shù)，支持長(zhǎng)短參數(shù)形式--all,-a，返回值是指針。
• pflag.Parse() 調(diào)用這個(gè)函數(shù)，把參數(shù)值設(shè)置到相應(yīng)指針的指向里。
• pflag.Usage() 用法打印。

這樣我們就實(shí)現(xiàn)了ls的基本功能。測(cè)試一下。

$ go build -o ls main.go
$ ./ls
folder go.mod go.sum main.go 
$ ./ls -a
. .. .git folder go.mod go.sum main.go 
$ ./ls ./ folder 
./:
folder go.mod go.sum main.go 
folder:
file1 file2
$ ./ls -h
Usage of ./ls:
  -a, --all    Include directory entries whose names begin with a dot (.).
  -h, --help   Show this help message.

是不是相當(dāng)給力啦！

小結(jié)

命令行讀取一共有五個(gè)常用的包

• os.Args 系統(tǒng)包提供的參數(shù)解析。
• flag：簡(jiǎn)單參數(shù)解析，昨天的每日一Go分享過(guò)。
• pflag：命令行參數(shù)解析。
• cobra：應(yīng)用命令行框架，改天再分享。
• viper：配置文件、環(huán)境變量、命令行、緩沖區(qū)。今天群里的每日一Go已經(jīng)分享了，就不多說(shuō)了。

命令行工具，你只要賦予可執(zhí)行權(quán)限，再放到bin里面，他就會(huì)成為一個(gè)系統(tǒng)命令，方便你做更多的事情，比如備份數(shù)據(jù)庫(kù)之類的動(dòng)作，提高你的運(yùn)維效率。 哦對(duì)了源碼在這：https://github.com/golang-minibear2333/cmd_utils

以上就是通過(guò)Golang實(shí)現(xiàn)linux命令ls命令(命令行工具構(gòu)建)的詳細(xì)內(nèi)容，更多關(guān)于Golang命令行工具的資料請(qǐng)關(guān)注腳本之家其它相關(guān)文章！

最新評(píng)論