beenull/ente
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

Merge branch 'add-caching-support' into lru-eviction

Browse source
This commit is contained in:
Abhinav 2022-08-10 14:12:48 +05:30
parent 7fd2843d41 421efaaa6c
commit 9fd54d8f4a
1 changed files with 12 additions and 2 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

14
main/utils/cache.ts
View file

@ -14,6 +14,7 @@ import {
rmdir,
} from 'promise-fs';
import getFolderSize from 'get-folder-size';
import crypto from 'crypto';
interface LeastRecentlyUsedResult {
atime: Date;
@ -41,7 +42,7 @@ class DiskCache {
constructor(private cacheBucketDir: string) {}
async put(cacheKey: string, response: Response): Promise<void> {
const cachePath = path.join(this.cacheBucketDir, cacheKey);
const cachePath = makeAssetCachePath(this.cacheBucketDir, cacheKey);
evictLeastRecentlyUsed(this.cacheBucketDir, MAX_CACHE_SIZE);
await writeFile(
cachePath,
@ -50,7 +51,7 @@ class DiskCache {
}
async match(cacheKey: string): Promise<Response> {
const cachePath = path.join(this.cacheBucketDir, cacheKey);
const cachePath = makeAssetCachePath(this.cacheBucketDir, cacheKey);
if (existsSync(cachePath)) {
touch(cachePath);
return new Response(await readFile(cachePath));
@ -60,6 +61,15 @@ class DiskCache {
}
}
function makeAssetCachePath(cacheDir: string, cacheKey: string) {
// hashing the key to prevent illegal filenames
const cacheKeyHash = crypto
.createHash('sha256')
.update(cacheKey)
.digest('hex');
return path.join(cacheDir, cacheKeyHash);
}
async function touch(path: string) {
const time = new Date();
try {