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