src/websocket/websocket.ts

   1 // Copyright (C) The Arvados Authors. All rights reserved.
   2 //
   3 // SPDX-License-Identifier: AGPL-3.0
   4
   5 import { RootStore } from '~/store/store';
   6 import { AuthService } from '~/services/auth-service/auth-service';
   7 import { Config } from '~/common/config';
   8 import { WebSocketService } from './websocket-service';
   9 import { ResourceEventMessage } from './resource-event-message';
  10 import { ResourceKind } from '~/models/resource';
  11 import { loadProcess } from '~/store/processes/processes-actions';
  12 // import { loadContainers } from '~/store/processes/processes-actions';
  13 import { LogEventType } from '~/models/log';
  14 import { addProcessLogsPanelItem } from '../store/process-logs-panel/process-logs-panel-actions';
  15 // import { FilterBuilder } from "~/services/api/filter-builder";
  16 import { subprocessPanelActions } from "~/store/subprocess-panel/subprocess-panel-actions";
  17 import { projectPanelActions } from "~/store/project-panel/project-panel-action";
  18
  19 export const initWebSocket = (config: Config, authService: AuthService, store: RootStore) => {
  20     if (config.websocketUrl) {
  21         const webSocketService = new WebSocketService(config.websocketUrl, authService);
  22         webSocketService.setMessageListener(messageListener(store));
  23         webSocketService.connect();
  24     } else {
  25         console.warn("WARNING: Websocket ExternalURL is not set on the cluster config.");
  26     }
  27 };
  28
  29 const messageListener = (store: RootStore) => (message: ResourceEventMessage) => {
  30     if (message.eventType === LogEventType.CREATE || message.eventType === LogEventType.UPDATE) {
  31         switch (message.objectKind) {
  32             case ResourceKind.CONTAINER_REQUEST:
  33                 if (store.getState().processPanel.containerRequestUuid === message.objectUuid) {
  34                     store.dispatch(loadProcess(message.objectUuid));
  35                 }
  36             case ResourceKind.CONTAINER:
  37                 store.dispatch(subprocessPanelActions.REQUEST_ITEMS());
  38                 store.dispatch(projectPanelActions.REQUEST_ITEMS());
  39                 return;
  40             default:
  41                 return;
  42         }
  43     } else {
  44         return store.dispatch(addProcessLogsPanelItem(message as ResourceEventMessage<{ text: string }>));
  45     }
  46 };