com.github.leeonky.map.ViewMapConverter Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of view-mapper Show documentation

view-mapper

There is a newer version: 0.7.5

package com.github.leeonky.map;

import ma.glasnost.orika.MappingContext;
import ma.glasnost.orika.metadata.Type;

import java.util.Map;

class ViewMapConverter extends ViewListConverter {
    private final String keyPropertyName;

    ViewMapConverter(Mapper mapper, Class view, String keyPropertyName, String propertyName) {
        super(mapper, view, propertyName);
        this.keyPropertyName = keyPropertyName;
    }

    @Override
    public Object convert(Object source, Type destinationType, MappingContext mappingContext) {
        Class rawType = destinationType.getRawType();
        Iterable collection = source instanceof Map ? wrapperEntry((Map) source) : (Iterable) source;
        return mapMap(collection, createMap(rawType), mappingContext);
    }

    @SuppressWarnings("unchecked")
    private Map mapMap(Iterable source, Map result, MappingContext mappingContext) {
        source.forEach(e -> result.put(getPropertyValue(e, keyPropertyName), map(getPropertyValue(e, propertyName), mappingContext)));
        return result;
    }

    @Override
    public String buildConvertId() {
        return String.format("ViewMapConverter:%s:%s:%s[%d]", keyPropertyName, propertyName, view.getName(), mapper.hashCode());
    }
}