HADOOP-15499. Performance severe drops when running RawErasureCoderBenchmark with NativeRSRawErasureCoder. Contributed by Sammi Chen.

This commit is contained in:
Sammi Chen 2018-06-11 13:53:37 +08:00
parent ccfb816d39
commit 18201b882a
7 changed files with 128 additions and 56 deletions

View File

@ -25,6 +25,7 @@
import java.io.IOException; import java.io.IOException;
import java.nio.ByteBuffer; import java.nio.ByteBuffer;
import java.util.concurrent.locks.ReentrantReadWriteLock;
/** /**
* Abstract native raw decoder for all native coders to extend with. * Abstract native raw decoder for all native coders to extend with.
@ -34,13 +35,20 @@ abstract class AbstractNativeRawDecoder extends RawErasureDecoder {
public static Logger LOG = public static Logger LOG =
LoggerFactory.getLogger(AbstractNativeRawDecoder.class); LoggerFactory.getLogger(AbstractNativeRawDecoder.class);
// Protect ISA-L coder data structure in native layer from being accessed and
// updated concurrently by the init, release and decode functions.
protected final ReentrantReadWriteLock decoderLock =
new ReentrantReadWriteLock();
public AbstractNativeRawDecoder(ErasureCoderOptions coderOptions) { public AbstractNativeRawDecoder(ErasureCoderOptions coderOptions) {
super(coderOptions); super(coderOptions);
} }
@Override @Override
protected synchronized void doDecode(ByteBufferDecodingState decodingState) protected void doDecode(ByteBufferDecodingState decodingState)
throws IOException { throws IOException {
decoderLock.readLock().lock();
try {
if (nativeCoder == 0) { if (nativeCoder == 0) {
throw new IOException(String.format("%s closed", throw new IOException(String.format("%s closed",
getClass().getSimpleName())); getClass().getSimpleName()));
@ -64,6 +72,9 @@ protected synchronized void doDecode(ByteBufferDecodingState decodingState)
performDecodeImpl(decodingState.inputs, inputOffsets, performDecodeImpl(decodingState.inputs, inputOffsets,
decodingState.decodeLength, decodingState.erasedIndexes, decodingState.decodeLength, decodingState.erasedIndexes,
decodingState.outputs, outputOffsets); decodingState.outputs, outputOffsets);
} finally {
decoderLock.readLock().unlock();
}
} }
protected abstract void performDecodeImpl(ByteBuffer[] inputs, protected abstract void performDecodeImpl(ByteBuffer[] inputs,

View File

@ -25,6 +25,7 @@
import java.io.IOException; import java.io.IOException;
import java.nio.ByteBuffer; import java.nio.ByteBuffer;
import java.util.concurrent.locks.ReentrantReadWriteLock;
/** /**
* Abstract native raw encoder for all native coders to extend with. * Abstract native raw encoder for all native coders to extend with.
@ -34,13 +35,20 @@ abstract class AbstractNativeRawEncoder extends RawErasureEncoder {
public static Logger LOG = public static Logger LOG =
LoggerFactory.getLogger(AbstractNativeRawEncoder.class); LoggerFactory.getLogger(AbstractNativeRawEncoder.class);
// Protect ISA-L coder data structure in native layer from being accessed and
// updated concurrently by the init, release and encode functions.
protected final ReentrantReadWriteLock encoderLock =
new ReentrantReadWriteLock();
public AbstractNativeRawEncoder(ErasureCoderOptions coderOptions) { public AbstractNativeRawEncoder(ErasureCoderOptions coderOptions) {
super(coderOptions); super(coderOptions);
} }
@Override @Override
protected synchronized void doEncode(ByteBufferEncodingState encodingState) protected void doEncode(ByteBufferEncodingState encodingState)
throws IOException { throws IOException {
encoderLock.readLock().lock();
try {
if (nativeCoder == 0) { if (nativeCoder == 0) {
throw new IOException(String.format("%s closed", throw new IOException(String.format("%s closed",
getClass().getSimpleName())); getClass().getSimpleName()));
@ -62,6 +70,9 @@ protected synchronized void doEncode(ByteBufferEncodingState encodingState)
performEncodeImpl(encodingState.inputs, inputOffsets, dataLen, performEncodeImpl(encodingState.inputs, inputOffsets, dataLen,
encodingState.outputs, outputOffsets); encodingState.outputs, outputOffsets);
} finally {
encoderLock.readLock().unlock();
}
} }
protected abstract void performEncodeImpl( protected abstract void performEncodeImpl(

View File

@ -36,19 +36,30 @@ public class NativeRSRawDecoder extends AbstractNativeRawDecoder {
public NativeRSRawDecoder(ErasureCoderOptions coderOptions) { public NativeRSRawDecoder(ErasureCoderOptions coderOptions) {
super(coderOptions); super(coderOptions);
initImpl(coderOptions.getNumDataUnits(), coderOptions.getNumParityUnits()); decoderLock.writeLock().lock();
try {
initImpl(coderOptions.getNumDataUnits(),
coderOptions.getNumParityUnits());
} finally {
decoderLock.writeLock().unlock();
}
} }
@Override @Override
protected synchronized void performDecodeImpl( protected void performDecodeImpl(
ByteBuffer[] inputs, int[] inputOffsets, int dataLen, int[] erased, ByteBuffer[] inputs, int[] inputOffsets, int dataLen, int[] erased,
ByteBuffer[] outputs, int[] outputOffsets) throws IOException { ByteBuffer[] outputs, int[] outputOffsets) throws IOException {
decodeImpl(inputs, inputOffsets, dataLen, erased, outputs, outputOffsets); decodeImpl(inputs, inputOffsets, dataLen, erased, outputs, outputOffsets);
} }
@Override @Override
public synchronized void release() { public void release() {
decoderLock.writeLock().lock();
try {
destroyImpl(); destroyImpl();
} finally {
decoderLock.writeLock().unlock();
}
} }
@Override @Override

View File

@ -36,19 +36,30 @@ public class NativeRSRawEncoder extends AbstractNativeRawEncoder {
public NativeRSRawEncoder(ErasureCoderOptions coderOptions) { public NativeRSRawEncoder(ErasureCoderOptions coderOptions) {
super(coderOptions); super(coderOptions);
initImpl(coderOptions.getNumDataUnits(), coderOptions.getNumParityUnits()); encoderLock.writeLock().lock();
try {
initImpl(coderOptions.getNumDataUnits(),
coderOptions.getNumParityUnits());
} finally {
encoderLock.writeLock().unlock();
}
} }
@Override @Override
protected synchronized void performEncodeImpl( protected void performEncodeImpl(
ByteBuffer[] inputs, int[] inputOffsets, int dataLen, ByteBuffer[] inputs, int[] inputOffsets, int dataLen,
ByteBuffer[] outputs, int[] outputOffsets) throws IOException { ByteBuffer[] outputs, int[] outputOffsets) throws IOException {
encodeImpl(inputs, inputOffsets, dataLen, outputs, outputOffsets); encodeImpl(inputs, inputOffsets, dataLen, outputs, outputOffsets);
} }
@Override @Override
public synchronized void release() { public void release() {
encoderLock.writeLock().lock();
try {
destroyImpl(); destroyImpl();
} finally {
encoderLock.writeLock().unlock();
}
} }
@Override @Override

View File

@ -36,19 +36,30 @@ public class NativeXORRawDecoder extends AbstractNativeRawDecoder {
public NativeXORRawDecoder(ErasureCoderOptions coderOptions) { public NativeXORRawDecoder(ErasureCoderOptions coderOptions) {
super(coderOptions); super(coderOptions);
initImpl(coderOptions.getNumDataUnits(), coderOptions.getNumParityUnits()); decoderLock.writeLock().lock();
try {
initImpl(coderOptions.getNumDataUnits(),
coderOptions.getNumParityUnits());
} finally {
decoderLock.writeLock().unlock();
}
} }
@Override @Override
protected synchronized void performDecodeImpl( protected void performDecodeImpl(
ByteBuffer[] inputs, int[] inputOffsets, int dataLen, int[] erased, ByteBuffer[] inputs, int[] inputOffsets, int dataLen, int[] erased,
ByteBuffer[] outputs, int[] outputOffsets) throws IOException { ByteBuffer[] outputs, int[] outputOffsets) throws IOException {
decodeImpl(inputs, inputOffsets, dataLen, erased, outputs, outputOffsets); decodeImpl(inputs, inputOffsets, dataLen, erased, outputs, outputOffsets);
} }
@Override @Override
public synchronized void release() { public void release() {
decoderLock.writeLock().lock();
try {
destroyImpl(); destroyImpl();
} finally {
decoderLock.writeLock().unlock();
}
} }
private native void initImpl(int numDataUnits, int numParityUnits); private native void initImpl(int numDataUnits, int numParityUnits);

View File

@ -36,19 +36,30 @@ public class NativeXORRawEncoder extends AbstractNativeRawEncoder {
public NativeXORRawEncoder(ErasureCoderOptions coderOptions) { public NativeXORRawEncoder(ErasureCoderOptions coderOptions) {
super(coderOptions); super(coderOptions);
initImpl(coderOptions.getNumDataUnits(), coderOptions.getNumParityUnits()); encoderLock.writeLock().lock();
try {
initImpl(coderOptions.getNumDataUnits(),
coderOptions.getNumParityUnits());
} finally {
encoderLock.writeLock().unlock();
}
} }
@Override @Override
protected synchronized void performEncodeImpl( protected void performEncodeImpl(
ByteBuffer[] inputs, int[] inputOffsets, int dataLen, ByteBuffer[] inputs, int[] inputOffsets, int dataLen,
ByteBuffer[] outputs, int[] outputOffsets) throws IOException { ByteBuffer[] outputs, int[] outputOffsets) throws IOException {
encodeImpl(inputs, inputOffsets, dataLen, outputs, outputOffsets); encodeImpl(inputs, inputOffsets, dataLen, outputs, outputOffsets);
} }
@Override @Override
public synchronized void release() { public void release() {
encoderLock.writeLock().lock();
try {
destroyImpl(); destroyImpl();
} finally {
encoderLock.writeLock().unlock();
}
} }
private native void initImpl(int numDataUnits, int numParityUnits); private native void initImpl(int numDataUnits, int numParityUnits);

View File

@ -230,6 +230,12 @@ public static void performBench(String opType, CODER coder,
throw e; throw e;
} finally { } finally {
executor.shutdown(); executor.shutdown();
if (encoder != null) {
encoder.release();
}
if (decoder != null) {
decoder.release();
}
} }
} }