rxjs
Version:
Reactive Extensions for modern JavaScript
45 lines • 1.77 kB
JavaScript
import { operate } from '../util/lift';
import { OperatorSubscriber } from './OperatorSubscriber';
import { innerFrom } from '../observable/from';
export const defaultThrottleConfig = {
leading: true,
trailing: false,
};
export function throttle(durationSelector, { leading, trailing } = defaultThrottleConfig) {
return operate((source, subscriber) => {
let hasValue = false;
let sendValue = null;
let throttled = null;
let isComplete = false;
const endThrottling = () => {
throttled === null || throttled === void 0 ? void 0 : throttled.unsubscribe();
throttled = null;
if (trailing) {
send();
isComplete && subscriber.complete();
}
};
const cleanupThrottling = () => {
throttled = null;
isComplete && subscriber.complete();
};
const startThrottle = (value) => (throttled = innerFrom(durationSelector(value)).subscribe(new OperatorSubscriber(subscriber, endThrottling, undefined, cleanupThrottling)));
const send = () => {
if (hasValue) {
subscriber.next(sendValue);
!isComplete && startThrottle(sendValue);
}
hasValue = false;
sendValue = null;
};
source.subscribe(new OperatorSubscriber(subscriber, (value) => {
hasValue = true;
sendValue = value;
!(throttled && !throttled.closed) && (leading ? send() : startThrottle(value));
}, undefined, () => {
isComplete = true;
!(trailing && hasValue && throttled && !throttled.closed) && subscriber.complete();
}));
});
}
//# sourceMappingURL=throttle.js.map