com.gitee.sunchenbin.mybatis.actable.utils.ColumnUtils Maven / Gradle / Ivy

Go to download

Show more of this group Show more artifacts with this name
Show all versions of mybatis-enhance-actable Show documentation

A.CTable is a Maven project based on Spring and Mybatis, which enhances the function of Mybatis

There is a newer version: 1.5.0.RELEASE

Show newest version

package com.gitee.sunchenbin.mybatis.actable.utils;

import com.gitee.sunchenbin.mybatis.actable.annotation.Column;
import com.gitee.sunchenbin.mybatis.actable.annotation.IsAutoIncrement;
import com.gitee.sunchenbin.mybatis.actable.annotation.IsKey;
import com.gitee.sunchenbin.mybatis.actable.annotation.IsNotNull;
import org.springframework.util.StringUtils;

import java.lang.reflect.Field;

public class ColumnUtils {
    
    public static String getColumnName(Field field){
        Column column = field.getAnnotation(Column.class);
        if(column == null){
            return null;
        }
        if (StringUtils.isEmpty(column.name())){
            char[] chars = field.getName().toCharArray();
            for (int i = 0; chars.length > i; i++){
                if(Character.isUpperCase(chars[i])){
                    StringBuilder sb = new StringBuilder(field.getName());
                    sb.insert(i,"_");
                    return sb.toString().toLowerCase();
                }
            }
            return field.getName().toLowerCase();
        }
        return column.name().toLowerCase();
    }

    public static boolean isKey(Field field){
        Column column = field.getAnnotation(Column.class);
        if(column == null){
            return false;
        }
        IsKey isKey = field.getAnnotation(IsKey.class);
        if(null != isKey){
            return true;
        }else if(column.isKey()){
            return true;
        }
        return false;
    }

    public static boolean isAutoIncrement(Field field){
        Column column = field.getAnnotation(Column.class);
        if(column == null){
            return false;
        }
        IsAutoIncrement isAutoIncrement = field.getAnnotation(IsAutoIncrement.class);
        if(null != isAutoIncrement){
            return true;
        }else if(column.isAutoIncrement()){
            return true;
        }
        return false;
    }

    public static boolean isNull(Field field){
        Column column = field.getAnnotation(Column.class);
        if(column == null){
            return false;
        }
        IsNotNull isNotNull = field.getAnnotation(IsNotNull.class);
        if(null != isNotNull){
            return false;
        }else if(column.isNull()){
            return true;
        }
        return false;
    }

}