手把手教你用Go语言开发一款简易目录生成器

开发 前端
有时候,我看着我的目录,或者我想找个文件时,尤其是目录很多时,总是有点难受。今天就教你用Go语言开发一款简易目录生成器。

[[377341]]

前言

有时候,我看着我的目录,或者我想找个文件时,尤其是目录很多时,总是有点难受。


比如这个目录,是我的一个教程目录。

但是我并不能一目了然的知道每个文件夹下都有啥,并且每次找一个文件也很费事。

所以,基于上述的需求,使用Go语言,终于撸了个目录生成器,感觉还行。

实现的效果


通过编写的脚本,可以将目录整合成.txt文件,并且下级目录使用4个空格缩进。

代码

先别管怎么实现的,先看代码。

  1. package main 
  2.  
  3. import ( 
  4.     "bytes" 
  5.     "flag" 
  6.     "fmt" 
  7.     "io/ioutil" 
  8.     "os" 
  9.     "path/filepath" 
  10.  
  11. func GenderNSymbol(char byte, count int) string { 
  12.     symbolSince := bytes.Repeat([]byte{char}, count
  13.     var symbol = string(symbolSince) 
  14.     return symbol 
  15.  
  16. // @title    GenderDirTree 
  17. // @description   生成目录树 
  18. // @param     path        string         "需要生成的目录" 
  19. // @param     count        int           "生成相同字符的个数" 
  20. // @param     char        byte           "生成相同字符的字符" 
  21. // @param     current_tier        int            "当前层数" 
  22. // @param     end_tier        int            "终止层数" 
  23. // @param     fileObj        int            "文件对象" 
  24. // @return    无        无         "无" 
  25. func GenderDirTree(path string, count intchar byte, current_tier int, end_tier int, fileObj *os.File) { 
  26.     if !(current_tier < end_tier) && end_tier != 0 { 
  27.         return 
  28.     current_tier++ 
  29.     files, err := ioutil.ReadDir(path) 
  30.     if err != nil { 
  31.         fmt.Println("错误:目录错误"
  32.         return 
  33.     for _, file := range files { 
  34.         var name = file.Name() 
  35.         //生成指定数目的相同符号 
  36.         var plac = GenderNSymbol(charcount
  37.         //符号+目录 
  38.         var sname = fmt.Sprintf("%s%s\n", plac, name
  39.         //输出当前目录 
  40.         fileObj.WriteString(sname) 
  41.         //fmt.Println(sname) 
  42.         //判断是否为目录,如果是,继续下次递归 
  43.         var isDir = file.IsDir() 
  44.         if isDir { 
  45.             //拼接传入的目录和循环的当前目录 
  46.             var nerPaht = filepath.Join(path, name
  47.             GenderDirTree(nerPaht, count+4, char, current_tier, end_tier, fileObj) 
  48.         } 
  49.  
  50. func main() { 
  51.     //终止层数,0表示无限层,>0表示指定层数 
  52.     var end_tier int 
  53.     //输入的目录 
  54.     //var path = `D:\0_教程\易锦教程` 
  55.     var path string 
  56.     flag.StringVar(&path, "path""""目录"
  57.     flag.IntVar(&end_tier, "tier", 0, "终止层数"
  58.     flag.Parse() 
  59.     //文件对象 
  60.     var wDirPath = filepath.Join(path, "目录.txt"
  61.     fileObj, _ := os.OpenFile(wDirPath, os.O_CREATE|os.O_TRUNC|os.O_WRONLY, 0666) 
  62.     defer fileObj.Close() 
  63.     //生成目录树 
  64.     GenderDirTree(path, 0, ' ', 0, end_tier, fileObj) 

去掉注释,其实也就30行代码,就实现了这个功能。

实现的功能

  • 自定义要生成的目录。
  • 自定义生成目录终止的层数。

打包

  1. go build main.go 

-help

采用flag包动态控制参数,参数如下。

  1. d:>main.exe -h 
  2. Usage of main.exe: 
  3.   -path string 
  4.         目录 
  5.   -tier int 
  6.         终止层数 

运行

语法

  1. main.exe -path <要生成的目录> -tier <终止层级> 
  2. 注意:-tier,终止层数,0表示无限层,>0表示指定层数,默认为0 

示例

此处-tier指定的是0,表示无限制生成,如果指定是1,就表示生成一层。

结果

会在要生成的目录下,多出一个目录.txt。


打开如下


主要代码理解

flag


fileObj


判断是否为目录,递归


总结

本次更像是日常使用中,为了解决实际问题的脚本,其实脚本使用Python更好。

但是Go有一个更大的优势就是可以编译为exe。

这次的难点主要是理解递归,自己调自己!

 

责任编辑:姜华 来源: Go语言进阶学习
相关推荐

2024-01-07 20:00:27

2021-01-10 08:14:01

Go语言TCP扫描器

2023-05-22 10:04:24

2021-02-01 08:41:06

Java考试系统

2021-02-04 15:52:46

Java考试系统

2017-09-14 09:09:04

php应用LibreOfficeWord转HTML

2021-01-04 09:55:26

Java移动互联网

2021-01-05 09:04:20

Javatxt文件

2021-08-13 09:01:31

Python小游戏Python基础

2021-08-23 07:32:57

生成器开发代码

2021-08-09 13:31:25

PythonExcel代码

2022-10-19 14:30:59

2023-03-14 07:34:47

代码生成器开发

2011-03-28 16:14:38

jQuery

2021-02-06 14:55:05

大数据pandas数据分析

2021-02-04 09:00:57

SQLDjango原生

2022-08-04 10:39:23

Jenkins集成CD

2022-02-17 10:26:17

JavaScript扫雷游戏前端

2009-04-22 09:17:19

LINQSQL基础

2022-12-05 10:47:37

点赞
收藏

51CTO技术栈公众号