[6a3a178] | 1 | import { Subscriber } from '../Subscriber';
|
---|
| 2 | import { Subscription } from '../Subscription';
|
---|
| 3 | import { Observable } from '../Observable';
|
---|
| 4 | import { Subject } from '../Subject';
|
---|
| 5 | export function groupBy(keySelector, elementSelector, durationSelector, subjectSelector) {
|
---|
| 6 | return (source) => source.lift(new GroupByOperator(keySelector, elementSelector, durationSelector, subjectSelector));
|
---|
| 7 | }
|
---|
| 8 | class GroupByOperator {
|
---|
| 9 | constructor(keySelector, elementSelector, durationSelector, subjectSelector) {
|
---|
| 10 | this.keySelector = keySelector;
|
---|
| 11 | this.elementSelector = elementSelector;
|
---|
| 12 | this.durationSelector = durationSelector;
|
---|
| 13 | this.subjectSelector = subjectSelector;
|
---|
| 14 | }
|
---|
| 15 | call(subscriber, source) {
|
---|
| 16 | return source.subscribe(new GroupBySubscriber(subscriber, this.keySelector, this.elementSelector, this.durationSelector, this.subjectSelector));
|
---|
| 17 | }
|
---|
| 18 | }
|
---|
| 19 | class GroupBySubscriber extends Subscriber {
|
---|
| 20 | constructor(destination, keySelector, elementSelector, durationSelector, subjectSelector) {
|
---|
| 21 | super(destination);
|
---|
| 22 | this.keySelector = keySelector;
|
---|
| 23 | this.elementSelector = elementSelector;
|
---|
| 24 | this.durationSelector = durationSelector;
|
---|
| 25 | this.subjectSelector = subjectSelector;
|
---|
| 26 | this.groups = null;
|
---|
| 27 | this.attemptedToUnsubscribe = false;
|
---|
| 28 | this.count = 0;
|
---|
| 29 | }
|
---|
| 30 | _next(value) {
|
---|
| 31 | let key;
|
---|
| 32 | try {
|
---|
| 33 | key = this.keySelector(value);
|
---|
| 34 | }
|
---|
| 35 | catch (err) {
|
---|
| 36 | this.error(err);
|
---|
| 37 | return;
|
---|
| 38 | }
|
---|
| 39 | this._group(value, key);
|
---|
| 40 | }
|
---|
| 41 | _group(value, key) {
|
---|
| 42 | let groups = this.groups;
|
---|
| 43 | if (!groups) {
|
---|
| 44 | groups = this.groups = new Map();
|
---|
| 45 | }
|
---|
| 46 | let group = groups.get(key);
|
---|
| 47 | let element;
|
---|
| 48 | if (this.elementSelector) {
|
---|
| 49 | try {
|
---|
| 50 | element = this.elementSelector(value);
|
---|
| 51 | }
|
---|
| 52 | catch (err) {
|
---|
| 53 | this.error(err);
|
---|
| 54 | }
|
---|
| 55 | }
|
---|
| 56 | else {
|
---|
| 57 | element = value;
|
---|
| 58 | }
|
---|
| 59 | if (!group) {
|
---|
| 60 | group = (this.subjectSelector ? this.subjectSelector() : new Subject());
|
---|
| 61 | groups.set(key, group);
|
---|
| 62 | const groupedObservable = new GroupedObservable(key, group, this);
|
---|
| 63 | this.destination.next(groupedObservable);
|
---|
| 64 | if (this.durationSelector) {
|
---|
| 65 | let duration;
|
---|
| 66 | try {
|
---|
| 67 | duration = this.durationSelector(new GroupedObservable(key, group));
|
---|
| 68 | }
|
---|
| 69 | catch (err) {
|
---|
| 70 | this.error(err);
|
---|
| 71 | return;
|
---|
| 72 | }
|
---|
| 73 | this.add(duration.subscribe(new GroupDurationSubscriber(key, group, this)));
|
---|
| 74 | }
|
---|
| 75 | }
|
---|
| 76 | if (!group.closed) {
|
---|
| 77 | group.next(element);
|
---|
| 78 | }
|
---|
| 79 | }
|
---|
| 80 | _error(err) {
|
---|
| 81 | const groups = this.groups;
|
---|
| 82 | if (groups) {
|
---|
| 83 | groups.forEach((group, key) => {
|
---|
| 84 | group.error(err);
|
---|
| 85 | });
|
---|
| 86 | groups.clear();
|
---|
| 87 | }
|
---|
| 88 | this.destination.error(err);
|
---|
| 89 | }
|
---|
| 90 | _complete() {
|
---|
| 91 | const groups = this.groups;
|
---|
| 92 | if (groups) {
|
---|
| 93 | groups.forEach((group, key) => {
|
---|
| 94 | group.complete();
|
---|
| 95 | });
|
---|
| 96 | groups.clear();
|
---|
| 97 | }
|
---|
| 98 | this.destination.complete();
|
---|
| 99 | }
|
---|
| 100 | removeGroup(key) {
|
---|
| 101 | this.groups.delete(key);
|
---|
| 102 | }
|
---|
| 103 | unsubscribe() {
|
---|
| 104 | if (!this.closed) {
|
---|
| 105 | this.attemptedToUnsubscribe = true;
|
---|
| 106 | if (this.count === 0) {
|
---|
| 107 | super.unsubscribe();
|
---|
| 108 | }
|
---|
| 109 | }
|
---|
| 110 | }
|
---|
| 111 | }
|
---|
| 112 | class GroupDurationSubscriber extends Subscriber {
|
---|
| 113 | constructor(key, group, parent) {
|
---|
| 114 | super(group);
|
---|
| 115 | this.key = key;
|
---|
| 116 | this.group = group;
|
---|
| 117 | this.parent = parent;
|
---|
| 118 | }
|
---|
| 119 | _next(value) {
|
---|
| 120 | this.complete();
|
---|
| 121 | }
|
---|
| 122 | _unsubscribe() {
|
---|
| 123 | const { parent, key } = this;
|
---|
| 124 | this.key = this.parent = null;
|
---|
| 125 | if (parent) {
|
---|
| 126 | parent.removeGroup(key);
|
---|
| 127 | }
|
---|
| 128 | }
|
---|
| 129 | }
|
---|
| 130 | export class GroupedObservable extends Observable {
|
---|
| 131 | constructor(key, groupSubject, refCountSubscription) {
|
---|
| 132 | super();
|
---|
| 133 | this.key = key;
|
---|
| 134 | this.groupSubject = groupSubject;
|
---|
| 135 | this.refCountSubscription = refCountSubscription;
|
---|
| 136 | }
|
---|
| 137 | _subscribe(subscriber) {
|
---|
| 138 | const subscription = new Subscription();
|
---|
| 139 | const { refCountSubscription, groupSubject } = this;
|
---|
| 140 | if (refCountSubscription && !refCountSubscription.closed) {
|
---|
| 141 | subscription.add(new InnerRefCountSubscription(refCountSubscription));
|
---|
| 142 | }
|
---|
| 143 | subscription.add(groupSubject.subscribe(subscriber));
|
---|
| 144 | return subscription;
|
---|
| 145 | }
|
---|
| 146 | }
|
---|
| 147 | class InnerRefCountSubscription extends Subscription {
|
---|
| 148 | constructor(parent) {
|
---|
| 149 | super();
|
---|
| 150 | this.parent = parent;
|
---|
| 151 | parent.count++;
|
---|
| 152 | }
|
---|
| 153 | unsubscribe() {
|
---|
| 154 | const parent = this.parent;
|
---|
| 155 | if (!parent.closed && !this.closed) {
|
---|
| 156 | super.unsubscribe();
|
---|
| 157 | parent.count -= 1;
|
---|
| 158 | if (parent.count === 0 && parent.attemptedToUnsubscribe) {
|
---|
| 159 | parent.unsubscribe();
|
---|
| 160 | }
|
---|
| 161 | }
|
---|
| 162 | }
|
---|
| 163 | }
|
---|
| 164 | //# sourceMappingURL=groupBy.js.map |
---|