MyException - 我的异常网
当前位置:我的异常网» SQL » 经过Hql、Sql生成Hibernate CountQuery 字符串

经过Hql、Sql生成Hibernate CountQuery 字符串

www.MyException.Cn  网友分享于:2015-02-08  浏览:0次
通过Hql、Sql生成Hibernate CountQuery 字符串

当我们在写Hibernate Orm组件的时候,经常会遇到分页查询这种操作,分页查询的原理不在熬述,比较重要的一点是需要计算查询的总数count,大部分人还是采用传统的hql/sql字符串截取或者拼接等方式实现。下面给出的代码是通过字符串模板、正则匹配等方式实现的,直接上代码:

public class QueryTemplateUtil {

 public static final String COUNT_QUERY_STRING = "select count(%s) from %s x";
 public static final String DELETE_ALL_QUERY_STRING = "delete from %s x";

 private static final String COUNT_REPLACEMENT_TEMPLATE = "select count(%s) $5$6$7";
 private static final String SIMPLE_COUNT_VALUE = "$2";
 private static final String COMPLEX_COUNT_VALUE = "$3$6";

 private static final Pattern COUNT_MATCH;

 private static final String IDENTIFIER = "[\\p{Alnum}._$]+";
 private static final String IDENTIFIER_GROUP = String.format("(%s)",
   IDENTIFIER);

 private static final String EQUALS_CONDITION_STRING = "%s.%s = ?";

 static {

  StringBuilder builder = new StringBuilder();
  builder.append("(?<=from)"); // from as starting delimiter
  builder.append("(?: )+"); // at least one space separating
  builder.append(IDENTIFIER_GROUP); // Entity name, can be qualified (any
  builder.append("(?: as)*"); // exclude possible "as" keyword
  builder.append("(?: )+"); // at least one space separating
  builder.append("(\\w*)"); // the actual alias


  builder = new StringBuilder();
  builder.append("(select\\s+((distinct )?(.+?)?)\\s+)?(from\\s+");
  builder.append(IDENTIFIER);
  builder.append("(?:\\s+as)?\\s+)");
  builder.append(IDENTIFIER_GROUP);
  builder.append("(.*)");

  COUNT_MATCH = compile(builder.toString(), CASE_INSENSITIVE);
 }

 /**
  * Private constructor to prevent instantiation.
  */
 private QueryTemplateUtil() {

 }

 /**
  * getCountQueryString:根据实体名称,count字段,查询条件生成count查询字符串
  * <p>
  * <b>例如:</b>{@code entityName="User"},{@code countQueryPlaceHolder="*"},
  * {@code conditionAttributes=["name","sex"]},生成的countQueryString是:
  * </p>
  * <p>
  * {@code select count(*) from User x where x.name=? and x.sex=?}
  * </p>
  * 
  * @Title: getCountQueryString
  * @param entityName
  *            实体名称,不允许{@literal null}或者{@literal empty}
  * @param countQueryPlaceHolder
  *            count查询占位字段,不允许{@literal null}或者{@literal empty}
  * @param conditionAttributes
  *            查询条件字段
  * @return String
  */
 public static String getCountQueryString(String entityName,
   String countQueryPlaceHolder, Iterable<String> conditionAttributes) {
  Assert.hasText(entityName, "实体名称不允许为null或empty!");
  StringBuilder sb = new StringBuilder(String.format(COUNT_QUERY_STRING,
    countQueryPlaceHolder, entityName));
  sb.append(" WHERE ");

  if (conditionAttributes != null) {
   for (String conditionAttribute : conditionAttributes) {
    sb.append(String.format(EQUALS_CONDITION_STRING, "x",
      conditionAttribute));
    sb.append(" AND ");
   }
  }

  sb.append("1 = 1");
  return sb.toString();
 }

 /**
  * getCountQueryString:根据实体名称,生成count查询字符串
  * <p>
  * <b>注意:</b>count占位字符默认为{@literal *},不带查询条件
  * </p>
  * 
  * @Title: getCountQueryString
  * @param entityName
  *            实体名称,不允许{@literal null}或者{@literal empty}
  * @return {@code select count(*) from SomeEntity x }
  */
 public static String getCountQueryString(String entityName) {
  return getCountQueryString(entityName, "*", null);
 }

 /**
  * getCountQueryString:根据实体名称、查询条件生成count查询字符串
  * 
  * @Title: getCountQueryString
  * @param entityName
  *            实体名称,不允许{@literal null}或者{@literal empty}
  * @param conditionAttributes
  *            查询条件字段
  * @return {@code select count(*) from SomeEntity x where condition1=? and condition2=? and 1=1}
  */
 public static String getCountQueryString(String entityName,
   Iterable<String> conditionAttributes) {
  return getCountQueryString(entityName, "*", conditionAttributes);

 }

 /**
  * getQueryString:通过实体名称与查询字符串模板,创建查询字符串
  * <p>
  * <b>参见:</b>{@link org.mk.mini.orm.util.DELETE_ALL_QUERY_STRING}
  * </p>
  * 
  * @Title: getQueryStringByTemplate
  * @param template
  *            查询字符串模板
  * @param entityName
  *            实体名称
  * @return 查询字符串
  */
 public static String getQueryStringByTemplate(String template,
   String entityName) {

  Assert.hasText(entityName, "实体名称不允许为null或empty!");

  return String.format(template, entityName);
 }

 

