JdbcHelper.java 76 KB

12345678910111213141516171819202122232425262728293031323334353637383940414243444546474849505152535455565758596061626364656667686970717273747576777879808182838485868788899091929394959697989910010110210310410510610710810911011111211311411511611711811912012112212312412512612712812913013113213313413513613713813914014114214314414514614714814915015115215315415515615715815916016116216316416516616716816917017117217317417517617717817918018118218318418518618718818919019119219319419519619719819920020120220320420520620720820921021121221321421521621721821922022122222322422522622722822923023123223323423523623723823924024124224324424524624724824925025125225325425525625725825926026126226326426526626726826927027127227327427527627727827928028128228328428528628728828929029129229329429529629729829930030130230330430530630730830931031131231331431531631731831932032132232332432532632732832933033133233333433533633733833934034134234334434534634734834935035135235335435535635735835936036136236336436536636736836937037137237337437537637737837938038138238338438538638738838939039139239339439539639739839940040140240340440540640740840941041141241341441541641741841942042142242342442542642742842943043143243343443543643743843944044144244344444544644744844945045145245345445545645745845946046146246346446546646746846947047147247347447547647747847948048148248348448548648748848949049149249349449549649749849950050150250350450550650750850951051151251351451551651751851952052152252352452552652752852953053153253353453553653753853954054154254354454554654754854955055155255355455555655755855956056156256356456556656756856957057157257357457557657757857958058158258358458558658758858959059159259359459559659759859960060160260360460560660760860961061161261361461561661761861962062162262362462562662762862963063163263363463563663763863964064164264364464564664764864965065165265365465565665765865966066166266366466566666766866967067167267367467567667767867968068168268368468568668768868969069169269369469569669769869970070170270370470570670770870971071171271371471571671771871972072172272372472572672772872973073173273373473573673773873974074174274374474574674774874975075175275375475575675775875976076176276376476576676776876977077177277377477577677777877978078178278378478578678778878979079179279379479579679779879980080180280380480580680780880981081181281381481581681781881982082182282382482582682782882983083183283383483583683783883984084184284384484584684784884985085185285385485585685785885986086186286386486586686786886987087187287387487587687787887988088188288388488588688788888989089189289389489589689789889990090190290390490590690790890991091191291391491591691791891992092192292392492592692792892993093193293393493593693793893994094194294394494594694794894995095195295395495595695795895996096196296396496596696796896997097197297397497597697797897998098198298398498598698798898999099199299399499599699799899910001001100210031004100510061007100810091010101110121013101410151016101710181019102010211022102310241025102610271028102910301031103210331034103510361037103810391040104110421043104410451046104710481049105010511052105310541055105610571058105910601061106210631064106510661067106810691070107110721073107410751076107710781079108010811082108310841085108610871088108910901091109210931094109510961097109810991100110111021103110411051106110711081109111011111112111311141115111611171118111911201121112211231124112511261127112811291130113111321133113411351136113711381139114011411142114311441145114611471148114911501151115211531154115511561157115811591160116111621163116411651166116711681169117011711172117311741175117611771178117911801181118211831184118511861187118811891190119111921193119411951196119711981199120012011202120312041205120612071208120912101211121212131214121512161217121812191220122112221223122412251226122712281229123012311232123312341235123612371238123912401241124212431244124512461247124812491250125112521253125412551256125712581259126012611262126312641265126612671268126912701271127212731274127512761277127812791280128112821283128412851286128712881289129012911292129312941295129612971298129913001301130213031304130513061307130813091310131113121313131413151316131713181319132013211322132313241325132613271328132913301331133213331334133513361337133813391340134113421343134413451346134713481349135013511352135313541355135613571358135913601361136213631364136513661367136813691370137113721373137413751376137713781379138013811382138313841385138613871388138913901391139213931394139513961397139813991400140114021403140414051406140714081409141014111412141314141415141614171418141914201421142214231424142514261427142814291430143114321433143414351436143714381439144014411442144314441445144614471448144914501451145214531454145514561457145814591460146114621463146414651466146714681469147014711472147314741475147614771478147914801481148214831484148514861487148814891490149114921493149414951496149714981499150015011502150315041505150615071508150915101511151215131514151515161517151815191520152115221523152415251526152715281529153015311532153315341535153615371538153915401541154215431544154515461547154815491550155115521553155415551556155715581559156015611562156315641565156615671568156915701571157215731574157515761577157815791580158115821583158415851586158715881589159015911592159315941595159615971598159916001601160216031604160516061607160816091610161116121613161416151616161716181619162016211622162316241625162616271628162916301631163216331634163516361637163816391640164116421643164416451646164716481649165016511652165316541655165616571658165916601661166216631664166516661667166816691670167116721673167416751676167716781679168016811682168316841685168616871688168916901691169216931694169516961697169816991700170117021703170417051706170717081709171017111712171317141715171617171718171917201721172217231724172517261727172817291730173117321733
  1. package com.kym.jdbc.template;
  2. import com.kym.DbUtil;
  3. import com.kym.JacksonUtil;
  4. import com.kym.jdbc.BasicQuery;
  5. import com.kym.jdbc.IHandler;
  6. import com.kym.jdbc.OBuilder;
  7. import com.kym.jdbc.SQLEntity;
  8. import com.kym.jdbc.annotations.DBF;
  9. import com.kym.jdbc.annotations.Entity;
  10. import com.kym.jdbc.annotations.FK;
  11. import com.kym.jdbc.annotations.JoinType;
  12. import com.kym.jdbc.annotations.Many;
  13. import com.kym.jdbc.annotations.OP;
  14. import com.kym.jdbc.annotations.One;
  15. import com.kym.jdbc.annotations.QE;
  16. import com.kym.jdbc.annotations.QF;
  17. import com.kym.jdbc.annotations.QueryType;
  18. import org.slf4j.Logger;
  19. import org.slf4j.LoggerFactory;
  20. import java.lang.reflect.Array;
  21. import java.lang.reflect.Field;
  22. import java.lang.reflect.Type;
  23. import java.math.BigDecimal;
  24. import java.sql.Timestamp;
  25. import java.util.ArrayList;
  26. import java.util.Collection;
  27. import java.util.Collections;
  28. import java.util.Comparator;
  29. import java.util.Date;
  30. import java.util.HashMap;
  31. import java.util.List;
  32. import java.util.Map;
  33. import java.util.Set;
  34. import java.util.function.Function;
  35. import java.util.stream.Collectors;
  36. /**
  37. * SQL动态解析器
  38. * <p>
  39. * jdbcTemplate模式
  40. *
  41. * @author yaop
  42. */
  43. public class JdbcHelper {
  44. private final Logger logger = LoggerFactory.getLogger(JdbcHelper.class);
  45. public static final int DIALECT_MYSQL = 1;
  46. public static final int DIALECT_MSSQL = 2;
  47. public static final int DIALECT_ORACLE = 3;
  48. public static final int DIALECT_PGSQL = 4;
  49. public static final String[] BASE_IGNORE = {"createAt", "updateAt", "createBy", "updateBy"};
  50. private static final String DEFAULT_MKF = "id";
  51. private String comma = "`";
  52. private int dialect = 1;
  53. /***
  54. * 表名前缀
  55. */
  56. public static String tbPrefix = "t";
  57. private static Function<String, String> func;
  58. /**
  59. * 设置解析器的SQL方言类型,默认MYSQL
  60. */
  61. public void setDialect(int dialectType) {
  62. this.dialect = dialectType;
  63. OBuilder.dialect = dialectType;
  64. switch (dialectType) {
  65. case IHandler.DIALECT_MYSQL -> this.comma = "`";
  66. case IHandler.DIALECT_MSSQL -> this.comma = "'";
  67. case IHandler.DIALECT_ORACLE -> this.comma = "\"";
  68. case IHandler.DIALECT_PGSQL -> this.comma = "\"";
  69. default -> {
  70. }
  71. }
  72. }
  73. public int getDialect() {
  74. return dialect;
  75. }
  76. /**
  77. * 设置表名前缀
  78. */
  79. public static void setFunc(Function<String, String> func) {
  80. JdbcHelper.func = func;
  81. }
  82. public static void setTablePrefix(String prefix) {
  83. JdbcHelper.tbPrefix = prefix;
  84. }
  85. public static String getTbPrefix() {
  86. return JdbcHelper.tbPrefix;
  87. }
  88. //region json处理
  89. //endregion
  90. //region select
  91. public SQLEntity selectList(String sql, Object... params) {
  92. SQLEntity entity = new SQLEntity();
  93. /* if (!DbUtil.isEmptyOrNull(params)) {
  94. for (Object param : params) {
  95. if (param.getClass() == String.class) {
  96. sql = sql.replaceFirst("\\?", "'" + DbUtil.injectDefend(param.toString()) + "'");
  97. } else if (param.getClass() == Date.class) {
  98. sql = sql.replaceFirst("\\?", "'" + new Timestamp(((Date) param).getTime()) + "'");
  99. } else {
  100. sql = sql.replaceFirst("\\?", DbUtil.injectDefend(param.toString()));
  101. }
  102. }
  103. }*/
  104. entity.sql = sql;
  105. entity.parameters = params;
  106. return entity;
  107. }
  108. /**
  109. * 聚合函数查询<p>
  110. * 注意:聚合函数的查询需要使用子查询</p>
  111. *
  112. * @param clz
  113. * @param func 聚合查询字段 count(1) sum(amt)
  114. * @param builder
  115. * @return
  116. */
  117. public SQLEntity selectFunc(Class<?> clz, String func, OBuilder builder) {
  118. SQLEntity entity = new SQLEntity();
  119. List<Object> parameters = new ArrayList<>(16);
  120. boolean distinct = distinctAggerate(func);
  121. StringBuilder sbr = new StringBuilder();
  122. boolean originEntity = !clz.getSimpleName().endsWith("Info");
  123. if (originEntity) {
  124. String funcField = func;
  125. if (!distinct) {
  126. funcField = func.contains("COUNT(") ? "1" : func.trim().split("\\(")[1].split("\\)")[0].trim();
  127. }
  128. sbr.append("SELECT ").append(funcField).append(" FROM ").append(getTableName(clz, clz.getSimpleName()));
  129. if (null != builder) {
  130. if (!DbUtil.isEmptyOrNull(builder.getWheres())) {
  131. List<OBuilder.Where> wheres = builder.getWheres();
  132. if (!DbUtil.isEmptyOrNull(wheres)) {
  133. sbr.append(" \r\n WHERE 1=1");
  134. wheres.forEach(wh -> {
  135. sbr.append(" AND ");
  136. appendBuilderWhere(sbr, wh, "", parameters);
  137. });
  138. }
  139. }
  140. //GROUP BY
  141. if (!DbUtil.isEmptyOrNull(builder.getGroupBys())) {
  142. sbr.append(" \r\n GROUP BY ");
  143. builder.getGroupBys().forEach(group -> {
  144. sbr.append(group).append(" ,");
  145. });
  146. sbr.deleteCharAt(sbr.length() - 1);
  147. }
  148. //ORDER BY
  149. if (!DbUtil.isEmptyOrNull(builder.getOrderBys())) {
  150. sbr.append(" ORDER BY ");
  151. builder.getOrderBys().forEach(order -> sbr.append(order).append(" ,"));
  152. sbr.deleteCharAt(sbr.length() - 1);
  153. }
  154. //count不分页
  155. if (!func.contains("COUNT(")) {
  156. //分页
  157. String result = appendWhereLimit(builder, sbr);
  158. if (null != result) {
  159. if (!distinct) {
  160. entity.sql = "SELECT " + func + " FROM (" + result + ") temp";
  161. } else {
  162. entity.sql = result;
  163. }
  164. }
  165. }
  166. }
  167. if (!distinct) {
  168. entity.sql = "SELECT " + func + " FROM (" + sbr.toString() + ") temp";
  169. } else {
  170. entity.sql = sbr.toString();
  171. }
  172. } else {
  173. //支持聚合查询的关联
  174. String queryDomain = clz.getSuperclass().getCanonicalName() + "$" + clz.getSuperclass().getSimpleName() + "BasicQuery";
  175. try {
  176. Object obj = Class.forName(queryDomain).newInstance();
  177. if (!DbUtil.isEmptyOrNull(func)) {
  178. DbUtil.setFieldValue(obj, obj.getClass().getField("pageSize"), -1);
  179. }
  180. return select((BasicQuery) obj, builder, null, null, func, true);
  181. } catch (InstantiationException | IllegalAccessException | ClassNotFoundException | NoSuchFieldException e) {
  182. logger.error("SQL2Helper ERR# initialize class,class:{}", queryDomain);
  183. throw new IllegalStateException(e);
  184. }
  185. }
  186. entity.parameters = parameters.toArray();
  187. return entity;
  188. }
  189. /**
  190. * 聚合函数查询
  191. */
  192. public SQLEntity selectDistinct(Class<?> clz, String func, OBuilder OBuilder) {
  193. return selectFunc(clz, func, OBuilder);
  194. }
  195. public SQLEntity selectListByWhere(Class<?> clz, OBuilder OBuilder, List<String> include, List<String> exclude, boolean isList) {
  196. return selectListByWhereV2(clz, OBuilder, include, exclude, null, isList);
  197. }
  198. public SQLEntity selectListByWhereV2(Class<?> clz, OBuilder OBuilder, List<String> include, List<String> exclude, String aggregate, boolean isList) {
  199. SQLEntity sqlEntity = new SQLEntity();
  200. List<Object> parameters = new ArrayList<>();
  201. Set<Field> fields = DbUtil.getEntityFields(clz);
  202. boolean isSimple = !clz.getSimpleName().endsWith("Info");
  203. if (isSimple) {
  204. Entity entity = clz.getAnnotation(Entity.class);
  205. StringBuilder sbr = new StringBuilder();
  206. if (!DbUtil.isEmptyOrNull(aggregate)) {
  207. sbr.append("SELECT ").append(aggregate).append(" FROM ").append(getTableName(entity));
  208. } else {
  209. if (DbUtil.isEmptyOrNull(include)) {
  210. if (DbUtil.isEmptyOrNull(exclude)) {
  211. sbr.append("SELECT * FROM ").append(getTableName(entity));
  212. } else {
  213. sbr.append("SELECT ");
  214. for (Field field : fields) {
  215. if (DbUtil.isUsageField(field)) {
  216. //richId rich2Id 不关联查
  217. String fileName = field.getName();
  218. if (!exclude.contains(fileName)) {
  219. sbr.append(this.comma).append(DbUtil.getColumnName(fileName)).append(this.comma).append(" ,");
  220. }
  221. }
  222. }
  223. sbr.deleteCharAt(sbr.length() - 1);
  224. sbr.append(" \n FROM ").append(getTableName(entity));
  225. }
  226. } else {
  227. sbr.append("SELECT ");
  228. for (Field field : fields) {
  229. if (DbUtil.isUsageField(field)) {
  230. String fileName = field.getName();
  231. if (include.contains(fileName)) {
  232. sbr.append(this.comma).append(DbUtil.getColumnName(fileName)).append(this.comma).append(" ,");
  233. }
  234. }
  235. }
  236. sbr.deleteCharAt(sbr.length() - 1);
  237. sbr.append(" \n FROM ").append(getTableName(entity));
  238. }
  239. }
  240. if (null != OBuilder && !DbUtil.isEmptyOrNull(OBuilder.getWheres())) {
  241. List<OBuilder.Where> wheres = OBuilder.getWheres();
  242. if (!DbUtil.isEmptyOrNull(wheres)) {
  243. sbr.append(" \r\n WHERE 1=1");
  244. wheres.forEach(wh -> {
  245. sbr.append(" AND ");
  246. appendBuilderWhere(sbr, wh, "", parameters);
  247. });
  248. }
  249. }
  250. if (isList) {
  251. //ORDER BY
  252. if (null != OBuilder && !DbUtil.isEmptyOrNull(OBuilder.getOrderBys())) {
  253. sbr.append(" ORDER BY ");
  254. OBuilder.getOrderBys().forEach(order -> sbr.append(order).append(" ,"));
  255. sbr.deleteCharAt(sbr.length() - 1);
  256. } else {
  257. sbr.append(" ORDER BY update_at DESC ");
  258. }
  259. //limit
  260. String result = appendWhereLimit(OBuilder, sbr);
  261. if (null != result) {
  262. sqlEntity.sql = result;
  263. }
  264. }
  265. sbr.append(";");
  266. sqlEntity.sql = sbr.toString();
  267. } else {
  268. String queryDomain = clz.getSuperclass().getCanonicalName() + "$" + clz.getSuperclass().getSimpleName() + "BasicQuery";
  269. Object obj;
  270. try {
  271. obj = Class.forName(queryDomain).newInstance();
  272. if (!DbUtil.isEmptyOrNull(aggregate)) {
  273. DbUtil.setFieldValue(obj, obj.getClass().getField("pageSize"), -1);
  274. }
  275. } catch (InstantiationException | IllegalAccessException | ClassNotFoundException | NoSuchFieldException e) {
  276. logger.error("SQL2Helper ERR# initialize class,class:{}", queryDomain);
  277. throw new IllegalStateException(e);
  278. }
  279. return select((BasicQuery) obj, OBuilder, include, exclude, aggregate, isList);
  280. }
  281. sqlEntity.parameters = parameters.toArray();
  282. return sqlEntity;
  283. }
  284. public SQLEntity selectListByQuery(BasicQuery query, List<String> include, List<String> exclude) {
  285. return select(query, null, include, exclude, null, true);
  286. }
  287. /**
  288. * 查询SQL解析生成
  289. *
  290. * @param query query条件对象
  291. * @param OBuilder where条件对象
  292. * @param include 指定查询字段集合
  293. * @param exclude 指定过滤不查询字段集合
  294. * @param aggregate 聚合查询条件
  295. */
  296. private SQLEntity select(BasicQuery query, OBuilder OBuilder, List<String> include, List<String> exclude, String aggregate, boolean isList) {
  297. SQLEntity sqlEntity = new SQLEntity();
  298. List<Object> parameters = new ArrayList<>(16);
  299. if (!DbUtil.isEmptyOrNull(query.includeFields)) {
  300. if (DbUtil.isEmptyOrNull(include)) {
  301. include = new ArrayList<>();
  302. }
  303. for (String includeField : query.includeFields) {
  304. if (!include.contains(includeField)) {
  305. include.add(includeField);
  306. }
  307. }
  308. }
  309. if (!DbUtil.isEmptyOrNull(query.excludeFields)) {
  310. if (DbUtil.isEmptyOrNull(exclude)) {
  311. exclude = new ArrayList<>();
  312. }
  313. for (String excludeField : query.excludeFields) {
  314. if (!exclude.contains(excludeField)) {
  315. exclude.add(excludeField);
  316. }
  317. }
  318. }
  319. StringBuilder sbr = new StringBuilder();
  320. int aliasIdx = 0;
  321. //表别名 @QF->inner @FK、@One->left
  322. Map<String, String> alias = new HashMap<>(4);
  323. //关联表名
  324. Map<String, String> tbs = new HashMap<>(4);
  325. QE qEAnnotation = query.getClass().getAnnotation(QE.class);
  326. String queryClassName = query.getClass().getSimpleName();
  327. if (null == qEAnnotation) {
  328. throw new IllegalStateException("query entity of " + queryClassName + " @Query Entity is not defined!");
  329. }
  330. Class<?> infoClass = qEAnnotation.clz();
  331. Entity entity = infoClass.getSuperclass().getAnnotation(Entity.class);
  332. // Field[] fields = infoClass.getFields();
  333. Set<Field> fields = DbUtil.getEntityFields(infoClass);
  334. Set<Field> qfields = DbUtil.getQueryFields(query.getClass());
  335. // Field[] qfields = query.getClass().getFields();
  336. //关联表别名
  337. //------1.查询字段
  338. for (Field field : fields) {
  339. String fieldName = field.getName();
  340. //常量字段过滤
  341. if (!DbUtil.isUsageField(field)) {
  342. continue;
  343. }
  344. One oneAnnotation = field.getAnnotation(One.class);
  345. Many manyAnnotation = field.getAnnotation(Many.class);
  346. boolean isExtenstionField = null != oneAnnotation || null != manyAnnotation;
  347. //忽略字段过滤
  348. if (!DbUtil.isEmptyOrNull(exclude) && exclude.contains(fieldName) && !isExtenstionField) {
  349. continue;
  350. }
  351. //指定查询
  352. if (!DbUtil.isEmptyOrNull(include) && !include.contains(fieldName) && !isExtenstionField) {
  353. continue;
  354. }
  355. DBF DBFAnnotation = field.getAnnotation(DBF.class);
  356. if (null != DBFAnnotation) {
  357. FK foreignKey = field.getAnnotation(FK.class);
  358. if (null != foreignKey) {
  359. Class<?> entityClass = foreignKey.clz();
  360. Entity ey = entityClass.getAnnotation(Entity.class);
  361. for (Field qfield : qfields) {
  362. if (qfield.isAnnotationPresent(QF.class)) {
  363. QF qf = qfield.getAnnotation(QF.class);
  364. String mkf = qf.pkf();
  365. if (!DbUtil.isEmptyOrNull(mkf) && mkf.equals(fieldName)) {
  366. if (null != DbUtil.getFieldValue(query, qfield)) {
  367. alias.putIfAbsent(fieldName, "i" + aliasIdx++);
  368. tbs.putIfAbsent(fieldName, getTableName(ey));
  369. }
  370. }
  371. }
  372. }
  373. alias.putIfAbsent(fieldName, "l" + aliasIdx++);
  374. tbs.putIfAbsent(fieldName, getTableName(ey));
  375. }
  376. } else {
  377. if (null != oneAnnotation) {
  378. String mkf = oneAnnotation.mkf();
  379. if (!DbUtil.isGenericType(field.getType()) && DEFAULT_MKF.equals(mkf)) {
  380. mkf = fieldName;
  381. }
  382. JoinType joinType = oneAnnotation.join();
  383. if (JoinType.LEFT == joinType) {
  384. alias.put(mkf, "l" + aliasIdx++);
  385. } else if (JoinType.INNER == joinType) {
  386. alias.put(mkf, "i" + aliasIdx++);
  387. } else if (JoinType.RIGHT == joinType) {
  388. alias.put(mkf, "r" + aliasIdx++);
  389. }
  390. }
  391. }
  392. }
  393. //非聚合查询
  394. if (DbUtil.isEmptyOrNull(aggregate)) {
  395. sbr.append("SELECT");
  396. //------1.查询主表字段
  397. boolean appendFields = false;
  398. for (Field field : fields) {
  399. String fieldName = field.getName();
  400. //常量字段过滤
  401. if (!DbUtil.isUsageField(field)) {
  402. continue;
  403. }
  404. boolean isExtenstionField = field.isAnnotationPresent(One.class) || field.isAnnotationPresent(Many.class);
  405. //忽略字段过滤
  406. if (!DbUtil.isEmptyOrNull(exclude) && exclude.contains(fieldName) && !isExtenstionField) {
  407. continue;
  408. }
  409. //指定查询
  410. if (!DbUtil.isEmptyOrNull(include) && !include.contains(fieldName) && !isExtenstionField) {
  411. continue;
  412. }
  413. DBF DBFAnnotation = field.getAnnotation(DBF.class);
  414. if (null != DBFAnnotation) {
  415. if (appendFields) {
  416. sbr.append(",");
  417. }
  418. sbr.append(" l.").append(this.comma).append(DbUtil.getColumnName(fieldName)).append(this.comma);
  419. appendFields = true;
  420. }
  421. }
  422. //-------2.查询关联字段
  423. for (Field field : fields) {
  424. Class<?> type = field.getType();
  425. String fieldName = field.getName();
  426. //常量字段过滤
  427. if (!DbUtil.isUsageField(field)) {
  428. continue;
  429. }
  430. boolean isExtenstionField = field.isAnnotationPresent(One.class) || field.isAnnotationPresent(Many.class);
  431. //忽略字段过滤
  432. if (!DbUtil.isEmptyOrNull(exclude) && exclude.contains(fieldName)) {
  433. continue;
  434. }
  435. //指定查询
  436. if (!DbUtil.isEmptyOrNull(include) && !include.contains(fieldName)) {
  437. continue;
  438. }
  439. Class<?> fieldClass = field.getType();
  440. One oneAnnotation = field.getAnnotation(One.class);
  441. if (null != oneAnnotation) {
  442. //忽略查询
  443. if (oneAnnotation.noQuery()) {
  444. continue;
  445. }
  446. String fkField = oneAnnotation.mkf();
  447. if (!DbUtil.isGenericType(field.getType()) && DEFAULT_MKF.equals(fkField)) {
  448. fkField = fieldName;
  449. }
  450. // join字段
  451. if (!DbUtil.isEmptyOrNull(fkField)) {
  452. //1.join查询对象
  453. if (!DbUtil.isGenericType(type) && type != String.class && !type.isArray() && !type.isAssignableFrom(List.class)) {
  454. // alias.putIfAbsent(fkField, "l" + aliasIdx++);
  455. Field[] joinFields = fieldClass.getFields();
  456. for (Field joinField : joinFields) {
  457. if (DbUtil.isUsageField(joinField)) {
  458. if (!"SELECT".contentEquals(sbr)) {
  459. sbr.append(",");
  460. }
  461. sbr.append(alias.get(fkField)).append(".").append(this.comma).append(DbUtil.getColumnName(joinField.getName())).append(this.comma).append(" AS ").append(this.comma).append(fieldName).append("_").append(DbUtil.getColumnName(joinField.getName())).append(this.comma);
  462. }
  463. }
  464. //2.join查询变量
  465. } else {
  466. if (!"SELECT".equals(sbr.toString().trim())) {
  467. sbr.append(",");
  468. }
  469. sbr.append(alias.get(fkField)).append(".").append(this.comma).append(DbUtil.getColumnName(oneAnnotation.tf())).append(this.comma).append(" AS ").append(this.comma).append(DbUtil.getColumnName(fieldName)).append(this.comma);
  470. }
  471. }
  472. }
  473. }
  474. } else {
  475. /* //聚合查询需要获取关联表别名
  476. for (Field field : fields) {
  477. if (field.isAnnotationPresent(Fk.class)) {
  478. alias.putIfAbsent(field.getName(), "l" + aliasIdx++);
  479. }
  480. }*/
  481. String aggregateField = aggregate;
  482. boolean distinct = distinctAggerate(aggregate);
  483. if (!distinct) {
  484. aggregateField = aggregate.contains("COUNT(") ? "l.*" : aggregate.trim().split("\\(")[1].split("\\)")[0].trim();
  485. }
  486. sbr.append("SELECT ").append(aggregateField);
  487. }
  488. sbr.append(" \n FROM ").append(this.comma).append(getTableName(entity)).append(this.comma).append(" AS l");
  489. List<String> joined = new ArrayList<>();
  490. //-------3.关联对象 JOINS
  491. //BUG:涉及到关联查询的对象字段需要inner join @QF
  492. Field[] qfs = query.getClass().getDeclaredFields();
  493. for (Field qf : qfields) {
  494. //qf 内联查询
  495. if (qf.isAnnotationPresent(QF.class)) {
  496. Object val = DbUtil.getFieldValue(query, qf);
  497. if (null != val) {
  498. String amkf = qf.getAnnotation(QF.class).pkf();
  499. if (!DbUtil.isEmptyOrNull(amkf)) {
  500. // if (!joinFields.contains(amkf)) {
  501. // joinFields.add(amkf);
  502. // alias.putIfAbsent(amkf, "i" + aliasIdx++);
  503. if (!joined.contains(alias.get(amkf))) {
  504. joined.add(alias.get(amkf));
  505. } else {
  506. continue;
  507. }
  508. Class<?> aliasClass = getFkDomainClass(qf, infoClass);
  509. sbr.append(" INNER JOIN ").append(getTableName(aliasClass, aliasClass.getSimpleName())).append(" AS ").append(alias.get(amkf)).append(" ON ").append("l.").append(this.comma).append(DbUtil.getColumnName(amkf)).append(this.comma).append("=").append(alias.get(amkf)).append(".").append(this.comma).append(DbUtil.getColumnName(qf.getAnnotation(QF.class).tkf())).append(this.comma);
  510. }
  511. }
  512. // }
  513. }
  514. }
  515. for (Field field : fields) {
  516. String fieldName = field.getName();
  517. //忽略查询
  518. if (!DbUtil.isEmptyOrNull(exclude) && exclude.contains(fieldName)) {
  519. continue;
  520. }
  521. One oneAnnotation = field.getAnnotation(One.class);
  522. if (null != oneAnnotation) {
  523. //忽略查询
  524. if (oneAnnotation.noQuery()) {
  525. continue;
  526. }
  527. if (!DbUtil.isUsageField(field)) {
  528. continue;
  529. }
  530. String mkf = oneAnnotation.mkf();
  531. if (!DbUtil.isGenericType(field.getType()) && DEFAULT_MKF.equals(mkf)) {
  532. mkf = fieldName;
  533. }
  534. //joins
  535. if (!DbUtil.isEmptyOrNull(mkf)) {
  536. if (!joined.contains(alias.get(mkf))) {
  537. joined.add(alias.get(mkf));
  538. } else {
  539. continue;
  540. }
  541. // joinFields.add(fkField);
  542. Class<?> aliasClass = getFkDomainClass(field, infoClass);
  543. // alias.putIfAbsent(fkField, "l" + aliasIdx++);
  544. if (oneAnnotation.join() == JoinType.LEFT) {
  545. sbr.append(" \n LEFT JOIN ");
  546. } else if (oneAnnotation.join() == JoinType.INNER) {
  547. sbr.append(" \n INNER JOIN ");
  548. } else {
  549. sbr.append(" \n RIGHT JOIN ");
  550. }
  551. sbr.append(getTableName(aliasClass, aliasClass.getSimpleName())).append(" AS ").append(alias.get(mkf)).append(" ON ").append("l.").append(this.comma).append(DbUtil.getColumnName(oneAnnotation.mkf())).append(this.comma).append("=").append(alias.get(mkf)).append(".").append(this.comma).append(DbUtil.getColumnName(oneAnnotation.tkf())).append(this.comma);
  552. }
  553. }
  554. }
  555. //-------4.查询条件
  556. sbr.append(" \n WHERE 1=1 ");
  557. Field[] queryFields = query.getClass().getDeclaredFields();
  558. //query条件和Builder条件仅支持一个(查询时只提供一个条件构造实体)
  559. if (null != OBuilder && !DbUtil.isEmptyOrNull(OBuilder.getWheres())) {
  560. List<OBuilder.Where> wheres = OBuilder.getWheres();
  561. wheres.forEach(wh -> {
  562. //联表查询字段
  563. String aliasName = alias.getOrDefault(getMkf(query, wh), "l");
  564. sbr.append(" AND ");
  565. appendBuilderWhere(sbr, wh, aliasName, parameters);
  566. });
  567. } else {
  568. for (Field qf : queryFields) {
  569. Object val = DbUtil.getFieldValue(query, qf);
  570. String fieldName = qf.getName();
  571. if (DbUtil.isEmptyOrNull(val)) {
  572. continue;
  573. }
  574. QF qfAnnotation = qf.getAnnotation(QF.class);
  575. if (null == qfAnnotation) {
  576. if (!fieldName.endsWith("Sort")) {
  577. sbr.append(" AND l.").append(this.comma).append(DbUtil.getColumnName(fieldName)).append(this.comma);
  578. if (String.class.equals(qf.getType())) {
  579. if (DIALECT_MYSQL == dialect) {
  580. sbr.append(" LIKE concat('%',?,'%')");
  581. } else if (DIALECT_PGSQL == dialect) {
  582. sbr.append(" like concat('%',?::text,'%')");
  583. } else if (DIALECT_MSSQL == dialect) {
  584. sbr.append(" LIKE '%'+?+'%')");
  585. } else if (DIALECT_ORACLE == dialect) {
  586. sbr.append(" LIKE '%'||?||'%')");
  587. }
  588. parameters.add(val);
  589. } else {
  590. sbr.append(" = ?");
  591. parameters.add(val);
  592. // .append(val);
  593. }
  594. }
  595. } else {
  596. //foreignKeys
  597. String mkf = qfAnnotation.pkf();
  598. if (qfAnnotation.qt() == QueryType.IGNORE) {
  599. continue;
  600. }
  601. if (!DbUtil.isEmptyOrNull(mkf)) {
  602. String tf = qfAnnotation.tf();
  603. if (DbUtil.isEmptyOrNull(tf)) {
  604. logger.error("SQL2Helper ERR# [" + fieldName + "] @QF with mkf define,but without tkf exist!!!");
  605. } else {
  606. if (qfAnnotation.qt() == QueryType.MAIN) {
  607. sbr.append(" AND ").append(" l").append(".").append(this.comma).append(DbUtil.getColumnName(mkf)).append(this.comma);
  608. } else {
  609. sbr.append(" AND ").append(alias.get(mkf)).append(".").append(this.comma).append(DbUtil.getColumnName(tf)).append(this.comma);
  610. }
  611. //字符串使用模糊查询
  612. if (String.class.equals(qf.getType())) {
  613. if (DIALECT_MYSQL == dialect) {
  614. sbr.append(" like concat('%',?,'%')");
  615. } else if (DIALECT_PGSQL == dialect) {
  616. sbr.append(" like concat('%',?::text,'%')");
  617. } else if (DIALECT_MSSQL == dialect) {
  618. sbr.append(" like '%'+'?'+'%')");
  619. } else if (DIALECT_ORACLE == dialect) {
  620. sbr.append(" like '%'||'?'||'%')");
  621. }
  622. parameters.add(val);
  623. } else {
  624. sbr.append(" =");
  625. // parameters.add(val);
  626. getValue(sbr, val, parameters);
  627. }
  628. }
  629. }
  630. //sql
  631. String whereSql = qfAnnotation.sql();
  632. String tbAlias = alias.getOrDefault(qfAnnotation.pkf(), "l");
  633. whereSql = appendWhereSql(whereSql, Collections.singletonList(val), tbAlias, parameters);
  634. if (!DbUtil.isEmptyOrNull(whereSql)) {
  635. sbr.append(" AND ").append(whereSql);
  636. }
  637. //op
  638. String operator = qfAnnotation.op();
  639. if (!DbUtil.isEmptyOrNull(operator)) {
  640. String tf = qfAnnotation.tf();
  641. if (DbUtil.isEmptyOrNull(tf)) {
  642. logger.error("SQL2Helper ERR# " + fieldName + "@QF with operator defined,but without tkf exist!!!");
  643. } else {
  644. //JIN JINT
  645. if (operator.equals(OP.JIN)) {
  646. int loop = 1;
  647. String each = qfAnnotation.each();
  648. if (val instanceof Collection<?>) {
  649. List<Object> list = JacksonUtil.toList(val);
  650. loop = list.size();
  651. parameters.addAll((Collection<?>) val);
  652. } else if (val.getClass().isArray()) {
  653. loop = Array.getLength(val);
  654. for (int i = 0; i < loop; i++) {
  655. parameters.add(Array.get(val, i));
  656. }
  657. } else {
  658. loop = 1;
  659. parameters.add(val);
  660. }
  661. // and ( json_contains(l.id_list,'1'))
  662. if (dialect == IHandler.DIALECT_MYSQL) {
  663. sbr.append(" AND ").append("( ");
  664. for (int i = 0; i < loop; i++) {
  665. sbr.append(" (JSON_CONTAINS(").append(tbAlias).append(".").append(this.comma).append(DbUtil.getColumnName(tf)).append(this.comma).append(",").append("JSON_ARRAY(?)) ");
  666. if (i != loop - 1) {
  667. sbr.append(each);
  668. }
  669. }
  670. sbr.append(") ");
  671. // owner_id_list::jsonb @> jsonb_build_array(2)
  672. // owner_id_list::jsonb @> '[2]'
  673. } else if (dialect == IHandler.DIALECT_PGSQL) {
  674. sbr.append(" AND ").append("( ");
  675. for (int i = 0; i < loop; i++) {
  676. sbr.append(" (").append(tbAlias).append(".").append(this.comma).append(DbUtil.getColumnName(tf)).append(this.comma).append("::jsonb @> ").append("jsonb_build_array(?)) ");
  677. if (i != loop - 1) {
  678. sbr.append(each);
  679. }
  680. }
  681. sbr.append(") ");
  682. }
  683. } else if (operator.equals(OP.JINT)) {
  684. //and (JSON_OVERLAPS(role,'["a","d"]')=1)
  685. if (dialect == IHandler.DIALECT_MYSQL) {
  686. sbr.append(" AND ").append("(").append(operator).append("JSON_OVERLAPS(").append(tbAlias).append(".").append(this.comma).append(DbUtil.getColumnName(tf)).append(this.comma).append(",").append("'?')=1").append(")");
  687. parameters.add(JacksonUtil.toJSONString(val));
  688. } else if (dialect == IHandler.DIALECT_PGSQL) {
  689. // //ARRAY[1,4,3] && ARRAY[2,1]
  690. sbr.append(" AND ").append("(").append(tbAlias).append(".").append(this.comma).append(DbUtil.getColumnName(tf)).append(this.comma).append("::jsonb && ").append("jsonb_build_array(?))");
  691. parameters.add(JacksonUtil.toJSONString(val));
  692. }
  693. } else if (operator.equals(OP.IN) || operator.equals(OP.NIN)) {
  694. sbr.append(" AND ").append("(").append(tbAlias).append(".").append(this.comma).append(DbUtil.getColumnName(tf)).append(this.comma).append(" ").append(operator).append(" (");
  695. if (val instanceof Collection<?>) {
  696. List<Object> list = JacksonUtil.toList(val);
  697. String collect = list.stream().map(k -> "?").collect(Collectors.joining(","));
  698. sbr.append(collect).append(") ) ");
  699. parameters.addAll((Collection<?>) val);
  700. } else if (val.getClass().isArray()) {
  701. int length = Array.getLength(val);
  702. for (int i = 0; i < length; i++) {
  703. sbr.append("?");
  704. if (i != length - 1) {
  705. sbr.append(",");
  706. }
  707. parameters.add(Array.get(val, i));
  708. }
  709. sbr.append("))");
  710. } else {
  711. throw new IllegalArgumentException("不支持的参数格式");
  712. }
  713. } else if (operator.equals(OP.RLK) || operator.equals(OP.LK) || operator.equals(OP.LLK)) {
  714. sbr.append(" AND ").append(tbAlias).append(".").append(this.comma).append(DbUtil.getColumnName(tf)).append(this.comma).append(" ").append(operator);
  715. parameters.add(val);
  716. } else {
  717. sbr.append(" AND ").append(tbAlias).append(".").append(this.comma).append(DbUtil.getColumnName(tf)).append(this.comma).append(" ").append(operator);
  718. if (qf.getType().isArray()) {
  719. sbr.append(" (");
  720. int length = Array.getLength(val);
  721. for (int i = 0; i < length; i++) {
  722. Object value = Array.get(val, i);
  723. getValue(sbr, value, parameters);
  724. if (i != length - 1) {
  725. sbr.append(",");
  726. }
  727. }
  728. sbr.append(")");
  729. } else if (val instanceof Collection<?>) {
  730. List<Object> list = JacksonUtil.toList(val);
  731. String collect = list.stream().map(k -> "?").collect(Collectors.joining(","));
  732. sbr.append(collect).append(") ");
  733. parameters.addAll((Collection<?>) val);
  734. } else {
  735. getValue(sbr, val, parameters);
  736. }
  737. }
  738. }
  739. }
  740. }
  741. }
  742. }
  743. //-------6.GROUP BY
  744. if (null != OBuilder && !DbUtil.isEmptyOrNull(OBuilder.getGroupBys())) {
  745. sbr.append(" GROUP BY ");
  746. OBuilder.getGroupBys().forEach(group -> {
  747. sbr.append(group).append(" ,");
  748. });
  749. sbr.deleteCharAt(sbr.length() - 1);
  750. }
  751. //-------5.having
  752. //todo
  753. //-------7.ORDER BY(query)
  754. boolean ordered = false;
  755. //聚合查询不需要排序
  756. if (DbUtil.isEmptyOrNull(aggregate) && isList) {
  757. //多个排序字段需要确定先后顺序
  758. Map<Integer, String> sortMap = new HashMap<>(16);
  759. if (DbUtil.isEmptyOrNull(query.sortFields)) {
  760. for (Field qf : queryFields) {
  761. Object val = DbUtil.getFieldValue(query, qf);
  762. if (null == val) {
  763. continue;
  764. }
  765. int sortValue = 0;
  766. if (val instanceof Integer) {
  767. sortValue = (int) val;
  768. }
  769. if (sortValue == 0) {
  770. continue;
  771. }
  772. String fieldName = qf.getName();
  773. List<Field> sorted = new ArrayList<>();
  774. QF qfAnnotation = qf.getAnnotation(QF.class);
  775. if (null == qfAnnotation) {
  776. if (fieldName.endsWith("Sort")) {
  777. if (!ordered) {
  778. ordered = true;
  779. sbr.append(" ORDER BY ");
  780. }
  781. // sbr.append(" l.")
  782. // .append(this.comma)
  783. // .append(DbUtil.getColumnName(fieldName.substring(0, fieldName.length() - 4)))
  784. // .append(this.comma);
  785. if (sortValue % 2 == BasicQuery.SORT_ASC) {
  786. sortMap.put(sortValue, String.format("%s ASC ", DbUtil.getColumnName(fieldName.substring(0, fieldName.length() - 4))));
  787. // sbr.append(" ASC ,");
  788. } else {
  789. // sbr.append(" DESC ,");
  790. sortMap.put(sortValue, String.format("%s DESC ", DbUtil.getColumnName(fieldName.substring(0, fieldName.length() - 4))));
  791. }
  792. }
  793. }
  794. if (!DbUtil.isEmptyOrNull(sortMap)) {
  795. Set<Integer> integers = sortMap.keySet();
  796. List<Integer> sorts = new ArrayList<>(integers);
  797. sorts.sort(Comparator.comparingInt(o -> o));
  798. sbr.append(sorts.stream().map(k -> "l." + sortMap.get(k)).collect(Collectors.joining(",")));
  799. }
  800. /* if (null == qfAnnotation) {
  801. if (fieldName.endsWith("Sort")) {
  802. if (!ordered) {
  803. ordered = true;
  804. sbr.append(" ORDER BY ");
  805. }
  806. sbr.append(" l.")
  807. .append(this.comma)
  808. .append(DbUtil.getColumnName(fieldName.substring(0, fieldName.length() - 4)))
  809. .append(this.comma);
  810. if (sortType == BasicQuery.SORT_ASC) {
  811. sbr.append(" ASC ,");
  812. } else if (sortType == BasicQuery.SORT_DESC) {
  813. sbr.append(" desc ,");
  814. }
  815. }
  816. }*/
  817. }
  818. } else {
  819. String[] sorts = query.sortFields;
  820. for (String sortField : sorts) {
  821. try {
  822. Field sf = query.getClass().getDeclaredField(sortField + "Sort");
  823. Object val = DbUtil.getFieldValue(query, sf);
  824. if (null == val) {
  825. continue;
  826. }
  827. int sortType = (int) val;
  828. if (!ordered) {
  829. ordered = true;
  830. sbr.append(" ORDER BY ");
  831. }
  832. sbr.append(" l.").append(this.comma).append(DbUtil.getColumnName(sortField)).append(this.comma);
  833. if (sortType == BasicQuery.SORT_ASC) {
  834. sbr.append(" ASC ,");
  835. } else if (sortType == BasicQuery.SORT_DESC) {
  836. sbr.append(" DESC ,");
  837. }
  838. } catch (NoSuchFieldException e) {
  839. logger.warn("SQL2Helper warn# sort field is not exist,{}", sortField);
  840. }
  841. }
  842. }
  843. }
  844. //-------7.ORDER BY(builder) 聚合查询不需要排序
  845. if (DbUtil.isEmptyOrNull(aggregate) && isList) {
  846. if (null != OBuilder && !DbUtil.isEmptyOrNull(OBuilder.getOrderBys())) {
  847. if (!ordered) {
  848. ordered = true;
  849. sbr.append(" ORDER BY ");
  850. }
  851. OBuilder.getOrderBys().forEach(order -> sbr.append(" l.").append(order).append(" ,"));
  852. }
  853. if (ordered && sbr.toString().endsWith(",")) {
  854. sbr.deleteCharAt(sbr.length() - 1);
  855. }
  856. //------7.ORDER BY default update_at DESC
  857. if (!ordered) {
  858. sbr.append(" ORDER BY l.update_at DESC");
  859. }
  860. }
  861. //-------8.limit(query)
  862. if (isList) {
  863. if (query.pageSize > 0) {
  864. if (DIALECT_MYSQL == dialect) {
  865. sbr.append(" LIMIT ");
  866. if (query.pageIndex > 0) {
  867. sbr.append((query.pageIndex - 1) * query.pageSize).append(",");
  868. }
  869. if (query.pageSize > 0) {
  870. sbr.append(query.pageSize);
  871. }
  872. } else if (DIALECT_MSSQL == dialect) {
  873. sbr.append(" OFFSET ");
  874. if (query.pageIndex > 0) {
  875. sbr.append((query.pageIndex - 1) * query.pageSize).append(" ROWS ");
  876. }
  877. if (query.pageSize > 0) {
  878. sbr.append(" FETCH NEXT ").append(query.pageSize).append(" ROWS ONLY ");
  879. }
  880. } else if (DIALECT_PGSQL == dialect) {
  881. sbr.append(" LIMIT ").append(query.pageSize);
  882. if (query.pageIndex > 0) {
  883. sbr.append(" OFFSET ").append((query.pageIndex - 1) * query.pageSize);
  884. }
  885. } else if (DIALECT_ORACLE == dialect) {
  886. StringBuilder sr = new StringBuilder();
  887. sr.append("SELECT * FROM (SELECT o.*,ROWNUM as rowno FROM (").append(sbr).append(") o ");
  888. if (query.pageSize > 0) {
  889. sr.append(" WHERE ROWNUM<=").append(query.pageSize).append(") o1");
  890. }
  891. if (query.pageIndex > 0) {
  892. sr.append(" WHERE o1.rowno>").append((query.pageIndex - 1) * query.pageSize).append(";");
  893. }
  894. //聚合类使用子查询
  895. if (!DbUtil.isEmptyOrNull(aggregate)) {
  896. sqlEntity.sql = "SELECT " + aggregate + " FROM (" + sr.toString() + ") temp";
  897. }
  898. sqlEntity.sql = sr.toString();
  899. }
  900. //-------8.limit(builder)
  901. } else {
  902. String result = appendWhereLimit(OBuilder, sbr);
  903. if (null != result) {
  904. //聚合类使用子查询
  905. if (!DbUtil.isEmptyOrNull(aggregate) && !distinctAggerate(aggregate)) {
  906. sbr.setLength(0);
  907. sbr.append("SELECT ").append(aggregate).append(" FROM (").append(result).append(") temp");
  908. }
  909. }
  910. }
  911. }
  912. sqlEntity.sql = sbr.toString();
  913. //聚合类使用子查询
  914. if (!DbUtil.isEmptyOrNull(aggregate) && !distinctAggerate(aggregate)) {
  915. String tmpSql = sbr.toString();
  916. sbr.setLength(0);
  917. sbr.append("SELECT ").append(aggregate).append(" FROM (").append(tmpSql).append(") temp");
  918. }
  919. sbr.append(";");
  920. sqlEntity.sql = sbr.toString();
  921. sqlEntity.parameters = parameters.toArray();
  922. return sqlEntity;
  923. }
  924. private boolean distinctAggerate(String aggregate) {
  925. return aggregate.toLowerCase().contains("distinct(");
  926. }
  927. private String getMkf(BasicQuery query, OBuilder.Where where) {
  928. String fieldName = where.key;
  929. if (DbUtil.isEmptyOrNull(fieldName)) {
  930. if (!DbUtil.isEmptyOrNull(where.sql)) {
  931. logger.warn("SQL#builder with sql foreign key is not support yet! {}", where.sql);
  932. }
  933. return null;
  934. }
  935. try {
  936. //info
  937. Class<?> infoClz = query.getClass().getAnnotation(QE.class).clz();
  938. Field field = infoClz.getField(DbUtil.getCamelName(fieldName));
  939. if (field.isAnnotationPresent(One.class)) {
  940. One one = field.getAnnotation(One.class);
  941. where.key = one.tf();
  942. return one.mkf();
  943. }
  944. } catch (NoSuchFieldException e) {
  945. try {
  946. //query
  947. Field field = query.getClass().getField(DbUtil.getCamelName(fieldName));
  948. if (field.isAnnotationPresent(QF.class)) {
  949. QF qf = field.getAnnotation(QF.class);
  950. where.key = qf.tf();
  951. return qf.pkf();
  952. }
  953. } catch (NoSuchFieldException e1) {
  954. return null;
  955. }
  956. }
  957. return null;
  958. }
  959. public SQLEntity selectOne(String sql, Object... params) {
  960. return selectList(sql, params);
  961. }
  962. public SQLEntity selectOneByQuery(BasicQuery query, List<String> include, List<String> exclude) {
  963. return select(query, null, include, exclude, null, false);
  964. }
  965. public SQLEntity selectById(Class<?> clz, Object id, List<String> include, List<String> exclude) {
  966. return selectListByWhere(clz, OBuilder.build().eq("id", id), include, exclude, false);
  967. }
  968. public SQLEntity selectOneByWhere(Class<?> clz, OBuilder OBuilder, List<String> include, List<String> exclude) {
  969. return selectListByWhere(clz, OBuilder, include, exclude, false);
  970. }
  971. public SQLEntity selectCount(Class<?> clz, OBuilder OBuilder) {
  972. return selectFunc(clz, "COUNT(1)", OBuilder);
  973. }
  974. public SQLEntity selectCountByQuery(BasicQuery query) {
  975. return select(query, null, null, null, "COUNT(1)", false);
  976. }
  977. //endregion select
  978. //region insert
  979. /**
  980. * 存储过程
  981. *
  982. * @param sql
  983. * @param params
  984. * @return
  985. */
  986. public SQLEntity call(String sql, Object... params) {
  987. SQLEntity sqlEntity = new SQLEntity();
  988. sqlEntity.sql = sql;
  989. sqlEntity.parameters = params;
  990. // if (!DbUtil.isEmptyOrNull(params)) {
  991. // sql = String.format(sql, params);
  992. // }
  993. // "CALL updateRole(#{id,jdbcType=VARCHAR,mode=IN},#{result,jdbcType=INTEGER,mode=OUT});";
  994. return sqlEntity;
  995. }
  996. /**
  997. * --------------insert method begin ---------------
  998. */
  999. public SQLEntity insertSelective(Object o, List<String> exclude) {
  1000. return insertWithCheck(o, true, false, exclude);
  1001. }
  1002. public SQLEntity insert(Object o, List<String> exclude) {
  1003. return insertWithCheck(o, false, false, exclude);
  1004. }
  1005. public SQLEntity insertWithGenKey(Object o, List<String> exclude) {
  1006. return insertWithCheck(o, false, true, exclude);
  1007. }
  1008. /***
  1009. * 插入方法
  1010. * @param o 插入实体对象
  1011. * @param ignoreNull 忽略空字段
  1012. * @param withAutoIncreamentKey 对象是否已赋值自增主键
  1013. * @param exclude 忽略字段
  1014. * @return
  1015. */
  1016. private SQLEntity insertWithCheck(Object o, boolean ignoreNull, boolean withAutoIncreamentKey, List<String> exclude) {
  1017. SQLEntity sqlEntity = new SQLEntity();
  1018. List<Object> parameters = new ArrayList<>();
  1019. StringBuilder sbr = new StringBuilder("INSERT INTO ");
  1020. sbr.append(getTableName(o.getClass(), o.getClass().getSimpleName()));
  1021. sbr.append("(");
  1022. StringBuilder vb = new StringBuilder("(");
  1023. //自带主键
  1024. if (withAutoIncreamentKey) {
  1025. sbr.append("id,");
  1026. try {
  1027. parameters.add(DbUtil.getFieldValue(o, DbUtil.getField(o, "id")));
  1028. vb.append("?").append(",");
  1029. } catch (NoSuchFieldException e) {
  1030. logger.error("SQL2Helper ERR# insert with primary key,but without a value!!");
  1031. throw new IllegalStateException("no primary key value exist.");
  1032. }
  1033. }
  1034. exclude.add("id");
  1035. Set<Field> fields = DbUtil.getEntityFields(o.getClass());
  1036. /* List<Field> fields1 = new ArrayList<>();
  1037. Entity declaredAnnotation = o.getClass().getDeclaredAnnotation(Entity.class);
  1038. if (null != declaredAnnotation) {
  1039. fields = o.getClass().getFields();
  1040. fields1.addAll(Arrays.asList( o.getClass().getFields()));
  1041. } else {
  1042. declaredAnnotation = o.getClass().getSuperclass().getDeclaredAnnotation(Entity.class);
  1043. if(null==declaredAnnotation){
  1044. throw new IllegalArgumentException("no @Entity define in class");
  1045. }
  1046. fields = o.getClass().getSuperclass().getDeclaredFields();
  1047. fields1.addAll(Arrays.asList( o.getClass().getFields()));
  1048. Field[] declaredFields = o.getClass().getSuperclass().getDeclaredFields();
  1049. fields1.addAll(Arrays.asList( o.getClass().getSuperclass().getDeclaredFields()));
  1050. }*/
  1051. for (Field field : fields) {
  1052. if (DbUtil.isUsageField(field)) {
  1053. String fieldName = field.getName();
  1054. if (!DbUtil.isEmptyOrNull(exclude)) {
  1055. //忽略属性
  1056. if (exclude.contains(fieldName)) {
  1057. continue;
  1058. }
  1059. }
  1060. Type type = field.getGenericType();
  1061. DBF DBF = field.getAnnotation(DBF.class);
  1062. if (null != DBF) {
  1063. Object value = DbUtil.getFieldValue(o, field);
  1064. if (ignoreNull) {
  1065. if (null != value) {
  1066. //TODO 校验值
  1067. checkFieldValue(field, value, true);
  1068. sbr.append(this.comma).append(DbUtil.getColumnName(fieldName)).append(this.comma).append(",");
  1069. appendValue(vb, type, value, parameters);
  1070. }
  1071. } else {
  1072. if (type == Date.class && null == value) {
  1073. continue;
  1074. }
  1075. sbr.append(this.comma).append(DbUtil.getColumnName(fieldName)).append(this.comma).append(",");
  1076. appendValue(vb, type, value, parameters);
  1077. }
  1078. }
  1079. }
  1080. }
  1081. sbr.deleteCharAt(sbr.length() - 1);
  1082. vb.deleteCharAt(vb.length() - 1);
  1083. sbr.append(")").append(" VALUES ");
  1084. sbr.append(vb.toString()).append(")");
  1085. sbr.append(";");
  1086. sqlEntity.sql = sbr.toString();
  1087. sqlEntity.parameters = parameters.toArray();
  1088. return sqlEntity;
  1089. }
  1090. //endregion insert
  1091. //region update
  1092. private SQLEntity updateNative(Object o, OBuilder OBuilder, boolean ignoreNull, List<String> includes, List<String> excludes) {
  1093. SQLEntity sqlEntity = new SQLEntity();
  1094. List<Object> parameters = new ArrayList<>();
  1095. StringBuilder sbr = new StringBuilder("UPDATE ");
  1096. sbr.append(getTableName(o.getClass(), o.getClass().getSimpleName())).append(" SET ");
  1097. Set<Field> fields = DbUtil.getEntityFields(o.getClass());
  1098. boolean hasUpdateField = false;
  1099. for (Field field : fields) {
  1100. if (DbUtil.isUsageField(field)) {
  1101. DBF dbf = field.getAnnotation(DBF.class);
  1102. if (null == dbf) {
  1103. continue;
  1104. }
  1105. String fieldName = field.getName();
  1106. if (!DbUtil.isEmptyOrNull(excludes)) {
  1107. if (excludes.contains(DbUtil.getColumnName(fieldName)) || excludes.contains(fieldName)) {
  1108. continue;
  1109. }
  1110. }
  1111. if (!DbUtil.isEmptyOrNull(includes)) {
  1112. if (!includes.contains(fieldName) && !includes.contains(DbUtil.getColumnName(fieldName))) {
  1113. continue;
  1114. }
  1115. }
  1116. if (dbf.canUpdate() && !isImmutableField(fieldName)) {
  1117. Type type = field.getGenericType();
  1118. Object value = DbUtil.getFieldValue(o, field);
  1119. if (ignoreNull) {
  1120. if (!DbUtil.isEmptyOrNull(value) && !"0".equals(value.toString())) {
  1121. checkFieldValue(field, value, false);
  1122. sbr.append(this.comma).append(DbUtil.getColumnName(fieldName)).append(this.comma).append("=");
  1123. appendValue(sbr, type, value, parameters);
  1124. hasUpdateField = true;
  1125. }
  1126. } else {
  1127. if (null == value && type == Date.class) {
  1128. continue;
  1129. }
  1130. sbr.append(this.comma).append(DbUtil.getColumnName(fieldName)).append(this.comma).append("=");
  1131. appendValue(sbr, type, value, parameters);
  1132. hasUpdateField = true;
  1133. }
  1134. }
  1135. }
  1136. }
  1137. if (!hasUpdateField) {
  1138. throw new RuntimeException("no field to UPDATE ,#SQL:{}" + sbr.toString());
  1139. }
  1140. if (sbr.toString().endsWith(",")) {
  1141. sbr.deleteCharAt(sbr.length() - 1);
  1142. }
  1143. //where
  1144. if (null != OBuilder && !DbUtil.isEmptyOrNull(OBuilder.getWheres())) {
  1145. List<OBuilder.Where> wheres = OBuilder.getWheres();
  1146. if (!DbUtil.isEmptyOrNull(wheres)) {
  1147. sbr.append(" WHERE ");
  1148. wheres.forEach(wh -> {
  1149. if (ignoreNull) {
  1150. if (null != wh.value) {
  1151. sbr.append(this.comma).append(DbUtil.getColumnName(wh.key)).append(this.comma).append(" ").append(wh.op).append(" ");
  1152. appendValue(sbr, wh.value, parameters);
  1153. sbr.append(" AND ");
  1154. }
  1155. } else {
  1156. sbr.append(this.comma).append(DbUtil.getColumnName(wh.key)).append(this.comma).append(" ").append(wh.op).append(" ");
  1157. appendValue(sbr, wh.value, parameters);
  1158. sbr.append(" AND ");
  1159. }
  1160. });
  1161. sbr.delete(sbr.length() - 5, sbr.length());
  1162. }
  1163. if (DIALECT_MYSQL == dialect) {
  1164. //mysql innodb 锁
  1165. if (OBuilder.forUpdate) {
  1166. sbr.append(" FOR UPDATE ");
  1167. }
  1168. if (OBuilder.forShare) {
  1169. sbr.append(" LOCK IN SHARE MODE");
  1170. }
  1171. }
  1172. } else {
  1173. try {
  1174. Field f = DbUtil.getField(o, "id");
  1175. sbr.append(" WHERE id=?");
  1176. parameters.add(DbUtil.getFieldValue(o, f));
  1177. } catch (Exception e) {
  1178. throw new IllegalArgumentException("tkf id is not exist!!!");
  1179. }
  1180. }
  1181. sbr.append(";");
  1182. sqlEntity.sql = sbr.toString();
  1183. sqlEntity.parameters = parameters.toArray();
  1184. return sqlEntity;
  1185. }
  1186. public SQLEntity updateByWhere(Object o, OBuilder builder) {
  1187. return updateNative(o, builder, false, null, null);
  1188. }
  1189. public SQLEntity updateByWhere(Object o, OBuilder builder, List<String> includes) {
  1190. return updateNative(o, builder, false, includes, null);
  1191. }
  1192. public SQLEntity updateByWhere(Object o, OBuilder builder, List<String> includes, List<String> excludes) {
  1193. return updateNative(o, builder, false, includes, excludes);
  1194. }
  1195. public SQLEntity update(String sql, Object... params) {
  1196. SQLEntity sqlEntity = new SQLEntity();
  1197. sqlEntity.sql = sql;
  1198. sqlEntity.parameters = params;
  1199. return sqlEntity;
  1200. }
  1201. public SQLEntity updateSelective(Object o, List<String> includes, List<String> excludes) {
  1202. return updateNative(o, null, true, includes, excludes);
  1203. }
  1204. public SQLEntity updateSelectiveByWhere(Object o, OBuilder OBuilder) {
  1205. return updateNative(o, OBuilder, true, null, null);
  1206. }
  1207. //endregion update
  1208. //region delete
  1209. public SQLEntity deleteById(Class<?> clz, Object id) {
  1210. SQLEntity sqlEntity = new SQLEntity();
  1211. List<Object> parameters = new ArrayList<>();
  1212. if (null != id) {
  1213. parameters.add(id);
  1214. } else {
  1215. throw new IllegalArgumentException("null value with id!!!");
  1216. }
  1217. sqlEntity.sql = "DELETE FROM " + getTableName(clz, clz.getSimpleName()) + " WHERE id= ?";
  1218. sqlEntity.parameters = parameters.toArray();
  1219. return sqlEntity;
  1220. }
  1221. public SQLEntity delete(Class<?> clz, OBuilder OBuilder) {
  1222. SQLEntity sqlEntity = new SQLEntity();
  1223. List<Object> parameters = new ArrayList<>();
  1224. StringBuilder sbr = new StringBuilder("DELETE FROM ");
  1225. sbr.append(getTableName(clz, clz.getSimpleName()));
  1226. if (null != OBuilder) {
  1227. List<OBuilder.Where> wheres = OBuilder.getWheres();
  1228. if (!DbUtil.isEmptyOrNull(wheres)) {
  1229. sbr.append(" WHERE 1=1");
  1230. wheres.forEach(wh -> {
  1231. sbr.append(" AND ");
  1232. appendBuilderWhere(sbr, wh, "", parameters);
  1233. });
  1234. }
  1235. } else {
  1236. throw new IllegalArgumentException(" delete with no where condition exist!!!");
  1237. }
  1238. sqlEntity.sql = sbr.toString();
  1239. sqlEntity.parameters = parameters.toArray();
  1240. return sqlEntity;
  1241. }
  1242. //endregion delete
  1243. private String appendWhereLimit(OBuilder OBuilder, StringBuilder sbr) {
  1244. if (null != OBuilder) {
  1245. if (OBuilder.limitDelta() > 0) {
  1246. if (DIALECT_MYSQL == dialect) {
  1247. sbr.append(" LIMIT ");
  1248. if (OBuilder.limitStart() > 0) {
  1249. sbr.append(OBuilder.limitStart()).append(",");
  1250. }
  1251. if (OBuilder.limitDelta() > 0) {
  1252. sbr.append(OBuilder.limitDelta());
  1253. }
  1254. return null;
  1255. } else if (DIALECT_MSSQL == dialect) {
  1256. sbr.append(" OFFSET ");
  1257. if (OBuilder.limitStart() >= 0) {
  1258. sbr.append(OBuilder.limitStart()).append(" ROWS ");
  1259. }
  1260. if (OBuilder.limitDelta() > 0) {
  1261. sbr.append(" FETCH NEXT ").append(OBuilder.limitDelta()).append(" ROWS ONLY ");
  1262. }
  1263. return null;
  1264. } else if (DIALECT_PGSQL == dialect) {
  1265. sbr.append(" LIMIT ");
  1266. if (OBuilder.limitDelta() > 0) {
  1267. sbr.append(OBuilder.limitDelta());
  1268. }
  1269. if (OBuilder.limitStart() >= 0) {
  1270. sbr.append(" OFFSET ").append(OBuilder.limitStart());
  1271. }
  1272. return null;
  1273. } else if (DIALECT_ORACLE == dialect) {
  1274. StringBuilder sr = new StringBuilder();
  1275. sr.append("SELECT * FROM (SELECT o.*,ROWNUM as rowno FROM (").append(sbr.toString()).append(") o ");
  1276. if (OBuilder.limitDelta() > 0) {
  1277. sr.append(" WHERE ROWNUM<=").append(OBuilder.limitDelta()).append(") o1");
  1278. }
  1279. if (OBuilder.limitStart() >= 0) {
  1280. sr.append(" WHERE o1.rowno>").append(OBuilder.limitStart());
  1281. }
  1282. return sr.toString();
  1283. }
  1284. }
  1285. }
  1286. return null;
  1287. }
  1288. /**
  1289. * 拼接Builder 条件
  1290. */
  1291. private void appendBuilderWhere(StringBuilder sbr, OBuilder.Where wh, String tbAlias, List<Object> parameters) {
  1292. if (DbUtil.isEmptyOrNull(wh.sql)) {
  1293. if (!DbUtil.isEmptyOrNull(tbAlias)) {
  1294. sbr.append(tbAlias).append(".");
  1295. }
  1296. sbr.append(this.comma).append(DbUtil.getColumnName(wh.key)).append(this.comma).append(" ").append(wh.op);
  1297. String op = wh.op;
  1298. Object value = wh.value;
  1299. if ("in".equals(op.trim()) || "not in".equals(op.trim())) {
  1300. sbr.append("(");
  1301. for (int i = 0; i < Array.getLength(value); i++) {
  1302. Object v = Array.get(value, i);
  1303. sbr.append("?");
  1304. parameters.add(v);
  1305. /* if (v.getClass() == String.class) {
  1306. sbr.append("'").append(DbUtil.injectDefend(v.toString())).append("'");
  1307. } else {
  1308. sbr.append(v);
  1309. }*/
  1310. if (i != Array.getLength(value) - 1) {
  1311. sbr.append(",");
  1312. }
  1313. }
  1314. sbr.append(")");
  1315. } else {
  1316. //转换取值方式
  1317. sbr.append("?");
  1318. parameters.add(value);
  1319. /* if (value.getClass() == String.class) {
  1320. sbr.append("'").append(value).append("'");
  1321. } else if (wh.value.getClass() == Date.class) {
  1322. sbr.append("'").append(new Timestamp(((Date) value).getTime())).append("'");
  1323. } else {
  1324. // sbr.append(value);
  1325. }*/
  1326. }
  1327. } else {
  1328. //别名判断
  1329. sbr.append(" (");
  1330. sbr.append(appendWhereSql(wh.sql, wh.sqlValues, tbAlias, parameters));
  1331. sbr.append(") ");
  1332. }
  1333. }
  1334. /**
  1335. * whereSql拼接,需要注意json类型处理
  1336. *
  1337. * @param whereSql
  1338. * @param val
  1339. * @param tbAlias
  1340. * @param parameters
  1341. * @return
  1342. */
  1343. private String appendWhereSql(String whereSql, List<Object> val, String tbAlias, List<Object> parameters) {
  1344. // parameters.addAll(val);
  1345. if (!DbUtil.isEmptyOrNull(whereSql)) {
  1346. if (!DbUtil.isEmptyOrNull(val)) {
  1347. if (whereSql.contains("?")) {
  1348. String regex = "\\?";
  1349. for (Object o : val) {
  1350. if (o.getClass() == String.class) {
  1351. whereSql = whereSql.replaceFirst(regex, "'" + DbUtil.injectDefend(o.toString()) + "'");
  1352. } else if (o.getClass() == Date.class) {
  1353. whereSql = whereSql.replaceFirst(regex, "'" + (new Timestamp(((Date) o).getTime())) + "'");
  1354. } else {
  1355. whereSql = whereSql.replaceFirst(regex, DbUtil.injectDefend(o.toString()));
  1356. }
  1357. }
  1358. while (whereSql.contains("?")) {
  1359. for (Object o : val) {
  1360. if (o.getClass() == String.class) {
  1361. whereSql = whereSql.replaceFirst(regex, "'" + DbUtil.injectDefend(o.toString()) + "'");
  1362. } else if (o.getClass() == Date.class) {
  1363. whereSql = whereSql.replaceFirst(regex, "'" + (new Timestamp(((Date) o).getTime())) + "'");
  1364. } else {
  1365. whereSql = whereSql.replaceFirst(regex, DbUtil.injectDefend(o.toString()));
  1366. }
  1367. }
  1368. }
  1369. } else {
  1370. if (val.size() == 1 && val.get(0) instanceof Boolean) {
  1371. if (!(Boolean) val.get(0)) {
  1372. return null;
  1373. }
  1374. }
  1375. }
  1376. }
  1377. String[] splits = whereSql.split(" AND ");
  1378. boolean containsComma = false;
  1379. if (splits.length > 1) {
  1380. StringBuilder sbd = new StringBuilder();
  1381. for (int i = 0; i < splits.length; i++) {
  1382. if (splits[i].startsWith("(")) {
  1383. containsComma = true;
  1384. sbd.append("(");
  1385. if (!DbUtil.isEmptyOrNull(tbAlias)) {
  1386. sbd.append(tbAlias).append(".");
  1387. }
  1388. sbd.append(splits[i].trim().substring(1));
  1389. } else {
  1390. if (!DbUtil.isEmptyOrNull(tbAlias)) {
  1391. sbd.append(tbAlias).append(".");
  1392. }
  1393. sbd.append(splits[i].trim());
  1394. }
  1395. if (i != splits.length - 1) {
  1396. sbd.append(" AND ");
  1397. }
  1398. }
  1399. whereSql = sbd.toString();
  1400. }
  1401. String[] splits1 = whereSql.split(" OR ");
  1402. if (splits1.length > 1) {
  1403. StringBuilder sbd = new StringBuilder();
  1404. for (int i = 0; i < splits1.length; i++) {
  1405. if (splits1[i].startsWith("(")) {
  1406. if (!containsComma) {
  1407. sbd.append("(");
  1408. if (!DbUtil.isEmptyOrNull(tbAlias)) {
  1409. sbd.append(tbAlias).append(".");
  1410. }
  1411. sbd.append(splits1[i].trim().substring(1));
  1412. }
  1413. } else {
  1414. if (!DbUtil.isEmptyOrNull(tbAlias)) {
  1415. sbd.append(tbAlias).append(".");
  1416. }
  1417. sbd.append(splits1[i].trim());
  1418. }
  1419. if (i != splits1.length - 1) {
  1420. sbd.append(" OR ");
  1421. }
  1422. }
  1423. whereSql = sbd.toString();
  1424. }
  1425. }
  1426. return whereSql;
  1427. }
  1428. private void appendValue(StringBuilder sbr, Type type, Object value, List<Object> parameters) {
  1429. sbr.append("?").append(",");
  1430. parameters.add(value);
  1431. // if (DbUtil.isGenericType(type)) {
  1432. // parameters.add(value);
  1433. // } else {
  1434. // parameters.add(JacksonUtil.toJSONString(value));
  1435. // }
  1436. /* if (type == String.class) {
  1437. if (null == value) {
  1438. sbr.append("?").append(", ");
  1439. parameters.add("");
  1440. } else {
  1441. sbr.append("?").append(", ");
  1442. parameters.add(value);
  1443. // sbr.append("'");
  1444. *//* String v = (String) value;
  1445. if (v.contains("'")) {
  1446. sbr.append(DbUtil.injectDefend(v.replaceAll("'", "\\\\'")));
  1447. } else {
  1448. sbr.append(DbUtil.injectDefend(v));
  1449. }
  1450. sbr.append("'").append(",");*//*
  1451. }
  1452. } else if (DbUtil.isGenericType(type)) {
  1453. sbr.append("?").append(", ");
  1454. parameters.add(value);
  1455. *//* if (null == value) {
  1456. sbr.append(0).append(",");
  1457. } else {
  1458. sbr.append(value).append(",");
  1459. }*//*
  1460. } else if (type == Date.class) {
  1461. sbr.append("?").append(", ");
  1462. parameters.add(value);
  1463. } else {
  1464. }*/
  1465. }
  1466. private void appendValue(StringBuilder sbr, Object value, List<Object> parameters) {
  1467. sbr.append(" ? ");
  1468. parameters.add(value);
  1469. /* if (null != value) {
  1470. if (value instanceof String) {
  1471. sbr.append("'");
  1472. String v = (String) value;
  1473. if (v.contains("'")) {
  1474. sbr.append(DbUtil.injectDefend(v.replaceAll("'", "\\\\'")));
  1475. } else {
  1476. sbr.append(DbUtil.injectDefend(v));
  1477. }
  1478. sbr.append("'").append(",");
  1479. } else if (DbUtil.isGenericType(value)) {
  1480. sbr.append(value).append(",");
  1481. } else if (value instanceof Date) {
  1482. sbr.append("'").append(new Timestamp(((Date) value).getTime())).append("'").append(",");
  1483. } else {
  1484. sbr.append("'").append(JacksonUtil.toJSONString(value)).append("'").append(",");
  1485. }
  1486. } else {
  1487. //类型默认值
  1488. sbr.append(" ").append(",");
  1489. }*/
  1490. }
  1491. private void getValue(StringBuilder sbr, Object value, List<Object> parameters) {
  1492. if (null != value) {
  1493. // if (value instanceof String) {
  1494. // sbr.append("'").append(value).append("'");
  1495. // } else if (DbUtil.isGenericType(value)) {
  1496. // sbr.append(value);
  1497. // } else if (value instanceof Date) {
  1498. // sbr.append("'").append(new Timestamp(((Date) value).getTime())).append("'");
  1499. // } else {
  1500. // sbr.append("'").append(JacksonUtil.toJSONString(value)).append("'");
  1501. // }
  1502. sbr.append("?");
  1503. parameters.add(value);
  1504. } else {
  1505. throw new IllegalArgumentException("append value with NULL");
  1506. }
  1507. }
  1508. private String getTableName(Entity anno) {
  1509. String tbName = anno.tbName();
  1510. if (!DbUtil.isEmptyOrNull(tbName)) {
  1511. return DbUtil.getColumnName(tbName);
  1512. }
  1513. Class<?> clz = anno.clz();
  1514. /* if (tbName.endsWith("Info")) {
  1515. tbName = tbName.substring(0, tbName.length() - 4);
  1516. }*/
  1517. return JdbcHelper.tbPrefix + DbUtil.getColumnName(clz.getSimpleName());
  1518. }
  1519. public static String getTableName(Class<?> clz, String tbName) {
  1520. if (clz.isAnnotationPresent(Entity.class)) {
  1521. Entity entity = clz.getAnnotation(Entity.class);
  1522. if (!DbUtil.isEmptyOrNull(entity.tbName())) {
  1523. return DbUtil.getColumnName(entity.tbName());
  1524. }
  1525. } else if (clz.getSuperclass().isAnnotationPresent(Entity.class)) {
  1526. Class<?> superClz = clz.getSuperclass();
  1527. Entity entity = superClz.getAnnotation(Entity.class);
  1528. if (!DbUtil.isEmptyOrNull(entity.tbName())) {
  1529. return DbUtil.getColumnName(entity.tbName());
  1530. }
  1531. }
  1532. if (tbName.endsWith("Info")) {
  1533. tbName = tbName.substring(0, tbName.length() - 4);
  1534. }
  1535. return JdbcHelper.tbPrefix + DbUtil.getColumnName(tbName);
  1536. }
  1537. /**
  1538. * 获取外键关联实体类
  1539. */
  1540. private Class<?> getFkDomainClass(Field f, Class<?> domainClass) {
  1541. try {
  1542. if (f.isAnnotationPresent(FK.class)) {
  1543. return f.getAnnotation(FK.class).clz();
  1544. }
  1545. if (f.isAnnotationPresent(One.class)) {
  1546. One oneAnnotation = f.getAnnotation(One.class);
  1547. if (null != oneAnnotation) {
  1548. if (oneAnnotation.te() != void.class) {
  1549. return oneAnnotation.te();
  1550. }
  1551. if (!DbUtil.isEmptyOrNull(oneAnnotation.mkf())) {
  1552. if (DbUtil.isGenericType(f.getType())) {
  1553. Field field = domainClass.getDeclaringClass().getDeclaredField(oneAnnotation.mkf());
  1554. if (field.isAnnotationPresent(FK.class)) {
  1555. return field.getAnnotation(FK.class).clz();
  1556. }
  1557. } else {
  1558. return f.getDeclaringClass();
  1559. }
  1560. }
  1561. }
  1562. }
  1563. if (f.isAnnotationPresent(QF.class)) {
  1564. QF qf = f.getAnnotation(QF.class);
  1565. if (!DbUtil.isEmptyOrNull(qf.pkf())) {
  1566. Field field = domainClass.getDeclaringClass().getDeclaredField(qf.pkf());
  1567. return field.getAnnotation(FK.class).clz();
  1568. }
  1569. }
  1570. } catch (NoSuchFieldException e) {
  1571. logger.error("SQL2Helper ERR# getFkDomainClass ,fileName:{},class:{}", f.getName(), domainClass.getSimpleName());
  1572. }
  1573. return Void.class;
  1574. }
  1575. private boolean isImmutableField(String fieldName) {
  1576. List<String> immutableFields = new ArrayList<>();
  1577. immutableFields.add("id");
  1578. immutableFields.add("createat");
  1579. immutableFields.add("updateat");
  1580. return immutableFields.contains(fieldName.toLowerCase());
  1581. }
  1582. /**
  1583. * 校验字段的值
  1584. */
  1585. private void checkFieldValue(Field field, Object value, boolean insertCheck) {
  1586. if (!field.isAnnotationPresent(DBF.class)) {
  1587. return;
  1588. }
  1589. DBF dbf = field.getAnnotation(DBF.class);
  1590. //插入SQL校验必填
  1591. if (insertCheck) {
  1592. if (dbf.required() && DbUtil.isEmptyOrNull(value)) {
  1593. throw new IllegalArgumentException("【" + dbf.comment() + "】取值不能为空");
  1594. }
  1595. }
  1596. int min = dbf.min();
  1597. int max = dbf.max();
  1598. if (dbf.min() > 0) {
  1599. if (value instanceof Integer || value instanceof Short || value instanceof Double || value instanceof Long) {
  1600. if (BigDecimal.valueOf(min).compareTo(new BigDecimal(String.valueOf(value))) > 0) {
  1601. throw new IllegalArgumentException("【" + dbf.comment() + "】最小值为" + min);
  1602. }
  1603. if (max > 0 && BigDecimal.valueOf(max).compareTo(new BigDecimal(value.toString())) < 0) {
  1604. throw new IllegalArgumentException("【" + dbf.comment() + "】最大值为" + max);
  1605. }
  1606. } else if (value instanceof String || value instanceof Character) {
  1607. if (!DbUtil.isEmptyOrNull(value)) {
  1608. int length = value.toString().length();
  1609. if (min > length) {
  1610. throw new IllegalArgumentException("【" + dbf.comment() + "】最小长度为" + min);
  1611. }
  1612. if (max > 0 && max < length) {
  1613. throw new IllegalArgumentException("【" + dbf.comment() + "】最大长度为" + max);
  1614. }
  1615. }
  1616. }
  1617. }
  1618. }
  1619. }