优化--mybatis增强器
chenjialing
parent
2142b7ffcd
commit
bb4e54bca6
|
|
@ -0,0 +1,16 @@
|
|||
package com.genersoft.iot.vmp.conf.dao;
|
||||
|
||||
import java.lang.annotation.ElementType;
|
||||
import java.lang.annotation.Retention;
|
||||
import java.lang.annotation.RetentionPolicy;
|
||||
import java.lang.annotation.Target;
|
||||
|
||||
/**
|
||||
* 通过注解控制实体中的ID属性
|
||||
*
|
||||
* @author
|
||||
*/
|
||||
@Target(ElementType.FIELD)
|
||||
@Retention(RetentionPolicy.RUNTIME)
|
||||
public @interface Id {
|
||||
}
|
||||
|
|
@ -0,0 +1,16 @@
|
|||
package com.genersoft.iot.vmp.conf.dao;
|
||||
|
||||
import java.lang.annotation.ElementType;
|
||||
import java.lang.annotation.Retention;
|
||||
import java.lang.annotation.RetentionPolicy;
|
||||
import java.lang.annotation.Target;
|
||||
|
||||
/**
|
||||
* 通过注解控制实体中的属性不存数据库
|
||||
*
|
||||
* @author
|
||||
*/
|
||||
@Target(ElementType.FIELD)
|
||||
@Retention(RetentionPolicy.RUNTIME)
|
||||
public @interface Invisible {
|
||||
}
|
||||
|
|
@ -0,0 +1,43 @@
|
|||
package com.genersoft.iot.vmp.conf.dao;
|
||||
|
||||
import com.google.common.base.CaseFormat;
|
||||
import org.apache.ibatis.mapping.SqlSource;
|
||||
import org.apache.ibatis.scripting.LanguageDriver;
|
||||
import org.apache.ibatis.scripting.xmltags.XMLLanguageDriver;
|
||||
import org.apache.ibatis.session.Configuration;
|
||||
|
||||
import java.lang.reflect.Field;
|
||||
import java.util.regex.Matcher;
|
||||
import java.util.regex.Pattern;
|
||||
|
||||
/**
|
||||
* mybatis 注解模式 select 支持传入对象 驱动
|
||||
* SELECT * FROM xxx_user where 1=1 (#{userObject})
|
||||
*
|
||||
* @author
|
||||
*/
|
||||
public class SelectByEntityExtendedLanguageDriver extends XMLLanguageDriver implements LanguageDriver {
|
||||
|
||||
private final Pattern inPattern = Pattern.compile("\\(#\\{(\\w+)\\}\\)");
|
||||
|
||||
@Override
|
||||
public SqlSource createSqlSource(Configuration configuration, String script, Class<?> parameterType) {
|
||||
|
||||
Matcher matcher = inPattern.matcher(script);
|
||||
if (matcher.find()) {
|
||||
StringBuffer ss = new StringBuffer();
|
||||
for (Field field : parameterType.getDeclaredFields()) {
|
||||
//如果不是加了忽略注解的字段就去拼接
|
||||
if (!field.isAnnotationPresent(Invisible.class)) {
|
||||
//and type !=''
|
||||
String temp = "<if test=\"__field != null and __field!='' \">and __column=#{__field}</if>";
|
||||
ss.append(temp.replaceAll("__field", field.getName()).replaceAll("__column",
|
||||
CaseFormat.LOWER_CAMEL.to(CaseFormat.LOWER_UNDERSCORE, field.getName())));
|
||||
}
|
||||
}
|
||||
script = matcher.replaceAll(ss.toString());
|
||||
script = "<script>" + script + "</script>";
|
||||
}
|
||||
return super.createSqlSource(configuration, script, parameterType);
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,59 @@
|
|||
package com.genersoft.iot.vmp.conf.dao;
|
||||
|
||||
import com.google.common.base.CaseFormat;
|
||||
import org.apache.ibatis.mapping.SqlSource;
|
||||
import org.apache.ibatis.scripting.LanguageDriver;
|
||||
import org.apache.ibatis.scripting.xmltags.XMLLanguageDriver;
|
||||
import org.apache.ibatis.session.Configuration;
|
||||
|
||||
import java.lang.reflect.Field;
|
||||
import java.util.regex.Matcher;
|
||||
import java.util.regex.Pattern;
|
||||
|
||||
/**
|
||||
* mybatis 注解模式 insert 支持传入对象 驱动
|
||||
* INSERT INTO xxx_user (#{userObject})
|
||||
*
|
||||
* @author
|
||||
*/
|
||||
public class SimpleInsertExtendedLanguageDriver extends XMLLanguageDriver implements LanguageDriver {
|
||||
private final Pattern inPattern = Pattern.compile("\\(#\\{(\\w+)\\}\\)");
|
||||
|
||||
@Override
|
||||
public SqlSource createSqlSource(Configuration configuration, String script, Class<?> parameterType) {
|
||||
Matcher matcher = inPattern.matcher(script);
|
||||
if (matcher.find()) {
|
||||
|
||||
// 组建 (xx, xx , xx)字段语句
|
||||
StringBuffer ss = new StringBuffer("( <trim suffixOverrides=\",\">");
|
||||
for (Field field : parameterType.getDeclaredFields()) {
|
||||
//如果不是加了忽略注解的字段就去拼接
|
||||
if (field.isAnnotationPresent(Invisible.class)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
String column = CaseFormat.LOWER_CAMEL.to(CaseFormat.LOWER_UNDERSCORE, field.getName());
|
||||
String temp = "<if test=\"__field != null\"> __column,</if>";
|
||||
ss.append(temp.replaceAll("__field", field.getName()).replaceAll("__column", column));
|
||||
}
|
||||
ss.append("</trim>) VALUES ( <trim suffixOverrides=\",\">");
|
||||
|
||||
// 组建 ("1", "xx"， "")值语句
|
||||
for (Field field : parameterType.getDeclaredFields()) {
|
||||
if (field.isAnnotationPresent(Invisible.class)) {
|
||||
continue;
|
||||
}
|
||||
|
||||
String temp = "<if test=\"__field != null\"> #{__field},</if>";
|
||||
ss.append(temp.replaceAll("__field", field.getName()));
|
||||
}
|
||||
|
||||
ss.append("</trim>) ");
|
||||
script = matcher.replaceAll(ss.toString());
|
||||
|
||||
script = "<script>" + script + "</script>";
|
||||
}
|
||||
return super.createSqlSource(configuration, script, parameterType);
|
||||
}
|
||||
|
||||
}
|
||||
|
|
@ -0,0 +1,33 @@
|
|||
package com.genersoft.iot.vmp.conf.dao;
|
||||
|
||||
import org.apache.ibatis.mapping.SqlSource;
|
||||
import org.apache.ibatis.scripting.LanguageDriver;
|
||||
import org.apache.ibatis.scripting.xmltags.XMLLanguageDriver;
|
||||
import org.apache.ibatis.session.Configuration;
|
||||
|
||||
import java.util.regex.Matcher;
|
||||
import java.util.regex.Pattern;
|
||||
|
||||
/**
|
||||
* mybatis 注解模式 select 支持传入list 驱动
|
||||
* SELECT * FROM xxx_user WHERE user_id IN (#{userIds})
|
||||
*
|
||||
* @author
|
||||
*/
|
||||
public class SimpleSelectInExtendedLanguageDriver extends XMLLanguageDriver implements LanguageDriver {
|
||||
|
||||
private final Pattern inPattern = Pattern.compile("\\(#\\{(\\w+)\\}\\)");
|
||||
|
||||
@Override
|
||||
public SqlSource createSqlSource(Configuration configuration, String script, Class<?> parameterType) {
|
||||
|
||||
Matcher matcher = inPattern.matcher(script);
|
||||
if (matcher.find()) {
|
||||
script = matcher
|
||||
.replaceAll("(<foreach collection=\"$1\" item=\"__item\" separator=\",\" >#{__item}</foreach>)");
|
||||
}
|
||||
|
||||
script = "<script>" + script + "</script>";
|
||||
return super.createSqlSource(configuration, script, parameterType);
|
||||
}
|
||||
}
|
||||
|
|
@ -0,0 +1,53 @@
|
|||
package com.genersoft.iot.vmp.conf.dao;
|
||||
|
||||
import com.google.common.base.CaseFormat;
|
||||
import org.apache.ibatis.mapping.SqlSource;
|
||||
import org.apache.ibatis.scripting.LanguageDriver;
|
||||
import org.apache.ibatis.scripting.xmltags.XMLLanguageDriver;
|
||||
import org.apache.ibatis.session.Configuration;
|
||||
|
||||
import java.lang.reflect.Field;
|
||||
import java.util.Date;
|
||||
import java.util.regex.Matcher;
|
||||
import java.util.regex.Pattern;
|
||||
|
||||
/**
|
||||
* mybatis 注解模式 update 支持传入对象 驱动
|
||||
* UPDATE xxx_user (#{userObject}) WHERE true and xxxx
|
||||
*
|
||||
* @author
|
||||
*/
|
||||
public class SimpleUpdateExtendedLanguageDriver extends XMLLanguageDriver implements LanguageDriver {
|
||||
private final Pattern inPattern = Pattern.compile("\\(#\\{(\\w+)\\}\\)");
|
||||
|
||||
@Override
|
||||
public SqlSource createSqlSource(Configuration configuration, String script, Class<?> parameterType) {
|
||||
Matcher matcher = inPattern.matcher(script);
|
||||
if (matcher.find()) {
|
||||
StringBuffer ss = new StringBuffer();
|
||||
ss.append("<set>");
|
||||
|
||||
for (Field field : parameterType.getDeclaredFields()) {
|
||||
if (!field.isAnnotationPresent(Id.class)){
|
||||
String temp = "";
|
||||
if(field.getType()== Date.class){
|
||||
temp = "<if test=\"__field != null \">__column=#{__field},</if>";
|
||||
}else {
|
||||
temp = "<if test=\"__field != null \">__column=#{__field},</if>";
|
||||
}
|
||||
ss.append(temp.replaceAll("__field", field.getName()).replaceAll("__column",
|
||||
CaseFormat.LOWER_CAMEL.to(CaseFormat.LOWER_UNDERSCORE, field.getName())));
|
||||
}
|
||||
|
||||
}
|
||||
|
||||
ss.deleteCharAt(ss.lastIndexOf(","));
|
||||
ss.append("</set>");
|
||||
|
||||
script = matcher.replaceAll(ss.toString());
|
||||
|
||||
script = "<script>" + script + "</script>";
|
||||
}
|
||||
return super.createSqlSource(configuration, script, parameterType);
|
||||
}
|
||||
}
|
||||
Loading…
Reference in New Issue