Examples with TableField com.baomidou.mybatisplus.annotation.TableField used on opensource projects

Example 11 with TableField

use of com.baomidou.mybatisplus.annotation.TableField in project kms by mahonelau.

the class QueryGenerator method getTableFieldName.

/**
 * 获取表字段名
 * @param clazz
 * @param name
 * @return
 */
private static String getTableFieldName(Class<?> clazz, String name) {
    try {
        // 如果字段加注解了@TableField(exist = false),不走DB查询
        Field field = null;
        try {
            field = clazz.getDeclaredField(name);
        } catch (NoSuchFieldException e) {
        // e.printStackTrace();
        }
        // 如果为空，则去父类查找字段
        if (field == null) {
            List<Field> allFields = getClassFields(clazz);
            List<Field> searchFields = allFields.stream().filter(a -> a.getName().equals(name)).collect(Collectors.toList());
            if (searchFields != null && searchFields.size() > 0) {
                field = searchFields.get(0);
            }
        }
        if (field != null) {
            TableField tableField = field.getAnnotation(TableField.class);
            if (tableField != null) {
                if (tableField.exist() == false) {
                    // 如果设置了TableField false 这个字段不需要处理
                    return null;
                } else {
                    String column = tableField.value();
                    // 如果设置了TableField value 这个字段是实体字段
                    if (!"".equals(column)) {
                        return column;
                    }
                }
            }
        }
    } catch (Exception e) {
        e.printStackTrace();
    }
    return name;
}