useReducer调度后如何获取状态？

Question

是否有任何方法可以在分派后获取更新状态？

想象一下，我正在尝试编写一些从上下文获取状态的钩子（其中值是useReducer的结果）根据状态调度动作并返回更新的状态：

const useHook = (state, dispatch) => {
    if (state.flag) {
        dispatch({
            type: `UNSET_FLAG`,
        })
    } else {
        dispatch({
            type: `SET_FLAG`,
        })
    }
}

const Foo = () => {
    const [state, dispatch] = useContext(MyContext);

    useHook(state, dispatch);

    return state.flag ? <Flagged /> : <Unflagged /> ;
}

在此先感谢您的任何意见和想法

P.S。出于描述目的，该示例只是我脑海中的第一件事。