@aws-amplify/datastore/src/storage/adapter/InMemoryStore.ts

AppSyncLocal support for aws-amplify

export class InMemoryStore {
	db = new Map<string, string>();

	getAllKeys = async () => {
		return Array.from(this.db.keys());
	};

	multiGet = async (keys: string[]) => {
		return keys.reduce(
			(res, k) => (res.push([k, this.db.get(k)!]), res),
			[] as [string, string][]
		);
	};

	multiRemove = async (keys: string[], callback?) => {
		keys.forEach((k) => this.db.delete(k));

		typeof callback === 'function' && callback();
	};

	multiSet = async (entries: string[][], callback?) => {
		entries.forEach(([key, value]) => {
			this.setItem(key, value);
		});

		typeof callback === 'function' && callback();
	};

	setItem = async (key: string, value: string) => {
		return this.db.set(key, value);
	};

	removeItem = async (key: string) => {
		return this.db.delete(key);
	};

	getItem = async (key: string) => {
		return this.db.get(key);
	};
}

export function createInMemoryStore() {
	return new InMemoryStore();
}