com.gitee.sunchenbin.mybatis.actable.manager.system.SysMysqlCreateTableManagerImpl Maven / Gradle / Ivy

Go to download
package com.gitee.sunchenbin.mybatis.actable.manager.system;

import java.lang.reflect.Field;
import java.util.ArrayList;
import java.util.HashMap;
import java.util.List;
import java.util.Map;
import java.util.Map.Entry;
import java.util.Set;

import org.apache.commons.lang.ArrayUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.beans.factory.annotation.Value;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;

import com.gitee.sunchenbin.mybatis.actable.annotation.Column;
import com.gitee.sunchenbin.mybatis.actable.annotation.LengthCount;
import com.gitee.sunchenbin.mybatis.actable.annotation.Table;
import com.gitee.sunchenbin.mybatis.actable.command.CreateTableParam;
import com.gitee.sunchenbin.mybatis.actable.command.SysMysqlColumns;
import com.gitee.sunchenbin.mybatis.actable.constants.MySqlTypeConstant;
import com.gitee.sunchenbin.mybatis.actable.dao.system.CreateMysqlTablesMapper;
import com.gitee.sunchenbin.mybatis.actable.utils.ClassTools;

/**
 * 项目启动时自动扫描配置的目录中的model，根据配置的规则自动创建或更新表 该逻辑只适用于mysql，其他数据库尚且需要另外扩展，因为sql的语法不同
 *
 * @author sunchenbin
 * @version 2016年6月23日 下午5:58:12
 */
@Transactional
@Service("sysMysqlCreateTableManager")
public class SysMysqlCreateTableManagerImpl implements SysMysqlCreateTableManager {

	private static final Logger log = LoggerFactory.getLogger(SysMysqlCreateTableManagerImpl.class);

	@Autowired
	private CreateMysqlTablesMapper createMysqlTablesMapper;

	/**
	 * 要扫描的model所在的pack
	 */
	@Value("#{configProperties['mybatis.model.pack']}")
	private String pack;

	/**
	 * 自动创建模式：update表示更新，create表示删除原表重新创建
	 */
	@Value("#{configProperties['mybatis.table.auto']}")
	private String tableAuto;

	/**
	 * 读取配置文件的三种状态（创建表、更新表、不做任何事情）
	 */
	public void createMysqlTable() {

		// 不做任何事情
		if ("none".equals(tableAuto)) {
			log.info("配置mybatis.table.auto=none，不需要做任何事情");
			return;
		}

		// 获取Mysql的类型，以及类型需要设置几个长度
		Map mySqlTypeAndLengthMap = mySqlTypeAndLengthMap();

		// 从包package中获取所有的Class
		Set> classes = ClassTools.getClasses(pack);

		// 用于存需要创建的表名+结构
		Map> newTableMap = new HashMap>();

		// 用于存需要更新字段类型等的表名+结构
		Map> modifyTableMap = new HashMap>();

		// 用于存需要增加字段的表名+结构
		Map> addTableMap = new HashMap>();

		// 用于存需要删除字段的表名+结构
		Map> removeTableMap = new HashMap>();

		// 用于存需要删除主键的表名+结构
		Map> dropKeyTableMap = new HashMap>();

		// 用于存需要删除唯一约束的表名+结构
		Map> dropUniqueTableMap = new HashMap>();

		// 构建出全部表的增删改的map
		allTableMapConstruct(mySqlTypeAndLengthMap, classes, newTableMap, modifyTableMap, addTableMap, removeTableMap,
				dropKeyTableMap, dropUniqueTableMap);

		// 根据传入的map，分别去创建或修改表结构
		createOrModifyTableConstruct(newTableMap, modifyTableMap, addTableMap, removeTableMap, dropKeyTableMap,
				dropUniqueTableMap);
	}

	/**
	 * 构建出全部表的增删改的map
	 * 
	 * @param mySqlTypeAndLengthMap
	 *            获取Mysql的类型，以及类型需要设置几个长度
	 * @param classes
	 *            从包package中获取所有的Class
	 * @param newTableMap
	 *            用于存需要创建的表名+结构
	 * @param modifyTableMap
	 *            用于存需要更新字段类型等的表名+结构
	 * @param addTableMap
	 *            用于存需要增加字段的表名+结构
	 * @param removeTableMap
	 *            用于存需要删除字段的表名+结构
	 * @param dropKeyTableMap
	 *            用于存需要删除主键的表名+结构
	 * @param dropUniqueTableMap
	 *            用于存需要删除唯一约束的表名+结构
	 */
	private void allTableMapConstruct(Map mySqlTypeAndLengthMap, Set> classes,
			Map> newTableMap, Map> modifyTableMap,
			Map> addTableMap, Map> removeTableMap,
			Map> dropKeyTableMap, Map> dropUniqueTableMap) {
		for (Class clas : classes) {

			Table table = clas.getAnnotation(Table.class);
			// 没有打注解不需要创建变
			if (null == table) {
				continue;
			}

			// 用于存新增表的字段
			List