com.apporiented.spring.override.MapMergeProcessor.java Source code

Introduction

Here is the source code for com.apporiented.spring.override.MapMergeProcessor.java

Source

/* 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.apporiented.spring.override;

import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.BeansException;
import org.springframework.beans.PropertyValue;
import org.springframework.beans.factory.config.*;
import org.springframework.beans.factory.support.ManagedMap;
import org.springframework.core.PriorityOrdered;
import org.springframework.util.Assert;
import org.springframework.util.StringUtils;

import java.util.Map;

/**
 * Bean factory post processor that handles merging maps.
 * @author Felix Gnass [fgnass at neteye dot de]
 */
public class MapMergeProcessor implements BeanFactoryPostProcessor, PriorityOrdered {

    private Logger log = LoggerFactory.getLogger(MapMergeProcessor.class);

    private String ref;

    private String property;

    private Map<?, ?> entries;

    private int order = 1;

    public int getOrder() {
        return this.order;
    }

    public void setOrder(int order) {
        this.order = order;
    }

    public void setRef(String ref) {
        this.ref = ref;
    }

    public void setProperty(String property) {
        this.property = property;
    }

    public void setEntries(Map<?, ?> entries) {
        this.entries = entries;
    }

    @SuppressWarnings({ "rawtypes", "unchecked" })
    public void postProcessBeanFactory(ConfigurableListableBeanFactory beanFactory) throws BeansException {

        BeanDefinition bd = beanFactory.getBeanDefinition(ref);
        if (property == null) {
            Assert.state(MapFactoryBean.class.getName().equals(bd.getBeanClassName()),
                    "Bean [" + ref + "] must be a MapFactoryBean");

            property = "sourceMap";
        }

        if (log.isInfoEnabled()) {
            String keys = StringUtils.collectionToCommaDelimitedString(entries.keySet());
            log.debug("Adding [" + keys + "] to " + ref + "." + property);
        }

        PropertyValue pv = bd.getPropertyValues().getPropertyValue(property);
        if (pv == null) {
            // No map set on the target bean, create a new one ...
            ManagedMap map = new ManagedMap();
            map.putAll(entries);
            bd.getPropertyValues().addPropertyValue(property, map);
        } else {
            Object value = pv.getValue();
            if (value instanceof RuntimeBeanReference) {
                RuntimeBeanReference ref = (RuntimeBeanReference) value;
                value = beanFactory.getBean(ref.getBeanName());
            }
            Assert.isInstanceOf(Map.class, value);
            Map map = (Map) value;
            map.putAll(entries);
        }
    }
}