golang 判断并生成文件/文件夹的操作

最新推荐文章于 2024-11-08 12:01:36 发布

bug不睡我不睡

最新推荐文章于 2024-11-08 12:01:36 发布

阅读量315

点赞数

文章标签： golang 开发语言后端

本文链接：https://blog.youkuaiyun.com/qq_53603794/article/details/130880285

版权

该篇文章展示了如何在Go语言中编写函数来检查文件目录是否存在，并在不存在时创建。主要函数包括PathExists用于判断路径是否存在，PathExistsAndCreate不仅检查还创建目录，以及CreateDir用于批量创建目录。代码中使用了os和os.Path包进行文件操作，以及zap日志库进行错误记录。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

package utils

import (
	"fmt"
	"go.uber.org/zap"
	"os"
)

//@author: [piexlmax](https://github.com/piexlmax)
//@function: PathExists
//@description: 文件目录是否存在
//@param: path string
//@return: bool, error

func PathExists(path string) (bool, error) {
	_, err := os.Stat(path)
	if err == nil {
		return true, nil
	}
	if os.IsNotExist(err) {
		return false, nil
	}
	return false, err
}

func PathExistsAndCreate(path string) (bool, error) {
	_, err := os.Stat(path)
	if err == nil {
		return true, nil
	}
	if os.IsNotExist(err) {
		// 创建文件夹
		err := os.MkdirAll(path, os.ModePerm)
		if err != nil {
			fmt.Printf("mkdir failed![%v]\n", err)
		} else {
			return true, nil
		}
	}
	return false, err
}

//@author: [piexlmax](https://github.com/piexlmax)
//@function: CreateDir
//@description: 批量创建文件夹
//@param: dirs ...string
//@return: err error

func CreateDir(dirs ...string) (err error) {
	for _, v := range dirs {
		exist, err := PathExists(v)
		if err != nil {
			return err
		}
		if !exist {
			global.GVA_LOG.Debug("create directory" + v)
			err = os.MkdirAll(v, os.ModePerm)
			if err != nil {
				global.GVA_LOG.Error("create directory"+v, zap.Any(" error:", err))
			}
		}
	}
	return err
}

//创建文件夹
func CreateFile(s string) {
	exist, err := PathExists(s)
	if err != nil {
		fmt.Printf("get dir error![%v]\n", err)
		return
	}

	if exist {
		fmt.Printf("has dir![%v]\n", s)
	} else {
		fmt.Printf("no dir![%v]\n", s)
		// 创建文件夹
		err := os.MkdirAll(s, os.ModePerm)
		if err != nil {
			fmt.Printf("mkdir failed![%v]\n", err)
		} else {
			fmt.Printf("mkdir success!\n")
		}
	}
}