1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20 package org.apache.hadoop.hbase.regionserver;
21
22 import static org.junit.Assert.assertNotNull;
23 import static org.junit.Assert.assertTrue;
24 import static org.junit.Assert.fail;
25
26 import java.io.IOException;
27 import java.io.InputStream;
28 import java.lang.ref.SoftReference;
29 import java.util.ArrayList;
30 import java.util.Collections;
31 import java.util.List;
32
33 import org.apache.commons.logging.Log;
34 import org.apache.commons.logging.LogFactory;
35 import org.apache.hadoop.fs.FSDataInputStream;
36 import org.apache.hadoop.fs.FileSystem;
37 import org.apache.hadoop.fs.FilterFileSystem;
38 import org.apache.hadoop.fs.Path;
39 import org.apache.hadoop.fs.PositionedReadable;
40 import org.apache.hadoop.hbase.HBaseTestingUtility;
41 import org.apache.hadoop.hbase.HColumnDescriptor;
42 import org.apache.hadoop.hbase.HConstants;
43 import org.apache.hadoop.hbase.HTableDescriptor;
44 import org.apache.hadoop.hbase.KeyValue;
45 import org.apache.hadoop.hbase.client.HBaseAdmin;
46 import org.apache.hadoop.hbase.client.HTable;
47 import org.apache.hadoop.hbase.io.hfile.HFile;
48 import org.apache.hadoop.hbase.io.hfile.HFileScanner;
49 import org.apache.hadoop.hbase.regionserver.StoreFile.BloomType;
50 import org.apache.hadoop.hbase.util.Bytes;
51 import org.junit.Test;
52
53
54
55
56
57
58 public class TestFSErrorsExposed {
59 private static final Log LOG = LogFactory.getLog(TestFSErrorsExposed.class);
60
61 HBaseTestingUtility util = new HBaseTestingUtility();
62
63
64
65
66
67 @Test
68 public void testHFileScannerThrowsErrors() throws IOException {
69 Path hfilePath = new Path(new Path(
70 HBaseTestingUtility.getTestDir("internalScannerExposesErrors"),
71 "regionname"), "familyname");
72 FaultyFileSystem fs = new FaultyFileSystem(util.getTestFileSystem());
73 StoreFile.Writer writer = StoreFile.createWriter(fs, hfilePath, 2*1024);
74 TestStoreFile.writeStoreFile(
75 writer, Bytes.toBytes("cf"), Bytes.toBytes("qual"));
76
77 StoreFile sf = new StoreFile(fs, writer.getPath(), false,
78 util.getConfiguration(), StoreFile.BloomType.NONE, false);
79 StoreFile.Reader reader = sf.createReader();
80 HFileScanner scanner = reader.getScanner(false, true);
81
82 FaultyInputStream inStream = fs.inStreams.get(0).get();
83 assertNotNull(inStream);
84
85 scanner.seekTo();
86
87 assertTrue(scanner.next());
88
89 inStream.startFaults();
90
91 try {
92 int scanned=0;
93 while (scanner.next()) {
94 scanned++;
95 }
96 fail("Scanner didn't throw after faults injected");
97 } catch (IOException ioe) {
98 LOG.info("Got expected exception", ioe);
99 assertTrue(ioe.getMessage().contains("Fault"));
100 }
101 reader.close();
102 }
103
104
105
106
107
108 @Test
109 public void testStoreFileScannerThrowsErrors() throws IOException {
110 Path hfilePath = new Path(new Path(
111 HBaseTestingUtility.getTestDir("internalScannerExposesErrors"),
112 "regionname"), "familyname");
113 FaultyFileSystem fs = new FaultyFileSystem(util.getTestFileSystem());
114 StoreFile.Writer writer = StoreFile.createWriter(fs, hfilePath, 2 * 1024);
115 TestStoreFile.writeStoreFile(
116 writer, Bytes.toBytes("cf"), Bytes.toBytes("qual"));
117
118 StoreFile sf = new StoreFile(fs, writer.getPath(), false,
119 util.getConfiguration(), BloomType.NONE, false);
120 List<StoreFileScanner> scanners = StoreFileScanner.getScannersForStoreFiles(
121 Collections.singletonList(sf), false, true);
122 KeyValueScanner scanner = scanners.get(0);
123
124 FaultyInputStream inStream = fs.inStreams.get(0).get();
125 assertNotNull(inStream);
126
127 scanner.seek(KeyValue.LOWESTKEY);
128
129 assertNotNull(scanner.next());
130
131 inStream.startFaults();
132
133 try {
134 int scanned=0;
135 while (scanner.next() != null) {
136 scanned++;
137 }
138 fail("Scanner didn't throw after faults injected");
139 } catch (IOException ioe) {
140 LOG.info("Got expected exception", ioe);
141 assertTrue(ioe.getMessage().contains("Could not iterate"));
142 }
143 scanner.close();
144 }
145
146
147
148
149
150
151 @Test
152 public void testFullSystemBubblesFSErrors() throws Exception {
153 try {
154 util.startMiniCluster(1);
155 byte[] tableName = Bytes.toBytes("table");
156 byte[] fam = Bytes.toBytes("fam");
157
158 HBaseAdmin admin = new HBaseAdmin(util.getConfiguration());
159 HTableDescriptor desc = new HTableDescriptor(tableName);
160 desc.addFamily(new HColumnDescriptor(
161 fam, 1, HColumnDescriptor.DEFAULT_COMPRESSION,
162 false, false, HConstants.FOREVER, "NONE"));
163 admin.createTable(desc);
164
165 HTable table = new HTable(tableName);
166
167
168 util.loadTable(table, fam);
169 table.flushCommits();
170 util.flush();
171 util.countRows(table);
172
173
174 util.getDFSCluster().shutdownDataNodes();
175
176 try {
177 util.countRows(table);
178 fail("Did not fail to count after removing data");
179 } catch (Exception e) {
180 LOG.info("Got expected error", e);
181 assertTrue(e.getMessage().contains("Could not seek"));
182 }
183
184 } finally {
185 util.shutdownMiniCluster();
186 }
187 }
188
189 static class FaultyFileSystem extends FilterFileSystem {
190 List<SoftReference<FaultyInputStream>> inStreams =
191 new ArrayList<SoftReference<FaultyInputStream>>();
192
193 public FaultyFileSystem(FileSystem testFileSystem) {
194 super(testFileSystem);
195 }
196
197 @Override
198 public FSDataInputStream open(Path p, int bufferSize) throws IOException {
199 FSDataInputStream orig = fs.open(p, bufferSize);
200 FaultyInputStream faulty = new FaultyInputStream(orig);
201 inStreams.add(new SoftReference<FaultyInputStream>(faulty));
202 return faulty;
203 }
204 }
205
206 static class FaultyInputStream extends FSDataInputStream {
207 boolean faultsStarted = false;
208
209 public FaultyInputStream(InputStream in) throws IOException {
210 super(in);
211 }
212
213 public void startFaults() {
214 faultsStarted = true;
215 }
216
217 public int read(long position, byte[] buffer, int offset, int length)
218 throws IOException {
219 injectFault();
220 return ((PositionedReadable)in).read(position, buffer, offset, length);
221 }
222
223 private void injectFault() throws IOException {
224 if (faultsStarted) {
225 throw new IOException("Fault injected");
226 }
227 }
228 }
229
230
231 }