首页 > 代码库 > Hibernate格式化sql语句实体类 BasicFormatterImpl 源码
Hibernate格式化sql语句实体类 BasicFormatterImpl 源码
package com.util; import java.util.HashSet; import java.util.LinkedList; import java.util.Set; import java.util.StringTokenizer; public class BasicFormatterImpl { private static final Set<String> BEGIN_CLAUSES = new HashSet<String>(); private static final Set<String> END_CLAUSES = new HashSet<String>(); private static final Set<String> LOGICAL = new HashSet<String>(); private static final Set<String> QUANTIFIERS = new HashSet<String>(); private static final Set<String> DML = new HashSet<String>(); private static final Set<String> MISC = new HashSet<String>(); static final String indentString = " "; static final String initial = "\n "; public String format(String source) { return new FormatProcess(source).perform().trim() ; } static { BEGIN_CLAUSES.add("left"); BEGIN_CLAUSES.add("right"); BEGIN_CLAUSES.add("inner"); BEGIN_CLAUSES.add("outer"); BEGIN_CLAUSES.add("group"); BEGIN_CLAUSES.add("order"); END_CLAUSES.add("where"); END_CLAUSES.add("set"); END_CLAUSES.add("having"); END_CLAUSES.add("join"); END_CLAUSES.add("from"); END_CLAUSES.add("by"); END_CLAUSES.add("join"); END_CLAUSES.add("into"); END_CLAUSES.add("union"); LOGICAL.add("and"); LOGICAL.add("or"); LOGICAL.add("when"); LOGICAL.add("else"); LOGICAL.add("end"); QUANTIFIERS.add("in"); QUANTIFIERS.add("all"); QUANTIFIERS.add("exists"); QUANTIFIERS.add("some"); QUANTIFIERS.add("any"); DML.add("insert"); DML.add("update"); DML.add("delete"); MISC.add("select"); MISC.add("on"); } private static class FormatProcess { boolean beginLine = true; boolean afterBeginBeforeEnd = false; boolean afterByOrSetOrFromOrSelect = false; boolean afterValues = false; boolean afterOn = false; boolean afterBetween = false; boolean afterInsert = false; int inFunction = 0; int parensSinceSelect = 0; private LinkedList<Integer> parenCounts = new LinkedList<Integer>(); private LinkedList<Boolean> afterByOrFromOrSelects = new LinkedList<Boolean>(); int indent = 1; StringBuffer result = new StringBuffer(); StringTokenizer tokens; String lastToken; String token; String lcToken; public FormatProcess(String sql) { this.tokens = new StringTokenizer(sql, "()+*/-=<>'`\"[], \n\r\f\t", true); } public String perform() { this.result.append("\n "); while (this.tokens.hasMoreTokens()) { this.token = this.tokens.nextToken(); this.lcToken = this.token.toLowerCase(); if ("'".equals(this.token)) { String t; do { t = this.tokens.nextToken(); this.token += t; } while ((!"'".equals(t)) && (this.tokens.hasMoreTokens())); } else if ("\"".equals(this.token)) { String t; do { t = this.tokens.nextToken(); this.token += t; } while (!"\"".equals(t)); } if ((this.afterByOrSetOrFromOrSelect) && (",".equals(this.token))) { commaAfterByOrFromOrSelect(); } else if ((this.afterOn) && (",".equals(this.token))) { commaAfterOn(); } else if ("(".equals(this.token)) { openParen(); } else if (")".equals(this.token)) { closeParen(); } else if (BasicFormatterImpl.BEGIN_CLAUSES.contains(this.lcToken)) { beginNewClause(); } else if (BasicFormatterImpl.END_CLAUSES.contains(this.lcToken)) { endNewClause(); } else if ("select".equals(this.lcToken)) { select(); } else if (BasicFormatterImpl.DML.contains(this.lcToken)) { updateOrInsertOrDelete(); } else if ("values".equals(this.lcToken)) { values(); } else if ("on".equals(this.lcToken)) { on(); } else if ((this.afterBetween) && (this.lcToken.equals("and"))) { misc(); this.afterBetween = false; } else if (BasicFormatterImpl.LOGICAL.contains(this.lcToken)) { logical(); } else if (isWhitespace(this.token)) { white(); } else { misc(); } if (!isWhitespace(this.token)) { this.lastToken = this.lcToken; } } return this.result.toString(); } private void commaAfterOn() { out(); this.indent -= 1; newline(); this.afterOn = false; this.afterByOrSetOrFromOrSelect = true; } private void commaAfterByOrFromOrSelect() { out(); newline(); } private void logical() { if ("end".equals(this.lcToken)) { this.indent -= 1; } newline(); out(); this.beginLine = false; } private void on() { this.indent += 1; this.afterOn = true; newline(); out(); this.beginLine = false; } private void misc() { out(); if ("between".equals(this.lcToken)) { this.afterBetween = true; } if (this.afterInsert) { newline(); this.afterInsert = false; } else { this.beginLine = false; if ("case".equals(this.lcToken)) this.indent += 1; } } private void white() { if (!this.beginLine) this.result.append(" "); } private void updateOrInsertOrDelete() { out(); this.indent += 1; this.beginLine = false; if ("update".equals(this.lcToken)) { newline(); } if ("insert".equals(this.lcToken)) this.afterInsert = true; } private void select() { out(); this.indent += 1; newline(); this.parenCounts.addLast(new Integer(this.parensSinceSelect)); this.afterByOrFromOrSelects.addLast(Boolean.valueOf(this.afterByOrSetOrFromOrSelect)); this.parensSinceSelect = 0; this.afterByOrSetOrFromOrSelect = true; } private void out() { this.result.append(this.token); } private void endNewClause() { if (!this.afterBeginBeforeEnd) { this.indent -= 1; if (this.afterOn) { this.indent -= 1; this.afterOn = false; } newline(); } out(); if (!"union".equals(this.lcToken)) { this.indent += 1; } newline(); this.afterBeginBeforeEnd = false; this.afterByOrSetOrFromOrSelect = (("by".equals(this.lcToken)) || ("set".equals(this.lcToken)) || ("from".equals(this.lcToken))); } private void beginNewClause() { if (!this.afterBeginBeforeEnd) { if (this.afterOn) { this.indent -= 1; this.afterOn = false; } this.indent -= 1; newline(); } out(); this.beginLine = false; this.afterBeginBeforeEnd = true; } private void values() { this.indent -= 1; newline(); out(); this.indent += 1; newline(); this.afterValues = true; } private void closeParen() { this.parensSinceSelect -= 1; if (this.parensSinceSelect < 0) { this.indent -= 1; this.parensSinceSelect = ((Integer)this.parenCounts.removeLast()).intValue(); this.afterByOrSetOrFromOrSelect = ((Boolean)this.afterByOrFromOrSelects.removeLast()).booleanValue(); } if (this.inFunction > 0) { this.inFunction -= 1; out(); } else { if (!this.afterByOrSetOrFromOrSelect) { this.indent -= 1; newline(); } out(); } this.beginLine = false; } private void openParen() { if ((isFunctionName(this.lastToken)) || (this.inFunction > 0)) { this.inFunction += 1; } this.beginLine = false; if (this.inFunction > 0) { out(); } else { out(); if (!this.afterByOrSetOrFromOrSelect) { this.indent += 1; newline(); this.beginLine = true; } } this.parensSinceSelect += 1; } private static boolean isFunctionName(String tok) { char begin = tok.charAt(0); boolean isIdentifier = (Character.isJavaIdentifierStart(begin)) || ('"' == begin); return (isIdentifier) && (!BasicFormatterImpl.LOGICAL.contains(tok)) && (!BasicFormatterImpl.END_CLAUSES.contains(tok)) && (!BasicFormatterImpl.QUANTIFIERS.contains(tok)) && (!BasicFormatterImpl.DML.contains(tok)) && (!BasicFormatterImpl.MISC.contains(tok)); } private static boolean isWhitespace(String token) { return " \n\r\f\t".indexOf(token) >= 0; } private void newline() { this.result.append("\n"); for (int i = 0; i < this.indent; i++) { this.result.append(" "); } this.beginLine = true; } } }
调用:
System.out.println( new BasicFormatterImpl().format("select dd,dd,ss from ta1 group by dd order by createtime desc limit 3 ") ) ;
Hibernate格式化sql语句实体类 BasicFormatterImpl 源码
声明:以上内容来自用户投稿及互联网公开渠道收集整理发布,本网站不拥有所有权,未作人工编辑处理,也不承担相关法律责任,若内容有误或涉及侵权可进行投诉: 投诉/举报 工作人员会在5个工作日内联系你,一经查实,本站将立刻删除涉嫌侵权内容。