index.js 3.8 KB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119
  1. function defaultEqualityCheck(a, b) {
  2. return a === b;
  3. }
  4. function areArgumentsShallowlyEqual(equalityCheck, prev, next) {
  5. if (prev === null || next === null || prev.length !== next.length) {
  6. return false;
  7. }
  8. // Do this in a for loop (and not a `forEach` or an `every`) so we can determine equality as fast as possible.
  9. var length = prev.length;
  10. for (var i = 0; i < length; i++) {
  11. if (!equalityCheck(prev[i], next[i])) {
  12. return false;
  13. }
  14. }
  15. return true;
  16. }
  17. export function defaultMemoize(func) {
  18. var equalityCheck = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : defaultEqualityCheck;
  19. var lastArgs = null;
  20. var lastResult = null;
  21. // we reference arguments instead of spreading them for performance reasons
  22. return function () {
  23. if (!areArgumentsShallowlyEqual(equalityCheck, lastArgs, arguments)) {
  24. // apply arguments instead of spreading for performance.
  25. lastResult = func.apply(null, arguments);
  26. }
  27. lastArgs = arguments;
  28. return lastResult;
  29. };
  30. }
  31. function getDependencies(funcs) {
  32. var dependencies = Array.isArray(funcs[0]) ? funcs[0] : funcs;
  33. if (!dependencies.every(function (dep) {
  34. return typeof dep === 'function';
  35. })) {
  36. var dependencyTypes = dependencies.map(function (dep) {
  37. return typeof dep;
  38. }).join(', ');
  39. throw new Error('Selector creators expect all input-selectors to be functions, ' + ('instead received the following types: [' + dependencyTypes + ']'));
  40. }
  41. return dependencies;
  42. }
  43. export function createSelectorCreator(memoize) {
  44. for (var _len = arguments.length, memoizeOptions = Array(_len > 1 ? _len - 1 : 0), _key = 1; _key < _len; _key++) {
  45. memoizeOptions[_key - 1] = arguments[_key];
  46. }
  47. return function () {
  48. for (var _len2 = arguments.length, funcs = Array(_len2), _key2 = 0; _key2 < _len2; _key2++) {
  49. funcs[_key2] = arguments[_key2];
  50. }
  51. var recomputations = 0;
  52. var resultFunc = funcs.pop();
  53. var dependencies = getDependencies(funcs);
  54. var memoizedResultFunc = memoize.apply(undefined, [function () {
  55. recomputations++;
  56. // apply arguments instead of spreading for performance.
  57. return resultFunc.apply(null, arguments);
  58. }].concat(memoizeOptions));
  59. // If a selector is called with the exact same arguments we don't need to traverse our dependencies again.
  60. var selector = defaultMemoize(function () {
  61. var params = [];
  62. var length = dependencies.length;
  63. for (var i = 0; i < length; i++) {
  64. // apply arguments instead of spreading and mutate a local list of params for performance.
  65. params.push(dependencies[i].apply(null, arguments));
  66. }
  67. // apply arguments instead of spreading for performance.
  68. return memoizedResultFunc.apply(null, params);
  69. });
  70. selector.resultFunc = resultFunc;
  71. selector.recomputations = function () {
  72. return recomputations;
  73. };
  74. selector.resetRecomputations = function () {
  75. return recomputations = 0;
  76. };
  77. return selector;
  78. };
  79. }
  80. export var createSelector = createSelectorCreator(defaultMemoize);
  81. export function createStructuredSelector(selectors) {
  82. var selectorCreator = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : createSelector;
  83. if (typeof selectors !== 'object') {
  84. throw new Error('createStructuredSelector expects first argument to be an object ' + ('where each property is a selector, instead received a ' + typeof selectors));
  85. }
  86. var objectKeys = Object.keys(selectors);
  87. return selectorCreator(objectKeys.map(function (key) {
  88. return selectors[key];
  89. }), function () {
  90. for (var _len3 = arguments.length, values = Array(_len3), _key3 = 0; _key3 < _len3; _key3++) {
  91. values[_key3] = arguments[_key3];
  92. }
  93. return values.reduce(function (composition, value, index) {
  94. composition[objectKeys[index]] = value;
  95. return composition;
  96. }, {});
  97. });
  98. }