1 // Copyright (C) The Arvados Authors. All rights reserved.
3 // SPDX-License-Identifier: AGPL-3.0
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, ResourceEventMessageType } 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';
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();
21 const messageListener = (store: RootStore) => (message: ResourceEventMessage) => {
22 if (message.eventType === ResourceEventMessageType.CREATE || message.eventType === ResourceEventMessageType.UPDATE) {
23 switch (message.objectKind) {
24 case ResourceKind.CONTAINER_REQUEST:
25 return store.dispatch(loadProcess(message.objectUuid));
26 case ResourceKind.CONTAINER:
27 return store.dispatch(loadContainers(
28 new FilterBuilder().addIn('uuid', [message.objectUuid]).getFilters()