HDDS-82. Merge ContainerData and ContainerStatus classes. Contributed by Bharat Viswanadham.
This commit is contained in:
parent
0b4c44bdee
commit
5e88126776
@ -152,6 +152,7 @@ enum ContainerLifeCycleState {
|
||||
OPEN = 1;
|
||||
CLOSING = 2;
|
||||
CLOSED = 3;
|
||||
INVALID = 4;
|
||||
}
|
||||
|
||||
message ContainerCommandRequestProto {
|
||||
|
@ -52,6 +52,17 @@ public class ContainerData {
|
||||
private ContainerType containerType;
|
||||
private String containerDBType;
|
||||
|
||||
|
||||
/**
|
||||
* Number of pending deletion blocks in container.
|
||||
*/
|
||||
private int numPendingDeletionBlocks;
|
||||
private AtomicLong readBytes;
|
||||
private AtomicLong writeBytes;
|
||||
private AtomicLong readCount;
|
||||
private AtomicLong writeCount;
|
||||
|
||||
|
||||
/**
|
||||
* Constructs a ContainerData Object.
|
||||
*
|
||||
@ -66,6 +77,34 @@ public ContainerData(long containerID,
|
||||
this.bytesUsed = new AtomicLong(0L);
|
||||
this.containerID = containerID;
|
||||
this.state = ContainerLifeCycleState.OPEN;
|
||||
this.numPendingDeletionBlocks = 0;
|
||||
this.readCount = new AtomicLong(0L);
|
||||
this.readBytes = new AtomicLong(0L);
|
||||
this.writeCount = new AtomicLong(0L);
|
||||
this.writeBytes = new AtomicLong(0L);
|
||||
}
|
||||
|
||||
/**
|
||||
* Constructs a ContainerData Object.
|
||||
*
|
||||
* @param containerID - ID
|
||||
* @param conf - Configuration
|
||||
* @param state - ContainerLifeCycleState
|
||||
* @param
|
||||
*/
|
||||
public ContainerData(long containerID, Configuration conf,
|
||||
ContainerLifeCycleState state) {
|
||||
this.metadata = new TreeMap<>();
|
||||
this.maxSize = conf.getLong(ScmConfigKeys.SCM_CONTAINER_CLIENT_MAX_SIZE_KEY,
|
||||
ScmConfigKeys.SCM_CONTAINER_CLIENT_MAX_SIZE_DEFAULT) * OzoneConsts.GB;
|
||||
this.bytesUsed = new AtomicLong(0L);
|
||||
this.containerID = containerID;
|
||||
this.state = state;
|
||||
this.numPendingDeletionBlocks = 0;
|
||||
this.readCount = new AtomicLong(0L);
|
||||
this.readBytes = new AtomicLong(0L);
|
||||
this.writeCount = new AtomicLong(0L);
|
||||
this.writeBytes = new AtomicLong(0L);
|
||||
}
|
||||
|
||||
/**
|
||||
@ -292,6 +331,14 @@ public synchronized boolean isOpen() {
|
||||
return ContainerLifeCycleState.OPEN == state;
|
||||
}
|
||||
|
||||
/**
|
||||
* checks if the container is invalid.
|
||||
* @return - boolean
|
||||
*/
|
||||
public boolean isValid() {
|
||||
return !(ContainerLifeCycleState.INVALID == state);
|
||||
}
|
||||
|
||||
/**
|
||||
* Marks this container as closed.
|
||||
*/
|
||||
@ -317,11 +364,119 @@ public void setBytesUsed(long used) {
|
||||
this.bytesUsed.set(used);
|
||||
}
|
||||
|
||||
public long addBytesUsed(long delta) {
|
||||
return this.bytesUsed.addAndGet(delta);
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the number of bytes used by the container.
|
||||
* @return the number of bytes used by the container.
|
||||
*/
|
||||
public long getBytesUsed() {
|
||||
return bytesUsed.get();
|
||||
}
|
||||
|
||||
/**
|
||||
* Increase the number of bytes used by the container.
|
||||
* @param used number of bytes used by the container.
|
||||
* @return the current number of bytes used by the container afert increase.
|
||||
*/
|
||||
public long incrBytesUsed(long used) {
|
||||
return this.bytesUsed.addAndGet(used);
|
||||
}
|
||||
|
||||
|
||||
/**
|
||||
* Decrease the number of bytes used by the container.
|
||||
* @param reclaimed the number of bytes reclaimed from the container.
|
||||
* @return the current number of bytes used by the container after decrease.
|
||||
*/
|
||||
public long decrBytesUsed(long reclaimed) {
|
||||
return this.bytesUsed.addAndGet(-1L * reclaimed);
|
||||
}
|
||||
|
||||
/**
|
||||
* Increase the count of pending deletion blocks.
|
||||
*
|
||||
* @param numBlocks increment number
|
||||
*/
|
||||
public void incrPendingDeletionBlocks(int numBlocks) {
|
||||
this.numPendingDeletionBlocks += numBlocks;
|
||||
}
|
||||
|
||||
/**
|
||||
* Decrease the count of pending deletion blocks.
|
||||
*
|
||||
* @param numBlocks decrement number
|
||||
*/
|
||||
public void decrPendingDeletionBlocks(int numBlocks) {
|
||||
this.numPendingDeletionBlocks -= numBlocks;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the number of pending deletion blocks.
|
||||
*/
|
||||
public int getNumPendingDeletionBlocks() {
|
||||
return this.numPendingDeletionBlocks;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the number of bytes read from the container.
|
||||
* @return the number of bytes read from the container.
|
||||
*/
|
||||
public long getReadBytes() {
|
||||
return readBytes.get();
|
||||
}
|
||||
|
||||
/**
|
||||
* Increase the number of bytes read from the container.
|
||||
* @param bytes number of bytes read.
|
||||
*/
|
||||
public void incrReadBytes(long bytes) {
|
||||
this.readBytes.addAndGet(bytes);
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the number of times the container is read.
|
||||
* @return the number of times the container is read.
|
||||
*/
|
||||
public long getReadCount() {
|
||||
return readCount.get();
|
||||
}
|
||||
|
||||
/**
|
||||
* Increase the number of container read count by 1.
|
||||
*/
|
||||
public void incrReadCount() {
|
||||
this.readCount.incrementAndGet();
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the number of bytes write into the container.
|
||||
* @return the number of bytes write into the container.
|
||||
*/
|
||||
public long getWriteBytes() {
|
||||
return writeBytes.get();
|
||||
}
|
||||
|
||||
/**
|
||||
* Increase the number of bytes write into the container.
|
||||
* @param bytes the number of bytes write into the container.
|
||||
*/
|
||||
public void incrWriteBytes(long bytes) {
|
||||
this.writeBytes.addAndGet(bytes);
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the number of writes into the container.
|
||||
* @return the number of writes into the container.
|
||||
*/
|
||||
public long getWriteCount() {
|
||||
return writeCount.get();
|
||||
}
|
||||
|
||||
/**
|
||||
* Increase the number of writes into the container by 1.
|
||||
*/
|
||||
public void incrWriteCount() {
|
||||
this.writeCount.incrementAndGet();
|
||||
}
|
||||
|
||||
|
||||
}
|
||||
|
@ -22,13 +22,14 @@
|
||||
import com.google.common.base.Preconditions;
|
||||
import org.apache.commons.io.FileUtils;
|
||||
import org.apache.hadoop.conf.Configuration;
|
||||
import org.apache.hadoop.fs.StorageType;
|
||||
import org.apache.hadoop.hdds.scm.ScmConfigKeys;
|
||||
import org.apache.hadoop.hdds.scm.container.common.helpers
|
||||
.StorageContainerException;
|
||||
import org.apache.hadoop.hdfs.server.datanode.StorageLocation;
|
||||
import org.apache.hadoop.hdds.protocol.DatanodeDetails;
|
||||
import org.apache.hadoop.hdds.protocol.datanode.proto.ContainerProtos;
|
||||
import org.apache.hadoop.hdds.protocol.datanode.proto.ContainerProtos
|
||||
.ContainerLifeCycleState;
|
||||
import org.apache.hadoop.hdds.protocol.proto
|
||||
.StorageContainerDatanodeProtocolProtos;
|
||||
import org.apache.hadoop.hdds.protocol.proto
|
||||
@ -39,8 +40,6 @@
|
||||
.StorageContainerDatanodeProtocolProtos.SCMNodeReport;
|
||||
import org.apache.hadoop.hdds.protocol.proto
|
||||
.StorageContainerDatanodeProtocolProtos.SCMStorageReport;
|
||||
import org.apache.hadoop.hdds.protocol.proto
|
||||
.StorageContainerDatanodeProtocolProtos.StorageTypeProto;
|
||||
import org.apache.hadoop.io.IOUtils;
|
||||
import org.apache.hadoop.ozone.OzoneConfigKeys;
|
||||
import org.apache.hadoop.ozone.OzoneConsts;
|
||||
@ -116,7 +115,7 @@ public class ContainerManagerImpl implements ContainerManager {
|
||||
|
||||
// TODO: consider primitive collection like eclipse-collections
|
||||
// to avoid autoboxing overhead
|
||||
private final ConcurrentSkipListMap<Long, ContainerStatus>
|
||||
private final ConcurrentSkipListMap<Long, ContainerData>
|
||||
containerMap = new ConcurrentSkipListMap<>();
|
||||
|
||||
// Use a non-fair RW lock for better throughput, we may revisit this decision
|
||||
@ -246,19 +245,20 @@ private void readContainerInfo(String containerName)
|
||||
// when loading the info we get a null, this often means last time
|
||||
// SCM was ending up at some middle phase causing that the metadata
|
||||
// was not populated. Such containers are marked as inactive.
|
||||
containerMap.put(containerID, new ContainerStatus(null));
|
||||
ContainerData cData = new ContainerData(containerID, conf,
|
||||
ContainerLifeCycleState.INVALID);
|
||||
containerMap.put(containerID, cData);
|
||||
return;
|
||||
}
|
||||
containerData = ContainerData.getFromProtBuf(containerDataProto, conf);
|
||||
|
||||
ContainerStatus containerStatus = new ContainerStatus(containerData);
|
||||
// Initialize pending deletion blocks count in in-memory
|
||||
// container status.
|
||||
MetadataStore metadata = KeyUtils.getDB(containerData, conf);
|
||||
List<Map.Entry<byte[], byte[]>> underDeletionBlocks = metadata
|
||||
.getSequentialRangeKVs(null, Integer.MAX_VALUE,
|
||||
MetadataKeyFilters.getDeletingKeyFilter());
|
||||
containerStatus.incrPendingDeletionBlocks(underDeletionBlocks.size());
|
||||
containerData.incrPendingDeletionBlocks(underDeletionBlocks.size());
|
||||
|
||||
List<Map.Entry<byte[], byte[]>> liveKeys = metadata
|
||||
.getRangeKVs(null, Integer.MAX_VALUE,
|
||||
@ -277,9 +277,9 @@ private void readContainerInfo(String containerName)
|
||||
return 0L;
|
||||
}
|
||||
}).sum();
|
||||
containerStatus.setBytesUsed(bytesUsed);
|
||||
containerData.setBytesUsed(bytesUsed);
|
||||
|
||||
containerMap.put(containerID, containerStatus);
|
||||
containerMap.put(containerID, containerData);
|
||||
} catch (IOException ex) {
|
||||
LOG.error("read failed for file: {} ex: {}", containerName,
|
||||
ex.getMessage());
|
||||
@ -287,7 +287,9 @@ private void readContainerInfo(String containerName)
|
||||
// TODO : Add this file to a recovery Queue.
|
||||
|
||||
// Remember that this container is busted and we cannot use it.
|
||||
containerMap.put(containerID, new ContainerStatus(null));
|
||||
ContainerData cData = new ContainerData(containerID, conf,
|
||||
ContainerLifeCycleState.INVALID);
|
||||
containerMap.put(containerID, cData);
|
||||
throw new StorageContainerException("Unable to read container info",
|
||||
UNABLE_TO_READ_METADATA_DB);
|
||||
} finally {
|
||||
@ -456,18 +458,19 @@ public void deleteContainer(long containerID,
|
||||
UNCLOSED_CONTAINER_IO);
|
||||
}
|
||||
|
||||
ContainerStatus status = containerMap.get(containerID);
|
||||
if (status == null) {
|
||||
ContainerData containerData = containerMap.get(containerID);
|
||||
if (containerData == null) {
|
||||
LOG.debug("No such container. ID: {}", containerID);
|
||||
throw new StorageContainerException("No such container. ID : " +
|
||||
containerID, CONTAINER_NOT_FOUND);
|
||||
}
|
||||
if (status.getContainer() == null) {
|
||||
|
||||
if(!containerData.isValid()) {
|
||||
LOG.debug("Invalid container data. ID: {}", containerID);
|
||||
throw new StorageContainerException("Invalid container data. Name : " +
|
||||
containerID, CONTAINER_NOT_FOUND);
|
||||
}
|
||||
ContainerUtils.removeContainer(status.getContainer(), conf, forceDelete);
|
||||
ContainerUtils.removeContainer(containerData, conf, forceDelete);
|
||||
containerMap.remove(containerID);
|
||||
} catch (StorageContainerException e) {
|
||||
throw e;
|
||||
@ -509,7 +512,7 @@ public void listContainer(long startContainerID, long count,
|
||||
|
||||
readLock();
|
||||
try {
|
||||
ConcurrentNavigableMap<Long, ContainerStatus> map;
|
||||
ConcurrentNavigableMap<Long, ContainerData> map;
|
||||
if (startContainerID == 0) {
|
||||
map = containerMap.tailMap(containerMap.firstKey(), true);
|
||||
} else {
|
||||
@ -517,9 +520,9 @@ public void listContainer(long startContainerID, long count,
|
||||
}
|
||||
|
||||
int currentCount = 0;
|
||||
for (ContainerStatus entry : map.values()) {
|
||||
for (ContainerData entry : map.values()) {
|
||||
if (currentCount < count) {
|
||||
data.add(entry.getContainer());
|
||||
data.add(entry);
|
||||
currentCount++;
|
||||
} else {
|
||||
return;
|
||||
@ -546,7 +549,7 @@ public ContainerData readContainer(long containerID)
|
||||
throw new StorageContainerException("Unable to find the container. ID: "
|
||||
+ containerID, CONTAINER_NOT_FOUND);
|
||||
}
|
||||
ContainerData cData = containerMap.get(containerID).getContainer();
|
||||
ContainerData cData = containerMap.get(containerID);
|
||||
if (cData == null) {
|
||||
throw new StorageContainerException("Invalid container data. ID: "
|
||||
+ containerID, CONTAINER_INTERNAL_ERROR);
|
||||
@ -584,8 +587,7 @@ public void closeContainer(long containerID)
|
||||
// I/O failure, this allows us to take quick action in case of container
|
||||
// issues.
|
||||
|
||||
ContainerStatus status = new ContainerStatus(containerData);
|
||||
containerMap.put(containerID, status);
|
||||
containerMap.put(containerID, containerData);
|
||||
}
|
||||
|
||||
@Override
|
||||
@ -614,7 +616,7 @@ public void updateContainer(long containerID, ContainerData data,
|
||||
|
||||
try {
|
||||
Path location = locationManager.getContainerPath();
|
||||
ContainerData orgData = containerMap.get(containerID).getContainer();
|
||||
ContainerData orgData = containerMap.get(containerID);
|
||||
if (orgData == null) {
|
||||
// updating a invalid container
|
||||
throw new StorageContainerException("Update a container with invalid" +
|
||||
@ -652,8 +654,7 @@ public void updateContainer(long containerID, ContainerData data,
|
||||
}
|
||||
|
||||
// Update the in-memory map
|
||||
ContainerStatus newStatus = new ContainerStatus(data);
|
||||
containerMap.replace(containerID, newStatus);
|
||||
containerMap.replace(containerID, data);
|
||||
} catch (IOException e) {
|
||||
// Restore the container file from backup
|
||||
if(containerFileBK != null && containerFileBK.exists() && deleted) {
|
||||
@ -699,17 +700,12 @@ protected File getContainerFile(ContainerData data) throws IOException {
|
||||
*/
|
||||
@Override
|
||||
public boolean isOpen(long containerID) throws StorageContainerException {
|
||||
final ContainerStatus status = containerMap.get(containerID);
|
||||
if (status == null) {
|
||||
throw new StorageContainerException(
|
||||
"Container status not found: " + containerID, CONTAINER_NOT_FOUND);
|
||||
}
|
||||
final ContainerData cData = status.getContainer();
|
||||
if (cData == null) {
|
||||
final ContainerData containerData = containerMap.get(containerID);
|
||||
if (containerData == null) {
|
||||
throw new StorageContainerException(
|
||||
"Container not found: " + containerID, CONTAINER_NOT_FOUND);
|
||||
}
|
||||
return cData.isOpen();
|
||||
return containerData.isOpen();
|
||||
}
|
||||
|
||||
/**
|
||||
@ -727,7 +723,7 @@ public void shutdown() throws IOException {
|
||||
|
||||
|
||||
@VisibleForTesting
|
||||
public ConcurrentSkipListMap<Long, ContainerStatus> getContainerMap() {
|
||||
public ConcurrentSkipListMap<Long, ContainerData> getContainerMap() {
|
||||
return containerMap;
|
||||
}
|
||||
|
||||
@ -847,9 +843,9 @@ public List<ContainerData> getContainerReports() throws IOException {
|
||||
// And we can never get the exact state since close might happen
|
||||
// after we iterate a point.
|
||||
return containerMap.entrySet().stream()
|
||||
.filter(containerStatus ->
|
||||
!containerStatus.getValue().getContainer().isOpen())
|
||||
.map(containerStatus -> containerStatus.getValue().getContainer())
|
||||
.filter(containerData ->
|
||||
!containerData.getValue().isOpen())
|
||||
.map(containerData -> containerData.getValue())
|
||||
.collect(Collectors.toList());
|
||||
}
|
||||
|
||||
@ -865,7 +861,7 @@ public ContainerReportsRequestProto getContainerReport() throws IOException {
|
||||
// No need for locking since containerMap is a ConcurrentSkipListMap
|
||||
// And we can never get the exact state since close might happen
|
||||
// after we iterate a point.
|
||||
List<ContainerStatus> containers = containerMap.values().stream()
|
||||
List<ContainerData> containers = containerMap.values().stream()
|
||||
.collect(Collectors.toList());
|
||||
|
||||
ContainerReportsRequestProto.Builder crBuilder =
|
||||
@ -875,18 +871,17 @@ public ContainerReportsRequestProto getContainerReport() throws IOException {
|
||||
crBuilder.setDatanodeDetails(datanodeDetails.getProtoBufMessage())
|
||||
.setType(ContainerReportsRequestProto.reportType.fullReport);
|
||||
|
||||
for (ContainerStatus container: containers) {
|
||||
for (ContainerData container: containers) {
|
||||
StorageContainerDatanodeProtocolProtos.ContainerInfo.Builder ciBuilder =
|
||||
StorageContainerDatanodeProtocolProtos.ContainerInfo.newBuilder();
|
||||
ciBuilder.setContainerID(container.getContainer().getContainerID())
|
||||
.setSize(container.getContainer().getMaxSize())
|
||||
.setUsed(container.getContainer().getBytesUsed())
|
||||
.setKeyCount(container.getContainer().getKeyCount())
|
||||
ciBuilder.setContainerID(container.getContainerID())
|
||||
.setSize(container.getMaxSize())
|
||||
.setUsed(container.getBytesUsed())
|
||||
.setKeyCount(container.getKeyCount())
|
||||
.setReadCount(container.getReadCount())
|
||||
.setWriteCount(container.getWriteCount())
|
||||
.setReadBytes(container.getReadBytes())
|
||||
.setWriteBytes(container.getWriteBytes())
|
||||
.setContainerID(container.getContainer().getContainerID());
|
||||
.setWriteBytes(container.getWriteBytes());
|
||||
|
||||
crBuilder.addReports(ciBuilder.build());
|
||||
}
|
||||
@ -943,8 +938,8 @@ public ContainerDeletionChoosingPolicy getContainerDeletionChooser() {
|
||||
public void incrPendingDeletionBlocks(int numBlocks, long containerId) {
|
||||
writeLock();
|
||||
try {
|
||||
ContainerStatus status = containerMap.get(containerId);
|
||||
status.incrPendingDeletionBlocks(numBlocks);
|
||||
ContainerData cData = containerMap.get(containerId);
|
||||
cData.incrPendingDeletionBlocks(numBlocks);
|
||||
} finally {
|
||||
writeUnlock();
|
||||
}
|
||||
@ -954,8 +949,8 @@ public void incrPendingDeletionBlocks(int numBlocks, long containerId) {
|
||||
public void decrPendingDeletionBlocks(int numBlocks, long containerId) {
|
||||
writeLock();
|
||||
try {
|
||||
ContainerStatus status = containerMap.get(containerId);
|
||||
status.decrPendingDeletionBlocks(numBlocks);
|
||||
ContainerData cData = containerMap.get(containerId);
|
||||
cData.decrPendingDeletionBlocks(numBlocks);
|
||||
} finally {
|
||||
writeUnlock();
|
||||
}
|
||||
@ -968,32 +963,37 @@ public void decrPendingDeletionBlocks(int numBlocks, long containerId) {
|
||||
*/
|
||||
@Override
|
||||
public void incrReadCount(long containerId) {
|
||||
ContainerStatus status = containerMap.get(containerId);
|
||||
status.incrReadCount();
|
||||
ContainerData cData = containerMap.get(containerId);
|
||||
cData.incrReadCount();
|
||||
}
|
||||
|
||||
public long getReadCount(long containerId) {
|
||||
ContainerStatus status = containerMap.get(containerId);
|
||||
return status.getReadCount();
|
||||
ContainerData cData = containerMap.get(containerId);
|
||||
return cData.getReadCount();
|
||||
}
|
||||
|
||||
/**
|
||||
* Increse the read counter for bytes read from the container.
|
||||
* Increase the read counter for bytes read from the container.
|
||||
*
|
||||
* @param containerId - ID of the container.
|
||||
* @param readBytes - bytes read from the container.
|
||||
*/
|
||||
@Override
|
||||
public void incrReadBytes(long containerId, long readBytes) {
|
||||
ContainerStatus status = containerMap.get(containerId);
|
||||
status.incrReadBytes(readBytes);
|
||||
ContainerData cData = containerMap.get(containerId);
|
||||
cData.incrReadBytes(readBytes);
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns number of bytes read from the container.
|
||||
* @param containerId
|
||||
* @return
|
||||
*/
|
||||
public long getReadBytes(long containerId) {
|
||||
readLock();
|
||||
try {
|
||||
ContainerStatus status = containerMap.get(containerId);
|
||||
return status.getReadBytes();
|
||||
ContainerData cData = containerMap.get(containerId);
|
||||
return cData.getReadBytes();
|
||||
} finally {
|
||||
readUnlock();
|
||||
}
|
||||
@ -1006,13 +1006,13 @@ public long getReadBytes(long containerId) {
|
||||
*/
|
||||
@Override
|
||||
public void incrWriteCount(long containerId) {
|
||||
ContainerStatus status = containerMap.get(containerId);
|
||||
status.incrWriteCount();
|
||||
ContainerData cData = containerMap.get(containerId);
|
||||
cData.incrWriteCount();
|
||||
}
|
||||
|
||||
public long getWriteCount(long containerId) {
|
||||
ContainerStatus status = containerMap.get(containerId);
|
||||
return status.getWriteCount();
|
||||
ContainerData cData = containerMap.get(containerId);
|
||||
return cData.getWriteCount();
|
||||
}
|
||||
|
||||
/**
|
||||
@ -1023,13 +1023,13 @@ public long getWriteCount(long containerId) {
|
||||
*/
|
||||
@Override
|
||||
public void incrWriteBytes(long containerId, long writeBytes) {
|
||||
ContainerStatus status = containerMap.get(containerId);
|
||||
status.incrWriteBytes(writeBytes);
|
||||
ContainerData cData = containerMap.get(containerId);
|
||||
cData.incrWriteBytes(writeBytes);
|
||||
}
|
||||
|
||||
public long getWriteBytes(long containerId) {
|
||||
ContainerStatus status = containerMap.get(containerId);
|
||||
return status.getWriteBytes();
|
||||
ContainerData cData = containerMap.get(containerId);
|
||||
return cData.getWriteBytes();
|
||||
}
|
||||
|
||||
/**
|
||||
@ -1041,8 +1041,8 @@ public long getWriteBytes(long containerId) {
|
||||
*/
|
||||
@Override
|
||||
public long incrBytesUsed(long containerId, long used) {
|
||||
ContainerStatus status = containerMap.get(containerId);
|
||||
return status.incrBytesUsed(used);
|
||||
ContainerData cData = containerMap.get(containerId);
|
||||
return cData.incrBytesUsed(used);
|
||||
}
|
||||
|
||||
/**
|
||||
@ -1054,13 +1054,13 @@ public long incrBytesUsed(long containerId, long used) {
|
||||
*/
|
||||
@Override
|
||||
public long decrBytesUsed(long containerId, long used) {
|
||||
ContainerStatus status = containerMap.get(containerId);
|
||||
return status.decrBytesUsed(used);
|
||||
ContainerData cData = containerMap.get(containerId);
|
||||
return cData.decrBytesUsed(used);
|
||||
}
|
||||
|
||||
public long getBytesUsed(long containerId) {
|
||||
ContainerStatus status = containerMap.get(containerId);
|
||||
return status.getBytesUsed();
|
||||
ContainerData cData = containerMap.get(containerId);
|
||||
return cData.getBytesUsed();
|
||||
}
|
||||
|
||||
/**
|
||||
@ -1071,8 +1071,8 @@ public long getBytesUsed(long containerId) {
|
||||
*/
|
||||
@Override
|
||||
public long getNumKeys(long containerId) {
|
||||
ContainerStatus status = containerMap.get(containerId);
|
||||
return status.getNumKeys(); }
|
||||
ContainerData cData = containerMap.get(containerId);
|
||||
return cData.getKeyCount(); }
|
||||
|
||||
/**
|
||||
* Get the container report state to send via HB to SCM.
|
||||
|
@ -1,217 +0,0 @@
|
||||
/**
|
||||
* Licensed to the Apache Software Foundation (ASF) under one
|
||||
* or more contributor license agreements. See the NOTICE file
|
||||
* distributed with this work for additional information
|
||||
* regarding copyright ownership. The ASF licenses this file
|
||||
* to you under the Apache License, Version 2.0 (the
|
||||
* "License"); you may not use this file except in compliance
|
||||
* with the License. You may obtain a copy of the License at
|
||||
*
|
||||
* http://www.apache.org/licenses/LICENSE-2.0
|
||||
*
|
||||
* Unless required by applicable law or agreed to in writing, software
|
||||
* distributed under the License is distributed on an "AS IS" BASIS,
|
||||
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
|
||||
* See the License for the specific language governing permissions and
|
||||
* limitations under the License.
|
||||
*/
|
||||
package org.apache.hadoop.ozone.container.common.impl;
|
||||
|
||||
import org.apache.hadoop.ozone.container.common.helpers.ContainerData;
|
||||
|
||||
import java.util.concurrent.atomic.AtomicLong;
|
||||
|
||||
/**
|
||||
* This class represents the state of a container. if the
|
||||
* container reading encountered an error when we boot up we will post that
|
||||
* info to a recovery queue and keep the info in the containerMap.
|
||||
* <p/>
|
||||
* if and when the issue is fixed, the expectation is that this entry will be
|
||||
* deleted by the recovery thread from the containerMap and will insert entry
|
||||
* instead of modifying this class.
|
||||
*/
|
||||
public class ContainerStatus {
|
||||
private final ContainerData containerData;
|
||||
|
||||
/**
|
||||
* Number of pending deletion blocks in container.
|
||||
*/
|
||||
private int numPendingDeletionBlocks;
|
||||
|
||||
private AtomicLong readBytes;
|
||||
|
||||
private AtomicLong writeBytes;
|
||||
|
||||
private AtomicLong readCount;
|
||||
|
||||
private AtomicLong writeCount;
|
||||
|
||||
/**
|
||||
* Creates a Container Status class.
|
||||
*
|
||||
* @param containerData - ContainerData.
|
||||
*/
|
||||
ContainerStatus(ContainerData containerData) {
|
||||
this.numPendingDeletionBlocks = 0;
|
||||
this.containerData = containerData;
|
||||
this.readCount = new AtomicLong(0L);
|
||||
this.readBytes = new AtomicLong(0L);
|
||||
this.writeCount = new AtomicLong(0L);
|
||||
this.writeBytes = new AtomicLong(0L);
|
||||
}
|
||||
|
||||
/**
|
||||
* Returns container if it is active. It is not active if we have had an
|
||||
* error and we are waiting for the background threads to fix the issue.
|
||||
*
|
||||
* @return ContainerData.
|
||||
*/
|
||||
public ContainerData getContainer() {
|
||||
return containerData;
|
||||
}
|
||||
|
||||
/**
|
||||
* Increase the count of pending deletion blocks.
|
||||
*
|
||||
* @param numBlocks increment number
|
||||
*/
|
||||
public void incrPendingDeletionBlocks(int numBlocks) {
|
||||
this.numPendingDeletionBlocks += numBlocks;
|
||||
}
|
||||
|
||||
/**
|
||||
* Decrease the count of pending deletion blocks.
|
||||
*
|
||||
* @param numBlocks decrement number
|
||||
*/
|
||||
public void decrPendingDeletionBlocks(int numBlocks) {
|
||||
this.numPendingDeletionBlocks -= numBlocks;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the number of pending deletion blocks.
|
||||
*/
|
||||
public int getNumPendingDeletionBlocks() {
|
||||
return this.numPendingDeletionBlocks;
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the number of bytes read from the container.
|
||||
* @return the number of bytes read from the container.
|
||||
*/
|
||||
public long getReadBytes() {
|
||||
return readBytes.get();
|
||||
}
|
||||
|
||||
/**
|
||||
* Increase the number of bytes read from the container.
|
||||
* @param bytes number of bytes read.
|
||||
*/
|
||||
public void incrReadBytes(long bytes) {
|
||||
this.readBytes.addAndGet(bytes);
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the number of times the container is read.
|
||||
* @return the number of times the container is read.
|
||||
*/
|
||||
public long getReadCount() {
|
||||
return readCount.get();
|
||||
}
|
||||
|
||||
/**
|
||||
* Increase the number of container read count by 1.
|
||||
*/
|
||||
public void incrReadCount() {
|
||||
this.readCount.incrementAndGet();
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the number of bytes write into the container.
|
||||
* @return the number of bytes write into the container.
|
||||
*/
|
||||
public long getWriteBytes() {
|
||||
return writeBytes.get();
|
||||
}
|
||||
|
||||
/**
|
||||
* Increase the number of bytes write into the container.
|
||||
* @param bytes the number of bytes write into the container.
|
||||
*/
|
||||
public void incrWriteBytes(long bytes) {
|
||||
this.writeBytes.addAndGet(bytes);
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the number of writes into the container.
|
||||
* @return the number of writes into the container.
|
||||
*/
|
||||
public long getWriteCount() {
|
||||
return writeCount.get();
|
||||
}
|
||||
|
||||
/**
|
||||
* Increase the number of writes into the container by 1.
|
||||
*/
|
||||
public void incrWriteCount() {
|
||||
this.writeCount.incrementAndGet();
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the number of bytes used by the container.
|
||||
* @return the number of bytes used by the container.
|
||||
*/
|
||||
public long getBytesUsed() {
|
||||
return containerData.getBytesUsed();
|
||||
}
|
||||
|
||||
/**
|
||||
* Increase the number of bytes used by the container.
|
||||
* @param used number of bytes used by the container.
|
||||
* @return the current number of bytes used by the container afert increase.
|
||||
*/
|
||||
public long incrBytesUsed(long used) {
|
||||
return containerData.addBytesUsed(used);
|
||||
}
|
||||
|
||||
/**
|
||||
* Set the number of bytes used by the container.
|
||||
* @param used the number of bytes used by the container.
|
||||
*/
|
||||
public void setBytesUsed(long used) {
|
||||
containerData.setBytesUsed(used);
|
||||
}
|
||||
|
||||
/**
|
||||
* Decrease the number of bytes used by the container.
|
||||
* @param reclaimed the number of bytes reclaimed from the container.
|
||||
* @return the current number of bytes used by the container after decrease.
|
||||
*/
|
||||
public long decrBytesUsed(long reclaimed) {
|
||||
return this.containerData.addBytesUsed(-1L * reclaimed);
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the maximum container size.
|
||||
* @return the maximum container size.
|
||||
*/
|
||||
public long getMaxSize() {
|
||||
return containerData.getMaxSize();
|
||||
}
|
||||
|
||||
/**
|
||||
* Set the maximum container size.
|
||||
* @param size the maximum container size.
|
||||
*/
|
||||
public void setMaxSize(long size) {
|
||||
this.containerData.setMaxSize(size);
|
||||
}
|
||||
|
||||
/**
|
||||
* Get the number of keys in the container.
|
||||
* @return the number of keys in the container.
|
||||
*/
|
||||
public long getNumKeys() {
|
||||
return containerData.getKeyCount();
|
||||
}
|
||||
}
|
@ -41,24 +41,24 @@ public class RandomContainerDeletionChoosingPolicy
|
||||
|
||||
@Override
|
||||
public List<ContainerData> chooseContainerForBlockDeletion(int count,
|
||||
Map<Long, ContainerStatus> candidateContainers)
|
||||
Map<Long, ContainerData> candidateContainers)
|
||||
throws StorageContainerException {
|
||||
Preconditions.checkNotNull(candidateContainers,
|
||||
"Internal assertion: candidate containers cannot be null");
|
||||
|
||||
int currentCount = 0;
|
||||
List<ContainerData> result = new LinkedList<>();
|
||||
ContainerStatus[] values = new ContainerStatus[candidateContainers.size()];
|
||||
ContainerData[] values = new ContainerData[candidateContainers.size()];
|
||||
// to get a shuffle list
|
||||
for (ContainerStatus entry : DFSUtil.shuffle(
|
||||
for (ContainerData entry : DFSUtil.shuffle(
|
||||
candidateContainers.values().toArray(values))) {
|
||||
if (currentCount < count) {
|
||||
result.add(entry.getContainer());
|
||||
result.add(entry);
|
||||
currentCount++;
|
||||
|
||||
LOG.debug("Select container {} for block deletion, "
|
||||
+ "pending deletion blocks num: {}.",
|
||||
entry.getContainer().getContainerID(),
|
||||
entry.getContainerID(),
|
||||
entry.getNumPendingDeletionBlocks());
|
||||
} else {
|
||||
break;
|
||||
|
@ -41,11 +41,11 @@ public class TopNOrderedContainerDeletionChoosingPolicy
|
||||
private static final Logger LOG =
|
||||
LoggerFactory.getLogger(TopNOrderedContainerDeletionChoosingPolicy.class);
|
||||
|
||||
/** customized comparator used to compare differentiate container status. **/
|
||||
private static final Comparator<ContainerStatus> CONTAINER_STATUS_COMPARATOR
|
||||
= new Comparator<ContainerStatus>() {
|
||||
/** customized comparator used to compare differentiate container data. **/
|
||||
private static final Comparator<ContainerData> CONTAINER_DATA_COMPARATOR
|
||||
= new Comparator<ContainerData>() {
|
||||
@Override
|
||||
public int compare(ContainerStatus c1, ContainerStatus c2) {
|
||||
public int compare(ContainerData c1, ContainerData c2) {
|
||||
return Integer.compare(c2.getNumPendingDeletionBlocks(),
|
||||
c1.getNumPendingDeletionBlocks());
|
||||
}
|
||||
@ -53,28 +53,28 @@ public int compare(ContainerStatus c1, ContainerStatus c2) {
|
||||
|
||||
@Override
|
||||
public List<ContainerData> chooseContainerForBlockDeletion(int count,
|
||||
Map<Long, ContainerStatus> candidateContainers)
|
||||
Map<Long, ContainerData> candidateContainers)
|
||||
throws StorageContainerException {
|
||||
Preconditions.checkNotNull(candidateContainers,
|
||||
"Internal assertion: candidate containers cannot be null");
|
||||
|
||||
List<ContainerData> result = new LinkedList<>();
|
||||
List<ContainerStatus> orderedList = new LinkedList<>();
|
||||
List<ContainerData> orderedList = new LinkedList<>();
|
||||
orderedList.addAll(candidateContainers.values());
|
||||
Collections.sort(orderedList, CONTAINER_STATUS_COMPARATOR);
|
||||
Collections.sort(orderedList, CONTAINER_DATA_COMPARATOR);
|
||||
|
||||
// get top N list ordered by pending deletion blocks' number
|
||||
int currentCount = 0;
|
||||
for (ContainerStatus entry : orderedList) {
|
||||
for (ContainerData entry : orderedList) {
|
||||
if (currentCount < count) {
|
||||
if (entry.getNumPendingDeletionBlocks() > 0) {
|
||||
result.add(entry.getContainer());
|
||||
result.add(entry);
|
||||
currentCount++;
|
||||
|
||||
LOG.debug(
|
||||
"Select container {} for block deletion, "
|
||||
+ "pending deletion blocks num: {}.",
|
||||
entry.getContainer().getContainerID(),
|
||||
entry.getContainerID(),
|
||||
entry.getNumPendingDeletionBlocks());
|
||||
} else {
|
||||
LOG.debug("Stop looking for next container, there is no"
|
||||
|
@ -20,7 +20,6 @@
|
||||
import org.apache.hadoop.hdds.scm.container.common.helpers
|
||||
.StorageContainerException;
|
||||
import org.apache.hadoop.ozone.container.common.helpers.ContainerData;
|
||||
import org.apache.hadoop.ozone.container.common.impl.ContainerStatus;
|
||||
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
@ -41,6 +40,6 @@ public interface ContainerDeletionChoosingPolicy {
|
||||
* @throws StorageContainerException
|
||||
*/
|
||||
List<ContainerData> chooseContainerForBlockDeletion(int count,
|
||||
Map<Long, ContainerStatus> candidateContainers)
|
||||
Map<Long, ContainerData> candidateContainers)
|
||||
throws StorageContainerException;
|
||||
}
|
||||
|
@ -65,7 +65,6 @@
|
||||
import java.util.List;
|
||||
import java.util.Map;
|
||||
import java.util.ArrayList;
|
||||
import java.util.UUID;
|
||||
|
||||
import static org.apache.hadoop.hdfs.DFSConfigKeys.DFS_DATANODE_DATA_DIR_KEY;
|
||||
import static org.apache.hadoop.ozone.OzoneConsts.CONTAINER_ROOT_PREFIX;
|
||||
@ -191,23 +190,23 @@ public void testCreateContainer() throws Exception {
|
||||
containerManager.createContainer(data);
|
||||
Assert.assertTrue(containerManager.getContainerMap()
|
||||
.containsKey(testContainerID));
|
||||
ContainerStatus status = containerManager
|
||||
ContainerData cData = containerManager
|
||||
.getContainerMap().get(testContainerID);
|
||||
|
||||
Assert.assertNotNull(status.getContainer());
|
||||
Assert.assertNotNull(status.getContainer().getContainerPath());
|
||||
Assert.assertNotNull(status.getContainer().getDBPath());
|
||||
Assert.assertNotNull(cData);
|
||||
Assert.assertNotNull(cData.getContainerPath());
|
||||
Assert.assertNotNull(cData.getDBPath());
|
||||
|
||||
|
||||
Assert.assertTrue(new File(status.getContainer().getContainerPath())
|
||||
Assert.assertTrue(new File(cData.getContainerPath())
|
||||
.exists());
|
||||
|
||||
Path meta = Paths.get(status.getContainer().getDBPath()).getParent();
|
||||
Path meta = Paths.get(cData.getDBPath()).getParent();
|
||||
Assert.assertTrue(meta != null && Files.exists(meta));
|
||||
|
||||
MetadataStore store = null;
|
||||
try {
|
||||
store = KeyUtils.getDB(status.getContainer(), conf);
|
||||
store = KeyUtils.getDB(cData, conf);
|
||||
Assert.assertNotNull(store);
|
||||
} finally {
|
||||
if (store != null) {
|
||||
@ -762,7 +761,7 @@ public void testUpdateContainer() throws IOException {
|
||||
|
||||
// Verify in-memory map
|
||||
ContainerData actualNewData = containerManager.getContainerMap()
|
||||
.get(testContainerID).getContainer();
|
||||
.get(testContainerID);
|
||||
Assert.assertEquals("shire_new",
|
||||
actualNewData.getAllMetadata().get("VOLUME"));
|
||||
Assert.assertEquals("bilbo_new",
|
||||
@ -805,7 +804,7 @@ public void testUpdateContainer() throws IOException {
|
||||
|
||||
// Verify in-memory map
|
||||
actualNewData = containerManager.getContainerMap()
|
||||
.get(testContainerID).getContainer();
|
||||
.get(testContainerID);
|
||||
Assert.assertEquals("shire_new_1",
|
||||
actualNewData.getAllMetadata().get("VOLUME"));
|
||||
Assert.assertEquals("bilbo_new_1",
|
||||
|
Loading…
Reference in New Issue
Block a user