File size: 1,711 Bytes
4e23b01 | 1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 69 70 71 72 73 | import { randomBytes } from 'node:crypto';
import { closeSync, fsyncSync, openSync } from 'node:fs';
import * as nodeFs from 'node:fs';
import { open, rename, unlink } from 'node:fs/promises';
export async function syncDir(dirPath: string): Promise<void> {
if (process.platform === 'win32') return;
const dirFh = await open(dirPath, 'r');
try {
await dirFh.sync();
} finally {
await dirFh.close();
}
}
export function syncDirSync(dirPath: string): void {
if (process.platform === 'win32') return;
const fd = openSync(dirPath, 'r');
try {
fsyncSync(fd);
} finally {
closeSync(fd);
}
}
function syncFd(fd: number): Promise<void> {
return new Promise<void>((resolve, reject) => {
nodeFs.fsync(fd, (err) => {
if (err) {
reject(err);
return;
}
resolve();
});
});
}
export async function atomicWrite(
filePath: string,
content: string | Uint8Array,
_syncOverride?: (fd: number) => Promise<void>,
): Promise<void> {
const hex = randomBytes(4).toString('hex');
const tmpPath = `${filePath}.tmp.${process.pid}.${hex}`;
let renamed = false;
try {
const fh = await open(tmpPath, 'w');
try {
await fh.writeFile(content);
await (_syncOverride ?? syncFd)(fh.fd);
} finally {
await fh.close();
}
if (process.platform === 'win32') {
try {
await unlink(filePath);
} catch (error) {
const code = (error as NodeJS.ErrnoException).code;
if (code !== 'ENOENT') throw error;
}
}
await rename(tmpPath, filePath);
renamed = true;
} finally {
if (!renamed) {
try {
await unlink(tmpPath);
} catch {
}
}
}
}
|