set-intersection.js 1.3 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546
  1. 'use strict';
  2. var aSet = require('../internals/a-set');
  3. var SetHelpers = require('../internals/set-helpers');
  4. var size = require('../internals/set-size');
  5. var getSetRecord = require('../internals/get-set-record');
  6. var iterateSet = require('../internals/set-iterate');
  7. var iterateSimple = require('../internals/iterate-simple');
  8. var Set = SetHelpers.Set;
  9. var add = SetHelpers.add;
  10. var has = SetHelpers.has;
  11. var nativeHas = SetHelpers.$has;
  12. var nativeKeys = SetHelpers.$keys;
  13. var isNativeSetRecord = function (record) {
  14. return record.has === nativeHas && record.keys === nativeKeys;
  15. };
  16. // `Set.prototype.intersection` method
  17. // https://github.com/tc39/proposal-set-methods
  18. module.exports = function intersection(other) {
  19. var O = aSet(this);
  20. var otherRec = getSetRecord(other);
  21. var result = new Set();
  22. // observable side effects
  23. if (!isNativeSetRecord(otherRec) && size(O) > otherRec.size) {
  24. iterateSimple(otherRec.getIterator(), function (e) {
  25. if (has(O, e)) add(result, e);
  26. });
  27. if (size(result) < 2) return result;
  28. var disordered = result;
  29. result = new Set();
  30. iterateSet(O, function (e) {
  31. if (has(disordered, e)) add(result, e);
  32. });
  33. } else {
  34. iterateSet(O, function (e) {
  35. if (otherRec.includes(e)) add(result, e);
  36. });
  37. }
  38. return result;
  39. };