com.mybatis.jpa.util.ColumnMetaResolver Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mybatis-jpa Show documentation
The plugins for mybatis, in order to provider the ability to handler jpa.
There is a newer version: 2.5.1
package com.mybatis.jpa.util;

import com.mybatis.jpa.annotation.CodeEnum;
import com.mybatis.jpa.annotation.MappedJdbcType;
import com.mybatis.jpa.type.CodeType;
import com.mybatis.jpa.type.IntCodeEnumTypeHandler;
import com.mybatis.jpa.type.StringCodeEnumTypeHandler;
import java.lang.reflect.Field;
import java.util.Objects;
import javax.persistence.EnumType;
import javax.persistence.Enumerated;
import org.apache.ibatis.builder.BuilderException;
import org.apache.ibatis.type.BooleanTypeHandler;
import org.apache.ibatis.type.EnumOrdinalTypeHandler;
import org.apache.ibatis.type.EnumTypeHandler;
import org.apache.ibatis.type.JdbcType;
import org.apache.ibatis.type.TypeHandler;

/**
 * @author sway.li
 **/
public class ColumnMetaResolver {

  public static JdbcType resolveJdbcType(String alias) {
    if (alias == null) {
      return null;
    }
    try {
      return JdbcType.valueOf(alias);
    } catch (IllegalArgumentException e) {
      throw new BuilderException("Error resolving JdbcType. Cause: " + e, e);
    }
  }

  public static String resolveJdbcAlias(Field field) {

    if (field.isAnnotationPresent(MappedJdbcType.class)) {
      MappedJdbcType jdbcType = field.getAnnotation(MappedJdbcType.class);
      return jdbcType.value().name();
    }

    if (field.isAnnotationPresent(CodeEnum.class)) {
      CodeEnum codeEnum = field.getAnnotation(CodeEnum.class);
      if (Objects.equals(codeEnum.value(), CodeType.INT)) {
        return JdbcType.INTEGER.name();
      }
      if (Objects.equals(codeEnum.value(), CodeType.STRING)) {
        return JdbcType.VARCHAR.name();
      }
    }
    if (field.getType().isEnum()) {
      if (field.isAnnotationPresent(Enumerated.class)) {
        Enumerated enumerated = field.getAnnotation(Enumerated.class);
        if (Objects.equals(enumerated.value(), EnumType.ORDINAL)) {
          return "INTEGER";
        }
      }
      return "VARCHAR";
    }

    Class fieldType = field.getType();

    if (Integer.class.equals(fieldType)) {
      return "INTEGER";
    }
    if (Double.class.equals(fieldType)) {
      return "DOUBLE";
    }
    if (Float.class.equals(fieldType)) {
      return "FLOAT";
    }
    if (String.class.equals(fieldType)) {
      return "VARCHAR";
    }
    if (java.util.Date.class.isAssignableFrom(fieldType)) {
      return "TIMESTAMP";
    }
    return null;
  }

  @SuppressWarnings({"unchecked", "rawtypes"})
  public static Class> resolveTypeHandler(Field field) {

    if (field.isAnnotationPresent(CodeEnum.class)) {
      CodeEnum codeEnum = field.getAnnotation(CodeEnum.class);
      if (Objects.equals(codeEnum.value(), CodeType.INT)) {
        IntCodeEnumTypeHandler typeHandler = new IntCodeEnumTypeHandler(field.getType());
        return (Class>) typeHandler.getClass();
      }
      if (Objects.equals(codeEnum.value(), CodeType.STRING)) {
        StringCodeEnumTypeHandler typeHandler = new StringCodeEnumTypeHandler(field.getType());
        return (Class>) typeHandler.getClass();
      }
    }

    if (field.getType().isEnum()) {
      if (field.isAnnotationPresent(Enumerated.class)) {
        Enumerated enumerated = field.getAnnotation(Enumerated.class);
        if (enumerated.value() == EnumType.ORDINAL) {
          EnumOrdinalTypeHandler> typeHandler = new EnumOrdinalTypeHandler(
              field.getType());
          return (Class>) typeHandler.getClass();

        }
      }
      EnumTypeHandler> typeHandler = new EnumTypeHandler(field.getType());
      return (Class>) typeHandler.getClass();
    }

    if (field.getType().equals(Boolean.class)) {
      return BooleanTypeHandler.class;
    }
    return null;
  }

  /**
   * 装配sql中动态参数的占位符 #{parameterName,jdbcType=,typeHandler=}
   */
  public static String resolveSqlPlaceholder(Field field) {

    return resolveSqlPlaceholder(field, "");
  }

  /**
   * 装配sql中动态参数的占位符 #{alias.parameterName,jdbcType=,typeHandler=}
   */
  public static String resolveSqlPlaceholder(Field field, String alias) {
    StringBuilder sqlParameter = new StringBuilder();
    sqlParameter.append("#{");
    if (alias != null && !"".equals(alias)) {
      sqlParameter.append(alias).append(".");
    }
    sqlParameter.append(field.getName());

    // jdbcType
    String jdbcType = resolveJdbcAlias(field);

    if (jdbcType != null) {
      sqlParameter.append(", jdbcType=").append(jdbcType);
    }
    // typeHandler
    Class> typeHandler = resolveTypeHandler(field);

    if (typeHandler != null) {
      sqlParameter.append(", typeHandler=").append(typeHandler.getName());
    }
    sqlParameter.append("} ");

    return sqlParameter.toString();
  }
}