56 lines
1.6 KiB
TypeScript
56 lines
1.6 KiB
TypeScript
import { useState, useEffect } from 'react'
|
||
|
||
export function useLocalStorage<T>(
|
||
key: string,
|
||
initialValue: T
|
||
): [T, (value: T | ((prev: T) => T)) => void] {
|
||
// 获取初始值
|
||
const [storedValue, setStoredValue] = useState<T>(() => {
|
||
if (typeof window === 'undefined') {
|
||
return initialValue
|
||
}
|
||
|
||
try {
|
||
const item = window.localStorage.getItem(key)
|
||
return item ? JSON.parse(item) : initialValue
|
||
} catch (error) {
|
||
console.warn(`Error reading localStorage key "${key}":`, error)
|
||
return initialValue
|
||
}
|
||
})
|
||
|
||
// 设置值的函数
|
||
const setValue = (value: T | ((prev: T) => T)) => {
|
||
try {
|
||
// 允许value是一个函数，类似于useState
|
||
const valueToStore =
|
||
value instanceof Function ? value(storedValue) : value
|
||
|
||
setStoredValue(valueToStore)
|
||
|
||
if (typeof window !== 'undefined') {
|
||
window.localStorage.setItem(key, JSON.stringify(valueToStore))
|
||
}
|
||
} catch (error) {
|
||
console.warn(`Error setting localStorage key "${key}":`, error)
|
||
}
|
||
}
|
||
|
||
// 监听localStorage变化
|
||
useEffect(() => {
|
||
const handleStorageChange = (event: StorageEvent) => {
|
||
if (event.key === key && event.newValue !== null) {
|
||
try {
|
||
setStoredValue(JSON.parse(event.newValue))
|
||
} catch (error) {
|
||
console.warn(`Error parsing localStorage change for key "${key}":`, error)
|
||
}
|
||
}
|
||
}
|
||
|
||
window.addEventListener('storage', handleStorageChange)
|
||
return () => window.removeEventListener('storage', handleStorageChange)
|
||
}, [key])
|
||
|
||
return [storedValue, setValue]
|
||
} |