加入收藏 | 设为首页 | 会员中心 | 我要投稿 李大同 (https://www.lidatong.com.cn/)- 科技、建站、经验、云计算、5G、大数据,站长网!
当前位置: 首页 > 大数据 > 正文

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

发布时间:2020-12-16 19:01:50 所属栏目:大数据 来源:网络整理
导读:package mainimport ("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
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("xEFxBBxBF")
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!")
}

(编辑:李大同)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!

    推荐文章
      热点阅读