fix bug

mybatis-plus/src/main/java/com/baomidou/mybatisplus/MybatisSqlSessionFactoryHolder.java

@@ -0,0 +1,32 @@
+package com.baomidou.mybatisplus;
+
+import org.apache.ibatis.session.SqlSession;
+import org.apache.ibatis.session.SqlSessionFactory;
+
+/**
+ * <p>
+ * MybatisSqlSessionFactoryHolder
+ * </p>
+ *
+ * @author Caratacus
+ * @Date 2016-10-26
+ */
+public class MybatisSqlSessionFactoryHolder {
+
+	private static SqlSession sqlSession;
+	private static SqlSessionFactory sqlSessionFactory;
+
+	public static void setSqlSessionFactory(SqlSessionFactory sqlSessionFactory) {
+		MybatisSqlSessionFactoryHolder.sqlSessionFactory = sqlSessionFactory;
+		MybatisSqlSessionFactoryHolder.sqlSession = sqlSessionFactory.openSession();
+	}
+
+	public static SqlSessionFactory getSqlSessionFactory() {
+		return sqlSessionFactory;
+	}
+
+	public static SqlSession getSqlSession() {
+		return sqlSession;
+	}
+
+}

mybatis-plus/src/main/java/com/baomidou/mybatisplus/spring/MybatisSqlSessionFactoryBean.java

@@ -16,6 +16,7 @@
 package com.baomidou.mybatisplus.spring;
 
 import com.baomidou.mybatisplus.MybatisConfiguration;
+import com.baomidou.mybatisplus.MybatisSqlSessionFactoryHolder;
 import com.baomidou.mybatisplus.MybatisXMLConfigBuilder;
 import com.baomidou.mybatisplus.MybatisXMLMapperBuilder;
 import com.baomidou.mybatisplus.annotations.FieldStrategy;
@@ -24,7 +25,6 @@ import com.baomidou.mybatisplus.mapper.DBType;
 import com.baomidou.mybatisplus.mapper.IMetaObjectHandler;
 import com.baomidou.mybatisplus.mapper.ISqlInjector;
 import com.baomidou.mybatisplus.toolkit.PackageHelper;
-import com.baomidou.mybatisplus.toolkit.TableInfoHelper;
 import org.apache.ibatis.cache.Cache;
 import org.apache.ibatis.executor.ErrorContext;
 import org.apache.ibatis.io.VFS;
@@ -598,7 +598,7 @@ public class MybatisSqlSessionFactoryBean implements FactoryBean<SqlSessionFacto
 		}
 		SqlSessionFactory sqlSessionFactory = this.sqlSessionFactoryBuilder.build(configuration);
 		// TODO 缓存 sqlSessionFactory
-		TableInfoHelper.cacheSqlSessionFactory(sqlSessionFactory);
+		MybatisSqlSessionFactoryHolder.setSqlSessionFactory(sqlSessionFactory);
 		return sqlSessionFactory;
 	}
 

mybatis-plus/src/main/java/com/baomidou/mybatisplus/toolkit/TableInfoHelper.java

@@ -16,16 +16,15 @@
 package com.baomidou.mybatisplus.toolkit;
 
 import com.baomidou.mybatisplus.MybatisConfiguration;
+import com.baomidou.mybatisplus.MybatisSqlSessionFactoryHolder;
 import com.baomidou.mybatisplus.annotations.TableField;
 import com.baomidou.mybatisplus.annotations.TableId;
 import com.baomidou.mybatisplus.annotations.TableName;
 import com.baomidou.mybatisplus.exceptions.MybatisPlusException;
-import org.apache.ibatis.session.SqlSessionFactory;
 
 import java.lang.reflect.Field;
 import java.lang.reflect.Modifier;
 import java.util.ArrayList;
-import java.util.Collection;
 import java.util.LinkedList;
 import java.util.List;
 import java.util.Map;
