GithubFollow/Surge_by_SukkaW
1
0
Fork 0
mirror of https://github.com/SukkaW/Surge.git synced 2025-12-12 09:10:35 +08:00
Code Issues Actions 2 Packages Projects Releases Wiki Activity

Cache SQLite statement

Browse Source
This commit is contained in:
SukkaW 2024-10-18 01:55:47 +08:00
parent a8c9cc5ac5
commit 4c1c237fc3
1 changed files with 21 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

33
Build/lib/cache-filesystem.ts
View File

@ -1,5 +1,5 @@
import createDb from 'better-sqlite3'; import createDb from 'better-sqlite3';
import type { Database } from 'better-sqlite3'; import type { Database, Statement } from 'better-sqlite3';
import os from 'node:os'; import os from 'node:os';
import path from 'node:path'; import path from 'node:path';
import { mkdirSync } from 'node:fs'; import { mkdirSync } from 'node:fs';
@ -80,7 +80,7 @@ function ensureETag(headers: IncomingHttpHeaders | Headers) {
} }
export class Cache<S = string> { export class Cache<S = string> {
db: Database; private db: Database;
/** Time before deletion */ /** Time before deletion */
tbd = 60 * 1000; tbd = 60 * 1000;
/** SQLite file path */ /** SQLite file path */
@ -89,6 +89,13 @@ export class Cache<S = string> {
tableName: string; tableName: string;
type: S extends string ? 'string' : 'buffer'; type: S extends string ? 'string' : 'buffer';
private statement: {
updateTtl: Statement<[number, string]>,
del: Statement<[string]>,
insert: Statement<[unknown]>,
get: Statement<[string], { ttl: number, value: S }>
};
constructor({ constructor({
cachePath = path.join(os.tmpdir() || '/tmp', 'hdc'), cachePath = path.join(os.tmpdir() || '/tmp', 'hdc'),
tbd, tbd,
@ -118,6 +125,14 @@ export class Cache<S = string> {
db.prepare(`CREATE TABLE IF NOT EXISTS ${this.tableName} (key TEXT PRIMARY KEY, value ${this.type === 'string' ? 'TEXT' : 'BLOB'}, ttl REAL NOT NULL);`).run(); db.prepare(`CREATE TABLE IF NOT EXISTS ${this.tableName} (key TEXT PRIMARY KEY, value ${this.type === 'string' ? 'TEXT' : 'BLOB'}, ttl REAL NOT NULL);`).run();
db.prepare(`CREATE INDEX IF NOT EXISTS cache_ttl ON ${this.tableName} (ttl);`).run(); db.prepare(`CREATE INDEX IF NOT EXISTS cache_ttl ON ${this.tableName} (ttl);`).run();
/** cache stmt */
this.statement = {
updateTtl: db.prepare(`UPDATE ${this.tableName} SET ttl = ? WHERE key = ?;`),
del: db.prepare(`DELETE FROM ${this.tableName} WHERE key = ?`),
insert: db.prepare(`INSERT INTO ${this.tableName} (key, value, ttl) VALUES ($key, $value, $valid) ON CONFLICT(key) DO UPDATE SET value = $value, ttl = $valid`),
get: db.prepare(`SELECT ttl, value FROM ${this.tableName} WHERE key = ? LIMIT 1`)
} as const;
const date = new Date(); const date = new Date();
// perform purge on startup // perform purge on startup
@ -142,13 +157,9 @@ export class Cache<S = string> {
} }
set(key: string, value: string, ttl = 60 * 1000): void { set(key: string, value: string, ttl = 60 * 1000): void {
const insert = this.db.prepare(
`INSERT INTO ${this.tableName} (key, value, ttl) VALUES ($key, $value, $valid) ON CONFLICT(key) DO UPDATE SET value = $value, ttl = $valid`
);
const valid = Date.now() + ttl; const valid = Date.now() + ttl;
insert.run({ this.statement.insert.run({
$key: key, $key: key,
key, key,
$value: value, $value: value,
@ -159,9 +170,7 @@ export class Cache<S = string> {
} }
get(key: string): S | null { get(key: string): S | null {
const rv = this.db.prepare<string, { value: S, ttl: number }>( const rv = this.statement.get.get(key);
`SELECT ttl, value FROM ${this.tableName} WHERE key = ? LIMIT 1`
).get(key);
if (!rv) return null; if (!rv) return null;
@ -179,11 +188,11 @@ export class Cache<S = string> {
} }
updateTtl(key: string, ttl: number): void { updateTtl(key: string, ttl: number): void {
this.db.prepare(`UPDATE ${this.tableName} SET ttl = ? WHERE key = ?;`).run(Date.now() + ttl, key); this.statement.updateTtl.run(Date.now() + ttl, key);
} }
del(key: string): void { del(key: string): void {
this.db.prepare(`DELETE FROM ${this.tableName} WHERE key = ?`).run(key); this.statement.del.run(key);
} }
async applyWithHttp304<T>( async applyWithHttp304<T>(