org.mule.modules.utils.mom.internal.CollectionInlinerMapObjectMapperInterceptor.java Source code

Introduction

Here is the source code for org.mule.modules.utils.mom.internal.CollectionInlinerMapObjectMapperInterceptor.java

Source

/**
 * Mule Modules Utils
 *
 * 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.modules.utils.mom.internal;

import java.util.HashMap;
import java.util.List;
import java.util.Map;

import net.sf.staccatocommons.defs.Applicable;
import net.sf.staccatocommons.defs.Applicable2;

import org.apache.commons.lang.StringUtils;

import com.zauberlabs.commons.mom.MapObjectMapperInterceptor;

/**
 * {@link MapObjectMapperInterceptor} that allows to wrap lists into
 * nested .*Collection elements in sabrix soap dtos when necessary
 *
 * @author flbulgarelli
 */
public class CollectionInlinerMapObjectMapperInterceptor implements MapObjectMapperInterceptor {
    @Override
    public Object unmap(final Object value, final Class<?> type, Applicable2<Object, Class<?>, Object> proceed) {
        if (sholdWrap(value, type)) {
            return proceed.apply(wrap(value, type), type);
        }
        return proceed.apply(value, type);
    }

    @SuppressWarnings({ "unchecked", "serial" })
    protected Map wrap(final Object value, final Class<?> type) {
        return new HashMap() {
            {
                put(propertyName(type), value);
            }
        };
    }

    @SuppressWarnings("unchecked")
    protected boolean sholdWrap(final Object value, final Class<?> type) {
        return value instanceof List && isCollection(type);
    }

    private boolean isCollection(final Class<?> type) {
        return type.getSimpleName().endsWith("Collection");
    }

    @Override
    public Object unamp(Object map, Applicable<Object, Object> proceed) {
        return proceed.apply(map);
    }

    @Override
    public Object map(Object value, Applicable<Object, Object> proceed) {
        return proceed.apply(value);
    }

    protected String propertyName(final Class<?> type) {
        return StringUtils.uncapitalize(type.getSimpleName().replace("Collection", ""));
    }
}