import { useCallback, useState } from 'react';
const getType = (value) => {
const type = Object.prototype.toString.call(value);
if (type === '[object Object]') {
return 'isObject'
} else if (type === '[object Array]') {
return 'isArray'
}
return typeof value;
}
const useMerge = (initialState) => {
const [state, setState] = useState(initialState);
const stateType = getType(initialState);
const handleMerge = useCallback((handleObject = {}) => {
const { id, type, value } = handleObject;
const merge = (state) => {
let mergeData = {};
if (stateType === 'isObject') {
mergeData = {
...state,
...value
}
}
if (stateType === 'isArray') {
if (type === 'remove') {
mergeData = state.filter((item) => {
return item.id !== id
})
}
if (type === 'add') {
mergeData = [...state, value];
}
if (type === 'update') {
mergeData = state.map((item) => {
if (item.id === id) {
return {
...item,
...value
}
}
return item;
})
}
}
return mergeData;
}
setState((prevState) => merge(prevState))
}, [])
return [state, handleMerge];
}
export default useMerge;
这是一个专为移动设备优化的页面(即为了让你能够在 Google 搜索结果里秒开这个页面),如果你希望参与 V2EX 社区的讨论,你可以继续到 V2EX 上打开本讨论主题的完整版本。
V2EX 是创意工作者们的社区,是一个分享自己正在做的有趣事物、交流想法,可以遇见新朋友甚至新机会的地方。
V2EX is a community of developers, designers and creative people.