| /** |
| * 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 org.apache.hadoop.hdfs.server.namenode.ha; |
| |
| import static org.junit.Assert.assertEquals; |
| import static org.junit.Assert.assertFalse; |
| |
| import java.io.File; |
| import java.io.IOException; |
| |
| import org.apache.hadoop.conf.Configuration; |
| import org.apache.hadoop.fs.CommonConfigurationKeysPublic; |
| import org.apache.hadoop.fs.Path; |
| import org.apache.hadoop.hdfs.DistributedFileSystem; |
| import org.apache.hadoop.hdfs.MiniDFSCluster; |
| import org.apache.hadoop.hdfs.qjournal.MiniJournalCluster; |
| import org.apache.hadoop.hdfs.qjournal.MiniQJMHACluster; |
| import org.apache.hadoop.hdfs.server.namenode.FSImage; |
| import org.apache.hadoop.hdfs.server.namenode.FSImageTestUtil; |
| import org.apache.hadoop.hdfs.server.namenode.NNStorage; |
| import org.junit.After; |
| import org.junit.Before; |
| import org.junit.Test; |
| |
| import com.google.common.collect.ImmutableList; |
| import org.mockito.internal.util.reflection.Whitebox; |
| |
| /** |
| * Test BootstrapStandby when QJM is used for shared edits. |
| */ |
| public class TestBootstrapStandbyWithQJM { |
| enum UpgradeState { |
| NORMAL, |
| RECOVER, |
| FORMAT |
| } |
| |
| private MiniDFSCluster cluster; |
| private MiniJournalCluster jCluster; |
| private int nnCount = 3; |
| |
| protected Configuration createConfig() { |
| Configuration conf = new Configuration(); |
| // Turn off IPC client caching, so that the suite can handle |
| // the restart of the daemons between test cases. |
| conf.setInt( |
| CommonConfigurationKeysPublic.IPC_CLIENT_CONNECTION_MAXIDLETIME_KEY, |
| 0); |
| return conf; |
| } |
| |
| @Before |
| public void setup() throws Exception { |
| Configuration conf = createConfig(); |
| |
| MiniQJMHACluster miniQjmHaCluster = |
| new MiniQJMHACluster.Builder(conf).setNumNameNodes(nnCount).build(); |
| cluster = miniQjmHaCluster.getDfsCluster(); |
| jCluster = miniQjmHaCluster.getJournalCluster(); |
| |
| // make nn0 active |
| cluster.transitionToActive(0); |
| // do sth to generate in-progress edit log data |
| DistributedFileSystem dfs = (DistributedFileSystem) |
| HATestUtil.configureFailoverFs(cluster, conf); |
| dfs.mkdirs(new Path("/test2")); |
| dfs.close(); |
| } |
| |
| @After |
| public void cleanup() throws IOException { |
| if (cluster != null) { |
| cluster.shutdown(); |
| cluster = null; |
| } |
| if (jCluster != null) { |
| jCluster.shutdown(); |
| jCluster = null; |
| } |
| } |
| |
| /** BootstrapStandby when the existing NN is standby */ |
| @Test |
| public void testBootstrapStandbyWithStandbyNN() throws Exception { |
| // make the first NN in standby state |
| cluster.transitionToStandby(0); |
| bootstrapStandbys(); |
| } |
| |
| /** BootstrapStandby when the existing NN is active */ |
| @Test |
| public void testBootstrapStandbyWithActiveNN() throws Exception { |
| // make the first NN in active state |
| cluster.transitionToActive(0); |
| bootstrapStandbys(); |
| } |
| |
| private void bootstrapStandbys() throws Exception { |
| // shutdown and bootstrap all the other nns, except the first (start 1, not 0) |
| for (int i = 1; i < nnCount; i++) { |
| Configuration otherNNConf = cluster.getConfiguration(i); |
| |
| // shut down other nn |
| cluster.shutdownNameNode(i); |
| |
| int rc = BootstrapStandby.run(new String[] { "-force" }, otherNNConf); |
| assertEquals(0, rc); |
| |
| // Should have copied over the namespace from the standby |
| FSImageTestUtil.assertNNHasCheckpoints(cluster, i, ImmutableList.of(0)); |
| } |
| FSImageTestUtil.assertNNFilesMatch(cluster); |
| } |
| |
| /** |
| * Test the bootstrapstandby while the other namenode is in upgrade state. |
| * Make sure a previous directory can be created. |
| */ |
| @Test |
| public void testUpgrade() throws Exception { |
| testUpgrade(UpgradeState.NORMAL); |
| } |
| |
| /** |
| * Similar with testUpgrade, but rename nn1's current directory to |
| * previous.tmp before bootstrapStandby, and make sure the nn1 is recovered |
| * first then converted into upgrade state. |
| */ |
| @Test |
| public void testUpgradeWithRecover() throws Exception { |
| testUpgrade(UpgradeState.RECOVER); |
| } |
| |
| /** |
| * Similar with testUpgrade, but rename nn1's current directory to a random |
| * name so that it's not formatted. Make sure the nn1 is formatted and then |
| * converted into upgrade state. |
| */ |
| @Test |
| public void testUpgradeWithFormat() throws Exception { |
| testUpgrade(UpgradeState.FORMAT); |
| } |
| |
| private void testUpgrade(UpgradeState state) throws Exception { |
| cluster.transitionToActive(0); |
| final Configuration confNN1 = cluster.getConfiguration(1); |
| |
| final File current = cluster.getNameNode(1).getFSImage().getStorage() |
| .getStorageDir(0).getCurrentDir(); |
| final File tmp = cluster.getNameNode(1).getFSImage().getStorage() |
| .getStorageDir(0).getPreviousTmp(); |
| // shut down nn1 |
| cluster.shutdownNameNode(1); |
| |
| // make NN0 in upgrade state |
| FSImage fsImage0 = cluster.getNameNode(0).getNamesystem().getFSImage(); |
| Whitebox.setInternalState(fsImage0, "isUpgradeFinalized", false); |
| |
| switch (state) { |
| case RECOVER: |
| // rename the current directory to previous.tmp in nn1 |
| NNStorage.rename(current, tmp); |
| break; |
| case FORMAT: |
| // rename the current directory to a random name so it's not formatted |
| final File wrongPath = new File(current.getParentFile(), "wrong"); |
| NNStorage.rename(current, wrongPath); |
| break; |
| default: |
| break; |
| } |
| |
| int rc = BootstrapStandby.run(new String[] { "-force" }, confNN1); |
| assertEquals(0, rc); |
| |
| // Should have copied over the namespace from the standby |
| FSImageTestUtil.assertNNHasCheckpoints(cluster, 1, |
| ImmutableList.of(0)); |
| FSImageTestUtil.assertNNFilesMatch(cluster); |
| |
| // make sure the NN1 is in upgrade state, i.e., the previous directory has |
| // been successfully created |
| cluster.restartNameNode(1); |
| assertFalse(cluster.getNameNode(1).getNamesystem().isUpgradeFinalized()); |
| } |
| |
| } |