-
Notifications
You must be signed in to change notification settings - Fork 1.4k
/
pods.store.ts
98 lines (77 loc) · 2.62 KB
/
pods.store.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
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
import countBy from "lodash/countBy";
import { action, observable } from "mobx";
import { KubeObjectStore } from "../../kube-object.store";
import { autobind, cpuUnitsToNumber, unitsToBytes } from "../../utils";
import { IPodMetrics, Pod, PodMetrics, podMetricsApi, podsApi } from "../../api/endpoints";
import { apiManager } from "../../api/api-manager";
import { WorkloadKubeObject } from "../../api/workload-kube-object";
@autobind()
export class PodsStore extends KubeObjectStore<Pod> {
api = podsApi;
@observable metrics: IPodMetrics = null;
@observable kubeMetrics = observable.array<PodMetrics>([]);
@action
async loadMetrics(pod: Pod) {
this.metrics = await podsApi.getMetrics([pod], pod.getNs());
}
loadContainerMetrics(pod: Pod) {
return podsApi.getMetrics([pod], pod.getNs(), "container, namespace");
}
async loadKubeMetrics(namespace?: string) {
try {
const metrics = await podMetricsApi.list({ namespace });
this.kubeMetrics.replace(metrics);
} catch (error) {
console.error("loadKubeMetrics failed", error);
}
}
getPodsByOwner(workload: WorkloadKubeObject): Pod[] {
if (!workload) return [];
return this.items.filter(pod => {
const owners = pod.getOwnerRefs();
if (!owners.length) return;
return owners.find(owner => owner.uid === workload.getId());
});
}
getPodsByOwnerId(workloadId: string): Pod[] {
return this.items.filter(pod => {
return pod.getOwnerRefs().find(owner => owner.uid === workloadId);
});
}
getPodsByNode(node: string) {
if (!this.isLoaded) return [];
return this.items.filter(pod => pod.spec.nodeName === node);
}
getStatuses(pods: Pod[]) {
return countBy(pods.map(pod => pod.getStatus()));
}
getPodKubeMetrics(pod: Pod) {
const containers = pod.getContainers();
const empty = { cpu: 0, memory: 0 };
const metrics = this.kubeMetrics.find(metric => {
return [
metric.getName() === pod.getName(),
metric.getNs() === pod.getNs()
].every(v => v);
});
if (!metrics) return empty;
return containers.reduce((total, container) => {
const metric = metrics.containers.find(item => item.name == container.name);
let cpu = "0";
let memory = "0";
if (metric && metric.usage) {
cpu = metric.usage.cpu || "0";
memory = metric.usage.memory || "0";
}
return {
cpu: total.cpu + cpuUnitsToNumber(cpu),
memory: total.memory + unitsToBytes(memory)
};
}, empty);
}
reset() {
this.metrics = null;
}
}
export const podsStore = new PodsStore();
apiManager.registerStore(podsStore);