Commit 710f5ae2 authored by Klaus Aehlig's avatar Klaus Aehlig
Browse files

Provide utility to atomically write a file



To keep our on-file data consistent at any moment, we change
file contents by atomically replacing the file with a new one.
Signed-off-by: default avatarKlaus Aehlig <aehlig@google.com>
Reviewed-by: default avatarMichele Tartara <mtartara@google.com>
parent 3ff2cf55
......@@ -60,6 +60,7 @@ module Ganeti.Utils
, resolveAddr
, setOwnerAndGroupFromNames
, formatOrdinal
, atomicWriteFile
) where
import Data.Char (toUpper, isAlphaNum, isDigit, isSpace)
......@@ -67,6 +68,8 @@ import Data.Function (on)
import Data.List
import qualified Data.Map as M
import Control.Monad (foldM)
import System.Directory (renameFile)
import System.FilePath.Posix (takeDirectory, takeBaseName)
import Debug.Trace
import Network.Socket
......@@ -485,3 +488,12 @@ formatOrdinal num
| otherwise = suffix "th"
where tens = num `mod` 10
suffix s = show num ++ s
-- | Atomically write a file, by first writing the contents into a temporary
-- file and then renaming it to the old position.
atomicWriteFile :: FilePath -> String -> IO ()
atomicWriteFile path contents = do
(tmppath, tmphandle) <- openTempFile (takeDirectory path) (takeBaseName path)
hPutStr tmphandle contents
hClose tmphandle
renameFile tmppath path
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment