diff --git a/src/main/__test__/cluster.test.ts b/src/main/__test__/cluster.test.ts index b3f0442cc2..4b11a19879 100644 --- a/src/main/__test__/cluster.test.ts +++ b/src/main/__test__/cluster.test.ts @@ -126,6 +126,7 @@ describe("create clusters", () => { }; jest.spyOn(Cluster.prototype, "isClusterAdmin").mockReturnValue(Promise.resolve(true)); + jest.spyOn(Cluster.prototype, "canUseWatchApi").mockReturnValue(Promise.resolve(true)); jest.spyOn(Cluster.prototype, "canI") .mockImplementation((attr: V1ResourceAttributes): Promise => { expect(attr.namespace).toBe("default"); diff --git a/src/main/cluster.ts b/src/main/cluster.ts index 956164e10c..cd92ab2650 100644 --- a/src/main/cluster.ts +++ b/src/main/cluster.ts @@ -48,6 +48,7 @@ export interface ClusterState { isAdmin: boolean; allowedNamespaces: string[] allowedResources: string[] + isGlobalWatchEnabled: boolean; } /** @@ -91,7 +92,6 @@ export class Cluster implements ClusterModel, ClusterState { */ @observable initializing = false; - /** * Is cluster object initialized * @@ -177,6 +177,12 @@ export class Cluster implements ClusterModel, ClusterState { * @observable */ @observable isAdmin = false; + /** + * Global watch-api accessibility , e.g. "/api/v1/services?watch=1" + * + * @observable + */ + @observable isGlobalWatchEnabled = false; /** * Preferences * @@ -353,9 +359,7 @@ export class Cluster implements ClusterModel, ClusterState { await this.refreshConnectionStatus(); if (this.accessible) { - await this.refreshAllowedResources(); - this.isAdmin = await this.isClusterAdmin(); - this.ready = true; + await this.refreshAccessibility(); this.ensureKubectl(); } this.activated = true; @@ -410,13 +414,11 @@ export class Cluster implements ClusterModel, ClusterState { await this.refreshConnectionStatus(); if (this.accessible) { - this.isAdmin = await this.isClusterAdmin(); - await this.refreshAllowedResources(); + await this.refreshAccessibility(); if (opts.refreshMetadata) { this.refreshMetadata(); } - this.ready = true; } this.pushState(); } @@ -433,6 +435,18 @@ export class Cluster implements ClusterModel, ClusterState { this.metadata = Object.assign(existingMetadata, metadata); } + /** + * @internal + */ + private async refreshAccessibility(): Promise { + this.isAdmin = await this.isClusterAdmin(); + this.isGlobalWatchEnabled = await this.canUseWatchApi({ resource: "*" }); + + await this.refreshAllowedResources(); + + this.ready = true; + } + /** * @internal */ @@ -571,6 +585,17 @@ export class Cluster implements ClusterModel, ClusterState { }); } + /** + * @internal + */ + async canUseWatchApi(customizeResource: V1ResourceAttributes = {}): Promise { + return this.canI({ + verb: "watch", + resource: "*", + ...customizeResource, + }); + } + toJSON(): ClusterModel { const model: ClusterModel = { id: this.id, @@ -604,6 +629,7 @@ export class Cluster implements ClusterModel, ClusterState { isAdmin: this.isAdmin, allowedNamespaces: this.allowedNamespaces, allowedResources: this.allowedResources, + isGlobalWatchEnabled: this.isGlobalWatchEnabled, }; return toJS(state, { diff --git a/src/renderer/api/kube-watch-api.ts b/src/renderer/api/kube-watch-api.ts index 8adf58676f..d3b2b4512a 100644 --- a/src/renderer/api/kube-watch-api.ts +++ b/src/renderer/api/kube-watch-api.ts @@ -5,12 +5,11 @@ import type { Cluster } from "../../main/cluster"; import type { IKubeWatchEvent, IKubeWatchEventStreamEnd, IWatchRoutePayload } from "../../main/routes/watch-route"; import type { KubeObject } from "./kube-object"; import type { KubeObjectStore } from "../kube-object.store"; -import type { NamespaceStore } from "../components/+namespaces/namespace.store"; import plimit from "p-limit"; import debounce from "lodash/debounce"; -import { comparer, computed, observable, reaction } from "mobx"; -import { autobind, EventEmitter } from "../utils"; +import { autorun, comparer, computed, IReactionDisposer, observable, reaction } from "mobx"; +import { autobind, EventEmitter, noop } from "../utils"; import { ensureObjectSelfLink, KubeApi, parseKubeApi } from "./kube-api"; import { KubeJsonApiData, KubeJsonApiError } from "./kube-json-api"; import { apiPrefix, isDebugging, isProduction } from "../../common/vars"; @@ -19,6 +18,7 @@ import { apiManager } from "./api-manager"; export { IKubeWatchEvent, IKubeWatchEventStreamEnd }; export interface IKubeWatchMessage { + namespace?: string; data?: IKubeWatchEvent error?: IKubeWatchEvent; api?: KubeApi; @@ -28,7 +28,7 @@ export interface IKubeWatchMessage { export interface IKubeWatchSubscribeStoreOptions { preload?: boolean; // preload store items, default: true waitUntilLoaded?: boolean; // subscribe only after loading all stores, default: true - cacheLoading?: boolean; // when enabled loading store will be skipped, default: false + loadOnce?: boolean; // check store.isLoaded to skip loading if done already, default: false } export interface IKubeWatchReconnectOptions { @@ -43,50 +43,49 @@ export interface IKubeWatchLog { @autobind() export class KubeWatchApi { - private cluster: Cluster; - private namespaceStore: NamespaceStore; - private requestId = 0; - private isConnected = false; private reader: ReadableStreamReader; - private subscribers = observable.map(); - - // events public onMessage = new EventEmitter<[IKubeWatchMessage]>(); + @observable.ref private cluster: Cluster; + @observable.ref private namespaces: string[] = []; + @observable subscribers = observable.map(); + @observable isConnected = false; + + @computed get isReady(): boolean { + return Boolean(this.cluster && this.namespaces); + } + @computed get isActive(): boolean { return this.apis.length > 0; } @computed get apis(): string[] { - const { cluster, namespaceStore } = this; - const activeApis = Array.from(this.subscribers.keys()); + if (!this.isReady) { + return []; + } - return activeApis.map(api => { - if (!cluster.isAllowedResource(api.kind)) { + return Array.from(this.subscribers.keys()).map(api => { + if (!this.isAllowedApi(api)) { return []; } - if (api.isNamespaced) { - return namespaceStore.getContextNamespaces().map(namespace => api.getWatchUrl(namespace)); - } else { - return api.getWatchUrl(); + if (api.isNamespaced && !this.cluster.isGlobalWatchEnabled) { + return this.namespaces.map(namespace => api.getWatchUrl(namespace)); } + + return api.getWatchUrl(); }).flat(); } - constructor() { - this.init(); - } - - private async init() { - const { getHostedCluster } = await import("../../common/cluster-store"); - const { namespaceStore } = await import("../components/+namespaces/namespace.store"); - - await namespaceStore.whenReady; - - this.cluster = getHostedCluster(); - this.namespaceStore = namespaceStore; + async init({ getCluster, getNamespaces }: { + getCluster: () => Cluster, + getNamespaces: () => string[], + }): Promise { + autorun(() => { + this.cluster = getCluster(); + this.namespaces = getNamespaces(); + }); this.bindAutoConnect(); } @@ -108,7 +107,7 @@ export class KubeWatchApi { } isAllowedApi(api: KubeApi): boolean { - return !!this?.cluster.isAllowedResource(api.kind); + return Boolean(this?.cluster.isAllowedResource(api.kind)); } subscribeApi(api: KubeApi | KubeApi[]): () => void { @@ -129,45 +128,66 @@ export class KubeWatchApi { }; } - subscribeStores(stores: KubeObjectStore[], options: IKubeWatchSubscribeStoreOptions = {}): () => void { - const { preload = true, waitUntilLoaded = true, cacheLoading = false } = options; + preloadStores(stores: KubeObjectStore[], { loadOnce = false } = {}) { const limitRequests = plimit(1); // load stores one by one to allow quick skipping when fast clicking btw pages const preloading: Promise[] = []; + + for (const store of stores) { + preloading.push(limitRequests(async () => { + if (store.isLoaded && loadOnce) return; // skip + + return store.loadAll(this.namespaces); + })); + } + + return { + loading: Promise.allSettled(preloading), + cancelLoading: () => limitRequests.clearQueue(), + }; + } + + subscribeStores(stores: KubeObjectStore[], options: IKubeWatchSubscribeStoreOptions = {}): () => void { + const { preload = true, waitUntilLoaded = true, loadOnce = false } = options; const apis = new Set(stores.map(store => store.getSubscribeApis()).flat()); const unsubscribeList: (() => void)[] = []; let isUnsubscribed = false; + const load = () => this.preloadStores(stores, { loadOnce }); + let preloading = preload && load(); + let cancelReloading: IReactionDisposer = noop; + const subscribe = () => { if (isUnsubscribed) return; apis.forEach(api => unsubscribeList.push(this.subscribeApi(api))); }; - if (preload) { - for (const store of stores) { - preloading.push(limitRequests(async () => { - if (cacheLoading && store.isLoaded) return; // skip - - return store.loadAll(); - })); - } - } - - if (waitUntilLoaded) { - Promise.all(preloading).then(subscribe, error => { - this.log({ - message: new Error("Loading stores has failed"), - meta: { stores, error, options }, + if (preloading) { + if (waitUntilLoaded) { + preloading.loading.then(subscribe, error => { + this.log({ + message: new Error("Loading stores has failed"), + meta: { stores, error, options }, + }); }); + } else { + subscribe(); + } + + // reload when context namespaces changes + cancelReloading = reaction(() => this.namespaces, () => { + preloading?.cancelLoading(); + preloading = load(); + }, { + equals: comparer.shallow, }); - } else { - subscribe(); } // unsubscribe return () => { if (isUnsubscribed) return; isUnsubscribed = true; - limitRequests.clearQueue(); + cancelReloading(); + preloading?.cancelLoading(); unsubscribeList.forEach(unsubscribe => unsubscribe()); }; } @@ -254,6 +274,10 @@ export class KubeWatchApi { const kubeEvent: IKubeWatchEvent = JSON.parse(json); const message = this.getMessage(kubeEvent); + if (!this.namespaces.includes(message.namespace)) { + continue; // skip updates from non-watching resources context + } + this.onMessage.emit(message); } catch (error) { return json; @@ -286,6 +310,7 @@ export class KubeWatchApi { message.api = api; message.store = apiManager.getStore(api); + message.namespace = namespace; } break; } diff --git a/src/renderer/components/+apps-releases/release.store.ts b/src/renderer/components/+apps-releases/release.store.ts index 6f7ed39fed..0ca6f45b39 100644 --- a/src/renderer/components/+apps-releases/release.store.ts +++ b/src/renderer/components/+apps-releases/release.store.ts @@ -58,11 +58,11 @@ export class ReleaseStore extends ItemStore { } @action - async loadAll() { + async loadAll(namespaces = namespaceStore.allowedNamespaces) { this.isLoading = true; try { - const items = await this.loadItems(namespaceStore.getContextNamespaces()); + const items = await this.loadItems(namespaces); this.items.replace(this.sortItems(items)); this.isLoaded = true; @@ -73,6 +73,10 @@ export class ReleaseStore extends ItemStore { } } + async loadSelectedNamespaces(): Promise { + return this.loadAll(namespaceStore.getContextNamespaces()); + } + async loadItems(namespaces: string[]) { return Promise .all(namespaces.map(namespace => helmReleasesApi.list(namespace))) @@ -82,7 +86,7 @@ export class ReleaseStore extends ItemStore { async create(payload: IReleaseCreatePayload) { const response = await helmReleasesApi.create(payload); - if (this.isLoaded) this.loadAll(); + if (this.isLoaded) this.loadSelectedNamespaces(); return response; } @@ -90,7 +94,7 @@ export class ReleaseStore extends ItemStore { async update(name: string, namespace: string, payload: IReleaseUpdatePayload) { const response = await helmReleasesApi.update(name, namespace, payload); - if (this.isLoaded) this.loadAll(); + if (this.isLoaded) this.loadSelectedNamespaces(); return response; } @@ -98,7 +102,7 @@ export class ReleaseStore extends ItemStore { async rollback(name: string, namespace: string, revision: number) { const response = await helmReleasesApi.rollback(name, namespace, revision); - if (this.isLoaded) this.loadAll(); + if (this.isLoaded) this.loadSelectedNamespaces(); return response; } diff --git a/src/renderer/components/+custom-resources/crd-resources.tsx b/src/renderer/components/+custom-resources/crd-resources.tsx index b9008b410d..2bae92b8d4 100644 --- a/src/renderer/components/+custom-resources/crd-resources.tsx +++ b/src/renderer/components/+custom-resources/crd-resources.tsx @@ -30,7 +30,7 @@ export class CrdResources extends React.Component { const { store } = this; if (store && !store.isLoading && !store.isLoaded) { - store.loadAll(); + store.loadSelectedNamespaces(); } }) ]); diff --git a/src/renderer/components/+events/kube-event-details.tsx b/src/renderer/components/+events/kube-event-details.tsx index 34b16103f0..60821d416d 100644 --- a/src/renderer/components/+events/kube-event-details.tsx +++ b/src/renderer/components/+events/kube-event-details.tsx @@ -14,7 +14,7 @@ export interface KubeEventDetailsProps { @observer export class KubeEventDetails extends React.Component { async componentDidMount() { - eventStore.loadAll(); + eventStore.loadSelectedNamespaces(); } render() { diff --git a/src/renderer/components/+namespaces/namespace-details.tsx b/src/renderer/components/+namespaces/namespace-details.tsx index 5dfa93cea7..e7397b6a5e 100644 --- a/src/renderer/components/+namespaces/namespace-details.tsx +++ b/src/renderer/components/+namespaces/namespace-details.tsx @@ -32,8 +32,8 @@ export class NamespaceDetails extends React.Component { } componentDidMount() { - resourceQuotaStore.loadAll(); - limitRangeStore.loadAll(); + resourceQuotaStore.loadSelectedNamespaces(); + limitRangeStore.loadSelectedNamespaces(); } render() { diff --git a/src/renderer/components/+namespaces/namespace.store.ts b/src/renderer/components/+namespaces/namespace.store.ts index 056e1ab24e..c56043766a 100644 --- a/src/renderer/components/+namespaces/namespace.store.ts +++ b/src/renderer/components/+namespaces/namespace.store.ts @@ -73,7 +73,7 @@ export class NamespaceStore extends KubeObjectStore { } private autoLoadAllowedNamespaces(): IReactionDisposer { - return reaction(() => this.allowedNamespaces, () => this.loadAll(), { + return reaction(() => this.allowedNamespaces, namespaces => this.loadAll(namespaces), { fireImmediately: true, equals: comparer.shallow, }); diff --git a/src/renderer/components/+nodes/node-details.tsx b/src/renderer/components/+nodes/node-details.tsx index 824affafee..d4208c4545 100644 --- a/src/renderer/components/+nodes/node-details.tsx +++ b/src/renderer/components/+nodes/node-details.tsx @@ -29,9 +29,7 @@ export class NodeDetails extends React.Component { }); async componentDidMount() { - if (!podsStore.isLoaded) { - podsStore.loadAll(); - } + podsStore.loadSelectedNamespaces(); } componentWillUnmount() { diff --git a/src/renderer/components/+user-management-roles-bindings/add-role-binding-dialog.tsx b/src/renderer/components/+user-management-roles-bindings/add-role-binding-dialog.tsx index ab06126bd2..808cef90d6 100644 --- a/src/renderer/components/+user-management-roles-bindings/add-role-binding-dialog.tsx +++ b/src/renderer/components/+user-management-roles-bindings/add-role-binding-dialog.tsx @@ -80,7 +80,7 @@ export class AddRoleBindingDialog extends React.Component { ]; this.isLoading = true; - await Promise.all(stores.map(store => store.loadAll())); + await Promise.all(stores.map(store => store.loadSelectedNamespaces())); this.isLoading = false; } diff --git a/src/renderer/components/+workloads-cronjobs/cronjob-details.tsx b/src/renderer/components/+workloads-cronjobs/cronjob-details.tsx index 6fd04ffe7e..4daff4abed 100644 --- a/src/renderer/components/+workloads-cronjobs/cronjob-details.tsx +++ b/src/renderer/components/+workloads-cronjobs/cronjob-details.tsx @@ -20,9 +20,7 @@ interface Props extends KubeObjectDetailsProps { @observer export class CronJobDetails extends React.Component { async componentDidMount() { - if (!jobStore.isLoaded) { - jobStore.loadAll(); - } + jobStore.loadSelectedNamespaces(); } render() { diff --git a/src/renderer/components/+workloads-daemonsets/daemonset-details.tsx b/src/renderer/components/+workloads-daemonsets/daemonset-details.tsx index 44ccbaac56..ab3269ede5 100644 --- a/src/renderer/components/+workloads-daemonsets/daemonset-details.tsx +++ b/src/renderer/components/+workloads-daemonsets/daemonset-details.tsx @@ -30,9 +30,7 @@ export class DaemonSetDetails extends React.Component { }); componentDidMount() { - if (!podsStore.isLoaded) { - podsStore.loadAll(); - } + podsStore.loadSelectedNamespaces(); } componentWillUnmount() { diff --git a/src/renderer/components/+workloads-deployments/deployment-details.tsx b/src/renderer/components/+workloads-deployments/deployment-details.tsx index 26cd4c0b23..e22137ea67 100644 --- a/src/renderer/components/+workloads-deployments/deployment-details.tsx +++ b/src/renderer/components/+workloads-deployments/deployment-details.tsx @@ -31,9 +31,7 @@ export class DeploymentDetails extends React.Component { }); componentDidMount() { - if (!podsStore.isLoaded) { - podsStore.loadAll(); - } + podsStore.loadSelectedNamespaces(); } componentWillUnmount() { diff --git a/src/renderer/components/+workloads-jobs/job-details.tsx b/src/renderer/components/+workloads-jobs/job-details.tsx index dfa16fe760..4ce4a9bc61 100644 --- a/src/renderer/components/+workloads-jobs/job-details.tsx +++ b/src/renderer/components/+workloads-jobs/job-details.tsx @@ -25,9 +25,7 @@ interface Props extends KubeObjectDetailsProps { @observer export class JobDetails extends React.Component { async componentDidMount() { - if (!podsStore.isLoaded) { - podsStore.loadAll(); - } + podsStore.loadSelectedNamespaces(); } render() { diff --git a/src/renderer/components/+workloads-replicasets/replicaset-details.tsx b/src/renderer/components/+workloads-replicasets/replicaset-details.tsx index 8c28d81a83..f427d78e9d 100644 --- a/src/renderer/components/+workloads-replicasets/replicaset-details.tsx +++ b/src/renderer/components/+workloads-replicasets/replicaset-details.tsx @@ -29,9 +29,7 @@ export class ReplicaSetDetails extends React.Component { }); async componentDidMount() { - if (!podsStore.isLoaded) { - podsStore.loadAll(); - } + podsStore.loadSelectedNamespaces(); } componentWillUnmount() { diff --git a/src/renderer/components/+workloads-statefulsets/statefulset-details.tsx b/src/renderer/components/+workloads-statefulsets/statefulset-details.tsx index d30633ff32..780572eb96 100644 --- a/src/renderer/components/+workloads-statefulsets/statefulset-details.tsx +++ b/src/renderer/components/+workloads-statefulsets/statefulset-details.tsx @@ -30,9 +30,7 @@ export class StatefulSetDetails extends React.Component { }); componentDidMount() { - if (!podsStore.isLoaded) { - podsStore.loadAll(); - } + podsStore.loadSelectedNamespaces(); } componentWillUnmount() { diff --git a/src/renderer/components/app.tsx b/src/renderer/components/app.tsx index 2b36bf0057..2236b3d6be 100755 --- a/src/renderer/components/app.tsx +++ b/src/renderer/components/app.tsx @@ -1,4 +1,5 @@ import React from "react"; +import { computed, observable, reaction } from "mobx"; import { disposeOnUnmount, observer } from "mobx-react"; import { Redirect, Route, Router, Switch } from "react-router"; import { history } from "../navigation"; @@ -42,7 +43,7 @@ import { ClusterPageMenuRegistration, clusterPageMenuRegistry } from "../../exte import { TabLayout, TabLayoutRoute } from "./layout/tab-layout"; import { StatefulSetScaleDialog } from "./+workloads-statefulsets/statefulset-scale-dialog"; import { eventStore } from "./+events/event.store"; -import { computed, reaction, observable } from "mobx"; +import { namespaceStore } from "./+namespaces/namespace.store"; import { nodesStore } from "./+nodes/nodes.store"; import { podsStore } from "./+workloads-pods/pods.store"; import { kubeWatchApi } from "../api/kube-watch-api"; @@ -74,6 +75,12 @@ export class App extends React.Component { window.location.reload(); }); whatInput.ask(); // Start to monitor user input device + + await namespaceStore.whenReady; + await kubeWatchApi.init({ + getCluster: getHostedCluster, + getNamespaces: namespaceStore.getContextNamespaces, + }); } componentDidMount() { diff --git a/src/renderer/components/dock/upgrade-chart.store.ts b/src/renderer/components/dock/upgrade-chart.store.ts index bc7445e29d..f609420d9d 100644 --- a/src/renderer/components/dock/upgrade-chart.store.ts +++ b/src/renderer/components/dock/upgrade-chart.store.ts @@ -80,7 +80,7 @@ export class UpgradeChartStore extends DockTabStore { const values = this.values.getData(tabId); await Promise.all([ - !releaseStore.isLoaded && releaseStore.loadAll(), + !releaseStore.isLoaded && releaseStore.loadSelectedNamespaces(), !values && this.loadValues(tabId) ]); } diff --git a/src/renderer/components/item-object-list/item-list-layout.tsx b/src/renderer/components/item-object-list/item-list-layout.tsx index b13d496064..ac0d2ea635 100644 --- a/src/renderer/components/item-object-list/item-list-layout.tsx +++ b/src/renderer/components/item-object-list/item-list-layout.tsx @@ -138,7 +138,7 @@ export class ItemListLayout extends React.Component { const { store, dependentStores } = this.props; const stores = Array.from(new Set([store, ...dependentStores])); - stores.forEach(store => store.loadAll()); + stores.forEach(store => store.loadAll(namespaceStore.getContextNamespaces())); } private filterCallbacks: { [type: string]: ItemsFilter } = { diff --git a/src/renderer/components/layout/sidebar.tsx b/src/renderer/components/layout/sidebar.tsx index b44dd4a759..20c401ab4f 100644 --- a/src/renderer/components/layout/sidebar.tsx +++ b/src/renderer/components/layout/sidebar.tsx @@ -40,9 +40,7 @@ interface Props { @observer export class Sidebar extends React.Component { async componentDidMount() { - if (!crdStore.isLoaded && isAllowedResource("customresourcedefinitions")) { - crdStore.loadAll(); - } + crdStore.loadSelectedNamespaces(); } renderCustomResources() { diff --git a/src/renderer/kube-object.store.ts b/src/renderer/kube-object.store.ts index 760ebd3335..d56e6bd912 100644 --- a/src/renderer/kube-object.store.ts +++ b/src/renderer/kube-object.store.ts @@ -106,17 +106,18 @@ export abstract class KubeObjectStore extends ItemSt } @action - async loadAll({ namespaces: contextNamespaces }: { namespaces?: string[] } = {}) { + async loadAll(namespaces: string[] = []): Promise { this.isLoading = true; try { - if (!contextNamespaces) { + if (!namespaces.length) { const { namespaceStore } = await import("./components/+namespaces/namespace.store"); - contextNamespaces = namespaceStore.getContextNamespaces(); + // load all available namespaces by default + namespaces.push(...namespaceStore.allowedNamespaces); } - let items = await this.loadItems({ namespaces: contextNamespaces, api: this.api }); + let items = await this.loadItems({ namespaces, api: this.api }); items = this.filterItemsOnLoad(items); items = this.sortItems(items); @@ -131,6 +132,12 @@ export abstract class KubeObjectStore extends ItemSt } } + async loadSelectedNamespaces(): Promise { + const { namespaceStore } = await import("./components/+namespaces/namespace.store"); + + return this.loadAll(namespaceStore.getContextNamespaces()); + } + protected resetOnError(error: any) { if (error) this.reset(); }