forked from tensorflow/tensorboard
-
Notifications
You must be signed in to change notification settings - Fork 0
/
baseStore.ts
75 lines (68 loc) · 2.27 KB
/
baseStore.ts
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
/* Copyright 2018 The TensorFlow Authors. All Rights Reserved.
+
+Licensed under the Apache License, Version 2.0 (the "License");
+you may not use this file except in compliance with the License.
+You may obtain a copy of the License at
+
+ http://www.apache.org/licenses/LICENSE-2.0
+
+Unless required by applicable law or agreed to in writing, software
+distributed under the License is distributed on an "AS IS" BASIS,
+WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
+See the License for the specific language governing permissions and
+limitations under the License.
+==============================================================================*/
namespace tf_backend {
export type Listener = () => void;
// A unique reference to a listener for an easier dereferencing.
export class ListenKey {
public readonly listener: Listener;
constructor(listener: Listener) {
this.listener = listener;
}
}
export abstract class BaseStore {
protected requestManager: RequestManager = new RequestManager(
1 /* simultaneous request */
);
private _listeners: Set<ListenKey> = new Set<ListenKey>();
public initialized: boolean = false;
/**
* Asynchronously load or reload the runs data. Listeners will be
* invoked if this causes the runs data to change.
*
* @see addListener
* @return {Promise<void>} a promise that resolves when new data have loaded.
*/
protected abstract load(): Promise<void>;
refresh(): Promise<void> {
return this.load().then(() => {
this.initialized = true;
});
}
/**
* Register a listener (nullary function) to be called when new runs are
* available.
*/
addListener(listener: Listener): ListenKey {
const key = new ListenKey(listener);
this._listeners.add(key);
return key;
}
/**
* Remove a listener registered with `addListener`.
*/
removeListenerByKey(listenKey: ListenKey): void {
this._listeners.delete(listenKey);
}
protected emitChange(): void {
this._listeners.forEach((listenKey) => {
try {
listenKey.listener();
} catch (e) {
// ignore exceptions on the listener side.
}
});
}
}
} // namespace tf_backend