mysql表结构同步

同步两张表之间的表结构。 模仿navict中同步数据库功能。 适合于开发和测试环境之间同步表结构使用 ,修改表结构会导致数据丢失。
生产环境慎用,单纯添加表、字段、索引不受影响。

执行后生成的部分sql

SET FOREIGN_KEY_CHECKS = 0;
ALTER TABLE `portal_user` ADD COLUMN `address_3` varchar(32)  NOT NULL after `address_2`;
ALTER TABLE `portal_user` ADD INDEX `index_address2` USING BTREE (`address_2`) comment '';
SET FOREIGN_KEY_CHECKS = 1;

SET FOREIGN_KEY_CHECKS = 0;
ALTER TABLE `portal_user` DROP INDEX `index_address2`;
ALTER TABLE `portal_user` CHANGE COLUMN `address_3` `address_3` varchar(32) CHARACTER SET utf8 NOT NULL after `user_age`;
ALTER TABLE `portal_user` CHANGE COLUMN `address_4` `address_4` varchar(32) CHARACTER SET utf8 NOT NULL DEFAULT '4' COMMENT 'address_4' after `address_3`;
ALTER TABLE `portal_user` ADD COLUMN `address_22` int(11) NOT NULL after `password`;
ALTER TABLE `portal_user` DROP COLUMN `address_2`;
ALTER TABLE `portal_user` ADD INDEX `index_address2` USING BTREE (`address_22`) comment '';
SET FOREIGN_KEY_CHECKS = 1;

基本原理
1、获取两个库所有表信息、字段信息和索引信息 -> 2、对比差异 -> 3、生成sql

部分代码

/*
 * Copyright (C) 2016 alchemystar, Inc. All Rights Reserved.
 */
package com.eblly.util;

import com.eblly.meta.Column;
import com.eblly.meta.Key;
import com.eblly.meta.MetaData;
import com.eblly.meta.Table;
import com.jfinal.kit.StrKit;
import lombok.Getter;
import lombok.Setter;

import java.util.Deque;
import java.util.Map;
import java.util.concurrent.LinkedBlockingDeque;

/**
 * @Author eblly
 */
@Getter
@Setter
public class CompareUtil {

    private MetaData sourceMetaData;
    private MetaData targetMetaData;

    private Deque<String> changeSqlDeque = new LinkedBlockingDeque<>();

    /**
     * @param source
     * @param target
     */
    public CompareUtil(MetaData source, MetaData target) {
        this.sourceMetaData = source;
        this.targetMetaData = target;
    }

    public void compare() {
        compareTables_Columns_Keys();
//        compareKeys();
    }


    // ================================= table ===============================================

    /**
     * 对比表和表中的字段
     */
    private void compareTables_Columns_Keys() {
        // 删除废弃表
        for (Table targetTable : targetMetaData.getTableMap().values()) {
            if (sourceMetaData.getTableMap().get(targetTable.getTableName()) == null) {
                // 如果对应的source没有这张表,直接drop Table

                String dropTableSql = "DROP TABLE " + addQuotes(targetTable.getTableName()) + " ;";
                changeSqlDeque.addLast(dropTableSql);
            }
        }

        /**
         * 创建表 & 对比字段 & 对比key
         */
        for (Table table : sourceMetaData.getTableMap().values()) {
            if (targetMetaData.getTableMap().get(table.getTableName()) == null) {
                // 如果对应的target没有这张表,直接把create Table拿出
                changeSqlDeque.addLast(table.getCreateTable() + ";");
            } else {

                Table tableTarget = targetMetaData.getTableMap().get(table.getTableName());
                // 这样就需要比较两者的字段
                compareColumns(table, tableTarget);

                // 对比两个表的keys
                compareKeys(table, tableTarget);
            }
        }
    }


    // ================================= column ===============================================

