hokein updated this revision to Diff 536698. hokein added a comment. Restrict the patch to only remove the writeFileAtomically API
Repository: rG LLVM Github Monorepo CHANGES SINCE LAST ACTION https://reviews.llvm.org/D153740/new/ https://reviews.llvm.org/D153740 Files: llvm/include/llvm/Support/FileUtilities.h llvm/lib/Support/FileUtilities.cpp llvm/unittests/Support/CMakeLists.txt llvm/unittests/Support/FileUtilitiesTest.cpp
Index: llvm/unittests/Support/FileUtilitiesTest.cpp =================================================================== --- llvm/unittests/Support/FileUtilitiesTest.cpp +++ /dev/null @@ -1,54 +0,0 @@ -//===- llvm/unittest/Support/FileUtilitiesTest.cpp - unit tests -----------===// -// -// Part of the LLVM Project, under the Apache License v2.0 with LLVM Exceptions. -// See https://llvm.org/LICENSE.txt for license information. -// SPDX-License-Identifier: Apache-2.0 WITH LLVM-exception -// -//===----------------------------------------------------------------------===// - -#include "llvm/Support/FileUtilities.h" -#include "llvm/Support/Errc.h" -#include "llvm/Support/ErrorHandling.h" -#include "llvm/Support/FileSystem.h" -#include "llvm/Support/MemoryBuffer.h" -#include "llvm/Support/Path.h" -#include "llvm/Testing/Support/SupportHelpers.h" -#include "gtest/gtest.h" -#include <fstream> - -using namespace llvm; -using namespace llvm::sys; - -using llvm::unittest::TempDir; - -#define ASSERT_NO_ERROR(x) \ - if (std::error_code ASSERT_NO_ERROR_ec = x) { \ - SmallString<128> MessageStorage; \ - raw_svector_ostream Message(MessageStorage); \ - Message << #x ": did not return errc::success.\n" \ - << "error number: " << ASSERT_NO_ERROR_ec.value() << "\n" \ - << "error message: " << ASSERT_NO_ERROR_ec.message() << "\n"; \ - GTEST_FATAL_FAILURE_(MessageStorage.c_str()); \ - } else { \ - } - -namespace { -TEST(writeFileAtomicallyTest, Test) { - // Create unique temporary directory for these tests - TempDir RootTestDirectory("writeFileAtomicallyTest", /*Unique*/ true); - - SmallString<128> FinalTestfilePath(RootTestDirectory.path()); - sys::path::append(FinalTestfilePath, "foo.txt"); - const std::string TempUniqTestFileModel = - std::string(FinalTestfilePath) + "-%%%%%%%%"; - const std::string TestfileContent = "fooFOOfoo"; - - llvm::Error Err = llvm::writeFileAtomically(TempUniqTestFileModel, FinalTestfilePath, TestfileContent); - ASSERT_FALSE(static_cast<bool>(Err)); - - std::ifstream FinalFileStream(std::string(FinalTestfilePath.str())); - std::string FinalFileContent; - FinalFileStream >> FinalFileContent; - ASSERT_EQ(FinalFileContent, TestfileContent); -} -} // anonymous namespace Index: llvm/unittests/Support/CMakeLists.txt =================================================================== --- llvm/unittests/Support/CMakeLists.txt +++ llvm/unittests/Support/CMakeLists.txt @@ -41,7 +41,6 @@ ExtensibleRTTITest.cpp FileCollectorTest.cpp FileOutputBufferTest.cpp - FileUtilitiesTest.cpp FormatVariadicTest.cpp FSUniqueIDTest.cpp GlobPatternTest.cpp Index: llvm/lib/Support/FileUtilities.cpp =================================================================== --- llvm/lib/Support/FileUtilities.cpp +++ llvm/lib/Support/FileUtilities.cpp @@ -267,64 +267,6 @@ return CompareFailed; } -void llvm::AtomicFileWriteError::log(raw_ostream &OS) const { - OS << "atomic_write_error: "; - switch (Error) { - case atomic_write_error::failed_to_create_uniq_file: - OS << "failed_to_create_uniq_file"; - return; - case atomic_write_error::output_stream_error: - OS << "output_stream_error"; - return; - case atomic_write_error::failed_to_rename_temp_file: - OS << "failed_to_rename_temp_file"; - return; - } - llvm_unreachable("unknown atomic_write_error value in " - "failed_to_rename_temp_file::log()"); -} - -llvm::Error llvm::writeFileAtomically(StringRef TempPathModel, - StringRef FinalPath, StringRef Buffer) { - return writeFileAtomically(TempPathModel, FinalPath, - [&Buffer](llvm::raw_ostream &OS) { - OS.write(Buffer.data(), Buffer.size()); - return llvm::Error::success(); - }); -} - -llvm::Error llvm::writeFileAtomically( - StringRef TempPathModel, StringRef FinalPath, - std::function<llvm::Error(llvm::raw_ostream &)> Writer) { - SmallString<128> GeneratedUniqPath; - int TempFD; - if (sys::fs::createUniqueFile(TempPathModel, TempFD, GeneratedUniqPath)) { - return llvm::make_error<AtomicFileWriteError>( - atomic_write_error::failed_to_create_uniq_file); - } - llvm::FileRemover RemoveTmpFileOnFail(GeneratedUniqPath); - - raw_fd_ostream OS(TempFD, /*shouldClose=*/true); - if (llvm::Error Err = Writer(OS)) { - return Err; - } - - OS.close(); - if (OS.has_error()) { - OS.clear_error(); - return llvm::make_error<AtomicFileWriteError>( - atomic_write_error::output_stream_error); - } - - if (sys::fs::rename(/*from=*/GeneratedUniqPath, /*to=*/FinalPath)) { - return llvm::make_error<AtomicFileWriteError>( - atomic_write_error::failed_to_rename_temp_file); - } - - RemoveTmpFileOnFail.releaseFile(); - return Error::success(); -} - Expected<FilePermissionsApplier> FilePermissionsApplier::create(StringRef InputFilename) { sys::fs::file_status Status; @@ -389,5 +331,3 @@ return Error::success(); } - -char llvm::AtomicFileWriteError::ID; Index: llvm/include/llvm/Support/FileUtilities.h =================================================================== --- llvm/include/llvm/Support/FileUtilities.h +++ llvm/include/llvm/Support/FileUtilities.h @@ -76,41 +76,6 @@ void releaseFile() { DeleteIt = false; } }; - enum class atomic_write_error { - failed_to_create_uniq_file = 0, - output_stream_error, - failed_to_rename_temp_file - }; - - class AtomicFileWriteError : public llvm::ErrorInfo<AtomicFileWriteError> { - public: - AtomicFileWriteError(atomic_write_error Error) : Error(Error) {} - - void log(raw_ostream &OS) const override; - - const atomic_write_error Error; - static char ID; - - private: - // Users are not expected to use error_code. - std::error_code convertToErrorCode() const override { - return llvm::inconvertibleErrorCode(); - } - }; - - // atomic_write_error + whatever the Writer can return - - /// Creates a unique file with name according to the given \p TempPathModel, - /// writes content of \p Buffer to the file and renames it to \p FinalPath. - /// - /// \returns \c AtomicFileWriteError in case of error. - llvm::Error writeFileAtomically(StringRef TempPathModel, StringRef FinalPath, - StringRef Buffer); - - llvm::Error - writeFileAtomically(StringRef TempPathModel, StringRef FinalPath, - std::function<llvm::Error(llvm::raw_ostream &)> Writer); - /// FilePermssionsApplier helps to copy permissions from an input file to /// an output one. It memorizes the status of the input file and can apply /// permissions and dates to the output file.
_______________________________________________ cfe-commits mailing list cfe-commits@lists.llvm.org https://lists.llvm.org/cgi-bin/mailman/listinfo/cfe-commits
