org.mule.expression.OutboundAttachmentsMap Maven / Gradle / Ivy

Show more of this group Show more artifacts with this name
Show all versions of mule-core
Mule server and core classes
There is a newer version: 3.9.0
/*
 * $Id: OutboundAttachmentsMap.java 20321 2010-11-24 15:21:24Z dfeist $
 * -------------------------------------------------------------------------------------
 * Copyright (c) MuleSoft, Inc.  All rights reserved.  http://www.mulesoft.com
 *
 * The software in this package is published under the terms of the CPAL v1.0
 * license, a copy of which has been included with this distribution in the
 * LICENSE.txt file.
 */
package org.mule.expression;

import org.mule.api.MuleMessage;

import java.util.Collection;
import java.util.HashMap;
import java.util.Map;
import java.util.Set;

import javax.activation.DataHandler;

/**
 * Creates a wrapper around Mule Message with a MAp facade used for allowing developers to add attachments to
 * an outgoing message in a transformer of component without needing to access the Mule API directly
 */
public class OutboundAttachmentsMap implements Map
{
    private MuleMessage message;

    public OutboundAttachmentsMap(MuleMessage message)
    {
        this.message = message;
    }

    public int size()
    {
        return message.getOutboundAttachmentNames().size();
    }

    public boolean isEmpty()
    {
        return message.getOutboundAttachmentNames().size() == 0;
    }

    public boolean containsKey(Object key)
    {
        return message.getOutboundAttachmentNames().contains(key.toString());
    }

    public boolean containsValue(Object value)
    {
        return values().contains(value);
    }

    public DataHandler get(Object key)
    {
        return message.getOutboundAttachment(key.toString());
    }

    public DataHandler put(String key, DataHandler value)
    {
        try
        {
            message.addOutboundAttachment(key, value);
            return value;
        }
        catch (Exception e)
        {
            //Not sure we can do anything else here
            throw new RuntimeException(e);
        }
    }

    public DataHandler put(String key, Object value, String contentType)
    {
        try
        {
            message.addOutboundAttachment(key, value, contentType);
            return get(key);
        }
        catch (Exception e)
        {
            //Not sure we can do anything else here
            throw new RuntimeException(e);
        }
    }


    public DataHandler remove(Object key)
    {
        DataHandler attachment = message.getOutboundAttachment(key.toString());
        try
        {
            message.removeOutboundAttachment(key.toString());
        }
        catch (Exception e)
        {
            //ignore (some message types may throw an exception if the attachment does not exist)
        }
        return attachment;
    }


    public void putAll(Map map)
    {
        for (Entry entry : map.entrySet())
        {
            put(entry.getKey(), entry.getValue());
        }
    }

    public void clear()
    {
        throw new UnsupportedOperationException("clear");
    }

    public Set keySet()
    {
        return message.getOutboundAttachmentNames();
    }

    public Collection values()
    {
        return getAttachments().values();
    }

    public Set> entrySet()
    {
        return getAttachments().entrySet();
    }

    //TODO Could optimise this to cache if no writes are made
    private Map getAttachments()
    {
        Map props = new HashMap();
        for (String s : message.getOutboundAttachmentNames())
        {
            props.put(s, message.getOutboundAttachment(s));
        }
        return props;
    }
}