AbstractExecAgentEventHandler.java

/*******************************************************************************
 * Copyright (c) 2012-2017 Codenvy, S.A.
 * All rights reserved. This program and the accompanying materials
 * are made available under the terms of the Eclipse Public License v1.0
 * which accompanies this distribution, and is available at
 * http://www.eclipse.org/legal/epl-v10.html
 *
 * Contributors:
 *   Codenvy, S.A. - initial API and implementation
 *******************************************************************************/
package org.eclipse.che.ide.api.machine.execagent;

import org.eclipse.che.api.machine.shared.dto.execagent.event.DtoWithPid;

import java.util.HashMap;
import java.util.HashSet;
import java.util.Map;
import java.util.Set;
import java.util.function.BiConsumer;
import java.util.function.Consumer;


public abstract class AbstractExecAgentEventHandler<P extends DtoWithPid> implements BiConsumer<String, P> {

    private final Map<String, Set<Consumer<P>>> operationRegistry = new HashMap<>();

    protected void handle(String endpointId, P params) {
        int pid = params.getPid();
        String key = endpointId + '@' + pid;

        if (!operationRegistry.containsKey(key)) {
            return;
        }

        for (Consumer<P> consumer : operationRegistry.get(key)) {
            consumer.accept(params);
        }
    }

    public void registerConsumer(String endpointId, int pid, Consumer<P> consumer) {
        String key = endpointId + '@' + pid;
        if (!operationRegistry.containsKey(key)) {
            operationRegistry.put(key, new HashSet<>());
        }

        operationRegistry.get(key).add(consumer);
    }

    public void unregisterConsumer(String endpointId, int pid, Consumer<P> consumer) {
        String key = endpointId + '@' + pid;
        if (operationRegistry.containsKey(key)) {
            operationRegistry.get(key).remove(consumer);
        }
    }

    public void unregisterConsumers(String endpointId, int pid) {
        String key = endpointId + '@' + pid;
        operationRegistry.remove(key);
    }
}