文件写入支持覆盖写入与追加写入支持

.gitignore

@@ -19,4 +19,5 @@
 .idea
 .vscode
 *_test.go
+*.txt
 

file.go

@@ -44,12 +44,13 @@ func File(workDir string, filePath string) *fileOperate {
 //
 // Date : 12:24 2022/5/22
 type fileOperate struct {
-	workDir     string
+	workDir         string
-	filePath    string
+	filePath        string
-	autoCreate  bool
+	autoCreate      bool
-	permission  os.FileMode
+	permission      os.FileMode
-	openFlag    int
+	openFlag        int
-	fileHandler *os.File
+	fileHandler     *os.File
+	clearHasContent bool // 文件写入时, 清空现有内容(覆盖写入)， 默认追加写入
 }
 
 // IsExist 文件是否存在
@@ -75,6 +76,7 @@ func (f *fileOperate) IsExist() (bool, bool, error) {
 // Date : 12:28 2022/5/22
 func (f *fileOperate) AutoCreate() *fileOperate {
 	f.autoCreate = true
+	f.openFlag = f.openFlag | os.O_CREATE
 	return f
 }
 
@@ -95,11 +97,26 @@ func (f *fileOperate) Permission(fileModel os.FileMode) *fileOperate {
 // Date : 13:40 2022/5/22
 func (f *fileOperate) SetOpenFlag(flagList ...int) *fileOperate {
 	for _, flag := range flagList {
+		if flag == os.O_CREATE {
+			f.AutoCreate()
+			continue
+		}
 		f.openFlag = f.openFlag | flag
 	}
 	return f
 }
 
+// SetClearHasContentForWrite 设置覆盖写入文件内容
+//
+// Author : go_developer@163.com<白茶清欢>
+//
+// Date : 16:41 2022/5/22
+func (f *fileOperate) SetClearHasContentForWrite() *fileOperate {
+	f.clearHasContent = true
+	f.openFlag = f.openFlag | os.O_TRUNC
+	return nil
+}
+
 // Create 创建文件
 //
 // Author : go_developer@163.com<白茶清欢>
@@ -210,6 +227,17 @@ func (f *fileOperate) Write(content []byte) error {
 			return errors.New("文件打开失败 : " + err.Error())
 		}
 	}
+	if f.clearHasContent {
+		// 覆盖写入
+		if err := os.Truncate(f.filePath, 0); nil != err {
+			return err
+		}
+		n, _ := f.fileHandler.Seek(0, io.SeekEnd)
+
+		_, writeErr := f.fileHandler.WriteAt(content, n)
+		return writeErr
+	}
+	// 追加写入
 	_, writeErr := f.fileHandler.Write(content)
 	return writeErr
 }