1 | import useForceUpdate from './useForceUpdate';
|
2 | import useStableMemo from './useStableMemo';
|
3 | export class ObservableSet extends Set {
|
4 | constructor(listener, init) {
|
5 | super(init);
|
6 | this.listener = listener;
|
7 | }
|
8 | add(value) {
|
9 | super.add(value);
|
10 |
|
11 |
|
12 | if (this.listener) this.listener(this);
|
13 | return this;
|
14 | }
|
15 | delete(value) {
|
16 | const result = super.delete(value);
|
17 | this.listener(this);
|
18 | return result;
|
19 | }
|
20 | clear() {
|
21 | super.clear();
|
22 | this.listener(this);
|
23 | }
|
24 | }
|
25 |
|
26 |
|
27 |
|
28 |
|
29 |
|
30 |
|
31 |
|
32 |
|
33 |
|
34 |
|
35 |
|
36 |
|
37 |
|
38 |
|
39 |
|
40 |
|
41 |
|
42 |
|
43 |
|
44 |
|
45 | function useSet(init) {
|
46 | const forceUpdate = useForceUpdate();
|
47 | return useStableMemo(() => new ObservableSet(forceUpdate, init), []);
|
48 | }
|
49 | export default useSet; |
\ | No newline at end of file |