gridool.taskqueue.sender.SenderResponseTaskQueue.java Source code

Introduction

Here is the source code for gridool.taskqueue.sender.SenderResponseTaskQueue.java

Source

/*
 * @(#)$Id$
 *
 * Copyright 2006-2008 Makoto YUI
 *
 * 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.
 * 
 * Contributors:
 *     Makoto YUI - initial implementation
 */
package gridool.taskqueue.sender;

import gridool.GridException;
import gridool.GridNode;
import gridool.GridResourceRegistry;
import gridool.GridTaskResult;
import gridool.communication.messages.GridTaskResponseMessage;
import gridool.marshaller.GridMarshaller;

import java.util.concurrent.BlockingQueue;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.ConcurrentMap;

import javax.annotation.Nonnull;
import javax.annotation.Nullable;

import org.apache.commons.logging.Log;
import org.apache.commons.logging.LogFactory;

/**
 * 
 * <DIV lang="en"></DIV>
 * <DIV lang="ja"></DIV>
 * 
 * @author Makoto YUI (yuin405@gmail.com)
 */
public final class SenderResponseTaskQueue implements TaskSenderListener {
    private static final Log LOG = LogFactory.getLog(SenderResponseTaskQueue.class);

    private final GridMarshaller marshaller;
    private final ConcurrentMap<String, BlockingQueue<GridTaskResult>> queueMap;
    private final GridResourceRegistry registry;

    public SenderResponseTaskQueue(@Nonnull GridResourceRegistry resourceRegistry) {
        this.marshaller = resourceRegistry.getMarshaller();
        this.queueMap = new ConcurrentHashMap<String, BlockingQueue<GridTaskResult>>(16);
        this.registry = resourceRegistry;
    }

    public void addResponseQueue(@Nonnull String jobId, @Nonnull BlockingQueue<GridTaskResult> queue) {
        queueMap.put(jobId, queue);
    }

    @Nullable
    @Deprecated
    public BlockingQueue<GridTaskResult> getResponseQueue(@Nonnull String jobId) {
        return queueMap.get(jobId);
    }

    @Override
    public void onResponse(@Nonnull String jobId, @Nonnull GridTaskResult result) {
        final BlockingQueue<GridTaskResult> queue = queueMap.get(jobId);
        if (queue != null) {
            if (LOG.isDebugEnabled()) {
                LOG.debug("Received a GridTaskResponseMessage for a task: " + result.getTaskId() + " of Job ["
                        + jobId + "] that was executed locally");
            }
            queue.add(result);
        } else {
            LOG.error("SenderResponseQueue is not found for a task: " + result.getTaskId() + " of Job [" + jobId
                    + "] that was executed locally");
        }
    }

    @Override
    public void onResponse(@Nonnull GridTaskResponseMessage response) {
        final String jobId = response.getJobId();
        final GridNode senderNode = response.getSenderNode();
        final BlockingQueue<GridTaskResult> queue = queueMap.get(jobId);
        if (queue != null) {
            if (LOG.isDebugEnabled()) {
                LOG.debug("Received a GridTaskResponseMessage for a task: " + response.getTaskId() + " of Job ["
                        + jobId + "] that was executed on " + (senderNode == null ? "localhost" : senderNode));
            }
            byte[] resultBytes = response.getMessage();
            String deployGroup = response.getDeploymentGroup();
            ClassLoader ldr = registry.getDeploymentGroupClassLoader(deployGroup);
            final GridTaskResult result;
            try {
                result = marshaller.unmarshall(resultBytes, ldr);
            } catch (GridException e) {
                throw new IllegalStateException("failed to unmarshall message from node: " + senderNode, e);
            }
            queue.add(result);
        } else {
            LOG.error("SenderResponseQueue is not found for a task: " + response.getTaskId() + " of Job [" + jobId
                    + "] that was executed on " + (senderNode == null ? "localhost" : senderNode));
        }
    }

}