com.github.youyinnn.youdbutils.dao.model.MappingHandler Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of you-db-utils Show documentation
There is a newer version: 2.2.3
package com.github.youyinnn.youdbutils.dao.model;

import java.util.ArrayList;
import java.util.HashMap;
import java.util.Set;

/**
 * 处理各种modelField集的映射,返回映射后的集.
 *
 * @author youyinnn
 */
public class MappingHandler {

    /**
     * 把传入的model字段列表映射成table字段列表
     *
     * @param modelName  the model name
     * @param modelField the model field
     * @return the array list
     * @throws NoSuchFieldException the no such field exception
     */
    public static ArrayList mappingHandle(String modelName, ArrayList modelField) throws NoSuchFieldException {

        if (modelField == null) {
            return null;
        }

        FieldMapping fieldMapping = ModelTableMessage.getFieldMapping(modelName);

        if (fieldMapping != null) {
            ArrayList result = new ArrayList<>();

            for (String mField : modelField) {
                if (fieldMapping.needToReplace(mField)) {
                    result.add(fieldMapping.getTableField(mField));
                } else {
                    result.add(mField);
                }
            }

            return result;
        } else {
            return modelField;
        }
    }

    /**
     * 把传入的model字段集合映射成table字段集合
     *
     * @param modelName  the model name
     * @param modelField the model field
     * @return the array list
     * @throws NoSuchFieldException the no such field exception
     */
    public static ArrayList mappingHandle(String modelName, Set modelField) throws NoSuchFieldException {
        return mappingHandle(modelName,new ArrayList<>(modelField));
    }

    /**
     * 把传入的model字段Map中的key值映射成table字段Map
     *
     * @param modelName  the model name
     * @param modelField the model field
     * @return the hash map
     * @throws NoSuchFieldException the no such field exception
     */
    public static HashMap mappingHandle(String modelName, HashMap modelField) throws NoSuchFieldException {

        if (modelField == null) {
            return null;
        }

        HashMap tableField = new HashMap<>(10);

        FieldMapping fieldMapping = ModelTableMessage.getFieldMapping(modelName);

        Set mFieldSet = modelField.keySet();

        for (String mField : mFieldSet) {
            Object value = modelField.get(mField);
            if (fieldMapping.needToReplace(mField)) {
                tableField.put(fieldMapping.getTableField(mField), value);
            } else {
                tableField.put(mField,value);
            }

        }

        return tableField;
    }

    /**
     * 把传入的model字段映射成table字段
     *
     * @param modelName  the model name
     * @param modelField the model field
     * @return the string
     * @throws NoSuchFieldException the no such field exception
     */
    public static String mappingHandle(String modelName, String modelField) throws NoSuchFieldException {

        FieldMapping fieldMapping = ModelTableMessage.getFieldMapping(modelName);

        if (fieldMapping.needToReplace(modelField)) {
            return fieldMapping.getTableField(modelField);
        }

        return modelField;
    }

}