优化--mybatis增强器

src/main/java/com/genersoft/iot/vmp/conf/dao/Id.java

@@ -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 {
+}

src/main/java/com/genersoft/iot/vmp/conf/dao/Invisible.java

@@ -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 {
+}

src/main/java/com/genersoft/iot/vmp/conf/dao/SelectByEntityExtendedLanguageDriver.java

@@ -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);
+	}
+}

src/main/java/com/genersoft/iot/vmp/conf/dao/SimpleInsertExtendedLanguageDriver.java

@@ -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);
+    }
+
+}

src/main/java/com/genersoft/iot/vmp/conf/dao/SimpleSelectInExtendedLanguageDriver.java

@@ -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);
+	}
+}

src/main/java/com/genersoft/iot/vmp/conf/dao/SimpleUpdateExtendedLanguageDriver.java

@@ -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);
+	}
+}