Skip to content

Commit

Permalink
HDFS-8378. Erasure Coding: Few improvements for the erasure coding wo…
Browse files Browse the repository at this point in the history
…rker. Contributed by Rakesh R.
  • Loading branch information
Walter Su authored and Zhe Zhang committed May 26, 2015
1 parent 8c95673 commit 9145809
Show file tree
Hide file tree
Showing 3 changed files with 35 additions and 30 deletions.
5 changes: 5 additions & 0 deletions hadoop-hdfs-project/hadoop-hdfs/CHANGES-HDFS-EC-7285.txt
Expand Up @@ -218,6 +218,8 @@

HDFS-8367. BlockInfoStriped uses EC schema. (Kai Sasaki via Kai Zheng)

HDFS-8352. Erasure Coding: test webhdfs read write stripe file. (waltersu4549)

HDFS-8417. Erasure Coding: Pread failed to read data starting from not-first stripe.
(Walter Su via jing9)

Expand All @@ -228,3 +230,6 @@

HDFS-8366. Erasure Coding: Make the timeout parameter of polling blocking queue
configurable in DFSStripedOutputStream. (Li Bo)

HDFS-8378. Erasure Coding: Few improvements for the erasure coding worker.
(Rakesh R via waltersu4549)
Expand Up @@ -728,6 +728,7 @@ assert getBlockPoolId().equals(bp) :
LOG.info("DatanodeCommand action: DNA_ERASURE_CODING_RECOVERY");
Collection<BlockECRecoveryInfo> ecTasks = ((BlockECRecoveryCommand) cmd).getECTasks();
dn.getErasureCodingWorker().processErasureCodingTasks(ecTasks);
break;
default:
LOG.warn("Unknown DatanodeCommand action: " + cmd.getAction());
}
Expand Down
Expand Up @@ -62,7 +62,6 @@
import org.apache.hadoop.hdfs.protocol.datatransfer.Sender;
import org.apache.hadoop.hdfs.protocol.datatransfer.sasl.DataEncryptionKeyFactory;
import org.apache.hadoop.hdfs.security.token.block.BlockTokenIdentifier;
import org.apache.hadoop.hdfs.server.common.HdfsServerConstants;
import org.apache.hadoop.hdfs.server.datanode.CachingStrategy;
import org.apache.hadoop.hdfs.server.datanode.DataNode;
import org.apache.hadoop.hdfs.server.protocol.BlockECRecoveryCommand.BlockECRecoveryInfo;
Expand All @@ -88,12 +87,12 @@
* commands.
*/
public final class ErasureCodingWorker {
private final Log LOG = DataNode.LOG;
private static final Log LOG = DataNode.LOG;

private final DataNode datanode;
private Configuration conf;
private final Configuration conf;

private ThreadPoolExecutor STRIPED_READ_TRHEAD_POOL;
private ThreadPoolExecutor STRIPED_READ_THREAD_POOL;
private final int STRIPED_READ_THRESHOLD_MILLIS;
private final int STRIPED_READ_BUFFER_SIZE;

Expand Down Expand Up @@ -121,7 +120,10 @@ private RawErasureDecoder newDecoder() {
}

private void initializeStripedReadThreadPool(int num) {
STRIPED_READ_TRHEAD_POOL = new ThreadPoolExecutor(1, num, 60,
if (LOG.isDebugEnabled()) {
LOG.debug("Using striped reads; pool threads=" + num);
}
STRIPED_READ_THREAD_POOL = new ThreadPoolExecutor(1, num, 60,
TimeUnit.SECONDS, new SynchronousQueue<Runnable>(),
new Daemon.DaemonFactory() {
private final AtomicInteger threadIndex = new AtomicInteger(0);
Expand All @@ -141,7 +143,7 @@ public void rejectedExecution(Runnable runnable, ThreadPoolExecutor e) {
super.rejectedExecution(runnable, e);
}
});
STRIPED_READ_TRHEAD_POOL.allowCoreThreadTimeOut(true);
STRIPED_READ_THREAD_POOL.allowCoreThreadTimeOut(true);
}

/**
Expand Down Expand Up @@ -231,23 +233,23 @@ private class ReconstructAndTransferBlock implements Runnable {

// sources
private final short[] liveIndices;
private DatanodeInfo[] sources;
private final DatanodeInfo[] sources;

private List<StripedReader> stripedReaders;
private final List<StripedReader> stripedReaders;

// targets
private DatanodeInfo[] targets;
private StorageType[] targetStorageTypes;
private final DatanodeInfo[] targets;
private final StorageType[] targetStorageTypes;

private short[] targetIndices;
private ByteBuffer[] targetBuffers;
private final short[] targetIndices;
private final ByteBuffer[] targetBuffers;

private Socket[] targetSockets;
private DataOutputStream[] targetOutputStreams;
private DataInputStream[] targetInputStreams;
private final Socket[] targetSockets;
private final DataOutputStream[] targetOutputStreams;
private final DataInputStream[] targetInputStreams;

private long[] blockOffset4Targets;
private long[] seqNo4Targets;
private final long[] blockOffset4Targets;
private final long[] seqNo4Targets;

private final int WRITE_PACKET_SIZE = 64 * 1024;
private DataChecksum checksum;
Expand All @@ -257,11 +259,11 @@ private class ReconstructAndTransferBlock implements Runnable {
private int bytesPerChecksum;
private int checksumSize;

private CachingStrategy cachingStrategy;
private final CachingStrategy cachingStrategy;

private Map<Future<Void>, Integer> futures = new HashMap<>();
private CompletionService<Void> readService =
new ExecutorCompletionService<>(STRIPED_READ_TRHEAD_POOL);
private final Map<Future<Void>, Integer> futures = new HashMap<>();
private final CompletionService<Void> readService =
new ExecutorCompletionService<>(STRIPED_READ_THREAD_POOL);

ReconstructAndTransferBlock(BlockECRecoveryInfo recoveryInfo) {
ECSchema schema = recoveryInfo.getECSchema();
Expand All @@ -277,7 +279,8 @@ private class ReconstructAndTransferBlock implements Runnable {

Preconditions.checkArgument(liveIndices.length >= dataBlkNum,
"No enough live striped blocks.");
Preconditions.checkArgument(liveIndices.length == sources.length);
Preconditions.checkArgument(liveIndices.length == sources.length,
"liveBlockIndices and source dns should match");

targets = recoveryInfo.getTargetDnInfos();
targetStorageTypes = recoveryInfo.getTargetStorageTypes();
Expand Down Expand Up @@ -336,7 +339,6 @@ public void run() {
if (nsuccess < dataBlkNum) {
String error = "Can't find minimum sources required by "
+ "recovery, block id: " + blockGroup.getBlockId();
LOG.warn(error);
throw new IOException(error);
}

Expand All @@ -358,7 +360,6 @@ public void run() {
boolean[] targetsStatus = new boolean[targets.length];
if (initTargetStreams(targetsStatus) == 0) {
String error = "All targets are failed.";
LOG.warn(error);
throw new IOException(error);
}

Expand All @@ -372,7 +373,6 @@ public void run() {
if (nsuccess < dataBlkNum) {
String error = "Can't read data from minimum number of sources "
+ "required by recovery, block id: " + blockGroup.getBlockId();
LOG.warn(error);
throw new IOException(error);
}

Expand All @@ -385,7 +385,6 @@ public void run() {
// step3: transfer data
if (transferData2Targets(targetsStatus) == 0) {
String error = "Transfer failed for all targets.";
LOG.warn(error);
throw new IOException(error);
}

Expand Down Expand Up @@ -906,11 +905,11 @@ private int initTargetStreams(boolean[] targetsStatus) {
}

private class StripedReader {
short index;
BlockReader blockReader;
ByteBuffer buffer;
private final short index;
private BlockReader blockReader;
private ByteBuffer buffer;

public StripedReader(short index) {
private StripedReader(short index) {
this.index = index;
}
}
Expand Down

0 comments on commit 9145809

Please sign in to comment.