golang遍历某个目录下的文件,并读取文件名到一个csv文件

rufidmx · · 2391 次点击 · · 开始浏览    
这是一个创建于 的文章,其中的信息可能已经有所发展或是发生改变。
package main

import (
"encoding/csv"
"fmt"
"os"
"path/filepath"
"container/list"
)
var outputFileName string = "filesName.csv"

func CheckErr(err error) {
if nil != err {
panic(err)
}
}

func GetFullPath(path string) string {
absolutePath, _ := filepath.Abs(path)
return absolutePath
}

func PrintFilesName(path string) {
fullPath := GetFullPath(path)

listStr := list.New()

filepath.Walk(fullPath, func(path string, fi os.FileInfo, err error) error {
if nil == fi {
return err
}
if fi.IsDir() {
return nil
}

name := fi.Name()
if outputFileName != name{
listStr.PushBack(name)
}

return nil
})

OutputFilesName(listStr)
}

func ConvertToSlice(listStr *list.List)[]string{
sli := []string{}
for el:= listStr.Front(); nil != el; el= el.Next(){
sli = append(sli, el.Value.(string))
}

return sli
}

func OutputFilesName(listStr *list.List) {
files := ConvertToSlice(listStr)
//sort.StringSlice(files).Sort()// sort  

f, err := os.Create(outputFileName)
//f, err := os.OpenFile(outputFileName, os.O_APPEND | os.O_CREATE, os.ModeAppend)
CheckErr(err)
defer f.Close()

f.WriteString("\xEF\xBB\xBF")
writer := csv.NewWriter(f)

length := len(files)
for i:= 0; i < length; i++{
writer.Write([]string{files[i]})
}

writer.Flush()
}

func main() {
var path string
if len(os.Args) > 1 {
path = os.Args[1]
} else {
path, _ = os.Getwd()
}
PrintFilesName(path)

fmt.Println("done!")
}



有疑问加站长微信联系(非本文作者)

入群交流(和以上内容无关):加入Go大咖交流群,或添加微信:liuxiaoyan-s 备注:入群;或加QQ群:1006366459

2391 次点击  
加入收藏 微博
暂无回复
添加一条新回复 (您需要 登录 后才能回复 没有账号 ?)
  • 请尽量让自己的回复能够对别人有帮助
  • 支持 Markdown 格式, **粗体**、~~删除线~~、`单行代码`
  • 支持 @ 本站用户;支持表情(输入 : 提示),见 Emoji cheat sheet
  • 图片支持拖拽、截图粘贴等方式上传