kim.zkp.quick.orm.table.CreateTable Maven / Gradle / Ivy

Go to download
/**
 * Copyright (c) 2017, ZhuKaipeng 朱开鹏 ([email protected]).

 * Licensed to the Apache Software Foundation (ASF) under one or more
 * contributor license agreements.  See the NOTICE file distributed with
 * this work for additional information regarding copyright ownership.
 * The ASF licenses this file to You under the Apache License, Version 2.0
 * (the "License"); you may not use this file except in compliance with
 * the License.  You may obtain a copy of the License at
 *
 *     http://www.apache.org/licenses/LICENSE-2.0
 *
 * Unless required by applicable law or agreed to in writing, software
 * distributed under the License is distributed on an "AS IS" BASIS,
 * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
 * See the License for the specific language governing permissions and
 * limitations under the License.
 */

package kim.zkp.quick.orm.table;

import java.io.File;
import java.util.ArrayList;
import java.util.HashSet;
import java.util.List;
import java.util.Set;
import java.util.regex.Matcher;
import java.util.regex.Pattern;

import com.xiaoleilu.hutool.log.Log;
import com.xiaoleilu.hutool.log.LogFactory;

import kim.zkp.quick.orm.annotation.Table;
import kim.zkp.quick.orm.session.Session;
import kim.zkp.quick.orm.sql.SqlInfo;
import kim.zkp.quick.orm.sql.builder.SqlBuilder;
import kim.zkp.quick.orm.sql.builder.SqlBuilderProcessor;
/**
 * class       :  CreateTable
 * @author     :  zhukaipeng
 * @version    :  1.0  
 * description :  加载Class创建表
 * @see        :  *
 */
public class CreateTable {
	private static final Log log = LogFactory.get();
	private Session session;
	private String packagePath;
	private SqlBuilderProcessor sqlBuilderProcessor;
	
	public CreateTable(Session session,SqlBuilderProcessor sqlBuilderProcessor,String packagePath) {
		super();
		this.session = session;
		this.sqlBuilderProcessor = sqlBuilderProcessor;
		this.packagePath = packagePath;
	}

	public void start(){
		File root = new File(System.getProperty("user.dir"));
		List allFileNames = getAllFileNames(root, "");
		Set fileNames = filterFile(allFileNames,packagePath);
		List> clzzs = loadClass(fileNames);
		createTable(clzzs);
	}
	
	private void createTable(List> clzzs){
		clzzs.forEach(clzz->{
			Table table = clzz.getAnnotation(Table.class);
			if (table == null || !table.create()) {
				return ;
			}
			log.info("准备创建表:{}",table.tableName());
			try {
				SqlInfo sqlInfo = sqlBuilderProcessor.getSql(SqlBuilder.SBType.CREATE_TABLE, clzz);
				log.info("建表SQL:{}",sqlInfo.getSql());
				try {
					session.sqlUpdate(sqlInfo.getSql());
					log.info("创建成功:{}",table.tableName());
				} catch (Exception e) {
					log.info("该表已存在:{}",table.tableName());
				}
			} catch (Exception e) {
				e.printStackTrace();
				log.error(e, "建表出现异常");
			}
		});
	}
	
	private List> loadClass(Set fileNames){
		List> clzzs = new ArrayList<>();
		fileNames.forEach(fileName->{
			try {
				clzzs.add(Thread.currentThread().getContextClassLoader().loadClass(fileName));
			} catch (Exception e) {
				log.error(e,"加载:{}出错",fileName);
			}
		});
		return clzzs;
	}
	
	private Set filterFile(List allFileNames,String packagePath){
		Set fileNames = new HashSet();
		Pattern p = Pattern.compile("("+packagePath+".*).class");
		allFileNames.forEach(fileName->{
			Matcher m = p.matcher(fileName);
			if (m.find()) {
				fileName = m.group(1);
				if (!fileName.contains("$")) {
					fileNames.add(fileName);
				}
			}
		});
		return fileNames;
	}

	private List getAllFileNames(File folder, String packageName){
		List listFile = new ArrayList();
		File[] files = folder.listFiles();
		for (int fileIndex = 0; fileIndex < files.length; fileIndex++) {
			File file = files[fileIndex];
			if (file.isDirectory()) {
				listFile.addAll(getAllFileNames(file, packageName + file.getName() + "."));
			} else {
				if (file.getName().endsWith(".class")) {
					listFile.add(packageName+file.getName());
				}
			}
		}
		return listFile;
	}

}