com.github.leeonky.map.ViewConverter 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.lang.reflect.Array;
import java.util.ArrayList;
import java.util.Collection;
import java.util.Map;

import static ma.glasnost.orika.metadata.TypeFactory.valueOf;

class ViewConverter extends BaseConverter {
    protected final Class view;
    protected final Mapper mapper;

    ViewConverter(Mapper mapper, Class view) {
        this.view = view;
        this.mapper = mapper;
    }

    @Override
    public Object convert(Object source, Type destinationType, MappingContext mappingContext) {
        Class rawType = destinationType.getRawType();
        if (Iterable.class.isAssignableFrom(rawType))
            return mapCollection((Iterable) source, createCollection(rawType), mappingContext);
        if (rawType.isArray())
            return mapCollection((Iterable) source, new ArrayList<>(), mappingContext).toArray((Object[]) Array.newInstance(rawType.getComponentType(), 0));
        if (Map.class.isAssignableFrom(rawType))
            return mapMap((Map) source, createMap(rawType), mappingContext);
        return map(source, mappingContext);
    }

    protected Object map(Object source, MappingContext mappingContext) {
        return mapper.findMapping(source, view).map(d -> {
            Object mappedObject = mappingContext.getMappedObject(source, valueOf(d));
            return mappedObject != null ? mappedObject : mapper.mapTo(source, d);
        }).orElse(null);
    }

    @SuppressWarnings("unchecked")
    private Map mapMap(Map source, Map result, MappingContext mappingContext) {
        source.forEach((k, v) -> result.put(k, map(v, mappingContext)));
        return result;
    }

    @SuppressWarnings("unchecked")
    private Collection mapCollection(Iterable source, Collection result, MappingContext mappingContext) {
        for (Object e : source)
            result.add(map(e, mappingContext));
        return result;
    }

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

}