com.codnos.dbgp.internal.handlers.DBGpEventsHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of dbgp-interfaces Show documentation
Interfaces and implementations for both client and server side of DBGp
The newest version!
/*
 * Copyright 2016 Codnos Ltd.
 *
 * Licensed under the Apache License, Version 2.0 (the "License");
 * you may not use this file except in compliance with the License.
 * You may obtain a copy of the License at
 *
 * http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package com.codnos.dbgp.internal.handlers;

import com.codnos.dbgp.internal.commands.Command;
import com.codnos.dbgp.internal.messages.Message;

import java.util.HashMap;
import java.util.Map;
import java.util.concurrent.ConcurrentHashMap;
import java.util.logging.Logger;

public class DBGpEventsHandler {
    private static final Logger LOGGER = Logger.getLogger(DBGpEventsHandler.class.getName());
    private final Map responseHandlers = new HashMap();
    private final Map messageMap = new ConcurrentHashMap();
    private final Object monitor = new Object();

    public void receive(Message message) {
        MessageHandler messageHandler = responseHandlers.remove(message.getHandlerKey());
        if (messageHandler != null) {
            messageHandler.handle(message);
        } else {
            synchronized (monitor) {
                LOGGER.fine(Thread.currentThread().getName() + ": got message with key: " + message.getHandlerKey());
                messageMap.put(message.getHandlerKey(), message);
                LOGGER.fine(Thread.currentThread().getName() + ": notifying about message with key: " + message.getHandlerKey());
                monitor.notifyAll();
                LOGGER.fine(Thread.currentThread().getName() + ": notified about message with key: " + message.getHandlerKey());
            }
        }
    }

    public void clearHandlers() {
        responseHandlers.clear();
    }

    public void registerMessageHandler(Command command, MessageHandler messageHandler) {
        responseHandlers.put(command.getHandlerKey(), messageHandler);
    }

    public void registerMessageHandler(String name, String transactionId, MessageHandler messageHandler) {
        responseHandlers.put(name + ":" + transactionId, messageHandler);
    }

    public  T getResponse(Command command) {
        int tryCount = 0;
        LOGGER.fine(Thread.currentThread().getName() + ": checking response for message with key: " + command.getHandlerKey());
        synchronized (monitor) {
            while (!messageMap.containsKey(command.getHandlerKey())) {

                LOGGER.fine(Thread.currentThread().getName() + ": about to wait for message with key: " + command.getHandlerKey());
                try {
                    monitor.wait(3000L);
                    LOGGER.fine(Thread.currentThread().getName() + ": got notified while waiting for message with key: " + command.getHandlerKey());
                    tryCount++;
                    if (tryCount > 10)
                        throw new RuntimeException("not waiting any longer!");
                } catch (InterruptedException e) {
                    e.printStackTrace();
                    return null;
                }
            }
            LOGGER.fine(Thread.currentThread().getName() + ": no longer waiting for message with key: " + command.getHandlerKey());
            T result = (T) messageMap.remove(command.getHandlerKey());
            LOGGER.fine(Thread.currentThread().getName() + ": after waiting for message with key: " + command.getHandlerKey() + " got response " + result + " with key " + (result != null ? ((Message) result).getHandlerKey() : null));
            return result;

        }
    }
}