-
Notifications
You must be signed in to change notification settings - Fork 11
/
ApolloLinkNetworkStatus.ts
70 lines (56 loc) · 1.5 KB
/
ApolloLinkNetworkStatus.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
import {
ApolloLink,
Observable,
Operation,
NextLink,
FetchResult
} from '@apollo/client';
import Dispatcher from './Dispatcher';
import ActionTypes from './ActionTypes';
export default class ApolloLinkNetworkStatus extends ApolloLink {
dispatcher: Dispatcher;
constructor(dispatcher: Dispatcher) {
super();
this.dispatcher = dispatcher;
}
request(operation: Operation, forward: NextLink): Observable<FetchResult> {
this.dispatcher.dispatch({
type: ActionTypes.REQUEST,
payload: {operation}
});
const subscriber = forward(operation);
return new Observable(observer => {
let isPending = true;
const subscription = subscriber.subscribe({
next: result => {
isPending = false;
this.dispatcher.dispatch({
type: ActionTypes.SUCCESS,
payload: {operation, result}
});
observer.next(result);
},
error: networkError => {
isPending = false;
this.dispatcher.dispatch({
type: ActionTypes.ERROR,
payload: {operation, networkError}
});
observer.error(networkError);
},
complete: observer.complete.bind(observer)
});
return () => {
if (isPending) {
this.dispatcher.dispatch({
type: ActionTypes.CANCEL,
payload: {operation}
});
}
if (subscription) {
subscription.unsubscribe();
}
};
});
}
}