From 052e1d5c47c4e536fde79074d53b0481c7d4f9b6 Mon Sep 17 00:00:00 2001 From: admin <weikou2014> Date: 星期三, 13 十月 2021 18:33:42 +0800 Subject: [PATCH] 自动化代码生成优化 --- src/main/resources/generater/dao/mongoDBDao.template | 56 +++++++++++++++++++++++++------------------------------- 1 files changed, 25 insertions(+), 31 deletions(-) diff --git a/src/main/resources/generater/dao/mongoDBDao.template b/src/main/resources/generater/dao/mongoDBDao.template index eeea6ba..32a5d1d 100644 --- a/src/main/resources/generater/dao/mongoDBDao.template +++ b/src/main/resources/generater/dao/mongoDBDao.template @@ -46,43 +46,37 @@ } private Query getQuery(DaoQuery daoQuery){ - List<Criteria> andList=new ArrayList<>(); - <#list daoQueryColumnDataList as value> - <#if (value.type!"") == "Date"> - if(daoQuery.min${value.columnName?cap_first}!=null){ - andList.add(Criteria.where("${value.columnName}").gte(daoQuery.min${value.columnName?cap_first})); - } - if(daoQuery.max${value.columnName?cap_first}!=null){ - andList.add(Criteria.where("${value.columnName}").lt(daoQuery.max${value.columnName?cap_first})); - } - <#else> - if(daoQuery.${value.columnName}!=null){ - <#if (value.queryType!"") == "equal"> - andList.add(Criteria.where("${value.columnName}").is(daoQuery.${value.columnName})); - <#else> - andList.add(Criteria.where("${value.columnName}").regex(daoQuery.${value.columnName})); - </#if> - } - </#if> - </#list> - Query query=new Query(); - Criteria[] ands=new Criteria[andList.size()]; - andList.toArray(ands); - if(ands.length>0){ - query.addCriteria(new Criteria().andOperator(ands)); - } - return query; + List<Criteria> andList=new ArrayList<>(); + <#list daoQueryColumnDataList as value> + if(daoQuery.${value.columnName}!=null){ + <#if (value.queryType!"") == "equal"> + andList.add(Criteria.where("${value.columnData.property}").is(daoQuery.${value.columnName})); + <#elseif (value.queryType!"") == "gt"> + andList.add(Criteria.where("${value.columnData.property}").gt(daoQuery.${value.columnName})); + <#elseif (value.queryType!"") == "gte"> + andList.add(Criteria.where("${value.columnData.property}").gte(daoQuery.${value.columnName})); + <#elseif (value.queryType!"") == "lt"> + andList.add(Criteria.where("${value.columnData.property}").lt(daoQuery.${value.columnName})); + <#elseif (value.queryType!"") == "lte"> + andList.add(Criteria.where("${value.columnData.property}").lte(daoQuery.${value.columnName})); + <#else> + andList.add(Criteria.where("${value.columnData.property}").regex(daoQuery.${value.columnName})); + </#if> + } + </#list> + Query query=new Query(); + Criteria[] ands=new Criteria[andList.size()]; + andList.toArray(ands); + if(ands.length>0){ + query.addCriteria(new Criteria().andOperator(ands)); + } + return query; } public static class DaoQuery{ <#if daoQueryColumnDataList??> <#list daoQueryColumnDataList as value> - <#if (value.type!"") == "Date"> - public ${value.type} min${value.columnName?cap_first}; - public ${value.type} max${value.columnName?cap_first}; - <#else> public ${value.type} ${value.columnName}; - </#if> </#list> </#if> public int start; -- Gitblit v1.8.0