    /**
     * 对比两张表的字段
     *
     * @param tableSource
     * @param tableTarget
     */
    private void compareColumns(Table tableSource, Table tableTarget) {
        // 移除废弃字段
        for (Column column : tableTarget.getColumnMap().values()) {
            if (tableSource.getColumnMap().get(column.getColumnName()) == null) {
                String sql =
                        "alter table `" + tableTarget.getTableName() + "` DROP COLUMN `" + column.getColumnName() + "`; ";
                changeSqlDeque.addLast(sql);
            }
        }

        String afterColumn = null;
        StringBuilder sqlStr;
        for (Column columnSource : tableSource.getColumnMap().values()) {
            sqlStr = new StringBuilder();

            String columnName = columnSource.getColumnName();
            Column columnTarget = tableTarget.getColumnMap().get(columnSource.getColumnName());

            if (columnTarget == null) {
                // ALTER TABLE `portal_user` ADD COLUMN `address_4` varchar(32)  NOT NULL DEFAULT '4' COMMENT 'address_4' after `address_3`;
                // 字段不存在 ,直接添加
                sqlStr.append("ALTER TABLE ")
                      .append("`")
                      .append(tableSource.getTableName())
                      .append("`");

                sqlStr.append(" ADD COLUMN ")
                      .append("`")
                      .append(columnName)
                      .append("` ");

                sqlStr.append(appendColumnTail(columnSource, afterColumn));
                changeSqlDeque.addLast(sqlStr.toString());
            } else {
                // 比较名字、顺序、是否允许为空、类型、默认值、注释等
                if (columnSource.equals(columnTarget)) {
                    continue;
                } else {
                    sqlStr.append("ALTER TABLE ")
                          .append("`")
                          .append(tableSource.getTableName())
                          .append("`");

                    //  修改字段
                    // ALTER TABLE `portal_user` CHANGE COLUMN `address_3` `address_3` varchar(32) NOT NULL after `user_age`;
                    sqlStr.append(" CHANGE COLUMN ")
                          .append("`")
                          .append(columnName)
                          .append("` ")
                          .append("`")
                          .append(columnName)
                          .append("` ");

                    sqlStr.append(appendColumnTail(columnSource, afterColumn));
                    changeSqlDeque.addLast(sqlStr.toString());
                }
            }

            afterColumn = columnName;
        }


    }

    /**
     * @param columnSource
     * @param afterColumn
     * @return
     */
    private StringBuilder appendColumnTail(Column columnSource, String afterColumn) {
        StringBuilder sqlStr = new StringBuilder();

        sqlStr.append(" ")
              .append(columnSource.getColumnType())
              .append(" ");

        if (columnSource.getIsNull().equals("NO")) {
            sqlStr.append(" NOT NULL ");
        } else {
            sqlStr.append(" NULL ");
        }

        if (columnSource.getExtra().toUpperCase().startsWith("AUTO_INCREMENT")) {
            sqlStr.append(" AUTO_INCREMENT ");
        }
        if (columnSource.getColumnDefault() != null) {
            sqlStr.append(" DEFAULT ")
                  .append(columnSource.getColumnDefault().equals("") ? "''" : columnSource.getColumnDefault())
                  .append(" ");
        }
        if (columnSource.getComment() != null) {
            sqlStr.append(" COMMENT ")
                  .append("'")
                  .append(columnSource.getComment())
                  .append("' ");
        }
        if (StrKit.notBlank(afterColumn)) {
            sqlStr.append(" after `")
                  .append(afterColumn)
                  .append("`");
        }

        sqlStr.append(";");

        return sqlStr;
    }

    // ==================================== index ============================================

    /**
     * @param tableSource
     * @param tableTarget
     */
    private void compareKeys(Table tableSource, Table tableTarget) {

        // ALTER TABLE `portal_user` DROP INDEX `index_userAge`;
        // ALTER TABLE `portal_user` ADD INDEX `index_userAge` USING BTREE (`user_age`,`user_name`) comment
        // 'index_userage';
        //ALTER TABLE `portal_user` ADD INDEX `index_address3` USING BTREE (`address_3`) comment 'index3';

        // 删除废弃 key
        for (Map.Entry<String, Key> keyEntry : tableTarget.getKeyMap().entrySet()) {
            String keyName = keyEntry.getKey();

            Key keySource = tableSource.getKeyMap().get(keyName);
            if (keySource == null) {

                String sql = "ALTER TABLE `" + tableTarget.getTableName() + "` DROP INDEX `" + keyName + "`;";
                changeSqlDeque.addFirst(sql);
            }
        }

        StringBuilder sqlStr;
        for (Map.Entry<String, Key> keyEntry : tableSource.getKeyMap().entrySet()) {
            sqlStr = new StringBuilder();

            String keyName = keyEntry.getKey();
            Key key = keyEntry.getValue();
            Key keyTarget = tableTarget.getKeyMap().get(keyName);

            if (keyTarget == null) {
                sqlStr.append(appendKey(tableSource, key));
                changeSqlDeque.addLast(sqlStr.toString());
            } else {
                if (key.equals(keyTarget)) {
                    continue;
                } else {
                    String sql = "ALTER TABLE `" + tableTarget.getTableName() + "` DROP INDEX `" + keyName + "`;";
                    changeSqlDeque.addFirst(sql);

                    sqlStr.append(appendKey(tableSource, key));
                    changeSqlDeque.addLast(sqlStr.toString());
                }
            }
        }
    }

