useNextFrame.js 1.9 KB

12345678910111213141516171819202122232425262728293031323334353637383940
  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 _raf = _interopRequireDefault(require("rc-util/lib/raf"));
  9. var React = _interopRequireWildcard(require("react"));
  10. 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); }
  11. 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; }
  12. var _default = exports.default = function _default() {
  13. var nextFrameRef = React.useRef(null);
  14. function cancelNextFrame() {
  15. _raf.default.cancel(nextFrameRef.current);
  16. }
  17. function nextFrame(callback) {
  18. var delay = arguments.length > 1 && arguments[1] !== undefined ? arguments[1] : 2;
  19. cancelNextFrame();
  20. var nextFrameId = (0, _raf.default)(function () {
  21. if (delay <= 1) {
  22. callback({
  23. isCanceled: function isCanceled() {
  24. return nextFrameId !== nextFrameRef.current;
  25. }
  26. });
  27. } else {
  28. nextFrame(callback, delay - 1);
  29. }
  30. });
  31. nextFrameRef.current = nextFrameId;
  32. }
  33. React.useEffect(function () {
  34. return function () {
  35. cancelNextFrame();
  36. };
  37. }, []);
  38. return [nextFrame, cancelNextFrame];
  39. };