equalsIterable.js.flow 1.7 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172
  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 equalsIterable
  8. * @flow
  9. */
  10. 'use strict';
  11. const enumerate = require("./enumerate");
  12. /**
  13. * Checks if two iterables are equal. A custom areEqual function may be provided
  14. * as an optional third argument.
  15. */
  16. function equalsIterable<T>(one: Iterable<T>, two: Iterable<T>, areEqual?: ?(one: T, two: T) => boolean): boolean {
  17. if (one === two) {
  18. return true;
  19. } // We might be able to short circuit by using the size or length fields.
  20. var oneSize = maybeGetSize(one);
  21. var twoSize = maybeGetSize(two);
  22. if (oneSize != null && twoSize != null && oneSize !== twoSize) {
  23. return false;
  24. } // Otherwise use the iterators to check equality. Here we cannot use for-of
  25. // because we need to advance the iterators at the same time.
  26. var oneIterator = enumerate(one);
  27. var oneItem = oneIterator.next();
  28. var twoIterator = enumerate(two);
  29. var twoItem = twoIterator.next();
  30. var safeAreEqual = areEqual || referenceEquality;
  31. while (!(oneItem.done || twoItem.done)) {
  32. if (!safeAreEqual(oneItem.value, twoItem.value)) {
  33. return false;
  34. }
  35. oneItem = oneIterator.next();
  36. twoItem = twoIterator.next();
  37. }
  38. return oneItem.done === twoItem.done;
  39. }
  40. function maybeGetSize(o: any): ?number {
  41. if (o == null) {
  42. return null;
  43. }
  44. if (typeof o.size === 'number') {
  45. return o.size;
  46. }
  47. if (typeof o.length === 'number') {
  48. return o.length;
  49. }
  50. return null;
  51. }
  52. function referenceEquality<T>(one: T, two: T): boolean {
  53. return one === two;
  54. }
  55. module.exports = equalsIterable;