UNPKG

common-hook

Version:
47 lines (46 loc) 1.76 kB
//@ts-nocheck var __awaiter = (this && this.__awaiter) || function (thisArg, _arguments, P, generator) { function adopt(value) { return value instanceof P ? value : new P(function (resolve) { resolve(value); }); } return new (P || (P = Promise))(function (resolve, reject) { function fulfilled(value) { try { step(generator.next(value)); } catch (e) { reject(e); } } function rejected(value) { try { step(generator["throw"](value)); } catch (e) { reject(e); } } function step(result) { result.done ? resolve(result.value) : adopt(result.value).then(fulfilled, rejected); } step((generator = generator.apply(thisArg, _arguments || [])).next()); }); }; import { useEffect } from "react"; import { isFunction } from "common-screw"; /** * @name 支持异步函数 * @description 例如:组件加载时进行异步的检查 * @example * useAsyncEffect(async () => {setPass(await mockCheck())}, []); */ export const useAsyncEffect = (effect, deps) => { const isAsyncGenerator = (val) => { return isFunction(val[Symbol.asyncIterator]); }; useEffect(() => { const e = effect(); let cancelled = false; function execute() { return __awaiter(this, void 0, void 0, function* () { if (isAsyncGenerator(e)) { while (true) { const result = yield e.next(); if (result.done || cancelled) { break; } } } else { yield e; } }); } execute(); return () => { cancelled = true; }; }, deps); };