封装storage存储class
export enum StorageType {
session = 'session',
local = 'local'
}
export class MyStorage {
private prefix: string
private static _instance: Map<string, MyStorage>
private storage: Storage
private engine = {
[StorageType.session]: sessionStorage,
[StorageType.local]: localStorage
}
constructor(prefix = '', type: StorageType = StorageType.local) {
this.prefix = prefix
this.storage = this.engine[type]
if (!this.storage) {
throw new Error("engine type error. it must be 'session','local'")
}
if (!MyStorage._instance) {
MyStorage._instance = new Map()
}
if (!MyStorage._instance?.has(type)) {
MyStorage._instance.set(type, this)
}
return MyStorage._instance.get(type) as MyStorage
}
private getFullKey(value: string) {
return this.prefix + value
}
set(key: string, value: any, expire = 0) {
try {
const wrapData = {
value,
expire,
time: Date.now()
}
const data = JSON.stringify(wrapData)
this.storage.setItem(this.getFullKey(key), data)
return true
} catch (error) {
console.error(error)
return false
}
}
setInto(key: string, value: Record<string, any>, expire = 0) {
try {
const preData = this.get(key)
const wrapData = {
value,
expire,
time: Date.now()
}
if (preData) {
Object.assign(wrapData.value, preData, value)
}
const data = JSON.stringify(wrapData)
this.storage.setItem(this.getFullKey(key), data)
return true
} catch (error) {
console.error(error)
return false
}
}
get(key: string) {
try {
const localData = this.storage.getItem(this.getFullKey(key))
if (localData === null) return localData
const data = JSON.parse(localData)
if (data.expire > 0 && Date.now() - data.time > data.expire * 1000) {
this.remove(key)
return null
}
return data.value
} catch (error) {
console.error(error)
return null
}
}
remove(key: string) {
this.storage.removeItem(this.getFullKey(key))
}
clear(): void {
Object.keys(this.storage).forEach(key => {
if (key.startsWith(this.prefix)) {
this.storage.removeItem(key)
}
})
}
}
封装 vue hook
import { MyStorage, StorageType } from 'MyStorage.ts'
import { ref} from 'vue'
export const useStorage = (key: string, prefix = 'test_', type = 'local') => {
const storageHandle = new MyStorage(prefix, type as StorageType)
const value = ref(storageHandle.get(key))
const setValue = (storage: any) => {
return (newValue: any) => {
value.value = newValue
storage.set(key, newValue)
}
}
return [value, setValue(storageHandle), storageHandle]
}
vue中使用
<script setup lang="ts">
import {useStorage} from '@/hook/useStorage'
onMounted(() => {
let [value2, setValue2, b] = useStorage('b', 'prefix_')
let [value3, setValue3, c] = useStorage('c', 'prefix_')
let [value, setValue, a] = useStorage('a', 'test2_', 'session')
console.log('c === b', b === c)
console.log('storage1', value)
console.log('storage2', value2)
setValue({
age: 18,
name: 'lili',
famary: {
p1: '11',
p2: '22',
}
})
setValue2({
b: 'bb'
})
setTimeout(() => {
setValue3({
c: 'after fetch save'
})
console.log('set storage1', value)
console.log('set storage2', value2)
console.log('set storage3', value3)
}, 3000)
})
</script>