nezha-nz-transfer/src/main/java/net/geedge/handler/AfterHandler.java

package net.geedge.handler;

import cn.hutool.log.Log;
import net.geedge.dao.SqlDao;
import org.springframework.beans.factory.annotation.Autowired;
import org.springframework.boot.CommandLineRunner;
import org.springframework.core.annotation.Order;
import org.springframework.stereotype.Component;

import java.util.List;

/**
 * 此类用于数据同步完成后 清除无用表
 * @author admin
 *
 */
@Component
@Order(3)
public class AfterHandler implements CommandLineRunner {

	private Log log = Log.get(); 
	
	@Autowired
	private SqlDao sqlDao;

	@Override
	public void run(String... args) throws Exception {

		String dropColumnSql = "ALTER TABLE `monitor_module` DROP COLUMN `type`;"
				+ "ALTER TABLE `monitor_endpoint` DROP COLUMN `configs`;"
				+ "ALTER TABLE `monitor_endpoint` DROP COLUMN `hash`;"
				+ "ALTER TABLE `monitor_endpoint` DROP COLUMN `enabled`;";
		sqlDao.execute(dropColumnSql);
		log.info("remove column successful");


		StringBuilder sb = new StringBuilder();
		List<String> sqls = sqlDao.allRemoveCopyTables();
		for (String sql : sqls) {
			sb.append(sql);
		}
		sqlDao.execute(sb.toString());
		log.info("remove table successful");
		log.info("-----------------transfer data success-----------------------");

	}

}