useEffectCleanupRegister.js 2.4 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152
  1. "use strict";
  2. var _interopRequireDefault = require("@babel/runtime/helpers/interopRequireDefault");
  3. var _typeof = require("@babel/runtime/helpers/typeof");
  4. Object.defineProperty(exports, "__esModule", {
  5. value: true
  6. });
  7. exports.default = void 0;
  8. var _objectSpread2 = _interopRequireDefault(require("@babel/runtime/helpers/objectSpread2"));
  9. var _warning = require("rc-util/lib/warning");
  10. var React = _interopRequireWildcard(require("react"));
  11. function _getRequireWildcardCache(e) { if ("function" != typeof WeakMap) return null; var r = new WeakMap(), t = new WeakMap(); return (_getRequireWildcardCache = function _getRequireWildcardCache(e) { return e ? t : r; })(e); }
  12. function _interopRequireWildcard(e, r) { if (!r && e && e.__esModule) return e; if (null === e || "object" != _typeof(e) && "function" != typeof e) return { default: e }; var t = _getRequireWildcardCache(r); if (t && t.has(e)) return t.get(e); var n = { __proto__: null }, a = Object.defineProperty && Object.getOwnPropertyDescriptor; for (var u in e) if ("default" !== u && Object.prototype.hasOwnProperty.call(e, u)) { var i = a ? Object.getOwnPropertyDescriptor(e, u) : null; i && (i.get || i.set) ? Object.defineProperty(n, u, i) : n[u] = e[u]; } return n.default = e, t && t.set(e, n), n; }
  13. var fullClone = (0, _objectSpread2.default)({}, React);
  14. var useInsertionEffect = fullClone.useInsertionEffect;
  15. // DO NOT register functions in useEffect cleanup function, or functions that registered will never be called.
  16. var useCleanupRegister = function useCleanupRegister(deps) {
  17. var effectCleanups = [];
  18. var cleanupFlag = false;
  19. function register(fn) {
  20. if (cleanupFlag) {
  21. if (process.env.NODE_ENV !== 'production') {
  22. (0, _warning.warning)(false, '[Ant Design CSS-in-JS] You are registering a cleanup function after unmount, which will not have any effect.');
  23. }
  24. return;
  25. }
  26. effectCleanups.push(fn);
  27. }
  28. React.useEffect(function () {
  29. // Compatible with strict mode
  30. cleanupFlag = false;
  31. return function () {
  32. cleanupFlag = true;
  33. if (effectCleanups.length) {
  34. effectCleanups.forEach(function (fn) {
  35. return fn();
  36. });
  37. }
  38. };
  39. }, deps);
  40. return register;
  41. };
  42. var useRun = function useRun() {
  43. return function (fn) {
  44. fn();
  45. };
  46. };
  47. // Only enable register in React 18
  48. var useEffectCleanupRegister = typeof useInsertionEffect !== 'undefined' ? useCleanupRegister : useRun;
  49. var _default = exports.default = useEffectCleanupRegister;