集成到Spring

abel533 · abel533 · commit 959b6ea59066 · 2014-12-10T17:30:03.000+08:00
diff --git a/src/main/java/com/isea533/mybatis/MappperBeanPostProcessor.java b/src/main/java/com/isea533/mybatis/MappperBeanPostProcessor.java
@@ -0,0 +1,84 @@
+package com.isea533.mybatis;
+
+import com.isea533.mybatis.mapperhelper.MapperHelper;
+import org.apache.ibatis.builder.annotation.ProviderSqlSource;
+import org.apache.ibatis.mapping.MappedStatement;
+import org.mybatis.spring.SqlSessionTemplate;
+import org.springframework.beans.BeansException;
+import org.springframework.beans.factory.config.BeanPostProcessor;
+
+import java.util.Collection;
+import java.util.Properties;
+
+/**
+ * @author liuzh
+ */
+public class MappperBeanPostProcessor implements BeanPostProcessor {
+
+    private final MapperHelper mapperHelper = new MapperHelper();
+
+    private boolean runed = false;
+
+    public void setProperties(Properties properties) {
+        String UUID = properties.getProperty("UUID");
+        if (UUID != null && UUID.length() > 0) {
+            mapperHelper.setUUID(UUID);
+        }
+        String IDENTITY = properties.getProperty("IDENTITY");
+        if (IDENTITY != null && IDENTITY.length() > 0) {
+            mapperHelper.setIDENTITY(IDENTITY);
+        }
+        String seqFormat = properties.getProperty("seqFormat");
+        if (seqFormat != null && seqFormat.length() > 0) {
+            mapperHelper.setSeqFormat(seqFormat);
+        }
+        String ORDER = properties.getProperty("ORDER");
+        if (ORDER != null && ORDER.length() > 0) {
+            mapperHelper.setBEFORE(ORDER);
+        }
+        String cameHumpMap = properties.getProperty("cameHumpMap");
+        if (cameHumpMap != null && cameHumpMap.length() > 0) {
+            mapperHelper.setCameHumpMap(cameHumpMap);
+        }
+        int mapperCount = 0;
+        String mapper = properties.getProperty("mappers");
+        if (mapper != null && mapper.length() > 0) {
+            String[] mappers = mapper.split(",");
+            for (String mapperClass : mappers) {
+                if (mapperClass.length() > 0) {
+                    mapperHelper.registerMapper(mapperClass);
+                    mapperCount++;
+                }
+            }
+        }
+        if (mapperCount == 0) {
+            throw new RuntimeException("通用Mapper没有配置任何有效的通用接口!");
+        }
+    }
+
+
+    @Override
+    public Object postProcessBeforeInitialization(Object bean, String beanName) throws BeansException {
+        return bean;
+    }
+
+    @Override
+    public Object postProcessAfterInitialization(Object bean, String beanName) throws BeansException {
+        if (!runed && bean instanceof SqlSessionTemplate) {
+            SqlSessionTemplate sqlSessionTemplate = (SqlSessionTemplate)bean;
+            Collection<MappedStatement> collection = sqlSessionTemplate.getConfiguration().getMappedStatements();
+            for (Object object : collection) {
+                if (object instanceof MappedStatement) {
+                    MappedStatement ms = (MappedStatement)object;
+                    if (mapperHelper.isMapperMethod(ms.getId())) {
+                        if (ms.getSqlSource() instanceof ProviderSqlSource) {
+                            mapperHelper.setSqlSource(ms);
+                        }
+                    }
+                }
+            }
+            runed = true;
+        }
+        return bean;
+    }
+}
diff --git a/src/main/resources/applicationContext.xml b/src/main/resources/applicationContext.xml
@@ -42,13 +42,6 @@
             </value>
           </property>
         </bean>
-        <bean class="com.isea533.mybatis.mapperhelper.MapperInterceptor">
-          <property name="properties">
-            <value>
-              mappers=com.isea533.mybatis.mapperhelper.Mapper
-            </value>
-          </property>
-        </bean>
       </array>
     </property>
   </bean>
@@ -80,4 +73,12 @@
   <bean id="transactionManager" class="org.springframework.jdbc.datasource.DataSourceTransactionManager">
     <property name="dataSource" ref="dataSource"/>
   </bean>
+
+  <bean id="mapperHelper" class="com.isea533.mybatis.MappperBeanPostProcessor">
+    <property name="properties">
+      <value>
+        mappers=com.isea533.mybatis.mapperhelper.Mapper
+      </value>
+    </property>
+  </bean>
 </beans>
