YARN-10562. Follow up changes for YARN-9833. Contributed by Jim Brennan.

This commit is contained in:
Eric Badger 2021-01-13 23:41:52 +00:00
parent d8fadec944
commit 768e2f42ba

View File

@ -22,6 +22,7 @@
import java.io.FileNotFoundException;
import java.io.IOException;
import java.util.ArrayList;
import java.util.Arrays;
import java.util.Collections;
import java.util.HashMap;
import java.util.HashSet;
@ -29,7 +30,6 @@
import java.util.Map;
import java.util.Set;
import java.util.concurrent.ConcurrentHashMap;
import java.util.concurrent.CopyOnWriteArrayList;
import java.util.concurrent.locks.ReentrantReadWriteLock;
import java.util.concurrent.locks.ReentrantReadWriteLock.ReadLock;
import java.util.concurrent.locks.ReentrantReadWriteLock.WriteLock;
@ -239,9 +239,9 @@ public DirectoryCollection(String[] dirs,
throw new YarnRuntimeException(e);
}
localDirs = new CopyOnWriteArrayList<>(dirs);
errorDirs = new CopyOnWriteArrayList<>();
fullDirs = new CopyOnWriteArrayList<>();
localDirs = new ArrayList<>(Arrays.asList(dirs));
errorDirs = new ArrayList<>();
fullDirs = new ArrayList<>();
directoryErrorInfo = new ConcurrentHashMap<>();
ReentrantReadWriteLock lock = new ReentrantReadWriteLock();
@ -287,7 +287,7 @@ List<String> getGoodDirs() {
List<String> getFailedDirs() {
this.readLock.lock();
try {
return ImmutableList.copyOf(
return Collections.unmodifiableList(
DirectoryCollection.concat(errorDirs, fullDirs));
} finally {
this.readLock.unlock();
@ -315,7 +315,7 @@ List<String> getFullDirs() {
List<String> getErroredDirs() {
this.readLock.lock();
try {
return Collections.unmodifiableList(errorDirs);
return ImmutableList.copyOf(errorDirs);
} finally {
this.readLock.unlock();
}