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 { FilterBuilder } from '~/common/api/filter-builder';
  14
  15 export const initWebSocket = (config: Config, authService: AuthService, store: RootStore) => {
  16     const webSocketService = new WebSocketService(config.websocketUrl, authService);
  17     webSocketService.setMessageListener(messageListener(store));
  18     webSocketService.connect();
  19 };
  20
  21 const messageListener = (store: RootStore) => (message: ResourceEventMessage) => {
  22     switch (message.objectKind) {
  23         case ResourceKind.CONTAINER_REQUEST:
  24             return store.dispatch(loadProcess(message.objectUuid));
  25         case ResourceKind.CONTAINER:
  26             return store.dispatch(loadContainers(
  27                 new FilterBuilder().addIn('uuid', [message.objectUuid]).getFilters()
  28             ));
  29         default:
  30             return;
  31     }
  32 };