71 lines
1.5 KiB
JavaScript
71 lines
1.5 KiB
JavaScript
import { useCallback, useEffect, useState } from "react";
|
|
import { storage } from "../libs/storage";
|
|
import { kissLog } from "../libs/log";
|
|
|
|
/**
|
|
*
|
|
* @param {*} key
|
|
* @param {*} defaultVal 需为调用hook外的常量
|
|
* @returns
|
|
*/
|
|
export function useStorage(key, defaultVal) {
|
|
const [loading, setLoading] = useState(false);
|
|
const [data, setData] = useState(null);
|
|
|
|
const save = useCallback(
|
|
async (val) => {
|
|
setData(val);
|
|
await storage.setObj(key, val);
|
|
},
|
|
[key]
|
|
);
|
|
|
|
const update = useCallback(
|
|
async (obj) => {
|
|
setData((pre = {}) => ({ ...pre, ...obj }));
|
|
await storage.putObj(key, obj);
|
|
},
|
|
[key]
|
|
);
|
|
|
|
const remove = useCallback(async () => {
|
|
setData(null);
|
|
await storage.del(key);
|
|
}, [key]);
|
|
|
|
const reload = useCallback(async () => {
|
|
try {
|
|
setLoading(true);
|
|
const val = await storage.getObj(key);
|
|
if (val) {
|
|
setData(val);
|
|
}
|
|
} catch (err) {
|
|
kissLog(err, "storage reload");
|
|
} finally {
|
|
setLoading(false);
|
|
}
|
|
}, [key]);
|
|
|
|
useEffect(() => {
|
|
(async () => {
|
|
try {
|
|
setLoading(true);
|
|
const val = await storage.getObj(key);
|
|
if (val) {
|
|
setData(val);
|
|
} else if (defaultVal) {
|
|
setData(defaultVal);
|
|
await storage.setObj(key, defaultVal);
|
|
}
|
|
} catch (err) {
|
|
kissLog(err, "storage load");
|
|
} finally {
|
|
setLoading(false);
|
|
}
|
|
})();
|
|
}, [key, defaultVal]);
|
|
|
|
return { data, save, update, remove, reload, loading };
|
|
}
|