big-data/hadoop
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

HADOOP-18416. fix ITestS3AIOStatisticsContext test failure (#4806)

Browse Source 
Follow on to HADOOP-17461.

Contributed by: Mehakmeet Singh
This commit is contained in:
Mehakmeet Singh 2022-09-08 21:03:18 +05:30 committed by GitHub
parent 832d0e0d76
commit 03961b10c2
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23
2 changed files with 15 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
hadoop-tools/hadoop-aws/src/test/java/org/apache/hadoop/fs/s3a/ITestS3AIOStatisticsContext.java
View File

@ -42,6 +42,7 @@
import static org.apache.hadoop.fs.contract.ContractTestUtils.assertCapabilities;
import static org.apache.hadoop.fs.contract.ContractTestUtils.dataset;
import static org.apache.hadoop.fs.contract.ContractTestUtils.writeDataset;
import static org.apache.hadoop.fs.s3a.S3ATestUtils.disablePrefetching;
import static org.apache.hadoop.fs.statistics.IOStatisticAssertions.assertThatStatisticCounter;
import static org.apache.hadoop.fs.statistics.IOStatisticAssertions.verifyStatisticCounterValue;
import static org.apache.hadoop.fs.statistics.StreamStatisticNames.STREAM_READ_BYTES;
@ -67,6 +68,7 @@ public class ITestS3AIOStatisticsContext extends AbstractS3ATestBase {
@Override
protected Configuration createConfiguration() {
Configuration configuration = super.createConfiguration();
disablePrefetching(configuration);
enableIOStatisticsContext();
return configuration;
}
@ -253,6 +255,7 @@ public void testThreadIOStatisticsForDifferentThreads()
// Worker thread work and wait for it to finish.
TestWorkerThread workerThread = new TestWorkerThread(path, null);
long workerThreadID = workerThread.getId();
LOG.info("Worker thread ID: {} ", workerThreadID);
workerThread.start();
workerThread.join();
@ -463,6 +466,8 @@ private class TestWorkerThread extends Thread implements Runnable {
@Override
public void run() {
// Setting the worker thread's name.
Thread.currentThread().setName("worker thread");
S3AFileSystem fs = getFileSystem();
byte[] data = new byte[BYTES_SMALL];
@ -470,6 +475,9 @@ public void run() {
if (ioStatisticsContext != null) {
IOStatisticsContext.setThreadIOStatisticsContext(ioStatisticsContext);
}
// Storing context in a field to not lose the reference in a GC.
IOStatisticsContext ioStatisticsContextWorkerThread =
getCurrentIOStatisticsContext();
// Write in the worker thread.
try (FSDataOutputStream out = fs.create(workerThreadPath)) {

11
hadoop-tools/hadoop-aws/src/test/java/org/apache/hadoop/fs/s3a/S3ATestUtils.java
View File

@ -1499,8 +1499,11 @@ public static S3AInputStream getS3AInputStream(
}
}
/**
* Disable Prefetching streams from S3AFileSystem in tests.
* @param conf Configuration to remove the prefetch property from.
*/
public static void disablePrefetching(Configuration conf) {
removeBaseAndBucketOverrides(conf, PREFETCH_ENABLED_KEY);
}
}