1 | "use strict";
|
2 |
|
3 | Object.defineProperty(exports, "__esModule", {
|
4 | value: true
|
5 | });
|
6 | exports.assertSimpleType = assertSimpleType;
|
7 | exports.makeStrongCache = makeStrongCache;
|
8 | exports.makeStrongCacheSync = makeStrongCacheSync;
|
9 | exports.makeWeakCache = makeWeakCache;
|
10 | exports.makeWeakCacheSync = makeWeakCacheSync;
|
11 |
|
12 | function _gensync() {
|
13 | const data = require("gensync");
|
14 |
|
15 | _gensync = function () {
|
16 | return data;
|
17 | };
|
18 |
|
19 | return data;
|
20 | }
|
21 |
|
22 | var _async = require("../gensync-utils/async");
|
23 |
|
24 | var _util = require("./util");
|
25 |
|
26 | const synchronize = gen => {
|
27 | return _gensync()(gen).sync;
|
28 | };
|
29 |
|
30 | function* genTrue() {
|
31 | return true;
|
32 | }
|
33 |
|
34 | function makeWeakCache(handler) {
|
35 | return makeCachedFunction(WeakMap, handler);
|
36 | }
|
37 |
|
38 | function makeWeakCacheSync(handler) {
|
39 | return synchronize(makeWeakCache(handler));
|
40 | }
|
41 |
|
42 | function makeStrongCache(handler) {
|
43 | return makeCachedFunction(Map, handler);
|
44 | }
|
45 |
|
46 | function makeStrongCacheSync(handler) {
|
47 | return synchronize(makeStrongCache(handler));
|
48 | }
|
49 |
|
50 | function makeCachedFunction(CallCache, handler) {
|
51 | const callCacheSync = new CallCache();
|
52 | const callCacheAsync = new CallCache();
|
53 | const futureCache = new CallCache();
|
54 | return function* cachedFunction(arg, data) {
|
55 | const asyncContext = yield* (0, _async.isAsync)();
|
56 | const callCache = asyncContext ? callCacheAsync : callCacheSync;
|
57 | const cached = yield* getCachedValueOrWait(asyncContext, callCache, futureCache, arg, data);
|
58 | if (cached.valid) return cached.value;
|
59 | const cache = new CacheConfigurator(data);
|
60 | const handlerResult = handler(arg, cache);
|
61 | let finishLock;
|
62 | let value;
|
63 |
|
64 | if ((0, _util.isIterableIterator)(handlerResult)) {
|
65 | const gen = handlerResult;
|
66 | value = yield* (0, _async.onFirstPause)(gen, () => {
|
67 | finishLock = setupAsyncLocks(cache, futureCache, arg);
|
68 | });
|
69 | } else {
|
70 | value = handlerResult;
|
71 | }
|
72 |
|
73 | updateFunctionCache(callCache, cache, arg, value);
|
74 |
|
75 | if (finishLock) {
|
76 | futureCache.delete(arg);
|
77 | finishLock.release(value);
|
78 | }
|
79 |
|
80 | return value;
|
81 | };
|
82 | }
|
83 |
|
84 | function* getCachedValue(cache, arg, data) {
|
85 | const cachedValue = cache.get(arg);
|
86 |
|
87 | if (cachedValue) {
|
88 | for (const {
|
89 | value,
|
90 | valid
|
91 | } of cachedValue) {
|
92 | if (yield* valid(data)) return {
|
93 | valid: true,
|
94 | value
|
95 | };
|
96 | }
|
97 | }
|
98 |
|
99 | return {
|
100 | valid: false,
|
101 | value: null
|
102 | };
|
103 | }
|
104 |
|
105 | function* getCachedValueOrWait(asyncContext, callCache, futureCache, arg, data) {
|
106 | const cached = yield* getCachedValue(callCache, arg, data);
|
107 |
|
108 | if (cached.valid) {
|
109 | return cached;
|
110 | }
|
111 |
|
112 | if (asyncContext) {
|
113 | const cached = yield* getCachedValue(futureCache, arg, data);
|
114 |
|
115 | if (cached.valid) {
|
116 | const value = yield* (0, _async.waitFor)(cached.value.promise);
|
117 | return {
|
118 | valid: true,
|
119 | value
|
120 | };
|
121 | }
|
122 | }
|
123 |
|
124 | return {
|
125 | valid: false,
|
126 | value: null
|
127 | };
|
128 | }
|
129 |
|
130 | function setupAsyncLocks(config, futureCache, arg) {
|
131 | const finishLock = new Lock();
|
132 | updateFunctionCache(futureCache, config, arg, finishLock);
|
133 | return finishLock;
|
134 | }
|
135 |
|
136 | function updateFunctionCache(cache, config, arg, value) {
|
137 | if (!config.configured()) config.forever();
|
138 | let cachedValue = cache.get(arg);
|
139 | config.deactivate();
|
140 |
|
141 | switch (config.mode()) {
|
142 | case "forever":
|
143 | cachedValue = [{
|
144 | value,
|
145 | valid: genTrue
|
146 | }];
|
147 | cache.set(arg, cachedValue);
|
148 | break;
|
149 |
|
150 | case "invalidate":
|
151 | cachedValue = [{
|
152 | value,
|
153 | valid: config.validator()
|
154 | }];
|
155 | cache.set(arg, cachedValue);
|
156 | break;
|
157 |
|
158 | case "valid":
|
159 | if (cachedValue) {
|
160 | cachedValue.push({
|
161 | value,
|
162 | valid: config.validator()
|
163 | });
|
164 | } else {
|
165 | cachedValue = [{
|
166 | value,
|
167 | valid: config.validator()
|
168 | }];
|
169 | cache.set(arg, cachedValue);
|
170 | }
|
171 |
|
172 | }
|
173 | }
|
174 |
|
175 | class CacheConfigurator {
|
176 | constructor(data) {
|
177 | this._active = true;
|
178 | this._never = false;
|
179 | this._forever = false;
|
180 | this._invalidate = false;
|
181 | this._configured = false;
|
182 | this._pairs = [];
|
183 | this._data = void 0;
|
184 | this._data = data;
|
185 | }
|
186 |
|
187 | simple() {
|
188 | return makeSimpleConfigurator(this);
|
189 | }
|
190 |
|
191 | mode() {
|
192 | if (this._never) return "never";
|
193 | if (this._forever) return "forever";
|
194 | if (this._invalidate) return "invalidate";
|
195 | return "valid";
|
196 | }
|
197 |
|
198 | forever() {
|
199 | if (!this._active) {
|
200 | throw new Error("Cannot change caching after evaluation has completed.");
|
201 | }
|
202 |
|
203 | if (this._never) {
|
204 | throw new Error("Caching has already been configured with .never()");
|
205 | }
|
206 |
|
207 | this._forever = true;
|
208 | this._configured = true;
|
209 | }
|
210 |
|
211 | never() {
|
212 | if (!this._active) {
|
213 | throw new Error("Cannot change caching after evaluation has completed.");
|
214 | }
|
215 |
|
216 | if (this._forever) {
|
217 | throw new Error("Caching has already been configured with .forever()");
|
218 | }
|
219 |
|
220 | this._never = true;
|
221 | this._configured = true;
|
222 | }
|
223 |
|
224 | using(handler) {
|
225 | if (!this._active) {
|
226 | throw new Error("Cannot change caching after evaluation has completed.");
|
227 | }
|
228 |
|
229 | if (this._never || this._forever) {
|
230 | throw new Error("Caching has already been configured with .never or .forever()");
|
231 | }
|
232 |
|
233 | this._configured = true;
|
234 | const key = handler(this._data);
|
235 | const fn = (0, _async.maybeAsync)(handler, `You appear to be using an async cache handler, but Babel has been called synchronously`);
|
236 |
|
237 | if ((0, _async.isThenable)(key)) {
|
238 | return key.then(key => {
|
239 | this._pairs.push([key, fn]);
|
240 |
|
241 | return key;
|
242 | });
|
243 | }
|
244 |
|
245 | this._pairs.push([key, fn]);
|
246 |
|
247 | return key;
|
248 | }
|
249 |
|
250 | invalidate(handler) {
|
251 | this._invalidate = true;
|
252 | return this.using(handler);
|
253 | }
|
254 |
|
255 | validator() {
|
256 | const pairs = this._pairs;
|
257 | return function* (data) {
|
258 | for (const [key, fn] of pairs) {
|
259 | if (key !== (yield* fn(data))) return false;
|
260 | }
|
261 |
|
262 | return true;
|
263 | };
|
264 | }
|
265 |
|
266 | deactivate() {
|
267 | this._active = false;
|
268 | }
|
269 |
|
270 | configured() {
|
271 | return this._configured;
|
272 | }
|
273 |
|
274 | }
|
275 |
|
276 | function makeSimpleConfigurator(cache) {
|
277 | function cacheFn(val) {
|
278 | if (typeof val === "boolean") {
|
279 | if (val) cache.forever();else cache.never();
|
280 | return;
|
281 | }
|
282 |
|
283 | return cache.using(() => assertSimpleType(val()));
|
284 | }
|
285 |
|
286 | cacheFn.forever = () => cache.forever();
|
287 |
|
288 | cacheFn.never = () => cache.never();
|
289 |
|
290 | cacheFn.using = cb => cache.using(() => assertSimpleType(cb()));
|
291 |
|
292 | cacheFn.invalidate = cb => cache.invalidate(() => assertSimpleType(cb()));
|
293 |
|
294 | return cacheFn;
|
295 | }
|
296 |
|
297 | function assertSimpleType(value) {
|
298 | if ((0, _async.isThenable)(value)) {
|
299 | throw new Error(`You appear to be using an async cache handler, ` + `which your current version of Babel does not support. ` + `We may add support for this in the future, ` + `but if you're on the most recent version of @babel/core and still ` + `seeing this error, then you'll need to synchronously handle your caching logic.`);
|
300 | }
|
301 |
|
302 | if (value != null && typeof value !== "string" && typeof value !== "boolean" && typeof value !== "number") {
|
303 | throw new Error("Cache keys must be either string, boolean, number, null, or undefined.");
|
304 | }
|
305 |
|
306 | return value;
|
307 | }
|
308 |
|
309 | class Lock {
|
310 | constructor() {
|
311 | this.released = false;
|
312 | this.promise = void 0;
|
313 | this._resolve = void 0;
|
314 | this.promise = new Promise(resolve => {
|
315 | this._resolve = resolve;
|
316 | });
|
317 | }
|
318 |
|
319 | release(value) {
|
320 | this.released = true;
|
321 |
|
322 | this._resolve(value);
|
323 | }
|
324 |
|
325 | } |
\ | No newline at end of file |