mirror of
https://github.com/actiontech/dble.git
synced 2026-01-24 13:58:29 -06:00
124 lines
4.0 KiB
Java
124 lines
4.0 KiB
Java
/*
|
|
* Copyright (c) 2013, OpenCloudDB/MyCAT and/or its affiliates. All rights reserved.
|
|
* DO NOT ALTER OR REMOVE COPYRIGHT NOTICES OR THIS FILE HEADER.
|
|
*
|
|
* This code is free software;Designed and Developed mainly by many Chinese
|
|
* opensource volunteers. you can redistribute it and/or modify it under the
|
|
* terms of the GNU General Public License version 2 only, as published by the
|
|
* Free Software Foundation.
|
|
*
|
|
* This code is distributed in the hope that it will be useful, but WITHOUT
|
|
* ANY WARRANTY; without even the implied warranty of MERCHANTABILITY or
|
|
* FITNESS FOR A PARTICULAR PURPOSE. See the GNU General Public License
|
|
* version 2 for more details (a copy is included in the LICENSE file that
|
|
* accompanied this code).
|
|
*
|
|
* You should have received a copy of the GNU General Public License version
|
|
* 2 along with this work; if not, write to the Free Software Foundation,
|
|
* Inc., 51 Franklin St, Fifth Floor, Boston, MA 02110-1301 USA.
|
|
*
|
|
* Any questions about this component can be directed to it's project Web address
|
|
* https://code.google.com/p/opencloudb/.
|
|
*
|
|
*/
|
|
package io.mycat.manager.response;
|
|
|
|
import java.util.Map;
|
|
import java.util.Set;
|
|
import java.util.concurrent.locks.ReentrantLock;
|
|
|
|
import org.slf4j.Logger;
|
|
import org.slf4j.LoggerFactory;
|
|
|
|
import io.mycat.MycatServer;
|
|
import io.mycat.backend.datasource.PhysicalDBNode;
|
|
import io.mycat.backend.datasource.PhysicalDBPool;
|
|
import io.mycat.config.ErrorCode;
|
|
import io.mycat.config.MycatCluster;
|
|
import io.mycat.config.MycatConfig;
|
|
import io.mycat.config.model.ERTable;
|
|
import io.mycat.config.model.FirewallConfig;
|
|
import io.mycat.config.model.SchemaConfig;
|
|
import io.mycat.config.model.UserConfig;
|
|
import io.mycat.manager.ManagerConnection;
|
|
import io.mycat.net.mysql.OkPacket;
|
|
|
|
/**
|
|
* @author mycat
|
|
*/
|
|
public final class RollbackConfig {
|
|
private static final Logger LOGGER = LoggerFactory.getLogger(RollbackConfig.class);
|
|
|
|
public static void execute(ManagerConnection c) {
|
|
final ReentrantLock lock = MycatServer.getInstance().getConfig()
|
|
.getLock();
|
|
lock.lock();
|
|
try {
|
|
if (rollback()) {
|
|
StringBuilder s = new StringBuilder();
|
|
s.append(c).append("Rollback config success by manager");
|
|
LOGGER.warn(s.toString());
|
|
OkPacket ok = new OkPacket();
|
|
ok.packetId = 1;
|
|
ok.affectedRows = 1;
|
|
ok.serverStatus = 2;
|
|
ok.message = "Rollback config success".getBytes();
|
|
ok.write(c);
|
|
} else {
|
|
c.writeErrMessage(ErrorCode.ER_YES, "Rollback config failure");
|
|
}
|
|
} finally {
|
|
lock.unlock();
|
|
}
|
|
}
|
|
|
|
private static boolean rollback() {
|
|
MycatConfig conf = MycatServer.getInstance().getConfig();
|
|
Map<String, UserConfig> users = conf.getBackupUsers();
|
|
Map<String, SchemaConfig> schemas = conf.getBackupSchemas();
|
|
Map<String, PhysicalDBNode> dataNodes = conf.getBackupDataNodes();
|
|
Map<String, PhysicalDBPool> dataHosts = conf.getBackupDataHosts();
|
|
MycatCluster cluster = conf.getBackupCluster();
|
|
FirewallConfig firewall = conf.getBackupFirewall();
|
|
Map<ERTable, Set<ERTable>> erRelations = conf.getBackupErRelations();
|
|
|
|
// 检查可回滚状态
|
|
if (!conf.canRollback()) {
|
|
return false;
|
|
}
|
|
|
|
// 如果回滚已经存在的pool
|
|
boolean rollbackStatus = true;
|
|
Map<String, PhysicalDBPool> cNodes = conf.getDataHosts();
|
|
for (PhysicalDBPool dn : dataHosts.values()) {
|
|
dn.init(dn.getActivedIndex());
|
|
if (!dn.isInitSuccess()) {
|
|
rollbackStatus = false;
|
|
break;
|
|
}
|
|
}
|
|
// 如果回滚不成功，则清理已初始化的资源。
|
|
if (!rollbackStatus) {
|
|
for (PhysicalDBPool dn : dataHosts.values()) {
|
|
dn.clearDataSources("rollbackup config");
|
|
dn.stopHeartbeat();
|
|
}
|
|
return false;
|
|
}
|
|
|
|
// 应用回滚
|
|
conf.rollback(users, schemas, dataNodes, dataHosts, erRelations, cluster, firewall);
|
|
|
|
// 处理旧的资源
|
|
for (PhysicalDBPool dn : cNodes.values()) {
|
|
dn.clearDataSources("clear old config ");
|
|
dn.stopHeartbeat();
|
|
}
|
|
|
|
//清理缓存
|
|
MycatServer.getInstance().getCacheService().clearCache();
|
|
MycatServer.getInstance().reloadMetaData();
|
|
return true;
|
|
}
|
|
|
|
} |