[Storage] Add wrapSync promise accessor

This commit is contained in:
redstonekasi 2023-01-30 14:58:06 +01:00 committed by Beef
parent 44652c4002
commit 48edf97b9a

View file

@ -5,6 +5,7 @@ import createEmitter from "./emitter";
const MMKVManager = RN.NativeModules.MMKVManager as MMKVManager; const MMKVManager = RN.NativeModules.MMKVManager as MMKVManager;
const emitterSymbol = Symbol("emitter accessor"); const emitterSymbol = Symbol("emitter accessor");
const syncAwaitSymbol = Symbol("wrapSync promise accessor");
export function createProxy(target: any = {}): { proxy: any, emitter: Emitter } { export function createProxy(target: any = {}): { proxy: any, emitter: Emitter } {
const emitter = createEmitter(); const emitter = createEmitter();
@ -91,11 +92,25 @@ export async function createStorage<T>(storeName: string): Promise<Awaited<T>> {
export function wrapSync<T extends Promise<any>>(store: T): Awaited<T> { export function wrapSync<T extends Promise<any>>(store: T): Awaited<T> {
let awaited: any = undefined; let awaited: any = undefined;
store.then((v) => (awaited = v));
return new Proxy( const awaitQueue: (() => void)[] = [];
{} as Awaited<T>, const awaitInit = (cb: () => void) => (awaited ? cb() : awaitQueue.push(cb));
Object.fromEntries(Object.getOwnPropertyNames(Reflect)
store.then((v) => {
awaited = v;
awaitQueue.forEach((cb) => cb());
});
return new Proxy({} as Awaited<T>, {
...Object.fromEntries(Object.getOwnPropertyNames(Reflect)
// @ts-expect-error // @ts-expect-error
.map((k) => [k, (t: T, ...a: any[]) => Reflect[k](awaited ?? t, ...a)])), .map((k) => [k, (t: T, ...a: any[]) => Reflect[k](awaited ?? t, ...a)])),
); get(target, prop, recv) {
if (prop === syncAwaitSymbol)
return awaitInit;
return Reflect.get(awaited ?? target, prop, recv);
} }
});
}
export const awaitSyncWrapper = (store: any) => new Promise<void>((res) => store[syncAwaitSymbol](res));