cn.net.vidyo.yd.common.utils.convert.EntityServiceColumnConverter Maven / Gradle / Ivy

package cn.net.vidyo.yd.common.utils.convert;

import cn.hutool.core.bean.BeanUtil;
import cn.net.vidyo.yd.common.data.service.EntityService;

import java.util.Map;

public class EntityServiceColumnConverter  implements MapConverter {
    EntityService entityService;
    String idField;
    String nameField;
    String addNameField;

    public EntityServiceColumnConverter(EntityService entityService, String idField, String nameField, String addNameField) {
        this.entityService = entityService;
        this.idField = idField;
        this.nameField = nameField;
        this.addNameField = addNameField;
    }
    @Override
    public Map convert(Map data) {
        if(data.containsKey(idField)){
            Object id = data.get(idField);
            Object entity = entityService.getById(id);
            if(entity!=null) {
                data.put(addNameField,BeanUtil.getFieldValue(entity, nameField));
            }
        }
        return data;
    }
}