go-常用代码段

获取当前应用路径

func getCurrentDirectory() string {
   dir, err := filepath.Abs(filepath.Dir(os.Args[0]))  //返回绝对路径  filepath.Dir(os.Args[0])去除最后一个元素的路径
   if err != nil {
      log.Fatal(err)
   }
   return strings.Replace(dir, "\\", "/", -1) //将\替换成/
}

遍历目录下文件和目录

//获取指定目录下的所有文件和目录
func GetFilesAndDirs(dirPth string) (files []string, dirs []string, err error) {
    dir, err := ioutil.ReadDir(dirPth)
	if err != nil {
    return nil, nil, err
	}

	PthSep := string(os.PathSeparator)
	//suffix = strings.ToUpper(suffix) //忽略后缀匹配的大小写

	for _, fi := range dir {
		if fi.IsDir() { // 目录, 递归遍历
			dirs = append(dirs, dirPth+PthSep+fi.Name())
			GetFilesAndDirs(dirPth + PthSep + fi.Name())
		} else {
			// 过滤指定格式
			// ok := strings.HasSuffix(fi.Name(), ".log")
			// if ok {
			// 	files = append(files, dirPth+PthSep+fi.Name())
			// }
			files = append(files, dirPth+PthSep+fi.Name())
		}
	}

	return files, dirs, nil
}