mirror of
https://github.com/mikefarah/yq.git
synced 2025-01-23 22:25:42 +00:00
Fixed #1062
This commit is contained in:
parent
84ddf1862f
commit
78b45a3eb0
@ -1,30 +1,28 @@
|
|||||||
package yqlib
|
package yqlib
|
||||||
|
|
||||||
import (
|
import (
|
||||||
|
"fmt"
|
||||||
"io"
|
"io"
|
||||||
"io/ioutil"
|
"io/ioutil"
|
||||||
"os"
|
"os"
|
||||||
)
|
)
|
||||||
|
|
||||||
func safelyRenameFile(from string, to string) {
|
func tryRenameFile(from string, to string) {
|
||||||
if renameError := os.Rename(from, to); renameError != nil {
|
if renameError := os.Rename(from, to); renameError != nil {
|
||||||
log.Debugf("Error renaming from %v to %v, attempting to copy contents", from, to)
|
log.Debugf("Error renaming from %v to %v, attempting to copy contents", from, to)
|
||||||
log.Debug(renameError.Error())
|
log.Debug(renameError.Error())
|
||||||
|
log.Debug("going to try copying instead")
|
||||||
// can't do this rename when running in docker to a file targeted in a mounted volume,
|
// can't do this rename when running in docker to a file targeted in a mounted volume,
|
||||||
// so gracefully degrade to copying the entire contents.
|
// so gracefully degrade to copying the entire contents.
|
||||||
if copyError := copyFileContents(from, to); copyError != nil {
|
if copyError := copyFileContents(from, to); copyError != nil {
|
||||||
log.Errorf("Failed copying from %v to %v", from, to)
|
panic(fmt.Errorf("failed copying from %v to %v: %w", from, to, copyError))
|
||||||
log.Error(copyError.Error())
|
|
||||||
} else {
|
} else {
|
||||||
removeErr := os.Remove(from)
|
tryRemoveTempFile(from)
|
||||||
if removeErr != nil {
|
|
||||||
log.Errorf("failed removing original file: %s", from)
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
func tryRemoveFile(filename string) {
|
func tryRemoveTempFile(filename string) {
|
||||||
log.Debug("Removing temp file: %v", filename)
|
log.Debug("Removing temp file: %v", filename)
|
||||||
removeErr := os.Remove(filename)
|
removeErr := os.Remove(filename)
|
||||||
if removeErr != nil {
|
if removeErr != nil {
|
||||||
|
@ -33,7 +33,7 @@ func (f *frontMatterHandlerImpl) GetContentReader() io.Reader {
|
|||||||
}
|
}
|
||||||
|
|
||||||
func (f *frontMatterHandlerImpl) CleanUp() {
|
func (f *frontMatterHandlerImpl) CleanUp() {
|
||||||
tryRemoveFile(f.yamlFrontMatterFilename)
|
tryRemoveTempFile(f.yamlFrontMatterFilename)
|
||||||
}
|
}
|
||||||
|
|
||||||
// Splits the given file by yaml front matter
|
// Splits the given file by yaml front matter
|
||||||
|
@ -66,7 +66,7 @@ yaml: doc
|
|||||||
}
|
}
|
||||||
test.AssertResult(t, expectedContent, string(contentBytes))
|
test.AssertResult(t, expectedContent, string(contentBytes))
|
||||||
|
|
||||||
tryRemoveFile(file)
|
tryRemoveTempFile(file)
|
||||||
fmHandler.CleanUp()
|
fmHandler.CleanUp()
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -103,7 +103,7 @@ yaml: doc
|
|||||||
}
|
}
|
||||||
test.AssertResult(t, expectedContent, string(contentBytes))
|
test.AssertResult(t, expectedContent, string(contentBytes))
|
||||||
|
|
||||||
tryRemoveFile(file)
|
tryRemoveTempFile(file)
|
||||||
fmHandler.CleanUp()
|
fmHandler.CleanUp()
|
||||||
}
|
}
|
||||||
|
|
||||||
@ -137,6 +137,6 @@ yaml: doc
|
|||||||
}
|
}
|
||||||
test.AssertResult(t, expectedContent, string(contentBytes))
|
test.AssertResult(t, expectedContent, string(contentBytes))
|
||||||
|
|
||||||
tryRemoveFile(file)
|
tryRemoveTempFile(file)
|
||||||
fmHandler.CleanUp()
|
fmHandler.CleanUp()
|
||||||
}
|
}
|
||||||
|
@ -44,8 +44,8 @@ func (w *writeInPlaceHandlerImpl) FinishWriteInPlace(evaluatedSuccessfully bool)
|
|||||||
safelyCloseFile(w.tempFile)
|
safelyCloseFile(w.tempFile)
|
||||||
if evaluatedSuccessfully {
|
if evaluatedSuccessfully {
|
||||||
log.Debug("Moving temp file to target")
|
log.Debug("Moving temp file to target")
|
||||||
safelyRenameFile(w.tempFile.Name(), w.inputFilename)
|
tryRenameFile(w.tempFile.Name(), w.inputFilename)
|
||||||
} else {
|
} else {
|
||||||
tryRemoveFile(w.tempFile.Name())
|
tryRemoveTempFile(w.tempFile.Name())
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
Loading…
Reference in New Issue
Block a user