1 Star 1 Fork 0

vincent/gcutil

Create your Gitee Account
Explore and code with more than 12 million developers,Free private repositories !:)
Sign up
Clone or Download
fileUtils.go 11.46 KB
Copy Edit Raw Blame History
vincent authored 2022-12-24 20:52 . feat(): 配置相关
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147148149150151152153154155156157158159160161162163164165166167168169170171172173174175176177178179180181182183184185186187188189190191192193194195196197198199200201202203204205206207208209210211212213214215216217218219220221222223224225226227228229230231232233234235236237238239240241242243244245246247248249250251252253254255256257258259260261262263264265266267268269270271272273274275276277278279280281282283284285286287288289290291292293294295296297298299300301302303304305306307308309310311312313314315316317318319320321322323324325326327328329330331332333334335336337338339340341342343344345346347348349350351352353354355356357358359360361362363364365366367368369370371372373374375376377378379380381382383384385386387388389390391392393394395396397398399400401402403404405406407408409410411412413414415416417418419420421422423424425426427428429430431432433434435436437438439440441442443444445446447448449450451452453454455456457458459460461462463464465466467468469470471472473474475476477478479480481482483484485486487488489490491492493494495496497498499500501502503504505506507508509510511512513514515516517518519520521522523524525526527528529530531532533534535536537538539540541542543544
package fileUtil
import (
"bufio"
"crypto/md5"
"encoding/hex"
"errors"
"fmt"
imgext "github.com/shamsher31/goimgext"
"io"
"io/fs"
"io/ioutil"
"log"
"mime/multipart"
"net/http"
"os"
"path"
pathUtil "path/filepath"
"runtime"
"strconv"
"strings"
)
func Join(str ...string) string {
return pathUtil.Join(str...)
}
func GetAbs(path string) string {
var err error
if !pathUtil.IsAbs(path) {
path = Join(CurrPath(), path)
path, err = pathUtil.Abs(path)
if err != nil {
return ""
}
}
return path
}
func LastPathName(fullpath string) string {
return pathUtil.Base(fullpath)
}
func GetFileMode(num int) os.FileMode {
var um, _ = strconv.ParseInt(strconv.Itoa(num), 8, 0)
return os.FileMode(um)
}
func CreatePathWithDefaultMode(path string) string {
if !strings.HasPrefix(path, string(os.PathSeparator)) {
path = pathUtil.Join(CurrPath(), path)
}
var r = false
if r, _ = Exist(path); !r {
r, _ = CreatePath(path, GetFileMode(700))
}
if r {
return path
} else {
return ""
}
}
func CreatePath(path string, perm os.FileMode) (bool, error) {
exist, err := Exist(path)
if !exist && err == nil {
if te := os.MkdirAll(pathUtil.Clean(path), perm); te != nil {
return false, te
} else {
return true, nil
}
} else {
return exist, err
}
}
func CurrPath() string {
dir, err := pathUtil.Abs(pathUtil.Dir(os.Args[0]))
if err != nil {
log.Fatal(err)
}
return dir
}
func CurrPath2() string {
_, file, _, _ := runtime.Caller(1)
return path.Dir(file)
}
func Exist(path string) (bool, error) {
_, err := os.Stat(path)
if err != nil && os.IsNotExist(err) {
return false, nil
} else if err != nil && !os.IsNotExist(err) {
return false, err
} else {
return true, nil
}
}
func ReadFile(filePath string) (string, error) {
exist, _ := Exist(filePath)
if !exist {
return "", fmt.Errorf("[%v] is not exist", filePath)
}
bytes, err := ioutil.ReadFile(filePath)
if err != nil {
return "", err
}
return string(bytes), nil
}
func WriteFile(filePath string, content string) (re error) {
defer func() {
if err := recover(); err != nil {
re = err.(error)
}
}()
f, err := os.OpenFile(filePath, os.O_WRONLY|os.O_CREATE|os.O_APPEND|os.O_SYNC, GetFileMode(666))
defer f.Close()
writer := bufio.NewWriter(f)
AppendStrWithWriter(writer, content)
return err
}
func AppendStrWithWriter(writer *bufio.Writer, content string) (w *bufio.Writer, re error) {
defer func() {
if err := recover(); err != nil {
re = err.(error)
}
}()
if writer != nil {
writer.WriteString(content)
writer.Flush()
return writer, nil
} else {
return writer, errors.New("writer is nil.")
}
}
type GetFilesOption struct {
subDir string //子目录
blankDir bool //是否包含空目录
hideObj bool //是否包含隐藏内容
abs bool //包括绝对路径
goNext bool //条件符合后是否查子目录
}
func NewGetFilesOption() GetFilesOption {
return GetFilesOption{
subDir: "",
blankDir: false,
hideObj: true,
abs: false,
goNext: false,
}
}
// WalkDir 获取符合条件的子目录
// TODO: 当碰到已符合条件的目录后,是否继续查这个目录下的子目录? 这个还未实现
func WalkDir(basePath string, filer func(name string) bool, option GetFilesOption) ([]string, error) {
exist, _ := Exist(basePath)
if !exist {
return nil, fmt.Errorf("[%v] is not exist", basePath)
}
result := make([]string, 0)
err := pathUtil.WalkDir(basePath, func(path string, info fs.DirEntry, err error) error {
curr, _ := pathUtil.Rel(basePath, path)
if "." == curr {
return nil
}
if !info.IsDir() {
return nil
}
// 过滤隐藏文件
if option.hideObj && strings.HasPrefix(info.Name(), ".") {
return nil
} else if filer(curr) {
if !option.abs && pathUtil.IsAbs(path) {
path, _ = pathUtil.Rel(basePath, path)
}
result = append(result, path)
}
return nil
})
return result, err
}
func Walk(basePath string, filer func(name string, isDir bool) bool, option GetFilesOption) ([]string, error) {
exist, _ := Exist(basePath)
if !exist {
return nil, fmt.Errorf("[%v] is not exist", basePath)
}
result := make([]string, 0)
err := pathUtil.Walk(basePath, func(path string, info fs.FileInfo, err error) error {
// 过滤隐藏文件
if option.hideObj && strings.HasPrefix(info.Name(), ".") {
return nil
} else if filer(info.Name(), info.IsDir()) {
if !option.abs && pathUtil.IsAbs(path) {
path, _ = pathUtil.Rel(basePath, path)
}
result = append(result, path)
}
return nil
})
return result, err
}
func GetFiles(path string, filter func(name string, isDir bool) bool, option GetFilesOption) ([]string, error) {
exist, _ := Exist(path)
if !exist {
return nil, fmt.Errorf("[%v] is not exist", path)
}
entries, err := os.ReadDir(pathUtil.Clean(pathUtil.ToSlash(path)))
if err != nil {
return nil, err
}
objs := make([]fs.FileInfo, 0, len(entries))
for _, entry := range entries {
info, err := entry.Info()
if err != nil {
return nil, err
}
objs = append(objs, info)
}
var result []string
for _, obj := range objs {
// 过滤隐藏文件
if option.hideObj && strings.HasPrefix(obj.Name(), ".") {
continue
}
isDir := obj.IsDir()
name := Join(option.subDir, obj.Name())
if filter != nil && !filter(name, isDir) {
continue
}
if !isDir {
result = append(result, name)
} else {
subOptions := option
subOptions.subDir = Join(option.subDir, obj.Name())
subResult, re := GetFiles(Join(path, obj.Name()), filter, subOptions)
if re == nil {
result = append(result, subResult...)
}
}
}
return result, nil
}
func GetSubDirs(path string, filter func(name string) bool) ([]string, error) {
exist, _ := Exist(path)
if !exist {
return nil, fmt.Errorf("[%v] is not exist", path)
}
entries, err := os.ReadDir(pathUtil.Clean(path))
if err != nil {
return nil, err
}
objs := make([]fs.FileInfo, 0, len(entries))
for _, entry := range entries {
info, err := entry.Info()
if err != nil {
return nil, err
}
objs = append(objs, info)
}
//objs, err := ioutil.ReadDir(pathUtil.Clean(path))
if err != nil {
return nil, err
}
result := make([]string, 0, len(objs))
for _, obj := range objs {
if obj.IsDir() {
name := obj.Name()
if filter == nil || filter(name) {
result = append(result, obj.Name())
}
}
}
return result, nil
}
// CopyFile over 是否覆盖
func CopyFile(source, target string, over bool) error {
if ex, _ := Exist(source); !ex {
return errors.New("the source is not exist.")
}
if ex, _ := Exist(target); ex && !over {
return errors.New("the target is exist and require not override.")
} else {
os.Remove(target)
}
tp, _ := pathUtil.Split(target)
if _, ce := CreatePath(tp, GetFileMode(700)); ce != nil {
return ce
} else {
src, err := os.Open(source)
if err != nil {
return err
}
defer src.Close()
dst, err := os.OpenFile(target, os.O_WRONLY|os.O_CREATE, GetFileMode(644))
if err != nil {
return err
}
defer dst.Close()
_, err = io.Copy(dst, src)
return err
}
}
func DeleteFile(target string) {
if ex, _ := Exist(target); ex {
os.RemoveAll(target)
}
}
func MoveFile(source, target string) {
if err := CopyFile(source, target, true); err != nil {
DeleteFile(source)
}
}
// GetSize 获取文件大小
func GetSize(f multipart.File) (int, error) {
content, err := ioutil.ReadAll(f)
return len(content), err
}
// GetExt 获取文件后缀
func GetExt(fileName string) string {
fileType := pathUtil.Ext(fileName)
if fileType != "" {
return fileType[1:]
} else {
return ""
}
}
func GetName(filePath string) string {
fileNameWithExt := path.Base(filePath)
fileExt := GetExt(fileNameWithExt)
tmp := strings.TrimSuffix(fileNameWithExt, fileExt)
// 删除末位的点
if len(tmp) > 1 && strings.HasSuffix(tmp, ".") {
tmp = tmp[0 : len(tmp)-1]
}
return tmp
}
// CheckExist 检查文件是否存在
func CheckExist(src string) bool {
_, err := os.Stat(src)
return os.IsNotExist(err)
}
// CheckPermission 检查文件权限
func CheckPermission(src string) bool {
_, err := os.Stat(src)
return os.IsPermission(err)
}
// IsNotExistMkDir 检查文件夹是否存在
// 如果不存在则新建文件夹
func IsNotExistMkDir(src string) error {
if exist := !CheckExist(src); exist == false {
if err := MkDir(src); err != nil {
return err
}
}
return nil
}
// MkDir 新建文件夹
func MkDir(src string) error {
err := os.MkdirAll(src, os.ModePerm)
//err := os.MkdirAll(src, 0777)
if err != nil {
return err
}
return nil
}
// Open 打开文件
func Open(name string, flag int, perm os.FileMode) (*os.File, error) {
f, err := os.OpenFile(name, flag, perm)
if err != nil {
return nil, err
}
return f, nil
}
// GetImgType 获取Img文件类型
func GetImgType(p string) (string, error) {
file, err := os.Open(p)
if err != nil {
log.Println(err)
os.Exit(1)
}
buff := make([]byte, 512)
_, err = file.Read(buff)
if err != nil {
log.Println(err)
os.Exit(1)
}
filetype := http.DetectContentType(buff)
ext := imgext.Get()
for i := 0; i < len(ext); i++ {
if strings.Contains(ext[i], filetype[6:len(filetype)]) {
return filetype, nil
}
}
return "", errors.New("Invalid image type")
}
// GetType 获取文件类型
func GetType(p string) (string, error) {
file, err := os.Open(p)
if err != nil {
log.Println(err)
os.Exit(1)
}
buff := make([]byte, 512)
_, err = file.Read(buff)
if err != nil {
log.Println(err)
}
filetype := http.DetectContentType(buff)
//ext := GetExt(p)
//var list = strings.Split(filetype, "/")
//filetype = list[0] + "/" + ext
return filetype, nil
}
func ReplaceStr(src, dst string, rmap map[string]string) error {
exist, e := Exist(src)
if e != nil {
return e
}
if !exist {
return fmt.Errorf("source file [%v] is not exist", src)
}
exist, e = Exist(dst)
if e != nil {
return e
}
if exist {
DeleteFile(dst)
}
in, err := os.Open(src)
if err != nil {
return err
}
defer in.Close()
out, err := os.OpenFile(dst, os.O_RDWR|os.O_CREATE, 0766)
if err != nil {
fmt.Println("Open write file fail:", err)
os.Exit(-1)
}
defer out.Close()
br := bufio.NewReader(in)
for {
line, _, err := br.ReadLine()
if err == io.EOF {
break
}
if err != nil {
return err
}
//newLine := strings.Replace(string(line), os.Args[2], os.Args[3], -1)
newLine := string(line)
for k, v := range rmap {
newLine = strings.ReplaceAll(newLine, k, v)
}
_, err = out.WriteString(newLine + "\n")
if err != nil {
return err
}
}
return nil
}
// FormatSize 转换文件大小的单位
func FormatSize(fileSize int64) (size string) {
if fileSize < 1024 {
//return strconv.FormatInt(fileSize, 10) + "B"
return fmt.Sprintf("%.2fB", float64(fileSize)/float64(1))
} else if fileSize < (1024 * 1024) {
return fmt.Sprintf("%.2fKB", float64(fileSize)/float64(1024))
} else if fileSize < (1024 * 1024 * 1024) {
return fmt.Sprintf("%.2fMB", float64(fileSize)/float64(1024*1024))
} else if fileSize < (1024 * 1024 * 1024 * 1024) {
return fmt.Sprintf("%.2fGB", float64(fileSize)/float64(1024*1024*1024))
} else if fileSize < (1024 * 1024 * 1024 * 1024 * 1024) {
return fmt.Sprintf("%.2fTB", float64(fileSize)/float64(1024*1024*1024*1024))
} else { //if fileSize < (1024 * 1024 * 1024 * 1024 * 1024 * 1024)
return fmt.Sprintf("%.2fEB", float64(fileSize)/float64(1024*1024*1024*1024*1024))
}
}
func FileMD5(filePath string) (string, error) {
file, err := os.Open(filePath)
if err != nil {
return "", err
}
hash := md5.New()
_, _ = io.Copy(hash, file)
return hex.EncodeToString(hash.Sum(nil)), nil
}
马建仓 AI 助手
尝试更多
代码解读
代码找茬
代码优化
Go
1
https://gitee.com/vincent78/gcutil.git
git@gitee.com:vincent78/gcutil.git
vincent78
gcutil
gcutil
v1.0.1

Search