1   /**
2    * Copyright 2009 The Apache Software Foundation
3    *
4    * Licensed to the Apache Software Foundation (ASF) under one
5    * or more contributor license agreements.  See the NOTICE file
6    * distributed with this work for additional information
7    * regarding copyright ownership.  The ASF licenses this file
8    * to you under the Apache License, Version 2.0 (the
9    * "License"); you may not use this file except in compliance
10   * with the License.  You may obtain a copy of the License at
11   *
12   *     http://www.apache.org/licenses/LICENSE-2.0
13   *
14   * Unless required by applicable law or agreed to in writing, software
15   * distributed under the License is distributed on an "AS IS" BASIS,
16   * WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
17   * See the License for the specific language governing permissions and
18   * limitations under the License.
19   */
20  
21  package org.apache.hadoop.hbase;
22  
23  import static org.junit.Assert.assertEquals;
24  
25  import org.apache.hadoop.conf.Configuration;
26  import org.apache.hadoop.hbase.client.HTable;
27  import org.apache.hadoop.hbase.client.Result;
28  import org.apache.hadoop.hbase.client.ResultScanner;
29  import org.apache.hadoop.hbase.client.Scan;
30  import org.apache.hadoop.hbase.util.Bytes;
31  import org.junit.After;
32  import org.junit.AfterClass;
33  import org.junit.Before;
34  import org.junit.BeforeClass;
35  import org.junit.Test;
36  
37  public class TestFullLogReconstruction {
38  
39    private final static HBaseTestingUtility
40        TEST_UTIL = new HBaseTestingUtility();
41  
42    private final static byte[] TABLE_NAME = Bytes.toBytes("tabletest");
43    private final static byte[] FAMILY = Bytes.toBytes("family");
44  
45    /**
46     * @throws java.lang.Exception
47     */
48    @BeforeClass
49    public static void setUpBeforeClass() throws Exception {
50      Configuration c = TEST_UTIL.getConfiguration();
51      c.setBoolean("dfs.support.append", true);
52      // quicker heartbeat interval for faster DN death notification
53      c.setInt("heartbeat.recheck.interval", 5000);
54      c.setInt("dfs.heartbeat.interval", 1);
55      c.setInt("dfs.socket.timeout", 5000);
56      // faster failover with cluster.shutdown();fs.close() idiom
57      c.setInt("ipc.client.connect.max.retries", 1);
58      c.setInt("dfs.client.block.recovery.retries", 1);
59      TEST_UTIL.startMiniCluster(2);
60    }
61  
62    /**
63     * @throws java.lang.Exception
64     */
65    @AfterClass
66    public static void tearDownAfterClass() throws Exception {
67      TEST_UTIL.shutdownMiniCluster();
68    }
69  
70    /**
71     * @throws java.lang.Exception
72     */
73    @Before
74    public void setUp() throws Exception {
75    }
76  
77    /**
78     * @throws java.lang.Exception
79     */
80    @After
81    public void tearDown() throws Exception {
82    }
83  
84    /**
85     * Test the whole reconstruction loop. Build a table with regions aaa to zzz
86     * and load every one of them multiple times with the same date and do a flush
87     * at some point. Kill one of the region servers and scan the table. We should
88     * see all the rows.
89     * @throws Exception
90     */
91    @Test
92    public void testReconstruction() throws Exception {
93  
94      HTable table = TEST_UTIL.createTable(TABLE_NAME, FAMILY);
95  
96      TEST_UTIL.createMultiRegions(table, Bytes.toBytes("family"));
97  
98      // Load up the table with simple rows and count them
99      int initialCount = TEST_UTIL.loadTable(table, FAMILY);
100     Scan scan = new Scan();
101     ResultScanner results = table.getScanner(scan);
102     int count = 0;
103     for (Result res : results) {
104       count++;
105     }
106     results.close();
107 
108     assertEquals(initialCount, count);
109 
110     for(int i = 0; i < 4; i++) {
111       TEST_UTIL.loadTable(table, FAMILY);
112     }
113 
114     TEST_UTIL.expireRegionServerSession(0);
115     scan = new Scan();
116     results = table.getScanner(scan);
117     int newCount = 0;
118     for (Result res : results) {
119       newCount++;
120     }
121     assertEquals(count, newCount);
122   }
123 }