package main
import (
 "fmt"
 "io/ioutil"
 "os"
 "path/filepath"
 "strings"
)
//獲取指定目錄下的所有檔案，不進入下一級目錄搜尋，可以匹配字尾過濾。
func ListDir(dirPth string, suffix string) (files []string, err error) {
 files = make([]string, 0, 10)
 dir, err := ioutil.ReadDir(dirPth)
 if err != nil {
  return nil, err
 }
 PthSep := string(os.PathSeparator)
 suffix = strings.ToUpper(suffix) //忽略字尾匹配的大小寫
 for _, fi := range dir {
  if fi.IsDir() { // 忽略目錄
   continue
  }
  if strings.HasSuffix(strings.ToUpper(fi.Name()), suffix) { //匹配檔案
   files = append(files, dirPth+PthSep+fi.Name())
  }
 }
 return files, nil
}
//獲取指定目錄及所有子目錄下的所有檔案，可以匹配字尾過濾。
func WalkDir(dirPth, suffix string) (files []string, err error) {
 files = make([]string, 0, 30)
 suffix = strings.ToUpper(suffix) //忽略字尾匹配的大小寫
 err = filepath.Walk(dirPth, func(filename string, fi os.FileInfo, err error) error { //遍歷目錄
  //if err != nil { //忽略錯誤
  // return err
  //}
  if fi.IsDir() { // 忽略目錄
   return nil
  }
  if strings.HasSuffix(strings.ToUpper(fi.Name()), suffix) {
   files = append(files, filename)
  }
  return nil
 })
 return files, err
}
func main() {
 files, err := ListDir("D:\Go", ".txt")
 fmt.Println(files, err)
 files, err = WalkDir("E:\Study", ".pdf")
 fmt.Println(files, err)
}