146 lines
2.8 KiB
Go
146 lines
2.8 KiB
Go
// 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
|
|
}
|