Functional updater form for store.set()

[jackwlee01]

store.set() should support the functional form, as per https://reactjs.org/docs/hooks-reference.html#functional-updates, in order update the state based on an instruction, instead of the current value (which may be out of date in an async scenario).