101 lines
2.2 KiB
Go
101 lines
2.2 KiB
Go
// Package util...
|
|
//
|
|
// Description : 文件相关工具
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 2021-04-26 6:00 下午
|
|
package util
|
|
|
|
import (
|
|
"io/ioutil"
|
|
"os"
|
|
"strings"
|
|
|
|
"github.com/pkg/errors"
|
|
yml "gopkg.in/yaml.v2"
|
|
)
|
|
|
|
// GetProjectPath 获取项目路径(可执行文件所在目录)
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 10:32 下午 2021/4/26
|
|
func GetProjectPath() (string, error) {
|
|
rootPath, err := os.Getwd()
|
|
if nil != err {
|
|
return "", err
|
|
}
|
|
pathArr := strings.Split(rootPath, "/")
|
|
if len(pathArr) > 0 {
|
|
if pathArr[len(pathArr)-1] == "test" {
|
|
rootPath = strings.Join(pathArr[0:len(pathArr)-1], "/")
|
|
}
|
|
}
|
|
return rootPath, nil
|
|
}
|
|
|
|
// ReadYmlConfig 读取yml配置问价,并解析到指定的结构体中
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 10:35 下午 2021/4/26
|
|
func ReadYmlConfig(filePath string, result interface{}) error {
|
|
if nil == result {
|
|
return errors.New("接收读取结果的数据指针为NIL")
|
|
}
|
|
var (
|
|
fileContent []byte
|
|
err error
|
|
)
|
|
if fileContent, err = ReadFileContent(filePath); nil != err {
|
|
return err
|
|
}
|
|
return yml.Unmarshal(fileContent, result)
|
|
}
|
|
|
|
// ReadFileContent 读取文件内容
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 10:37 下午 2021/4/26
|
|
func ReadFileContent(filePath string) ([]byte, error) {
|
|
if exist, isFile := IsFileExist(filePath); !exist || !isFile {
|
|
//文件不存在或者是一个目录
|
|
return nil, errors.New(filePath + " 文件不存在或者是一个目录!")
|
|
}
|
|
//打开文件
|
|
var (
|
|
f *os.File
|
|
err error
|
|
)
|
|
if f, err = os.Open(filePath); nil != err {
|
|
return nil, err
|
|
}
|
|
|
|
return ioutil.ReadAll(f)
|
|
}
|
|
|
|
// IsFileExist 判断文件是否存在
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 10:37 下午 2021/4/26
|
|
func IsFileExist(filePath string) (bool, bool) {
|
|
f, err := os.Stat(filePath)
|
|
return nil == err || os.IsExist(err), (nil == err || os.IsExist(err)) && !f.IsDir()
|
|
}
|
|
|
|
// GetFileType 获取文件类型
|
|
//
|
|
// Author : go_developer@163.com<白茶清欢>
|
|
//
|
|
// Date : 2:30 下午 2021/11/29
|
|
func GetFileType(fileName string) string {
|
|
fileArr := strings.Split(fileName, ".")
|
|
if len(fileArr) < 2 {
|
|
return ""
|
|
}
|
|
return strings.ToLower(fileArr[len(fileArr)-1])
|
|
}
|