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

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

import com.gitee.sunchenbin.mybatis.actable.annotation.Index;
import com.gitee.sunchenbin.mybatis.actable.annotation.Unique;
import com.gitee.sunchenbin.mybatis.actable.command.*;
import com.gitee.sunchenbin.mybatis.actable.constants.Constants;
import com.gitee.sunchenbin.mybatis.actable.constants.MySqlCharsetConstant;
import com.gitee.sunchenbin.mybatis.actable.constants.MySqlEngineConstant;
import com.gitee.sunchenbin.mybatis.actable.constants.MySqlTypeConstant;
import com.gitee.sunchenbin.mybatis.actable.dao.system.CreateMysqlTablesMapper;
import com.gitee.sunchenbin.mybatis.actable.manager.util.ConfigurationUtil;
import com.gitee.sunchenbin.mybatis.actable.utils.ClassTools;
import com.gitee.sunchenbin.mybatis.actable.utils.ColumnUtils;
import org.slf4j.Logger;
import org.slf4j.LoggerFactory;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.stereotype.Service;
import org.springframework.transaction.annotation.Transactional;
import org.springframework.util.StringUtils;

import java.lang.reflect.Field;
import java.util.*;
import java.util.Map.Entry;

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

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

	@Autowired
	private CreateMysqlTablesMapper createMysqlTablesMapper;

	@Autowired
	private ConfigurationUtil springContextUtil;

	/**
	 * 要扫描的model所在的pack
	 */
	private static String pack = null;

	/**
	 * 自动创建模式：update表示更新，create表示删除原表重新创建
	 */
	private static String tableAuto = null;

	/**
	 * 读取配置文件的三种状态（创建表、更新表、不做任何事情）
	 */
	public void createMysqlTable() {
		// 读取配置信息
		pack = springContextUtil.getConfig(Constants.MODEL_PACK_KEY);
		tableAuto = springContextUtil.getConfig(Constants.TABLE_AUTO_KEY);

		// 不做任何事情
		if (!"none".equals(tableAuto) && !"update".equals(tableAuto) && !"create".equals(tableAuto) && !"add".equals(tableAuto)) {
			log.warn("配置mybatis.table.auto错误无法识别，当前配置只支持[none/update/create/add]三种类型!");
			return;
		}

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

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

		// 初始化用于存储各种操作表结构的容器
		Map> baseTableMap = initTableMap();

		// 表名集合
		List tableNames = new ArrayList();

		// 循环全部的model
		for (Class clas : classes) {

			// 没有打注解不需要创建表
			if (!ColumnUtils.hasTableAnnotation(clas)) {
				log.warn("{}，没有@Table注解直接跳过", clas.getName());
				continue;
			}
			// 禁止出现重名表
			checkTableName(tableNames, clas);
			// 构建出全部表的增删改的map
			buildTableMapConstruct(clas, baseTableMap);
		}

		// 根据传入的map，分别去创建或修改表结构
		createOrModifyTableConstruct(baseTableMap);
	}

	private void checkTableName(List tableNames, Class clas) {
		String tableName = ColumnUtils.getTableName(clas);
		if (tableNames.contains(tableName)){
			throw new RuntimeException(tableName + "表名出现重复，禁止创建！");
		}
		tableNames.add(tableName);
	}

	/**
	 * 初始化用于存储各种操作表结构的容器
	 *
	 * @return 初始化map
	 */
	private Map> initTableMap() {
		Map> baseTableMap = new HashMap>();
		// 1.用于存需要创建的表名+（字段结构/表信息）
		baseTableMap.put(Constants.NEW_TABLE_MAP, new HashMap());
		// 2.用于存需要更新字段类型等的表名+结构
		baseTableMap.put(Constants.MODIFY_TABLE_MAP, new HashMap());
		// 3.用于存需要增加字段的表名+结构
		baseTableMap.put(Constants.ADD_TABLE_MAP, new HashMap());
		// 4.用于存需要删除字段的表名+结构
		baseTableMap.put(Constants.REMOVE_TABLE_MAP, new HashMap());
		// 5.用于存需要删除主键的表名+结构
		baseTableMap.put(Constants.DROPKEY_TABLE_MAP, new HashMap());
		// 6.用于存需要删除唯一约束的表名+结构
		baseTableMap.put(Constants.DROPINDEXANDUNIQUE_TABLE_MAP, new HashMap());
		// 7.用于存需要增加的索引
		baseTableMap.put(Constants.ADDINDEX_TABLE_MAP, new HashMap());
		// 8.用于存需要增加的唯一约束
		baseTableMap.put(Constants.ADDUNIQUE_TABLE_MAP, new HashMap());
		// 9.更新表注释
		baseTableMap.put(Constants.MODIFY_TABLE_PROPERTY_MAP, new HashMap());
		return baseTableMap;
	}

	/**
	 * 构建出全部表的增删改的map
	 *
	 * @param clas
	 *            package中的model的Class
	 * @param baseTableMap
	 *            用于存储各种操作表结构的容器
	 */
	private void buildTableMapConstruct(Class clas, Map> baseTableMap) {

		// 获取model的tablename
		String tableName = ColumnUtils.getTableName(clas);

		// 获取表注释
		String tableComment = ColumnUtils.getTableComment(clas);

		// 获取表字符集
		MySqlCharsetConstant tableCharset = ColumnUtils.getTableCharset(clas);

		// 获取表引擎
		MySqlEngineConstant tableEngine = ColumnUtils.getTableEngine(clas);

		// 1. 用于存表的全部字段
		List