PromiseMap.js.flow 1.4 KB

1234567891011121314151617181920212223242526272829303132333435363738394041424344454647484950515253
  1. /**
  2. * Copyright (c) 2013-present, Facebook, Inc.
  3. *
  4. * This source code is licensed under the MIT license found in the
  5. * LICENSE file in the root directory of this source tree.
  6. *
  7. * @providesModule PromiseMap
  8. * @flow
  9. */
  10. 'use strict';
  11. const Deferred = require('./Deferred');
  12. const invariant = require('./invariant');
  13. /**
  14. * A map of asynchronous values that can be get or set in any order. Unlike a
  15. * normal map, setting the value for a particular key more than once throws.
  16. * Also unlike a normal map, a key can either be resolved or rejected.
  17. */
  18. class PromiseMap<Tvalue, Treason> {
  19. _deferred: { [key: string]: Deferred<any, any> };
  20. constructor() {
  21. this._deferred = {};
  22. }
  23. get(key: string): Promise<any> {
  24. return getDeferred(this._deferred, key).getPromise();
  25. }
  26. resolveKey(key: string, value: Tvalue): void {
  27. const entry = getDeferred(this._deferred, key);
  28. invariant(!entry.isSettled(), 'PromiseMap: Already settled `%s`.', key);
  29. entry.resolve(value);
  30. }
  31. rejectKey(key: string, reason: Treason): void {
  32. const entry = getDeferred(this._deferred, key);
  33. invariant(!entry.isSettled(), 'PromiseMap: Already settled `%s`.', key);
  34. entry.reject(reason);
  35. }
  36. }
  37. function getDeferred(entries: { [key: string]: Deferred<any, any> }, key: string): Deferred<any, any> {
  38. if (!entries.hasOwnProperty(key)) {
  39. entries[key] = new Deferred();
  40. }
  41. return entries[key];
  42. }
  43. module.exports = PromiseMap;