querymethods.util.TkMapperUtil Maven / Gradle / Ivy

Go to download
Show more of this group Show more artifacts with this name
Show all versions of mybatis-querymethods Show documentation
There is a newer version: 0.1.4-RELEASE
package querymethods.util;

import java.util.ArrayList;
import java.util.Collections;
import java.util.List;

import org.apache.ibatis.mapping.MappedStatement;
import org.apache.ibatis.mapping.ResultMap;
import org.apache.ibatis.mapping.SqlCommandType;
import org.apache.ibatis.reflection.MetaObject;

import querymethods.springdata.query.parser.PartTree;
import tk.mybatis.mapper.entity.EntityTable;
import tk.mybatis.mapper.mapperhelper.EntityHelper;
import tk.mybatis.mapper.mapperhelper.MapperTemplate;
import tk.mybatis.mapper.mapperhelper.SqlHelper;
import tk.mybatis.mapper.util.MetaObjectUtil;
import tk.mybatis.mapper.util.StringUtil;

/**
 * TkMapper工具类
 * 
 * @see tk.mybatis.mapper.provider.ExampleProvider
 * 
 * @author OYGD
 *
 */
public class TkMapperUtil {

  private TkMapperUtil() {}

  public static String selectCountByExample(Class entityClass) {
    StringBuilder sql = new StringBuilder();
    sql.append(SqlHelper.selectCount(entityClass));
    sql.append(SqlHelper.fromTable(entityClass, tableName(entityClass)));
    sql.append(SqlHelper.exampleWhereClause());
    sql.append(SqlHelper.exampleForUpdate());
    return sql.toString();
  }

  /**
   * 根据Example查询
   *
   * @param ms
   * @return
   */
  public static String selectByExample(MappedStatement ms, Class entityClass, PartTree tree) {
    if (StringUtil.isEmpty(tree.getQueryProperty())) {
      // 将返回值修改为实体类型
      // tk.mybatis.mapper.provider.ExampleProvider#selectByExample
      TkMapperUtil.setResultType(ms, entityClass);
    }

    StringBuilder sql = new StringBuilder("SELECT ");
    sql.append("distinct");
    // 支持查询指定列
    sql.append(SqlHelper.exampleSelectColumns(entityClass));
    sql.append(SqlHelper.fromTable(entityClass, tableName(entityClass)));
    sql.append(SqlHelper.exampleWhereClause());
    sql.append(SqlHelper.exampleOrderBy(entityClass));
    sql.append(SqlHelper.exampleForUpdate());
    return sql.toString();
  }

  /**
   * 根据Example删除
   *
   * @param ms
   * @return
   */
  public static String deleteByExample(MappedStatement ms, Class entityClass) {
    StringBuilder sql = new StringBuilder();
    if (SqlHelper.hasLogicDeleteColumn(entityClass)) {
      sql.append(SqlHelper.updateTable(entityClass, tableName(entityClass)));
      sql.append("");
      sql.append(SqlHelper.logicDeleteColumnEqualsValue(entityClass, true));
      sql.append("");
      MetaObjectUtil.forObject(ms).setValue("sqlCommandType", SqlCommandType.UPDATE);
    } else {
      sql.append(SqlHelper.deleteFromTable(entityClass, tableName(entityClass)));
    }
    sql.append(SqlHelper.exampleWhereClause());
    return sql.toString();
  }

  /**
   * 获取实体类的表名
   *
   * @param entityClass
   * @return
   */
  public static String tableName(Class entityClass) {
    EntityTable entityTable = EntityHelper.getEntityTable(entityClass);
    String prefix = entityTable.getPrefix();
    if (StringUtil.isNotEmpty(prefix)) {
      return prefix + "." + entityTable.getName();
    }
    return entityTable.getName();
  }

  /**
   * 设置返回值类型 - 为了让typeHandler在select时有效，改为设置resultMap
   *
   * @param ms
   * @param entityClass
   * @see MapperTemplate#setResultType
   */
  public static void setResultType(MappedStatement ms, Class entityClass) {
    EntityTable entityTable = EntityHelper.getEntityTable(entityClass);
    List resultMaps = new ArrayList();
    resultMaps.add(entityTable.getResultMap(ms.getConfiguration()));
    MetaObject metaObject = MetaObjectUtil.forObject(ms);
    metaObject.setValue("resultMaps", Collections.unmodifiableList(resultMaps));
  }
}