mirror of
https://github.com/scm-manager/scm-manager.git
synced 2025-11-17 18:51:10 +01:00
Enhance error handling
This commit is contained in:
@@ -1,5 +1,8 @@
|
|||||||
package sonia.scm.store;
|
package sonia.scm.store;
|
||||||
|
|
||||||
|
import org.slf4j.Logger;
|
||||||
|
import org.slf4j.LoggerFactory;
|
||||||
|
|
||||||
import java.io.IOException;
|
import java.io.IOException;
|
||||||
import java.nio.file.Files;
|
import java.nio.file.Files;
|
||||||
import java.nio.file.Path;
|
import java.nio.file.Path;
|
||||||
@@ -7,20 +10,15 @@ import java.util.UUID;
|
|||||||
|
|
||||||
public final class CopyOnWrite {
|
public final class CopyOnWrite {
|
||||||
|
|
||||||
|
private static final Logger LOG = LoggerFactory.getLogger(CopyOnWrite.class);
|
||||||
|
|
||||||
private CopyOnWrite() {}
|
private CopyOnWrite() {}
|
||||||
|
|
||||||
public static void withTemporaryFile(FileWriter creator, Path targetFile) {
|
public static void withTemporaryFile(FileWriter writer, Path targetFile) {
|
||||||
validateInput(targetFile);
|
validateInput(targetFile);
|
||||||
|
Path temporaryFile = createTemporaryFile(targetFile);
|
||||||
try {
|
executeCallback(writer, targetFile, temporaryFile);
|
||||||
Path temporaryFile = Files.createFile(targetFile.getParent().resolve(UUID.randomUUID().toString()));
|
|
||||||
|
|
||||||
creator.write(temporaryFile);
|
|
||||||
|
|
||||||
replaceOriginalFile(targetFile, temporaryFile);
|
replaceOriginalFile(targetFile, temporaryFile);
|
||||||
} catch (Exception ex) {
|
|
||||||
throw new StoreException("could not write file", ex);
|
|
||||||
}
|
|
||||||
}
|
}
|
||||||
|
|
||||||
private static void validateInput(Path targetFile) {
|
private static void validateInput(Path targetFile) {
|
||||||
@@ -32,38 +30,72 @@ public final class CopyOnWrite {
|
|||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private static void replaceOriginalFile(Path targetFile, Path temporaryFile) throws IOException {
|
private static Path createTemporaryFile(Path targetFile) {
|
||||||
|
Path temporaryFile = targetFile.getParent().resolve(UUID.randomUUID().toString());
|
||||||
|
try {
|
||||||
|
Files.createFile(temporaryFile);
|
||||||
|
} catch (IOException ex) {
|
||||||
|
LOG.error("Error creating temporary file {} to replace file {}", temporaryFile, targetFile);
|
||||||
|
throw new StoreException("could create temporary file", ex);
|
||||||
|
}
|
||||||
|
return temporaryFile;
|
||||||
|
}
|
||||||
|
|
||||||
|
private static void executeCallback(FileWriter writer, Path targetFile, Path temporaryFile) {
|
||||||
|
try {
|
||||||
|
writer.write(temporaryFile);
|
||||||
|
} catch (RuntimeException e) {
|
||||||
|
throw e;
|
||||||
|
} catch (Exception ex) {
|
||||||
|
LOG.error("Error writing to temporary file {}. Target file {} has not been modified", temporaryFile, targetFile);
|
||||||
|
throw new StoreException("could not write temporary file", ex);
|
||||||
|
}
|
||||||
|
}
|
||||||
|
|
||||||
|
private static void replaceOriginalFile(Path targetFile, Path temporaryFile) {
|
||||||
Path backupFile = backupOriginalFile(targetFile);
|
Path backupFile = backupOriginalFile(targetFile);
|
||||||
try {
|
try {
|
||||||
Files.move(temporaryFile, targetFile);
|
Files.move(temporaryFile, targetFile);
|
||||||
if (backupFile != null) {
|
if (backupFile != null) {
|
||||||
Files.delete(backupFile);
|
Files.delete(backupFile);
|
||||||
}
|
}
|
||||||
} catch (RuntimeException | IOException e) {
|
} catch (IOException e) {
|
||||||
|
LOG.error("Error renaming temporary file {} to target file {}", temporaryFile, targetFile);
|
||||||
restoreBackup(targetFile, backupFile);
|
restoreBackup(targetFile, backupFile);
|
||||||
throw e;
|
throw new StoreException("could rename temporary file to target file", e);
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private static Path backupOriginalFile(Path targetFile) throws IOException {
|
private static Path backupOriginalFile(Path targetFile) {
|
||||||
Path directory = targetFile.getParent();
|
Path directory = targetFile.getParent();
|
||||||
if (Files.exists(targetFile)) {
|
if (Files.exists(targetFile)) {
|
||||||
Path backupFile = directory.resolve(UUID.randomUUID().toString());
|
Path backupFile = directory.resolve(UUID.randomUUID().toString());
|
||||||
|
try {
|
||||||
Files.move(targetFile, backupFile);
|
Files.move(targetFile, backupFile);
|
||||||
|
} catch (IOException e) {
|
||||||
|
LOG.error("Could not backup original file {}. Aborting here so that original file will not be overwritten.", targetFile);
|
||||||
|
throw new StoreException("could not create backup of file", e);
|
||||||
|
}
|
||||||
return backupFile;
|
return backupFile;
|
||||||
} else {
|
} else {
|
||||||
return null;
|
return null;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
private static void restoreBackup(Path targetFile, Path backupFile) throws IOException {
|
private static void restoreBackup(Path targetFile, Path backupFile) {
|
||||||
if (backupFile != null) {
|
if (backupFile != null) {
|
||||||
|
try {
|
||||||
Files.move(backupFile, targetFile);
|
Files.move(backupFile, targetFile);
|
||||||
|
LOG.info("Recovered original file {} from backup", targetFile);
|
||||||
|
} catch (IOException e) {
|
||||||
|
LOG.error("Could not replace original file {} with backup file {} after failure", targetFile, backupFile);
|
||||||
|
}
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|
||||||
@FunctionalInterface
|
@FunctionalInterface
|
||||||
public interface FileWriter {
|
public interface FileWriter {
|
||||||
|
@SuppressWarnings("squid:S00112") // We do not want to limit exceptions here
|
||||||
void write(Path t) throws Exception;
|
void write(Path t) throws Exception;
|
||||||
}
|
}
|
||||||
}
|
}
|
||||||
|
|||||||
@@ -73,6 +73,19 @@ class CopyOnWriteTest {
|
|||||||
Assertions.assertThat(unchangedOriginalFile).hasContent("this should be kept");
|
Assertions.assertThat(unchangedOriginalFile).hasContent("this should be kept");
|
||||||
}
|
}
|
||||||
|
|
||||||
|
@Test
|
||||||
|
void shouldNotWrapRuntimeExceptions(@TempDirectory.TempDir Path tempDir) throws IOException {
|
||||||
|
Path someFile = tempDir.resolve("something.txt");
|
||||||
|
|
||||||
|
assertThrows(
|
||||||
|
NullPointerException.class,
|
||||||
|
() -> withTemporaryFile(
|
||||||
|
file -> {
|
||||||
|
throw new NullPointerException("test");
|
||||||
|
},
|
||||||
|
someFile));
|
||||||
|
}
|
||||||
|
|
||||||
@Test
|
@Test
|
||||||
void shouldKeepBackupIfTemporaryFileIsMissing(@TempDirectory.TempDir Path tempDir) throws IOException {
|
void shouldKeepBackupIfTemporaryFileIsMissing(@TempDirectory.TempDir Path tempDir) throws IOException {
|
||||||
Path backedUpFile = tempDir.resolve("notToBeDeleted.txt");
|
Path backedUpFile = tempDir.resolve("notToBeDeleted.txt");
|
||||||
|
|||||||
Reference in New Issue
Block a user