forked from Mirrors/elk
37 lines
854 B
TypeScript
37 lines
854 B
TypeScript
import type { Driver } from 'unstorage'
|
|
import memory from 'unstorage/drivers/memory'
|
|
import { defineDriver } from 'unstorage'
|
|
|
|
export interface CacheDriverOptions {
|
|
driver: Driver
|
|
}
|
|
|
|
export default defineDriver((driver: Driver = memory()) => {
|
|
const memoryDriver = memory()
|
|
return {
|
|
...driver,
|
|
async hasItem(key: string) {
|
|
if (await memoryDriver.hasItem(key))
|
|
return true
|
|
|
|
return driver.hasItem(key)
|
|
},
|
|
async setItem(key: string, value: any) {
|
|
await Promise.all([
|
|
memoryDriver.setItem(key, value),
|
|
driver.setItem?.(key, value),
|
|
])
|
|
},
|
|
async getItem(key: string) {
|
|
let value = await memoryDriver.getItem(key)
|
|
|
|
if (value !== null)
|
|
return value
|
|
|
|
value = await driver.getItem(key)
|
|
memoryDriver.setItem(key, value)
|
|
|
|
return value
|
|
},
|
|
}
|
|
})
|