TRKJ/EdgeNode
2
0
Fork 0
mirror of https://github.com/TeaOSLab/EdgeNode.git synced 2025-11-07 10:40:26 +08:00
Code Issues Packages Projects Releases Wiki Activity

对缓存文件关闭事件也增加写入统计

Browse Source
This commit is contained in:
GoEdgeLab
2023-08-02 08:51:31 +08:00
parent 76e764e4e8
commit be80256aab
2 changed files with 18 additions and 0 deletions
Download Patch File Download Diff File Expand all files Collapse all files

8
internal/caches/writer_file.go
View File

@@ -138,18 +138,24 @@ func (this *FileWriter) Close() error {
err := this.WriteHeaderLength(types.Int(this.headerSize)) err := this.WriteHeaderLength(types.Int(this.headerSize))
if err != nil { if err != nil {
fsutils.WriteBegin()
_ = this.rawWriter.Close() _ = this.rawWriter.Close()
fsutils.WriteEnd()
_ = os.Remove(path) _ = os.Remove(path)
return err return err
} }
err = this.WriteBodyLength(this.bodySize) err = this.WriteBodyLength(this.bodySize)
if err != nil { if err != nil {
fsutils.WriteBegin()
_ = this.rawWriter.Close() _ = this.rawWriter.Close()
fsutils.WriteEnd()
_ = os.Remove(path) _ = os.Remove(path)
return err return err
} }
fsutils.WriteBegin()
err = this.rawWriter.Close() err = this.rawWriter.Close()
fsutils.WriteEnd()
if err != nil { if err != nil {
_ = os.Remove(path) _ = os.Remove(path)
} else if strings.HasSuffix(path, FileTmpSuffix) { } else if strings.HasSuffix(path, FileTmpSuffix) {
@@ -168,7 +174,9 @@ func (this *FileWriter) Discard() error {
this.endFunc() this.endFunc()
}) })
fsutils.WriteBegin()
_ = this.rawWriter.Close() _ = this.rawWriter.Close()
fsutils.WriteEnd()
err := os.Remove(this.rawWriter.Name()) err := os.Remove(this.rawWriter.Name())
return err return err

10
internal/caches/writer_partial_file.go
View File

@@ -181,7 +181,9 @@ func (this *PartialFileWriter) Close() error {
this.ranges.BodySize = this.bodySize this.ranges.BodySize = this.bodySize
err := this.ranges.WriteToFile(this.rangePath) err := this.ranges.WriteToFile(this.rangePath)
if err != nil { if err != nil {
fsutils.WriteBegin()
_ = this.rawWriter.Close() _ = this.rawWriter.Close()
fsutils.WriteEnd()
this.remove() this.remove()
return err return err
} }
@@ -190,19 +192,25 @@ func (this *PartialFileWriter) Close() error {
if this.isNew { if this.isNew {
err = this.WriteHeaderLength(types.Int(this.headerSize)) err = this.WriteHeaderLength(types.Int(this.headerSize))
if err != nil { if err != nil {
fsutils.WriteBegin()
_ = this.rawWriter.Close() _ = this.rawWriter.Close()
fsutils.WriteEnd()
this.remove() this.remove()
return err return err
} }
err = this.WriteBodyLength(this.bodySize) err = this.WriteBodyLength(this.bodySize)
if err != nil { if err != nil {
fsutils.WriteBegin()
_ = this.rawWriter.Close() _ = this.rawWriter.Close()
fsutils.WriteEnd()
this.remove() this.remove()
return err return err
} }
} }
fsutils.WriteBegin()
err = this.rawWriter.Close() err = this.rawWriter.Close()
fsutils.WriteEnd()
if err != nil { if err != nil {
this.remove() this.remove()
} }
@@ -216,7 +224,9 @@ func (this *PartialFileWriter) Discard() error {
this.endFunc() this.endFunc()
}) })
fsutils.WriteBegin()
_ = this.rawWriter.Close() _ = this.rawWriter.Close()
fsutils.WriteEnd()
_ = os.Remove(this.rangePath) _ = os.Remove(this.rangePath)