    /**
     * @param tableSource
     * @param key
     * @return
     */
    private String appendKey(Table tableSource, Key key) {
        StringBuilder sqlStr = new StringBuilder();

        sqlStr.append("ALTER TABLE `")
              .append(tableSource.getTableName())
              .append("` ADD INDEX `")
              .append(key.getIndexName())
              .append("` USING ")
              .append(key.getIndexType())
              .append(" (");

        StringBuilder columnStr = new StringBuilder();
        for (String columnName : key.getColumnList()) {
            columnStr.append("`")
                     .append(columnName)
                     .append("`,");
        }

        // 去除最后一个","
        sqlStr.append(columnStr.toString(), 0, columnStr.toString().length() - 1);

        sqlStr.append(")");

        if (StrKit.notBlank(key.getIndexComment())) {
            sqlStr.append(" comment '")
                  .append(key.getIndexComment())
                  .append("'");
        }

        sqlStr.append(";");
        return sqlStr.toString();
    }

    /**
     * @param name
     * @return
     */
    public static String addQuotes(String name) {
        return " `" + name + "`  ";
    }
}

代码
https://gitee.com/vimda2/mysqlSync

最后编辑于
©著作权归作者所有,转载或内容合作请联系作者
  • 序言:七十年代末,一起剥皮案震惊了整个滨河市,随后出现的几起案子,更是在滨河造成了极大的恐慌,老刑警刘岩,带你破解...
    沈念sama阅读 213,014评论 6 492
  • 序言:滨河连续发生了三起死亡事件,死亡现场离奇诡异,居然都是意外死亡,警方通过查阅死者的电脑和手机,发现死者居然都...
    沈念sama阅读 90,796评论 3 386
  • 文/潘晓璐 我一进店门,熙熙楼的掌柜王于贵愁眉苦脸地迎上来,“玉大人,你说我怎么就摊上这事。” “怎么了?”我有些...
    开封第一讲书人阅读 158,484评论 0 348
  • 文/不坏的土叔 我叫张陵,是天一观的道长。 经常有香客问我,道长,这世上最难降的妖魔是什么? 我笑而不...
    开封第一讲书人阅读 56,830评论 1 285
  • 正文 为了忘掉前任,我火速办了婚礼,结果婚礼上,老公的妹妹穿的比我还像新娘。我一直安慰自己,他们只是感情好,可当我...
    茶点故事阅读 65,946评论 6 386
  • 文/花漫 我一把揭开白布。 她就那样静静地躺着,像睡着了一般。 火红的嫁衣衬着肌肤如雪。 梳的纹丝不乱的头发上,一...
    开封第一讲书人阅读 50,114评论 1 292
  • 那天,我揣着相机与录音,去河边找鬼。 笑死,一个胖子当着我的面吹牛,可吹牛的内容都是我干的。 我是一名探鬼主播,决...
    沈念sama阅读 39,182评论 3 412
  • 文/苍兰香墨 我猛地睁开眼,长吁一口气:“原来是场噩梦啊……” “哼!你这毒妇竟也来了?” 一声冷哼从身侧响起,我...
    开封第一讲书人阅读 37,927评论 0 268
  • 序言:老挝万荣一对情侣失踪,失踪者是张志新(化名)和其女友刘颖,没想到半个月后,有当地人在树林里发现了一具尸体,经...
    沈念sama阅读 44,369评论 1 303
  • 正文 独居荒郊野岭守林人离奇死亡,尸身上长有42处带血的脓包…… 初始之章·张勋 以下内容为张勋视角 年9月15日...
    茶点故事阅读 36,678评论 2 327
  • 正文 我和宋清朗相恋三年,在试婚纱的时候发现自己被绿了。 大学时的朋友给我发了我未婚夫和他白月光在一起吃饭的照片。...
    茶点故事阅读 38,832评论 1 341
  • 序言:一个原本活蹦乱跳的男人离奇死亡,死状恐怖,灵堂内的尸体忽然破棺而出,到底是诈尸还是另有隐情,我是刑警宁泽,带...
    沈念sama阅读 34,533评论 4 335
  • 正文 年R本政府宣布,位于F岛的核电站,受9级特大地震影响,放射性物质发生泄漏。R本人自食恶果不足惜,却给世界环境...
    茶点故事阅读 40,166评论 3 317
  • 文/蒙蒙 一、第九天 我趴在偏房一处隐蔽的房顶上张望。 院中可真热闹,春花似锦、人声如沸。这庄子的主人今日做“春日...
    开封第一讲书人阅读 30,885评论 0 21
  • 文/苍兰香墨 我抬头看了看天上的太阳。三九已至,却和暖如春,着一层夹袄步出监牢的瞬间,已是汗流浃背。 一阵脚步声响...
    开封第一讲书人阅读 32,128评论 1 267
  • 我被黑心中介骗来泰国打工, 没想到刚下飞机就差点儿被人妖公主榨干…… 1. 我叫王不留,地道东北人。 一个月前我还...
    沈念sama阅读 46,659评论 2 362
  • 正文 我出身青楼,却偏偏与公主长得像,于是被迫代替她去往敌国和亲。 传闻我的和亲对象是个残疾皇子,可洞房花烛夜当晚...
    茶点故事阅读 43,738评论 2 351