增加文件相关操作
This commit is contained in:
parent
c890e06f01
commit
da134e32a1
145
file.go
Normal file
145
file.go
Normal file
@ -0,0 +1,145 @@
|
||||
// Package command ...
|
||||
//
|
||||
// Description : 文件相关操作
|
||||
//
|
||||
// Author : go_developer@163.com<白茶清欢>
|
||||
//
|
||||
// Date : 2022-05-22 12:23
|
||||
package command
|
||||
|
||||
import (
|
||||
"errors"
|
||||
"os"
|
||||
"strings"
|
||||
)
|
||||
|
||||
// File ...
|
||||
//
|
||||
// Author : go_developer@163.com<白茶清欢>
|
||||
//
|
||||
// Date : 12:25 2022/5/22
|
||||
func File(workDir string, filePath string) *fileOperate {
|
||||
if len(workDir) == 0 {
|
||||
workDir, _ = os.Getwd()
|
||||
}
|
||||
if !strings.HasPrefix(filePath, "/") {
|
||||
filePath = strings.ReplaceAll(workDir+"/"+filePath, "//", "/")
|
||||
}
|
||||
return &fileOperate{
|
||||
workDir: workDir,
|
||||
filePath: filePath,
|
||||
permission: 0644,
|
||||
openFlag: os.O_RDONLY,
|
||||
}
|
||||
}
|
||||
|
||||
// file ...
|
||||
//
|
||||
// Author : go_developer@163.com<白茶清欢>
|
||||
//
|
||||
// Date : 12:24 2022/5/22
|
||||
type fileOperate struct {
|
||||
workDir string
|
||||
filePath string
|
||||
autoCreate bool
|
||||
permission os.FileMode
|
||||
openFlag int
|
||||
fileHandler *os.File
|
||||
}
|
||||
|
||||
// IsExist 文件是否存在
|
||||
//
|
||||
// Author : go_developer@163.com<白茶清欢>
|
||||
//
|
||||
// Date : 12:30 2022/5/22
|
||||
func (f *fileOperate) IsExist() (bool, bool, error) {
|
||||
fileState, err := os.Stat(f.filePath)
|
||||
if err != nil {
|
||||
if os.IsNotExist(err) {
|
||||
return false, false, nil
|
||||
}
|
||||
return false, false, err
|
||||
}
|
||||
return true, fileState.IsDir(), nil
|
||||
}
|
||||
|
||||
// AutoCreate 文件不存在时自动创建
|
||||
//
|
||||
// Author : go_developer@163.com<白茶清欢>
|
||||
//
|
||||
// Date : 12:28 2022/5/22
|
||||
func (f *fileOperate) AutoCreate() *fileOperate {
|
||||
f.autoCreate = true
|
||||
return f
|
||||
}
|
||||
|
||||
// Permission 设置文件权限
|
||||
//
|
||||
// Author : go_developer@163.com<白茶清欢>
|
||||
//
|
||||
// Date : 13:00 2022/5/22
|
||||
func (f *fileOperate) Permission(fileModel os.FileMode) *fileOperate {
|
||||
f.permission = fileModel
|
||||
return f
|
||||
}
|
||||
|
||||
// SetOpenFlag ...
|
||||
//
|
||||
// Author : go_developer@163.com<白茶清欢>
|
||||
//
|
||||
// Date : 13:40 2022/5/22
|
||||
func (f *fileOperate) SetOpenFlag(flagList ...int) *fileOperate {
|
||||
for _, flag := range flagList {
|
||||
f.openFlag = f.openFlag | flag
|
||||
}
|
||||
return f
|
||||
}
|
||||
|
||||
// Create 创建文件
|
||||
//
|
||||
// Author : go_developer@163.com<白茶清欢>
|
||||
//
|
||||
// Date : 13:04 2022/5/22
|
||||
func (f *fileOperate) Create() error {
|
||||
|
||||
var (
|
||||
fileExist bool
|
||||
isDir bool
|
||||
err error
|
||||
)
|
||||
|
||||
if fileExist, isDir, err = f.IsExist(); nil != err {
|
||||
return err
|
||||
}
|
||||
|
||||
if isDir {
|
||||
return errors.New("路径已存在, 且为目录 : " + f.filePath)
|
||||
}
|
||||
|
||||
if !fileExist {
|
||||
// 文件不存在
|
||||
if !f.autoCreate {
|
||||
return errors.New("文件不存在, 且不允许自动创建 : " + f.filePath)
|
||||
}
|
||||
}
|
||||
f.fileHandler, err = os.OpenFile(f.filePath, f.openFlag, f.permission)
|
||||
return err
|
||||
}
|
||||
|
||||
// Open 打开文件
|
||||
//
|
||||
// Author : go_developer@163.com<白茶清欢>
|
||||
//
|
||||
// Date : 13:22 2022/5/22
|
||||
func (f *fileOperate) Open() error {
|
||||
return f.Create()
|
||||
}
|
||||
|
||||
// GetFileHandler 获取文件处理句柄
|
||||
//
|
||||
// Author : go_developer@163.com<白茶清欢>
|
||||
//
|
||||
// Date : 13:23 2022/5/22
|
||||
func (f *fileOperate) GetFileHandler() *os.File {
|
||||
return f.fileHandler
|
||||
}
|
Loading…
Reference in New Issue
Block a user