 /**
  * createCountQueryFor:通过给定的原始sql或hql创建count查询字符串.
  * 
  * @Title: createCountQueryFor
  * @param originalQuery
  *            原始查询字符串,支持sql与hql 不能为 {@literal null} 或者 {@literal empty}
  * @return
  */
 public static String createCountQueryFor(String originalQuery) {

  Assert.hasText(originalQuery, "查询字符串不允许为null或empty");

  Matcher matcher = COUNT_MATCH.matcher(originalQuery);
  String variable = matcher.matches() ? matcher.group(4) : null;
  boolean useVariable = ExtraStringUtils.hasText(variable)
    && !variable.startsWith("new")
    && !variable.startsWith("count(");

  return matcher.replaceFirst(String.format(COUNT_REPLACEMENT_TEMPLATE,
    useVariable ? SIMPLE_COUNT_VALUE : COMPLEX_COUNT_VALUE));
 }
}

 

实例就不用写了,注释已经写的很清楚,关于Assert和ExtraStringUtils只是做了断言判断与字符串判断,需要copy该代码的同学们可以改造一下,写的不好,请指正。

文章评论

初级 vs 高级开发者 哪个性价比更高?
初级 vs 高级开发者 哪个性价比更高?
如何成为一名黑客
如何成为一名黑客
程序员的鄙视链
程序员的鄙视链
Web开发者需具备的8个好习惯
Web开发者需具备的8个好习惯
当下全球最炙手可热的八位少年创业者
当下全球最炙手可热的八位少年创业者
我跳槽是因为他们的显示器更大
我跳槽是因为他们的显示器更大
Web开发人员为什么越来越懒了?
Web开发人员为什么越来越懒了?
总结2014中国互联网十大段子
总结2014中国互联网十大段子
什么才是优秀的用户界面设计
什么才是优秀的用户界面设计
团队中“技术大拿”并非越多越好
团队中“技术大拿”并非越多越好
中美印日四国程序员比较
中美印日四国程序员比较
代码女神横空出世
代码女神横空出世
聊聊HTTPS和SSL/TLS协议
聊聊HTTPS和SSL/TLS协议
不懂技术不要对懂技术的人说这很容易实现
不懂技术不要对懂技术的人说这很容易实现
科技史上最臭名昭著的13大罪犯
科技史上最臭名昭著的13大罪犯
看13位CEO、创始人和高管如何提高工作效率
看13位CEO、创始人和高管如何提高工作效率
为啥Android手机总会越用越慢?
为啥Android手机总会越用越慢?
2013年美国开发者薪资调查报告
2013年美国开发者薪资调查报告
一个程序员的时间管理
一个程序员的时间管理
10个帮程序员减压放松的网站
10个帮程序员减压放松的网站
写给自己也写给你 自己到底该何去何从
写给自己也写给你 自己到底该何去何从
做程序猿的老婆应该注意的一些事情
做程序猿的老婆应该注意的一些事情
 程序员的样子
程序员的样子
程序猿的崛起——Growth Hacker
程序猿的崛起——Growth Hacker
为什么程序员都是夜猫子
为什么程序员都是夜猫子
程序员和编码员之间的区别
程序员和编码员之间的区别
10个调试和排错的小建议
10个调试和排错的小建议
那些争议最大的编程观点
那些争议最大的编程观点
Java 与 .NET 的平台发展之争
Java 与 .NET 的平台发展之争
那些性感的让人尖叫的程序员
那些性感的让人尖叫的程序员
我的丈夫是个程序员
我的丈夫是个程序员
程序员都该阅读的书
程序员都该阅读的书
程序员最害怕的5件事 你中招了吗?
程序员最害怕的5件事 你中招了吗?
程序员应该关注的一些事儿
程序员应该关注的一些事儿
要嫁就嫁程序猿—钱多话少死的早
要嫁就嫁程序猿—钱多话少死的早
漫画:程序员的工作
漫画:程序员的工作
程序员的一天:一寸光阴一寸金
程序员的一天:一寸光阴一寸金
“肮脏的”IT工作排行榜
“肮脏的”IT工作排行榜
鲜为人知的编程真相
鲜为人知的编程真相
Google伦敦新总部 犹如星级庄园
Google伦敦新总部 犹如星级庄园
“懒”出效率是程序员的美德
“懒”出效率是程序员的美德
程序员眼里IE浏览器是什么样的
程序员眼里IE浏览器是什么样的
每天工作4小时的程序员
每天工作4小时的程序员
十大编程算法助程序员走上高手之路
十大编程算法助程序员走上高手之路
60个开发者不容错过的免费资源库
60个开发者不容错过的免费资源库
程序员周末都喜欢做什么?
程序员周末都喜欢做什么?
如何区分一个程序员是“老手“还是“新手“?
如何区分一个程序员是“老手“还是“新手“?
程序员必看的十大电影
程序员必看的十大电影
老美怎么看待阿里赴美上市
老美怎么看待阿里赴美上市
软件开发程序错误异常ExceptionCopyright © 2009-2015 MyException 版权所有