@@ -41,184 +40,170 @@ import java.util.concurrent.ConcurrentHashMap;
  */
 public class TableInfoHelper {
 
-    /**
-     * 缓存反射类表信息
-     */
-    private static Map<String, TableInfo> tableInfoCache = new ConcurrentHashMap<String, TableInfo>();
-
-    /**
-     * <p>
-     * 获取实体映射表信息
-     * <p>
-     *
-     * @param clazz
-     *            反射实体类
-     * @return
-     */
-    public static TableInfo getTableInfo(Class<?> clazz) {
-        return tableInfoCache.get(clazz.getName());
-    }
-
-    /**
-     * <p>
-     * 实体类反射获取表信息【初始化】
-     * <p>
-     *
-     * @param clazz
-     *            反射实体类
-     * @return
-     */
-    public synchronized static TableInfo initTableInfo(Class<?> clazz) {
-        TableInfo ti = tableInfoCache.get(clazz.getName());
-        if (ti != null) {
-            return ti;
-        }
-        TableInfo tableInfo = new TableInfo();
+	/**
+	 * 缓存反射类表信息
+	 */
+	private static Map<String, TableInfo> tableInfoCache = new ConcurrentHashMap<String, TableInfo>();
+
+	/**
+	 * <p>
+	 * 获取实体映射表信息
+	 * <p>
+	 *
+	 * @param clazz
+	 *            反射实体类
+	 * @return
+	 */
+	public static TableInfo getTableInfo(Class<?> clazz) {
+		return tableInfoCache.get(clazz.getName());
+	}
+
+	/**
+	 * <p>
+	 * 实体类反射获取表信息【初始化】
+	 * <p>
+	 *
+	 * @param clazz
+	 *            反射实体类
+	 * @return
+	 */
+	public synchronized static TableInfo initTableInfo(Class<?> clazz) {
+		TableInfo ti = tableInfoCache.get(clazz.getName());
+		if (ti != null) {
+			return ti;
+		}
+		TableInfo tableInfo = new TableInfo();
 
 		/* 表名 */
-        TableName table = clazz.getAnnotation(TableName.class);
-        if (table != null && StringUtils.isNotEmpty(table.value())) {
-            tableInfo.setTableName(table.value());
-        } else {
-            tableInfo.setTableName(StringUtils.camelToUnderline(clazz.getSimpleName()));
-        }
+		TableName table = clazz.getAnnotation(TableName.class);
+		if (table != null && StringUtils.isNotEmpty(table.value())) {
+			tableInfo.setTableName(table.value());
+		} else {
+			tableInfo.setTableName(StringUtils.camelToUnderline(clazz.getSimpleName()));
+		}
 
 		/* 表结果集映射 */
-        if (table != null && StringUtils.isNotEmpty(table.resultMap())) {
-            tableInfo.setResultMap(table.resultMap());
-        }
-
-        List<TableFieldInfo> fieldList = new ArrayList<TableFieldInfo>();
-        List<Field> list = getAllFields(clazz);
-        for (Field field : list) {
-            /**
-             * 主键ID
-             */
-            TableId tableId = field.getAnnotation(TableId.class);
-            if (tableId != null) {
-                if (tableInfo.getKeyColumn() == null) {
-                    tableInfo.setIdType(tableId.type());
-                    if (StringUtils.isNotEmpty(tableId.value())) {
+		if (table != null && StringUtils.isNotEmpty(table.resultMap())) {
+			tableInfo.setResultMap(table.resultMap());
+		}
+
+		List<TableFieldInfo> fieldList = new ArrayList<TableFieldInfo>();
+		List<Field> list = getAllFields(clazz);
+		for (Field field : list) {
+			/**
+			 * 主键ID
+			 */
+			TableId tableId = field.getAnnotation(TableId.class);
+			if (tableId != null) {
+				if (tableInfo.getKeyColumn() == null) {
+					tableInfo.setIdType(tableId.type());
+					if (StringUtils.isNotEmpty(tableId.value())) {
 						/* 自定义字段 */
-                        tableInfo.setKeyColumn(tableId.value());
-                        tableInfo.setKeyRelated(true);
-                    } else if (MybatisConfiguration.DB_COLUMN_UNDERLINE) {
+						tableInfo.setKeyColumn(tableId.value());
+						tableInfo.setKeyRelated(true);
+					} else if (MybatisConfiguration.DB_COLUMN_UNDERLINE) {
 						/* 开启字段下划线申明 */
-                        tableInfo.setKeyColumn(StringUtils.camelToUnderline(field.getName()));
-                    } else {
-                        tableInfo.setKeyColumn(field.getName());
-                    }
-                    tableInfo.setKeyProperty(field.getName());
-                    continue;
-                } else {
+						tableInfo.setKeyColumn(StringUtils.camelToUnderline(field.getName()));
+					} else {
+						tableInfo.setKeyColumn(field.getName());
+					}
+					tableInfo.setKeyProperty(field.getName());
+					continue;
+				} else {
 					/* 发现设置多个主键注解抛出异常 */
-                    throw new MybatisPlusException("There must be only one, Discover multiple @TableId annotation in " + clazz);
-                }
-            }
+					throw new MybatisPlusException("There must be only one, Discover multiple @TableId annotation in " + clazz);
+				}
+			}
 
 			/* 获取注解属性，自定义字段 */
-            TableField tableField = field.getAnnotation(TableField.class);
-            if (tableField != null) {
-                String columnName = field.getName();
-                if (StringUtils.isNotEmpty(tableField.value())) {
-                    columnName = tableField.value();
-                }
+			TableField tableField = field.getAnnotation(TableField.class);
+			if (tableField != null) {
+				String columnName = field.getName();
+				if (StringUtils.isNotEmpty(tableField.value())) {
+					columnName = tableField.value();
+				}
 
 				/*
 				 * el 语法支持，可以传入多个参数以逗号分开
 				 */
-                String el = field.getName();
-                if (StringUtils.isNotEmpty(tableField.el())) {
-                    el = tableField.el();
-                }
-                String[] columns = columnName.split(";");
-                String[] els = el.split(";");
-                if (null != columns && null != els && columns.length == els.length) {
-                    for (int i = 0; i < columns.length; i++) {
-                        fieldList.add(new TableFieldInfo(true, columns[i], field.getName(), els[i], tableField.validate()));
-                    }
-                } else {
-                    String errorMsg = "Class: %s, Field: %s, 'value' 'el' Length must be consistent.";
-                    throw new MybatisPlusException(String.format(errorMsg, clazz.getName(), field.getName()));
-                }
-
-                continue;
-            }
-
-            /**
-             * 字段, 使用 camelToUnderline 转换驼峰写法为下划线分割法, 如果已指定 TableField , 便不会执行这里
-             */
-            if (MybatisConfiguration.DB_COLUMN_UNDERLINE) {
+				String el = field.getName();
+				if (StringUtils.isNotEmpty(tableField.el())) {
+					el = tableField.el();
+				}
+				String[] columns = columnName.split(";");
+				String[] els = el.split(";");
+				if (null != columns && null != els && columns.length == els.length) {
+					for (int i = 0; i < columns.length; i++) {
+						fieldList.add(new TableFieldInfo(true, columns[i], field.getName(), els[i], tableField.validate()));
+					}
+				} else {
+					String errorMsg = "Class: %s, Field: %s, 'value' 'el' Length must be consistent.";
+					throw new MybatisPlusException(String.format(errorMsg, clazz.getName(), field.getName()));
+				}
+
+				continue;
+			}
+
+			/**
+			 * 字段, 使用 camelToUnderline 转换驼峰写法为下划线分割法, 如果已指定 TableField , 便不会执行这里
+			 */
+			if (MybatisConfiguration.DB_COLUMN_UNDERLINE) {
 				/* 开启字段下划线申明 */
-                fieldList.add(new TableFieldInfo(true, StringUtils.camelToUnderline(field.getName()), field.getName()));
-            } else {
-                fieldList.add(new TableFieldInfo(field.getName()));
-            }
-        }
+				fieldList.add(new TableFieldInfo(true, StringUtils.camelToUnderline(field.getName()), field.getName()));
+			} else {
+				fieldList.add(new TableFieldInfo(field.getName()));
+			}
+		}
 
 		/* 字段列表 */
-        tableInfo.setFieldList(fieldList);
+		tableInfo.setFieldList(fieldList);
 
 		/*
 		 * 未发现主键注解，跳过注入
 		 */
-        if (null == tableInfo.getKeyColumn()) {
-            return null;
-        }
-
+		if (null == tableInfo.getKeyColumn()) {
+			return null;
+		}
+		// 缓存
+		tableInfo.setSqlSessionFactory(MybatisSqlSessionFactoryHolder.getSqlSessionFactory());
 		/*
 		 * 注入
 		 */
-        tableInfoCache.put(clazz.getName(), tableInfo);
-        return tableInfo;
-    }
-
-    /**
-     * 获取该类的所有属性列表
-     *
-     * @param clazz
-     *            反射类
-     * @return
-     */
-    private static List<Field> getAllFields(Class<?> clazz) {
-        List<Field> result = new LinkedList<Field>();
-        Field[] fields = clazz.getDeclaredFields();
-        for (Field field : fields) {
+		tableInfoCache.put(clazz.getName(), tableInfo);
+		return tableInfo;
+	}
+
+	/**
+	 * 获取该类的所有属性列表
+	 *
+	 * @param clazz
+	 *            反射类
+	 * @return
+	 */
+	private static List<Field> getAllFields(Class<?> clazz) {
+		List<Field> result = new LinkedList<Field>();
+		Field[] fields = clazz.getDeclaredFields();
+		for (Field field : fields) {
 
 			/* 过滤 transient关键字修饰的属性 */
-            if (Modifier.isTransient(field.getModifiers())) {
-                continue;
-            }
+			if (Modifier.isTransient(field.getModifiers())) {
+				continue;
+			}
 
 			/* 过滤注解非表字段属性 */
-            TableField tableField = field.getAnnotation(TableField.class);
-            if (tableField == null || tableField.exist()) {
-                result.add(field);
-            }
-        }
+			TableField tableField = field.getAnnotation(TableField.class);
+			if (tableField == null || tableField.exist()) {
+				result.add(field);
+			}
+		}
 
 		/* 处理父类字段 */
-        Class<?> superClass = clazz.getSuperclass();
-        if (superClass.equals(Object.class)) {
-            return result;
-        }
-        result.addAll(getAllFields(superClass));
-        return result;
-    }
-
-    /**
-     * 缓存sqlSessionFactory
-     *
-     * @param sqlSessionFactory
-     * @return
-     */
-    public static void cacheSqlSessionFactory(SqlSessionFactory sqlSessionFactory) {
-        Collection<TableInfo> values = tableInfoCache.values();
-        for (TableInfo tableInfo : values) {
-            if (null == tableInfo.getSqlSessionFactory()) {
-                tableInfo.setSqlSessionFactory(sqlSessionFactory);
-            }
-        }
-    }
+		Class<?> superClass = clazz.getSuperclass();
+		if (superClass.equals(Object.class)) {
+			return result;
+		}
+		result.addAll(getAllFields(superClass));
+		return result;
+	}
 
 }